update interrupt

Author: codeSTACKr

src/memory_agent/graph.ts

@@ -2,7 +2,6 @@ import { ChatOpenAI } from "@langchain/openai";
 import {
   task,
   entrypoint,
-  interrupt,
   addMessages,
   getPreviousState,
 } from "@langchain/langgraph";
@@ -13,7 +12,7 @@ import {
 } from "@langchain/core/messages";
 import { initializeTools } from "./tools.js";
 import { createCheckpointer } from "./checkpointer.js";
-import { AGENT_PROMPT, HELP_PHRASE } from "./prompts.js";
+import { AGENT_PROMPT } from "./prompts.js";
 import { ToolCall } from "@langchain/core/messages/tool";
 
 // Create workflow
@@ -73,17 +72,6 @@ export const agent = entrypoint(
 
     while (true) {
       if (!llmResponse.tool_calls?.length) {
-        if (llmResponse.content.toString().includes(HELP_PHRASE)) {
-          const humanResponse = interrupt({
-            current_response: llmResponse.content,
-            action: "Please provide guidance on how to answer this question.",
-          });
-          currentMessages = addMessages(currentMessages, [
-            llmResponse,
-            humanResponse,
-          ]);
-          continue;
-        }
         break;
       }
 

src/memory_agent/prompts.ts

@@ -1,7 +1,5 @@
 // Define default prompts
 
-export const HELP_PHRASE = `I am calling my supervisor for assistance.`;
-
 export const AGENT_PROMPT = `You are an HR Support Assistant. Your role is to help employees with questions about company policies.
 You have access to a database of HR policies through the searchPolicies tool.
 
@@ -21,6 +19,4 @@ Think through this step by step:
 1. What specific policy information is needed?
 2. Do I have this information in the current policies?
 3. Should I search for more policies or request human help?
-4. How should I format my response?
-
-If you get stuck, you can always respond directly with a message containing the following: "${HELP_PHRASE}"`;
+4. How should I format my response?`

src/memory_agent/tools.ts

@@ -1,11 +1,10 @@
-import { LangGraphRunnableConfig } from "@langchain/langgraph";
+import { LangGraphRunnableConfig, interrupt } from "@langchain/langgraph";
 import { tool } from "@langchain/core/tools";
 import { z } from "zod";
 import { MongoDBAtlasVectorSearch } from "@langchain/mongodb";
 import { OpenAIEmbeddings } from "@langchain/openai";
 import { Document } from "@langchain/core/documents";
 import { MongoClient } from "mongodb";
-import { HELP_PHRASE } from "./prompts.js";
 
 /**
  * Initialize tools within a function so that they have access to the current
@@ -77,9 +76,8 @@ export function initializeTools(config?: LangGraphRunnableConfig) {
    */
   async function humanAssistance(opts: { query: string }): Promise<string> {
     const { query } = opts;
-
-    // This will be intercepted by the interrupt handler
-    return `Be sure to include the following in your response: "${HELP_PHRASE} ${query}"`;
+    const humanResponse = await interrupt({query});
+    return humanResponse.content.toString();
   }
 
   const humanAssistanceTool = tool(humanAssistance, {