track status code

index.ts

@@ -61,6 +61,7 @@ const requestTraces: {
   arguments?: any;
   provider?: string;
   duration_ms?: number;
+  status_code?: number;
 }[] = [];
 
 let lastTraceBatchTime = Date.now();
@@ -246,11 +247,13 @@ app.get('/', (c) => {
   }
 });
 
-async function storeStreamedResponse(streamToLog: ReadableStream<Uint8Array>, contentType: string | null, targetUrl: string, traceIndex: number) {
+async function storeStreamedResponse(streamToLog: ReadableStream<Uint8Array>, contentType: string | null, targetUrl: string, traceIndex: number, statusCode: number) {
     const reader = streamToLog.getReader();
     const chunks: Uint8Array[] = [];
     
     try {
+        requestTraces[traceIndex].status_code = statusCode;
+        
         while (true) {
             const { done, value } = await reader.read();
             if (done) break;
@@ -483,7 +486,7 @@ app.all('*', async (c) => {
         const contentType = response.headers.get('content-type');
 
         if (shouldCreateTrace && traceIndex >= 0) {
-          storeStreamedResponse(streamForStorage, contentType, targetUrl, traceIndex).catch(err => {
+          storeStreamedResponse(streamForStorage, contentType, targetUrl, traceIndex, response.status).catch(err => {
                console.error("Error in background stream storage:", err);
           });
         }
@@ -498,6 +501,7 @@ app.all('*', async (c) => {
         }
         if (shouldCreateTrace && traceIndex >= 0) {
           requestTraces[traceIndex].timestamp_end = new Date().toISOString();
+          requestTraces[traceIndex].status_code = response.status;
           
           // Calculate duration if we have both timestamps
           if (requestTraces[traceIndex].timestamp_start && requestTraces[traceIndex].timestamp_end) {