refactor: make toolkits pass in runnables

langchain-ai · Oct 16, 2023 · 80ed3c5 · 80ed3c5
1 parent 319c24f
commit 80ed3c5
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/langchain/src/agents/toolkits/json/json.ts b/langchain/src/agents/toolkits/json/json.ts
@@ -56,7 +56,7 @@ export function createJsonAgent(
   });
   const chain = new LLMChain({ prompt, llm });
   const agent = new ZeroShotAgent({
-    llmChain: chain,
+    runnable: chain,
     allowedTools: tools.map((t) => t.name),
   });
   return AgentExecutor.fromAgentAndTools({

diff --git a/langchain/src/agents/toolkits/openapi/openapi.ts b/langchain/src/agents/toolkits/openapi/openapi.ts
@@ -88,7 +88,7 @@ export function createOpenApiAgent(
     llm,
   });
   const toolNames = tools.map((tool) => tool.name);
-  const agent = new ZeroShotAgent({ llmChain: chain, allowedTools: toolNames });
+  const agent = new ZeroShotAgent({ runnable: chain, allowedTools: toolNames });
   return AgentExecutor.fromAgentAndTools({
     agent,
     tools,

diff --git a/langchain/src/agents/toolkits/sql/sql.ts b/langchain/src/agents/toolkits/sql/sql.ts
@@ -70,7 +70,7 @@ export function createSqlAgent(
   });
   const chain = new LLMChain({ prompt, llm });
   const agent = new ZeroShotAgent({
-    llmChain: chain,
+    runnable: chain,
     allowedTools: tools.map((t) => t.name),
   });
   return AgentExecutor.fromAgentAndTools({

diff --git a/langchain/src/agents/toolkits/vectorstore/vectorstore.ts b/langchain/src/agents/toolkits/vectorstore/vectorstore.ts
@@ -92,7 +92,7 @@ export function createVectorStoreAgent(
   });
   const chain = new LLMChain({ prompt, llm });
   const agent = new ZeroShotAgent({
-    llmChain: chain,
+    runnable: chain,
     allowedTools: tools.map((t) => t.name),
   });
   return AgentExecutor.fromAgentAndTools({
@@ -120,7 +120,7 @@ export function createVectorStoreRouterAgent(
   });
   const chain = new LLMChain({ prompt, llm });
   const agent = new ZeroShotAgent({
-    llmChain: chain,
+    runnable: chain,
     allowedTools: tools.map((t) => t.name),
   });
   return AgentExecutor.fromAgentAndTools({