From f6b40792a98a15b70957ecff7077154ec6a4a35e Mon Sep 17 00:00:00 2001
From: Christina Tong <christina@letta.com>
Date: Fri, 20 Mar 2026 11:12:13 -0700
Subject: [PATCH] fix(ws): set agent context for skills/tools (#1471)

---
 src/websocket/listener/turn.ts | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/websocket/listener/turn.ts b/src/websocket/listener/turn.ts
index 17612b5..3530e61 100644
--- a/src/websocket/listener/turn.ts
+++ b/src/websocket/listener/turn.ts
@@ -9,6 +9,7 @@ import type { ApprovalResult } from "../../agent/approval-execution";
 import { fetchRunErrorDetail } from "../../agent/approval-recovery";
 import { getResumeData } from "../../agent/check-approval";
 import { getClient } from "../../agent/client";
+import { setConversationId, setCurrentAgentId } from "../../agent/context";
 import {
   getStreamToolContextId,
   type sendMessageStream,
@@ -152,6 +153,10 @@ export async function handleIncomingMessage(
       return;
     }
 
+    // Set agent context for tools that need it (e.g., Skill tool)
+    setCurrentAgentId(agentId);
+    setConversationId(conversationId);
+
     if (isDebugEnabled()) {
       console.log(
         `[Listen] Handling message: agentId=${agentId}, requestedConversationId=${requestedConversationId}, conversationId=${conversationId}`,