feat: Log request and response headers in logger

Author: ericc-ch

src/lib/logger.ts

@@ -44,6 +44,7 @@ export const logger = {
     endpoint: string,
     method: string,
     payload: unknown,
+    headers?: Record<string, string>,
   ): Promise<void> {
     if (!this.options.enabled || !this.options.filePath) return
 
@@ -52,6 +53,9 @@ export const logger = {
       `## Request - ${timestamp}`,
       `Endpoint: ${endpoint}`,
       `Method: ${method}`,
+      headers ?
+        `Headers:\n\`\`\`json\n${JSON.stringify(headers, null, 2)}\n\`\`\``
+      : "",
       `Payload:`,
       `\`\`\`json`,
       JSON.stringify(payload, null, 2),
@@ -66,13 +70,20 @@ export const logger = {
     }
   },
 
-  async logResponse(endpoint: string, response: unknown): Promise<void> {
+  async logResponse(
+    endpoint: string,
+    response: unknown,
+    headers?: Record<string, string>,
+  ): Promise<void> {
     if (!this.options.enabled || !this.options.filePath) return
 
     const timestamp = new Date().toISOString()
     const logEntry = [
       `## Response - ${timestamp}`,
       `Endpoint: ${endpoint}`,
+      headers ?
+        `Headers:\n\`\`\`json\n${JSON.stringify(headers, null, 2)}\n\`\`\``
+      : "",
       `Response:`,
       `\`\`\`json`,
       JSON.stringify(response, null, 2),

src/routes/chat-completions/handler.ts

@@ -38,8 +38,11 @@ function createCondensedStreamingResponse(
 export async function handlerStreaming(c: Context) {
   const payload = await c.req.json<ChatCompletionsPayload>()
 
+  // Convert request headers to a regular object from Headers
+  const requestHeaders = c.req.header()
+
   // Log the request at the beginning for both streaming and non-streaming cases
-  await logger.logRequest("/chat/completions", "POST", payload)
+  await logger.logRequest("/chat/completions", "POST", payload, requestHeaders)
 
   if (payload.stream) {
     const response = await chatCompletionsStream(payload)
@@ -52,7 +55,7 @@ export async function handlerStreaming(c: Context) {
       for await (const chunk of response) {
         await stream.writeSSE(chunk as SSEMessage)
 
-        if (!logger.options.enabled) continue // Changed from return to continue
+        if (!logger.options.enabled) continue
 
         // Check if chunk data is "DONE" or not a valid JSON string
         if (!chunk.data || chunk.data === "[DONE]") {
@@ -83,15 +86,18 @@ export async function handlerStreaming(c: Context) {
           collectedContent,
         )
 
-        await logger.logResponse("/chat/completions", condensedResponse)
+        await logger.logResponse("/chat/completions", condensedResponse, {})
       }
     })
   }
 
   const response = await chatCompletions(payload)
 
-  // Log the non-streaming response
-  await logger.logResponse("/chat/completions", response)
+  // Get response headers if any
+  const responseHeaders = {} // Empty placeholder for response headers
+
+  // Log the non-streaming response with headers
+  await logger.logResponse("/chat/completions", response, responseHeaders)
 
   return c.json(response)
 }