From 7e3551caa3341f4b1c0abec28dd954158cdfed57 Mon Sep 17 00:00:00 2001
From: haritabh-z01 <haritabh.z01+github@gmail.com>
Date: Tue, 16 Dec 2025 11:31:09 +0530
Subject: [PATCH] chore: improve logging for formatters

---
 .../pipeline/providers/formatting/ollama-formatter.ts | 11 +++++++++++
 .../providers/formatting/openrouter-formatter.ts      | 11 +++++++++++
 2 files changed, 22 insertions(+)

diff --git a/apps/desktop/src/pipeline/providers/formatting/ollama-formatter.ts b/apps/desktop/src/pipeline/providers/formatting/ollama-formatter.ts
index 5ff4661..e9c5dea 100644
--- a/apps/desktop/src/pipeline/providers/formatting/ollama-formatter.ts
+++ b/apps/desktop/src/pipeline/providers/formatting/ollama-formatter.ts
@@ -17,6 +17,12 @@ export class OllamaFormatter implements FormattingProvider {
       // Construct the formatter prompt using the same function as OpenRouter
       const { systemPrompt } = constructFormatterPrompt(context);
 
+      logger.pipeline.debug("Formatting request", {
+        model: this.model,
+        systemPrompt,
+        userPrompt: text,
+      });
+
       // Use Ollama's chat endpoint for system/user message structure
       const response = await fetch(`${this.ollamaUrl}/api/chat`, {
         method: "POST",
@@ -42,6 +48,11 @@ export class OllamaFormatter implements FormattingProvider {
       const data = await response.json();
       const aiResponse = data.message?.content ?? "";
 
+      logger.pipeline.debug("Formatting raw response", {
+        model: this.model,
+        rawResponse: aiResponse,
+      });
+
       // Extract formatted text from XML tags (same as OpenRouter)
       const match = aiResponse.match(
         /<formatted_text>([\s\S]*?)<\/formatted_text>/,
diff --git a/apps/desktop/src/pipeline/providers/formatting/openrouter-formatter.ts b/apps/desktop/src/pipeline/providers/formatting/openrouter-formatter.ts
index 7ef68a5..b1b28c5 100644
--- a/apps/desktop/src/pipeline/providers/formatting/openrouter-formatter.ts
+++ b/apps/desktop/src/pipeline/providers/formatting/openrouter-formatter.ts
@@ -31,6 +31,12 @@ export class OpenRouterProvider implements FormattingProvider {
       // Build user prompt with context
       const userPrompt = text;
 
+      logger.pipeline.info("Formatting request", {
+        model: this.model,
+        systemPrompt,
+        userPrompt,
+      });
+
       const { text: aiResponse } = await generateText({
         model: this.provider(this.model),
         messages: [
@@ -47,6 +53,11 @@ export class OpenRouterProvider implements FormattingProvider {
         maxTokens: 2000,
       });
 
+      logger.pipeline.debug("Formatting raw response", {
+        model: this.model,
+        rawResponse: aiResponse,
+      });
+
       // Extract formatted text from XML tags
       const match = aiResponse.match(
         /<formatted_text>([\s\S]*?)<\/formatted_text>/,