From 4c01d7d8f1460cc3b417879eff050d45500d166e Mon Sep 17 00:00:00 2001
From: perf3ct <jonfuller2012@gmail.com>
Date: Sat, 5 Jul 2025 00:11:15 +0000
Subject: [PATCH] fix(llm): resolve compilation issues due to additional stages

---
 .../services/llm/pipeline/stages/error_recovery_stage.ts   | 7 ++++++-
 apps/server/src/services/ws.ts                             | 3 +++
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/apps/server/src/services/llm/pipeline/stages/error_recovery_stage.ts b/apps/server/src/services/llm/pipeline/stages/error_recovery_stage.ts
index e49eeccdf..c753a57ad 100644
--- a/apps/server/src/services/llm/pipeline/stages/error_recovery_stage.ts
+++ b/apps/server/src/services/llm/pipeline/stages/error_recovery_stage.ts
@@ -98,7 +98,12 @@ export class ErrorRecoveryStage extends BasePipelineStage<ToolExecutionInput, {
         log.info(`========== ERROR RECOVERY STAGE PROCESSING ==========`);
         log.info(`Processing ${response.tool_calls.length} tool calls with advanced error recovery`);
 
-        const recoveredToolCalls = [];
+        const recoveredToolCalls: Array<{
+            toolCall: any;
+            message: Message;
+            attempts: number;
+            recovered: boolean;
+        }> = [];
         const updatedMessages = [...input.messages];
 
         // Process each tool call with recovery
diff --git a/apps/server/src/services/ws.ts b/apps/server/src/services/ws.ts
index ad1aafe0f..2692f47eb 100644
--- a/apps/server/src/services/ws.ts
+++ b/apps/server/src/services/ws.ts
@@ -48,6 +48,9 @@ interface Message {
     chatNoteId?: string;
     content?: string;
     thinking?: string;
+    interactionId?: string;
+    response?: string;
+    timestamp?: number;
     toolExecution?: {
         action?: string;
         tool?: string;