@langchain/classic 1.0.23 → 1.0.24-dev-1773698445534

package/dist/agents/toolkits/index.js

@@ -5,7 +5,6 @@ import { VectorStoreRouterToolkit, VectorStoreToolkit, createVectorStoreAgent, c
 import { createRetrieverTool } from "./conversational_retrieval/tool.js";
 import { OpenAIAgentTokenBufferMemory } from "./conversational_retrieval/token_buffer_memory.js";
 import { createConversationalRetrievalAgent } from "./conversational_retrieval/openai_functions.js";
-
 //#region src/agents/toolkits/index.ts
 var toolkits_exports = /* @__PURE__ */ __exportAll({
 	JsonToolkit: () => JsonToolkit,
@@ -21,7 +20,7 @@ var toolkits_exports = /* @__PURE__ */ __exportAll({
 	createVectorStoreAgent: () => createVectorStoreAgent,
 	createVectorStoreRouterAgent: () => createVectorStoreRouterAgent
 });
-
 //#endregion
 export { JsonToolkit, OpenAIAgentTokenBufferMemory, OpenApiToolkit, RequestsToolkit, VectorStoreRouterToolkit, VectorStoreToolkit, createConversationalRetrievalAgent, createJsonAgent, createOpenApiAgent, createRetrieverTool, createVectorStoreAgent, createVectorStoreRouterAgent, toolkits_exports };
+
 //# sourceMappingURL=index.js.map

package/dist/agents/toolkits/json/json.cjs

@@ -1,11 +1,10 @@
-const require_runtime = require('../../../_virtual/_rolldown/runtime.cjs');
-const require_json = require('../../../tools/json.cjs');
-const require_prompt = require('./prompt.cjs');
-const require_llm_chain = require('../../../chains/llm_chain.cjs');
-const require_index = require('../../mrkl/index.cjs');
-const require_executor = require('../../executor.cjs');
+require("../../../_virtual/_rolldown/runtime.cjs");
+const require_json = require("../../../tools/json.cjs");
+const require_prompt = require("./prompt.cjs");
+const require_llm_chain = require("../../../chains/llm_chain.cjs");
+const require_index = require("../../mrkl/index.cjs");
+const require_executor = require("../../executor.cjs");
 let _langchain_core_tools = require("@langchain/core/tools");
-
 //#region src/agents/toolkits/json/json.ts
 /**
 * Represents a toolkit for working with JSON data. It initializes the
@@ -58,8 +57,8 @@ function createJsonAgent(llm, toolkit, args) {
 		returnIntermediateSteps: true
 	});
 }
-
 //#endregion
 exports.JsonToolkit = JsonToolkit;
 exports.createJsonAgent = createJsonAgent;
+
 //# sourceMappingURL=json.cjs.map

package/dist/agents/toolkits/json/json.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"json.cjs","names":["BaseToolkit","JsonListKeysTool","JsonGetValueTool","JSON_PREFIX","JSON_SUFFIX","ZeroShotAgent","LLMChain","AgentExecutor"],"sources":["../../../../src/agents/toolkits/json/json.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { type ToolInterface, BaseToolkit } from \"@langchain/core/tools\";\nimport {\n  JsonGetValueTool,\n  JsonListKeysTool,\n  JsonSpec,\n} from \"../../../tools/json.js\";\nimport { JSON_PREFIX, JSON_SUFFIX } from \"./prompt.js\";\nimport { LLMChain } from \"../../../chains/llm_chain.js\";\nimport { ZeroShotCreatePromptArgs, ZeroShotAgent } from \"../../mrkl/index.js\";\nimport { AgentExecutor } from \"../../executor.js\";\n\n/**\n * Represents a toolkit for working with JSON data. It initializes the\n * JSON tools based on the provided JSON specification.\n * @example\n * ```typescript\n * const toolkit = new JsonToolkit(new JsonSpec());\n * const executor = createJsonAgent(model, toolkit);\n * const result = await executor.invoke({\n *   input: 'What are the required parameters in the request body to the /completions endpoint?'\n * });\n * ```\n */\nexport class JsonToolkit extends BaseToolkit {\n  tools: ToolInterface[];\n\n  constructor(public jsonSpec: JsonSpec) {\n    super();\n    this.tools = [\n      new JsonListKeysTool(jsonSpec),\n      new JsonGetValueTool(jsonSpec),\n    ];\n  }\n}\n\n/**\n * Creates a JSON agent using a language model, a JSON toolkit, and\n * optional prompt arguments. It creates a prompt for the agent using the\n * JSON tools and the provided prefix and suffix. It then creates a\n * ZeroShotAgent with the prompt and the JSON tools, and returns an\n * AgentExecutor for executing the agent with the tools.\n * @param llm The language model used to create the JSON agent.\n * @param toolkit The JSON toolkit used to create the JSON agent.\n * @param args Optional prompt arguments used to create the JSON agent.\n * @returns An AgentExecutor for executing the created JSON agent with the tools.\n */\nexport function createJsonAgent(\n  llm: BaseLanguageModelInterface,\n  toolkit: JsonToolkit,\n  args?: ZeroShotCreatePromptArgs\n) {\n  const {\n    prefix = JSON_PREFIX,\n    suffix = JSON_SUFFIX,\n    inputVariables = [\"input\", \"agent_scratchpad\"],\n  } = args ?? {};\n  const { tools } = toolkit;\n  const prompt = ZeroShotAgent.createPrompt(tools, {\n    prefix,\n    suffix,\n    inputVariables,\n  });\n  const chain = new LLMChain({ prompt, llm });\n  const agent = new ZeroShotAgent({\n    llmChain: chain,\n    allowedTools: tools.map((t) => t.name),\n  });\n  return AgentExecutor.fromAgentAndTools({\n    agent,\n    tools,\n    returnIntermediateSteps: true,\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;AAwBA,IAAa,cAAb,cAAiCA,kCAAY;CAC3C;CAEA,YAAY,AAAO,UAAoB;AACrC,SAAO;EADU;AAEjB,OAAK,QAAQ,CACX,IAAIC,8BAAiB,SAAS,EAC9B,IAAIC,8BAAiB,SAAS,CAC/B;;;;;;;;;;;;;;AAeL,SAAgB,gBACd,KACA,SACA,MACA;CACA,MAAM,EACJ,SAASC,4BACT,SAASC,4BACT,iBAAiB,CAAC,SAAS,mBAAmB,KAC5C,QAAQ,EAAE;CACd,MAAM,EAAE,UAAU;CAOlB,MAAM,QAAQ,IAAIC,4BAAc;EAC9B,UAFY,IAAIC,2BAAS;GAAE,QALdD,4BAAc,aAAa,OAAO;IAC/C;IACA;IACA;IACD,CAAC;GACmC;GAAK,CAAC;EAGzC,cAAc,MAAM,KAAK,MAAM,EAAE,KAAK;EACvC,CAAC;AACF,QAAOE,+BAAc,kBAAkB;EACrC;EACA;EACA,yBAAyB;EAC1B,CAAC"}
+{"version":3,"file":"json.cjs","names":["BaseToolkit","JsonListKeysTool","JsonGetValueTool","JSON_PREFIX","JSON_SUFFIX","ZeroShotAgent","LLMChain","AgentExecutor"],"sources":["../../../../src/agents/toolkits/json/json.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { type ToolInterface, BaseToolkit } from \"@langchain/core/tools\";\nimport {\n  JsonGetValueTool,\n  JsonListKeysTool,\n  JsonSpec,\n} from \"../../../tools/json.js\";\nimport { JSON_PREFIX, JSON_SUFFIX } from \"./prompt.js\";\nimport { LLMChain } from \"../../../chains/llm_chain.js\";\nimport { ZeroShotCreatePromptArgs, ZeroShotAgent } from \"../../mrkl/index.js\";\nimport { AgentExecutor } from \"../../executor.js\";\n\n/**\n * Represents a toolkit for working with JSON data. It initializes the\n * JSON tools based on the provided JSON specification.\n * @example\n * ```typescript\n * const toolkit = new JsonToolkit(new JsonSpec());\n * const executor = createJsonAgent(model, toolkit);\n * const result = await executor.invoke({\n *   input: 'What are the required parameters in the request body to the /completions endpoint?'\n * });\n * ```\n */\nexport class JsonToolkit extends BaseToolkit {\n  tools: ToolInterface[];\n\n  constructor(public jsonSpec: JsonSpec) {\n    super();\n    this.tools = [\n      new JsonListKeysTool(jsonSpec),\n      new JsonGetValueTool(jsonSpec),\n    ];\n  }\n}\n\n/**\n * Creates a JSON agent using a language model, a JSON toolkit, and\n * optional prompt arguments. It creates a prompt for the agent using the\n * JSON tools and the provided prefix and suffix. It then creates a\n * ZeroShotAgent with the prompt and the JSON tools, and returns an\n * AgentExecutor for executing the agent with the tools.\n * @param llm The language model used to create the JSON agent.\n * @param toolkit The JSON toolkit used to create the JSON agent.\n * @param args Optional prompt arguments used to create the JSON agent.\n * @returns An AgentExecutor for executing the created JSON agent with the tools.\n */\nexport function createJsonAgent(\n  llm: BaseLanguageModelInterface,\n  toolkit: JsonToolkit,\n  args?: ZeroShotCreatePromptArgs\n) {\n  const {\n    prefix = JSON_PREFIX,\n    suffix = JSON_SUFFIX,\n    inputVariables = [\"input\", \"agent_scratchpad\"],\n  } = args ?? {};\n  const { tools } = toolkit;\n  const prompt = ZeroShotAgent.createPrompt(tools, {\n    prefix,\n    suffix,\n    inputVariables,\n  });\n  const chain = new LLMChain({ prompt, llm });\n  const agent = new ZeroShotAgent({\n    llmChain: chain,\n    allowedTools: tools.map((t) => t.name),\n  });\n  return AgentExecutor.fromAgentAndTools({\n    agent,\n    tools,\n    returnIntermediateSteps: true,\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAwBA,IAAa,cAAb,cAAiCA,sBAAAA,YAAY;CAC3C;CAEA,YAAY,UAA2B;AACrC,SAAO;AADU,OAAA,WAAA;AAEjB,OAAK,QAAQ,CACX,IAAIC,aAAAA,iBAAiB,SAAS,EAC9B,IAAIC,aAAAA,iBAAiB,SAAS,CAC/B;;;;;;;;;;;;;;AAeL,SAAgB,gBACd,KACA,SACA,MACA;CACA,MAAM,EACJ,SAASC,eAAAA,aACT,SAASC,eAAAA,aACT,iBAAiB,CAAC,SAAS,mBAAmB,KAC5C,QAAQ,EAAE;CACd,MAAM,EAAE,UAAU;CAOlB,MAAM,QAAQ,IAAIC,cAAAA,cAAc;EAC9B,UAFY,IAAIC,kBAAAA,SAAS;GAAE,QALdD,cAAAA,cAAc,aAAa,OAAO;IAC/C;IACA;IACA;IACD,CAAC;GACmC;GAAK,CAAC;EAGzC,cAAc,MAAM,KAAK,MAAM,EAAE,KAAK;EACvC,CAAC;AACF,QAAOE,iBAAAA,cAAc,kBAAkB;EACrC;EACA;EACA,yBAAyB;EAC1B,CAAC"}

package/dist/agents/toolkits/json/json.js

@@ -4,7 +4,6 @@ import { LLMChain } from "../../../chains/llm_chain.js";
 import { ZeroShotAgent } from "../../mrkl/index.js";
 import { AgentExecutor } from "../../executor.js";
 import { BaseToolkit } from "@langchain/core/tools";
-
 //#region src/agents/toolkits/json/json.ts
 /**
 * Represents a toolkit for working with JSON data. It initializes the
@@ -57,7 +56,7 @@ function createJsonAgent(llm, toolkit, args) {
 		returnIntermediateSteps: true
 	});
 }
-
 //#endregion
 export { JsonToolkit, createJsonAgent };
+
 //# sourceMappingURL=json.js.map

package/dist/agents/toolkits/json/json.js.map

@@ -1 +1 @@
-{"version":3,"file":"json.js","names":[],"sources":["../../../../src/agents/toolkits/json/json.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { type ToolInterface, BaseToolkit } from \"@langchain/core/tools\";\nimport {\n  JsonGetValueTool,\n  JsonListKeysTool,\n  JsonSpec,\n} from \"../../../tools/json.js\";\nimport { JSON_PREFIX, JSON_SUFFIX } from \"./prompt.js\";\nimport { LLMChain } from \"../../../chains/llm_chain.js\";\nimport { ZeroShotCreatePromptArgs, ZeroShotAgent } from \"../../mrkl/index.js\";\nimport { AgentExecutor } from \"../../executor.js\";\n\n/**\n * Represents a toolkit for working with JSON data. It initializes the\n * JSON tools based on the provided JSON specification.\n * @example\n * ```typescript\n * const toolkit = new JsonToolkit(new JsonSpec());\n * const executor = createJsonAgent(model, toolkit);\n * const result = await executor.invoke({\n *   input: 'What are the required parameters in the request body to the /completions endpoint?'\n * });\n * ```\n */\nexport class JsonToolkit extends BaseToolkit {\n  tools: ToolInterface[];\n\n  constructor(public jsonSpec: JsonSpec) {\n    super();\n    this.tools = [\n      new JsonListKeysTool(jsonSpec),\n      new JsonGetValueTool(jsonSpec),\n    ];\n  }\n}\n\n/**\n * Creates a JSON agent using a language model, a JSON toolkit, and\n * optional prompt arguments. It creates a prompt for the agent using the\n * JSON tools and the provided prefix and suffix. It then creates a\n * ZeroShotAgent with the prompt and the JSON tools, and returns an\n * AgentExecutor for executing the agent with the tools.\n * @param llm The language model used to create the JSON agent.\n * @param toolkit The JSON toolkit used to create the JSON agent.\n * @param args Optional prompt arguments used to create the JSON agent.\n * @returns An AgentExecutor for executing the created JSON agent with the tools.\n */\nexport function createJsonAgent(\n  llm: BaseLanguageModelInterface,\n  toolkit: JsonToolkit,\n  args?: ZeroShotCreatePromptArgs\n) {\n  const {\n    prefix = JSON_PREFIX,\n    suffix = JSON_SUFFIX,\n    inputVariables = [\"input\", \"agent_scratchpad\"],\n  } = args ?? {};\n  const { tools } = toolkit;\n  const prompt = ZeroShotAgent.createPrompt(tools, {\n    prefix,\n    suffix,\n    inputVariables,\n  });\n  const chain = new LLMChain({ prompt, llm });\n  const agent = new ZeroShotAgent({\n    llmChain: chain,\n    allowedTools: tools.map((t) => t.name),\n  });\n  return AgentExecutor.fromAgentAndTools({\n    agent,\n    tools,\n    returnIntermediateSteps: true,\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAwBA,IAAa,cAAb,cAAiC,YAAY;CAC3C;CAEA,YAAY,AAAO,UAAoB;AACrC,SAAO;EADU;AAEjB,OAAK,QAAQ,CACX,IAAI,iBAAiB,SAAS,EAC9B,IAAI,iBAAiB,SAAS,CAC/B;;;;;;;;;;;;;;AAeL,SAAgB,gBACd,KACA,SACA,MACA;CACA,MAAM,EACJ,SAAS,aACT,SAAS,aACT,iBAAiB,CAAC,SAAS,mBAAmB,KAC5C,QAAQ,EAAE;CACd,MAAM,EAAE,UAAU;CAOlB,MAAM,QAAQ,IAAI,cAAc;EAC9B,UAFY,IAAI,SAAS;GAAE,QALd,cAAc,aAAa,OAAO;IAC/C;IACA;IACA;IACD,CAAC;GACmC;GAAK,CAAC;EAGzC,cAAc,MAAM,KAAK,MAAM,EAAE,KAAK;EACvC,CAAC;AACF,QAAO,cAAc,kBAAkB;EACrC;EACA;EACA,yBAAyB;EAC1B,CAAC"}
+{"version":3,"file":"json.js","names":[],"sources":["../../../../src/agents/toolkits/json/json.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { type ToolInterface, BaseToolkit } from \"@langchain/core/tools\";\nimport {\n  JsonGetValueTool,\n  JsonListKeysTool,\n  JsonSpec,\n} from \"../../../tools/json.js\";\nimport { JSON_PREFIX, JSON_SUFFIX } from \"./prompt.js\";\nimport { LLMChain } from \"../../../chains/llm_chain.js\";\nimport { ZeroShotCreatePromptArgs, ZeroShotAgent } from \"../../mrkl/index.js\";\nimport { AgentExecutor } from \"../../executor.js\";\n\n/**\n * Represents a toolkit for working with JSON data. It initializes the\n * JSON tools based on the provided JSON specification.\n * @example\n * ```typescript\n * const toolkit = new JsonToolkit(new JsonSpec());\n * const executor = createJsonAgent(model, toolkit);\n * const result = await executor.invoke({\n *   input: 'What are the required parameters in the request body to the /completions endpoint?'\n * });\n * ```\n */\nexport class JsonToolkit extends BaseToolkit {\n  tools: ToolInterface[];\n\n  constructor(public jsonSpec: JsonSpec) {\n    super();\n    this.tools = [\n      new JsonListKeysTool(jsonSpec),\n      new JsonGetValueTool(jsonSpec),\n    ];\n  }\n}\n\n/**\n * Creates a JSON agent using a language model, a JSON toolkit, and\n * optional prompt arguments. It creates a prompt for the agent using the\n * JSON tools and the provided prefix and suffix. It then creates a\n * ZeroShotAgent with the prompt and the JSON tools, and returns an\n * AgentExecutor for executing the agent with the tools.\n * @param llm The language model used to create the JSON agent.\n * @param toolkit The JSON toolkit used to create the JSON agent.\n * @param args Optional prompt arguments used to create the JSON agent.\n * @returns An AgentExecutor for executing the created JSON agent with the tools.\n */\nexport function createJsonAgent(\n  llm: BaseLanguageModelInterface,\n  toolkit: JsonToolkit,\n  args?: ZeroShotCreatePromptArgs\n) {\n  const {\n    prefix = JSON_PREFIX,\n    suffix = JSON_SUFFIX,\n    inputVariables = [\"input\", \"agent_scratchpad\"],\n  } = args ?? {};\n  const { tools } = toolkit;\n  const prompt = ZeroShotAgent.createPrompt(tools, {\n    prefix,\n    suffix,\n    inputVariables,\n  });\n  const chain = new LLMChain({ prompt, llm });\n  const agent = new ZeroShotAgent({\n    llmChain: chain,\n    allowedTools: tools.map((t) => t.name),\n  });\n  return AgentExecutor.fromAgentAndTools({\n    agent,\n    tools,\n    returnIntermediateSteps: true,\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;AAwBA,IAAa,cAAb,cAAiC,YAAY;CAC3C;CAEA,YAAY,UAA2B;AACrC,SAAO;AADU,OAAA,WAAA;AAEjB,OAAK,QAAQ,CACX,IAAI,iBAAiB,SAAS,EAC9B,IAAI,iBAAiB,SAAS,CAC/B;;;;;;;;;;;;;;AAeL,SAAgB,gBACd,KACA,SACA,MACA;CACA,MAAM,EACJ,SAAS,aACT,SAAS,aACT,iBAAiB,CAAC,SAAS,mBAAmB,KAC5C,QAAQ,EAAE;CACd,MAAM,EAAE,UAAU;CAOlB,MAAM,QAAQ,IAAI,cAAc;EAC9B,UAFY,IAAI,SAAS;GAAE,QALd,cAAc,aAAa,OAAO;IAC/C;IACA;IACA;IACD,CAAC;GACmC;GAAK,CAAC;EAGzC,cAAc,MAAM,KAAK,MAAM,EAAE,KAAK;EACvC,CAAC;AACF,QAAO,cAAc,kBAAkB;EACrC;EACA;EACA,yBAAyB;EAC1B,CAAC"}

package/dist/agents/toolkits/json/prompt.cjs

@@ -1,4 +1,3 @@
-
 //#region src/agents/toolkits/json/prompt.ts
 const JSON_PREFIX = `You are an agent designed to interact with JSON.
 Your goal is to return a final answer by interacting with the JSON.
@@ -24,8 +23,8 @@ const JSON_SUFFIX = `Begin!"
 Question: {input}
 Thought: I should look at the keys that exist to see what I can query. I should use the 'json_list_keys' tool with an empty string as the input.
 {agent_scratchpad}`;
-
 //#endregion
 exports.JSON_PREFIX = JSON_PREFIX;
 exports.JSON_SUFFIX = JSON_SUFFIX;
+
 //# sourceMappingURL=prompt.cjs.map

package/dist/agents/toolkits/json/prompt.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"prompt.cjs","names":[],"sources":["../../../../src/agents/toolkits/json/prompt.ts"],"sourcesContent":["export const JSON_PREFIX = `You are an agent designed to interact with JSON.\nYour goal is to return a final answer by interacting with the JSON.\nYou have access to the following tools which help you learn more about the JSON you are interacting with.\nOnly use the below tools. Only use the information returned by the below tools to construct your final answer.\nDo not make up any information that is not contained in the JSON.\nYour input to the tools should be in the form of in json pointer syntax (e.g. /key1/0/key2).\nYou must escape a slash in a key with a ~1, and escape a tilde with a ~0.\nFor example, to access the key /foo, you would use /~1foo\nYou should only use keys that you know for a fact exist. You must validate that a key exists by seeing it previously when calling 'json_list_keys'.\nIf you have not seen a key in one of those responses, you cannot use it.\nYou should only add one key at a time to the path. You cannot add multiple keys at once.\nIf you encounter a null or undefined value, go back to the previous key, look at the available keys, and try again.\n\nIf the question does not seem to be related to the JSON, just return \"I don't know\" as the answer.\nAlways begin your interaction with the 'json_list_keys' with an empty string as the input to see what keys exist in the JSON.\n\nNote that sometimes the value at a given path is large. In this case, you will get an error \"Value is a large dictionary, should explore its keys directly\".\nIn this case, you should ALWAYS follow up by using the 'json_list_keys' tool to see what keys exist at that path.\nDo not simply refer the user to the JSON or a section of the JSON, as this is not a valid answer. Keep digging until you find the answer and explicitly return it.`;\n\nexport const JSON_SUFFIX = `Begin!\"\n\nQuestion: {input}\nThought: I should look at the keys that exist to see what I can query. I should use the 'json_list_keys' tool with an empty string as the input.\n{agent_scratchpad}`;\n"],"mappings":";;AAAA,MAAa,cAAc;;;;;;;;;;;;;;;;;;;AAoB3B,MAAa,cAAc"}
+{"version":3,"file":"prompt.cjs","names":[],"sources":["../../../../src/agents/toolkits/json/prompt.ts"],"sourcesContent":["export const JSON_PREFIX = `You are an agent designed to interact with JSON.\nYour goal is to return a final answer by interacting with the JSON.\nYou have access to the following tools which help you learn more about the JSON you are interacting with.\nOnly use the below tools. Only use the information returned by the below tools to construct your final answer.\nDo not make up any information that is not contained in the JSON.\nYour input to the tools should be in the form of in json pointer syntax (e.g. /key1/0/key2).\nYou must escape a slash in a key with a ~1, and escape a tilde with a ~0.\nFor example, to access the key /foo, you would use /~1foo\nYou should only use keys that you know for a fact exist. You must validate that a key exists by seeing it previously when calling 'json_list_keys'.\nIf you have not seen a key in one of those responses, you cannot use it.\nYou should only add one key at a time to the path. You cannot add multiple keys at once.\nIf you encounter a null or undefined value, go back to the previous key, look at the available keys, and try again.\n\nIf the question does not seem to be related to the JSON, just return \"I don't know\" as the answer.\nAlways begin your interaction with the 'json_list_keys' with an empty string as the input to see what keys exist in the JSON.\n\nNote that sometimes the value at a given path is large. In this case, you will get an error \"Value is a large dictionary, should explore its keys directly\".\nIn this case, you should ALWAYS follow up by using the 'json_list_keys' tool to see what keys exist at that path.\nDo not simply refer the user to the JSON or a section of the JSON, as this is not a valid answer. Keep digging until you find the answer and explicitly return it.`;\n\nexport const JSON_SUFFIX = `Begin!\"\n\nQuestion: {input}\nThought: I should look at the keys that exist to see what I can query. I should use the 'json_list_keys' tool with an empty string as the input.\n{agent_scratchpad}`;\n"],"mappings":";AAAA,MAAa,cAAc;;;;;;;;;;;;;;;;;;;AAoB3B,MAAa,cAAc"}

package/dist/agents/toolkits/json/prompt.js

@@ -23,7 +23,7 @@ const JSON_SUFFIX = `Begin!"
 Question: {input}
 Thought: I should look at the keys that exist to see what I can query. I should use the 'json_list_keys' tool with an empty string as the input.
 {agent_scratchpad}`;
-
 //#endregion
 export { JSON_PREFIX, JSON_SUFFIX };
+
 //# sourceMappingURL=prompt.js.map

package/dist/agents/toolkits/openapi/openapi.cjs

@@ -1,12 +1,11 @@
-const require_runtime = require('../../../_virtual/_rolldown/runtime.cjs');
-const require_llm_chain = require('../../../chains/llm_chain.cjs');
-const require_index = require('../../mrkl/index.cjs');
-const require_executor = require('../../executor.cjs');
-const require_json = require('../json/json.cjs');
-const require_prompt = require('./prompt.cjs');
-const require_requests = require('../../../tools/requests.cjs');
+require("../../../_virtual/_rolldown/runtime.cjs");
+const require_llm_chain = require("../../../chains/llm_chain.cjs");
+const require_index = require("../../mrkl/index.cjs");
+const require_executor = require("../../executor.cjs");
+const require_json = require("../json/json.cjs");
+const require_prompt = require("./prompt.cjs");
+const require_requests = require("../../../tools/requests.cjs");
 let _langchain_core_tools = require("@langchain/core/tools");
-
 //#region src/agents/toolkits/openapi/openapi.ts
 /**
 * Represents a toolkit for making HTTP requests. It initializes the
@@ -95,9 +94,9 @@ function createOpenApiAgent(llm, openApiToolkit, args) {
 		returnIntermediateSteps: true
 	});
 }
-
 //#endregion
 exports.OpenApiToolkit = OpenApiToolkit;
 exports.RequestsToolkit = RequestsToolkit;
 exports.createOpenApiAgent = createOpenApiAgent;
+
 //# sourceMappingURL=openapi.cjs.map

package/dist/agents/toolkits/openapi/openapi.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"openapi.cjs","names":["BaseToolkit","RequestsGetTool","RequestsPostTool","createJsonAgent","JsonToolkit","DynamicTool","JSON_EXPLORER_DESCRIPTION","OPENAPI_PREFIX","OPENAPI_SUFFIX","ZeroShotAgent","LLMChain","AgentExecutor"],"sources":["../../../../src/agents/toolkits/openapi/openapi.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { ToolInterface } from \"@langchain/core/tools\";\nimport { DynamicTool, BaseToolkit } from \"@langchain/core/tools\";\nimport { JsonSpec } from \"../../../tools/json.js\";\nimport { AgentExecutor } from \"../../executor.js\";\nimport {\n  OPENAPI_PREFIX,\n  OPENAPI_SUFFIX,\n  JSON_EXPLORER_DESCRIPTION,\n} from \"./prompt.js\";\nimport { LLMChain } from \"../../../chains/llm_chain.js\";\nimport { ZeroShotCreatePromptArgs, ZeroShotAgent } from \"../../mrkl/index.js\";\nimport {\n  Headers,\n  RequestsGetTool,\n  RequestsPostTool,\n} from \"../../../tools/requests.js\";\nimport { createJsonAgent, JsonToolkit } from \"../json/json.js\";\n\n/**\n * Represents a toolkit for making HTTP requests. It initializes the\n * request tools based on the provided headers.\n */\nexport class RequestsToolkit extends BaseToolkit {\n  tools: ToolInterface[];\n\n  constructor(headers?: Headers) {\n    super();\n    this.tools = [new RequestsGetTool(headers), new RequestsPostTool(headers)];\n  }\n}\n\n/**\n * Extends the `RequestsToolkit` class and adds a dynamic tool for\n * exploring JSON data. It creates a JSON agent using the `JsonToolkit`\n * and the provided language model, and adds the JSON explorer tool to the\n * toolkit.\n * @example\n * ```typescript\n * const toolkit = new OpenApiToolkit(\n *   new JsonSpec({\n *   }),\n *   new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *   {\n *     \"Content-Type\": \"application/json\",\n *     Authorization: `Bearer ${process.env.OPENAI_API_KEY}`,\n *   },\n * );\n *\n * const result = await toolkit.invoke({\n *   input:\n *     \"Make a POST request to openai /completions. The prompt should be 'tell me a joke.'\",\n * });\n * console.log(`Got output ${result.output}`);\n * ```\n */\nexport class OpenApiToolkit extends RequestsToolkit {\n  constructor(\n    jsonSpec: JsonSpec,\n    llm: BaseLanguageModelInterface,\n    headers?: Headers\n  ) {\n    super(headers);\n    const jsonAgent = createJsonAgent(llm, new JsonToolkit(jsonSpec));\n    this.tools = [\n      ...this.tools,\n      new DynamicTool({\n        name: \"json_explorer\",\n        func: async (input: string) => {\n          const result = await jsonAgent.call({ input });\n          return result.output as string;\n        },\n        description: JSON_EXPLORER_DESCRIPTION,\n      }),\n    ];\n  }\n}\n\n/**\n * Creates an OpenAPI agent using a language model, an OpenAPI toolkit,\n * and optional prompt arguments. It creates a prompt for the agent using\n * the OpenAPI tools and the provided prefix and suffix. It then creates a\n * ZeroShotAgent with the prompt and the OpenAPI tools, and returns an\n * AgentExecutor for executing the agent with the tools.\n * @param llm The language model to use.\n * @param openApiToolkit The OpenAPI toolkit to use.\n * @param args Optional arguments for creating the prompt.\n * @returns An AgentExecutor for executing the agent with the tools.\n *\n * @security **Security Notice** This agent provides access to external APIs.\n * Use with caution as this agent can make API calls with arbitrary headers.\n * Exposing this agent to users could lead to security vulnerabilities. Consider\n * limiting access to what endpoints it can hit, what actions can be taken, and\n * more.\n *\n * @link See https://js.langchain.com/docs/security for more information.\n */\nexport function createOpenApiAgent(\n  llm: BaseLanguageModelInterface,\n  openApiToolkit: OpenApiToolkit,\n  args?: ZeroShotCreatePromptArgs\n) {\n  const {\n    prefix = OPENAPI_PREFIX,\n    suffix = OPENAPI_SUFFIX,\n    inputVariables = [\"input\", \"agent_scratchpad\"],\n  } = args ?? {};\n\n  const { tools } = openApiToolkit;\n  const prompt = ZeroShotAgent.createPrompt(tools, {\n    prefix,\n    suffix,\n    inputVariables,\n  });\n  const chain = new LLMChain({\n    prompt,\n    llm,\n  });\n  const toolNames = tools.map((tool) => tool.name);\n  const agent = new ZeroShotAgent({ llmChain: chain, allowedTools: toolNames });\n  return AgentExecutor.fromAgentAndTools({\n    agent,\n    tools,\n    returnIntermediateSteps: true,\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;AAuBA,IAAa,kBAAb,cAAqCA,kCAAY;CAC/C;CAEA,YAAY,SAAmB;AAC7B,SAAO;AACP,OAAK,QAAQ,CAAC,IAAIC,iCAAgB,QAAQ,EAAE,IAAIC,kCAAiB,QAAQ,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;AA4B9E,IAAa,iBAAb,cAAoC,gBAAgB;CAClD,YACE,UACA,KACA,SACA;AACA,QAAM,QAAQ;EACd,MAAM,YAAYC,6BAAgB,KAAK,IAAIC,yBAAY,SAAS,CAAC;AACjE,OAAK,QAAQ,CACX,GAAG,KAAK,OACR,IAAIC,kCAAY;GACd,MAAM;GACN,MAAM,OAAO,UAAkB;AAE7B,YADe,MAAM,UAAU,KAAK,EAAE,OAAO,CAAC,EAChC;;GAEhB,aAAaC;GACd,CAAC,CACH;;;;;;;;;;;;;;;;;;;;;;AAuBL,SAAgB,mBACd,KACA,gBACA,MACA;CACA,MAAM,EACJ,SAASC,+BACT,SAASC,+BACT,iBAAiB,CAAC,SAAS,mBAAmB,KAC5C,QAAQ,EAAE;CAEd,MAAM,EAAE,UAAU;CAWlB,MAAM,QAAQ,IAAIC,4BAAc;EAAE,UALpB,IAAIC,2BAAS;GACzB,QANaD,4BAAc,aAAa,OAAO;IAC/C;IACA;IACA;IACD,CAAC;GAGA;GACD,CAAC;EAEiD,cADjC,MAAM,KAAK,SAAS,KAAK,KAAK;EAC4B,CAAC;AAC7E,QAAOE,+BAAc,kBAAkB;EACrC;EACA;EACA,yBAAyB;EAC1B,CAAC"}
+{"version":3,"file":"openapi.cjs","names":["BaseToolkit","RequestsGetTool","RequestsPostTool","createJsonAgent","JsonToolkit","DynamicTool","JSON_EXPLORER_DESCRIPTION","OPENAPI_PREFIX","OPENAPI_SUFFIX","ZeroShotAgent","LLMChain","AgentExecutor"],"sources":["../../../../src/agents/toolkits/openapi/openapi.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { ToolInterface } from \"@langchain/core/tools\";\nimport { DynamicTool, BaseToolkit } from \"@langchain/core/tools\";\nimport { JsonSpec } from \"../../../tools/json.js\";\nimport { AgentExecutor } from \"../../executor.js\";\nimport {\n  OPENAPI_PREFIX,\n  OPENAPI_SUFFIX,\n  JSON_EXPLORER_DESCRIPTION,\n} from \"./prompt.js\";\nimport { LLMChain } from \"../../../chains/llm_chain.js\";\nimport { ZeroShotCreatePromptArgs, ZeroShotAgent } from \"../../mrkl/index.js\";\nimport {\n  Headers,\n  RequestsGetTool,\n  RequestsPostTool,\n} from \"../../../tools/requests.js\";\nimport { createJsonAgent, JsonToolkit } from \"../json/json.js\";\n\n/**\n * Represents a toolkit for making HTTP requests. It initializes the\n * request tools based on the provided headers.\n */\nexport class RequestsToolkit extends BaseToolkit {\n  tools: ToolInterface[];\n\n  constructor(headers?: Headers) {\n    super();\n    this.tools = [new RequestsGetTool(headers), new RequestsPostTool(headers)];\n  }\n}\n\n/**\n * Extends the `RequestsToolkit` class and adds a dynamic tool for\n * exploring JSON data. It creates a JSON agent using the `JsonToolkit`\n * and the provided language model, and adds the JSON explorer tool to the\n * toolkit.\n * @example\n * ```typescript\n * const toolkit = new OpenApiToolkit(\n *   new JsonSpec({\n *   }),\n *   new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *   {\n *     \"Content-Type\": \"application/json\",\n *     Authorization: `Bearer ${process.env.OPENAI_API_KEY}`,\n *   },\n * );\n *\n * const result = await toolkit.invoke({\n *   input:\n *     \"Make a POST request to openai /completions. The prompt should be 'tell me a joke.'\",\n * });\n * console.log(`Got output ${result.output}`);\n * ```\n */\nexport class OpenApiToolkit extends RequestsToolkit {\n  constructor(\n    jsonSpec: JsonSpec,\n    llm: BaseLanguageModelInterface,\n    headers?: Headers\n  ) {\n    super(headers);\n    const jsonAgent = createJsonAgent(llm, new JsonToolkit(jsonSpec));\n    this.tools = [\n      ...this.tools,\n      new DynamicTool({\n        name: \"json_explorer\",\n        func: async (input: string) => {\n          const result = await jsonAgent.call({ input });\n          return result.output as string;\n        },\n        description: JSON_EXPLORER_DESCRIPTION,\n      }),\n    ];\n  }\n}\n\n/**\n * Creates an OpenAPI agent using a language model, an OpenAPI toolkit,\n * and optional prompt arguments. It creates a prompt for the agent using\n * the OpenAPI tools and the provided prefix and suffix. It then creates a\n * ZeroShotAgent with the prompt and the OpenAPI tools, and returns an\n * AgentExecutor for executing the agent with the tools.\n * @param llm The language model to use.\n * @param openApiToolkit The OpenAPI toolkit to use.\n * @param args Optional arguments for creating the prompt.\n * @returns An AgentExecutor for executing the agent with the tools.\n *\n * @security **Security Notice** This agent provides access to external APIs.\n * Use with caution as this agent can make API calls with arbitrary headers.\n * Exposing this agent to users could lead to security vulnerabilities. Consider\n * limiting access to what endpoints it can hit, what actions can be taken, and\n * more.\n *\n * @link See https://js.langchain.com/docs/security for more information.\n */\nexport function createOpenApiAgent(\n  llm: BaseLanguageModelInterface,\n  openApiToolkit: OpenApiToolkit,\n  args?: ZeroShotCreatePromptArgs\n) {\n  const {\n    prefix = OPENAPI_PREFIX,\n    suffix = OPENAPI_SUFFIX,\n    inputVariables = [\"input\", \"agent_scratchpad\"],\n  } = args ?? {};\n\n  const { tools } = openApiToolkit;\n  const prompt = ZeroShotAgent.createPrompt(tools, {\n    prefix,\n    suffix,\n    inputVariables,\n  });\n  const chain = new LLMChain({\n    prompt,\n    llm,\n  });\n  const toolNames = tools.map((tool) => tool.name);\n  const agent = new ZeroShotAgent({ llmChain: chain, allowedTools: toolNames });\n  return AgentExecutor.fromAgentAndTools({\n    agent,\n    tools,\n    returnIntermediateSteps: true,\n  });\n}\n"],"mappings":";;;;;;;;;;;;;AAuBA,IAAa,kBAAb,cAAqCA,sBAAAA,YAAY;CAC/C;CAEA,YAAY,SAAmB;AAC7B,SAAO;AACP,OAAK,QAAQ,CAAC,IAAIC,iBAAAA,gBAAgB,QAAQ,EAAE,IAAIC,iBAAAA,iBAAiB,QAAQ,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;AA4B9E,IAAa,iBAAb,cAAoC,gBAAgB;CAClD,YACE,UACA,KACA,SACA;AACA,QAAM,QAAQ;EACd,MAAM,YAAYC,aAAAA,gBAAgB,KAAK,IAAIC,aAAAA,YAAY,SAAS,CAAC;AACjE,OAAK,QAAQ,CACX,GAAG,KAAK,OACR,IAAIC,sBAAAA,YAAY;GACd,MAAM;GACN,MAAM,OAAO,UAAkB;AAE7B,YADe,MAAM,UAAU,KAAK,EAAE,OAAO,CAAC,EAChC;;GAEhB,aAAaC,eAAAA;GACd,CAAC,CACH;;;;;;;;;;;;;;;;;;;;;;AAuBL,SAAgB,mBACd,KACA,gBACA,MACA;CACA,MAAM,EACJ,SAASC,eAAAA,gBACT,SAASC,eAAAA,gBACT,iBAAiB,CAAC,SAAS,mBAAmB,KAC5C,QAAQ,EAAE;CAEd,MAAM,EAAE,UAAU;CAWlB,MAAM,QAAQ,IAAIC,cAAAA,cAAc;EAAE,UALpB,IAAIC,kBAAAA,SAAS;GACzB,QANaD,cAAAA,cAAc,aAAa,OAAO;IAC/C;IACA;IACA;IACD,CAAC;GAGA;GACD,CAAC;EAEiD,cADjC,MAAM,KAAK,SAAS,KAAK,KAAK;EAC4B,CAAC;AAC7E,QAAOE,iBAAAA,cAAc,kBAAkB;EACrC;EACA;EACA,yBAAyB;EAC1B,CAAC"}

package/dist/agents/toolkits/openapi/openapi.js

@@ -5,7 +5,6 @@ import { JsonToolkit, createJsonAgent } from "../json/json.js";
 import { JSON_EXPLORER_DESCRIPTION, OPENAPI_PREFIX, OPENAPI_SUFFIX } from "./prompt.js";
 import { RequestsGetTool, RequestsPostTool } from "../../../tools/requests.js";
 import { BaseToolkit, DynamicTool } from "@langchain/core/tools";
-
 //#region src/agents/toolkits/openapi/openapi.ts
 /**
 * Represents a toolkit for making HTTP requests. It initializes the
@@ -94,7 +93,7 @@ function createOpenApiAgent(llm, openApiToolkit, args) {
 		returnIntermediateSteps: true
 	});
 }
-
 //#endregion
 export { OpenApiToolkit, RequestsToolkit, createOpenApiAgent };
+
 //# sourceMappingURL=openapi.js.map

package/dist/agents/toolkits/openapi/openapi.js.map

@@ -1 +1 @@
-{"version":3,"file":"openapi.js","names":[],"sources":["../../../../src/agents/toolkits/openapi/openapi.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { ToolInterface } from \"@langchain/core/tools\";\nimport { DynamicTool, BaseToolkit } from \"@langchain/core/tools\";\nimport { JsonSpec } from \"../../../tools/json.js\";\nimport { AgentExecutor } from \"../../executor.js\";\nimport {\n  OPENAPI_PREFIX,\n  OPENAPI_SUFFIX,\n  JSON_EXPLORER_DESCRIPTION,\n} from \"./prompt.js\";\nimport { LLMChain } from \"../../../chains/llm_chain.js\";\nimport { ZeroShotCreatePromptArgs, ZeroShotAgent } from \"../../mrkl/index.js\";\nimport {\n  Headers,\n  RequestsGetTool,\n  RequestsPostTool,\n} from \"../../../tools/requests.js\";\nimport { createJsonAgent, JsonToolkit } from \"../json/json.js\";\n\n/**\n * Represents a toolkit for making HTTP requests. It initializes the\n * request tools based on the provided headers.\n */\nexport class RequestsToolkit extends BaseToolkit {\n  tools: ToolInterface[];\n\n  constructor(headers?: Headers) {\n    super();\n    this.tools = [new RequestsGetTool(headers), new RequestsPostTool(headers)];\n  }\n}\n\n/**\n * Extends the `RequestsToolkit` class and adds a dynamic tool for\n * exploring JSON data. It creates a JSON agent using the `JsonToolkit`\n * and the provided language model, and adds the JSON explorer tool to the\n * toolkit.\n * @example\n * ```typescript\n * const toolkit = new OpenApiToolkit(\n *   new JsonSpec({\n *   }),\n *   new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *   {\n *     \"Content-Type\": \"application/json\",\n *     Authorization: `Bearer ${process.env.OPENAI_API_KEY}`,\n *   },\n * );\n *\n * const result = await toolkit.invoke({\n *   input:\n *     \"Make a POST request to openai /completions. The prompt should be 'tell me a joke.'\",\n * });\n * console.log(`Got output ${result.output}`);\n * ```\n */\nexport class OpenApiToolkit extends RequestsToolkit {\n  constructor(\n    jsonSpec: JsonSpec,\n    llm: BaseLanguageModelInterface,\n    headers?: Headers\n  ) {\n    super(headers);\n    const jsonAgent = createJsonAgent(llm, new JsonToolkit(jsonSpec));\n    this.tools = [\n      ...this.tools,\n      new DynamicTool({\n        name: \"json_explorer\",\n        func: async (input: string) => {\n          const result = await jsonAgent.call({ input });\n          return result.output as string;\n        },\n        description: JSON_EXPLORER_DESCRIPTION,\n      }),\n    ];\n  }\n}\n\n/**\n * Creates an OpenAPI agent using a language model, an OpenAPI toolkit,\n * and optional prompt arguments. It creates a prompt for the agent using\n * the OpenAPI tools and the provided prefix and suffix. It then creates a\n * ZeroShotAgent with the prompt and the OpenAPI tools, and returns an\n * AgentExecutor for executing the agent with the tools.\n * @param llm The language model to use.\n * @param openApiToolkit The OpenAPI toolkit to use.\n * @param args Optional arguments for creating the prompt.\n * @returns An AgentExecutor for executing the agent with the tools.\n *\n * @security **Security Notice** This agent provides access to external APIs.\n * Use with caution as this agent can make API calls with arbitrary headers.\n * Exposing this agent to users could lead to security vulnerabilities. Consider\n * limiting access to what endpoints it can hit, what actions can be taken, and\n * more.\n *\n * @link See https://js.langchain.com/docs/security for more information.\n */\nexport function createOpenApiAgent(\n  llm: BaseLanguageModelInterface,\n  openApiToolkit: OpenApiToolkit,\n  args?: ZeroShotCreatePromptArgs\n) {\n  const {\n    prefix = OPENAPI_PREFIX,\n    suffix = OPENAPI_SUFFIX,\n    inputVariables = [\"input\", \"agent_scratchpad\"],\n  } = args ?? {};\n\n  const { tools } = openApiToolkit;\n  const prompt = ZeroShotAgent.createPrompt(tools, {\n    prefix,\n    suffix,\n    inputVariables,\n  });\n  const chain = new LLMChain({\n    prompt,\n    llm,\n  });\n  const toolNames = tools.map((tool) => tool.name);\n  const agent = new ZeroShotAgent({ llmChain: chain, allowedTools: toolNames });\n  return AgentExecutor.fromAgentAndTools({\n    agent,\n    tools,\n    returnIntermediateSteps: true,\n  });\n}\n"],"mappings":";;;;;;;;;;;;;AAuBA,IAAa,kBAAb,cAAqC,YAAY;CAC/C;CAEA,YAAY,SAAmB;AAC7B,SAAO;AACP,OAAK,QAAQ,CAAC,IAAI,gBAAgB,QAAQ,EAAE,IAAI,iBAAiB,QAAQ,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;AA4B9E,IAAa,iBAAb,cAAoC,gBAAgB;CAClD,YACE,UACA,KACA,SACA;AACA,QAAM,QAAQ;EACd,MAAM,YAAY,gBAAgB,KAAK,IAAI,YAAY,SAAS,CAAC;AACjE,OAAK,QAAQ,CACX,GAAG,KAAK,OACR,IAAI,YAAY;GACd,MAAM;GACN,MAAM,OAAO,UAAkB;AAE7B,YADe,MAAM,UAAU,KAAK,EAAE,OAAO,CAAC,EAChC;;GAEhB,aAAa;GACd,CAAC,CACH;;;;;;;;;;;;;;;;;;;;;;AAuBL,SAAgB,mBACd,KACA,gBACA,MACA;CACA,MAAM,EACJ,SAAS,gBACT,SAAS,gBACT,iBAAiB,CAAC,SAAS,mBAAmB,KAC5C,QAAQ,EAAE;CAEd,MAAM,EAAE,UAAU;CAWlB,MAAM,QAAQ,IAAI,cAAc;EAAE,UALpB,IAAI,SAAS;GACzB,QANa,cAAc,aAAa,OAAO;IAC/C;IACA;IACA;IACD,CAAC;GAGA;GACD,CAAC;EAEiD,cADjC,MAAM,KAAK,SAAS,KAAK,KAAK;EAC4B,CAAC;AAC7E,QAAO,cAAc,kBAAkB;EACrC;EACA;EACA,yBAAyB;EAC1B,CAAC"}
+{"version":3,"file":"openapi.js","names":[],"sources":["../../../../src/agents/toolkits/openapi/openapi.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { ToolInterface } from \"@langchain/core/tools\";\nimport { DynamicTool, BaseToolkit } from \"@langchain/core/tools\";\nimport { JsonSpec } from \"../../../tools/json.js\";\nimport { AgentExecutor } from \"../../executor.js\";\nimport {\n  OPENAPI_PREFIX,\n  OPENAPI_SUFFIX,\n  JSON_EXPLORER_DESCRIPTION,\n} from \"./prompt.js\";\nimport { LLMChain } from \"../../../chains/llm_chain.js\";\nimport { ZeroShotCreatePromptArgs, ZeroShotAgent } from \"../../mrkl/index.js\";\nimport {\n  Headers,\n  RequestsGetTool,\n  RequestsPostTool,\n} from \"../../../tools/requests.js\";\nimport { createJsonAgent, JsonToolkit } from \"../json/json.js\";\n\n/**\n * Represents a toolkit for making HTTP requests. It initializes the\n * request tools based on the provided headers.\n */\nexport class RequestsToolkit extends BaseToolkit {\n  tools: ToolInterface[];\n\n  constructor(headers?: Headers) {\n    super();\n    this.tools = [new RequestsGetTool(headers), new RequestsPostTool(headers)];\n  }\n}\n\n/**\n * Extends the `RequestsToolkit` class and adds a dynamic tool for\n * exploring JSON data. It creates a JSON agent using the `JsonToolkit`\n * and the provided language model, and adds the JSON explorer tool to the\n * toolkit.\n * @example\n * ```typescript\n * const toolkit = new OpenApiToolkit(\n *   new JsonSpec({\n *   }),\n *   new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *   {\n *     \"Content-Type\": \"application/json\",\n *     Authorization: `Bearer ${process.env.OPENAI_API_KEY}`,\n *   },\n * );\n *\n * const result = await toolkit.invoke({\n *   input:\n *     \"Make a POST request to openai /completions. The prompt should be 'tell me a joke.'\",\n * });\n * console.log(`Got output ${result.output}`);\n * ```\n */\nexport class OpenApiToolkit extends RequestsToolkit {\n  constructor(\n    jsonSpec: JsonSpec,\n    llm: BaseLanguageModelInterface,\n    headers?: Headers\n  ) {\n    super(headers);\n    const jsonAgent = createJsonAgent(llm, new JsonToolkit(jsonSpec));\n    this.tools = [\n      ...this.tools,\n      new DynamicTool({\n        name: \"json_explorer\",\n        func: async (input: string) => {\n          const result = await jsonAgent.call({ input });\n          return result.output as string;\n        },\n        description: JSON_EXPLORER_DESCRIPTION,\n      }),\n    ];\n  }\n}\n\n/**\n * Creates an OpenAPI agent using a language model, an OpenAPI toolkit,\n * and optional prompt arguments. It creates a prompt for the agent using\n * the OpenAPI tools and the provided prefix and suffix. It then creates a\n * ZeroShotAgent with the prompt and the OpenAPI tools, and returns an\n * AgentExecutor for executing the agent with the tools.\n * @param llm The language model to use.\n * @param openApiToolkit The OpenAPI toolkit to use.\n * @param args Optional arguments for creating the prompt.\n * @returns An AgentExecutor for executing the agent with the tools.\n *\n * @security **Security Notice** This agent provides access to external APIs.\n * Use with caution as this agent can make API calls with arbitrary headers.\n * Exposing this agent to users could lead to security vulnerabilities. Consider\n * limiting access to what endpoints it can hit, what actions can be taken, and\n * more.\n *\n * @link See https://js.langchain.com/docs/security for more information.\n */\nexport function createOpenApiAgent(\n  llm: BaseLanguageModelInterface,\n  openApiToolkit: OpenApiToolkit,\n  args?: ZeroShotCreatePromptArgs\n) {\n  const {\n    prefix = OPENAPI_PREFIX,\n    suffix = OPENAPI_SUFFIX,\n    inputVariables = [\"input\", \"agent_scratchpad\"],\n  } = args ?? {};\n\n  const { tools } = openApiToolkit;\n  const prompt = ZeroShotAgent.createPrompt(tools, {\n    prefix,\n    suffix,\n    inputVariables,\n  });\n  const chain = new LLMChain({\n    prompt,\n    llm,\n  });\n  const toolNames = tools.map((tool) => tool.name);\n  const agent = new ZeroShotAgent({ llmChain: chain, allowedTools: toolNames });\n  return AgentExecutor.fromAgentAndTools({\n    agent,\n    tools,\n    returnIntermediateSteps: true,\n  });\n}\n"],"mappings":";;;;;;;;;;;;AAuBA,IAAa,kBAAb,cAAqC,YAAY;CAC/C;CAEA,YAAY,SAAmB;AAC7B,SAAO;AACP,OAAK,QAAQ,CAAC,IAAI,gBAAgB,QAAQ,EAAE,IAAI,iBAAiB,QAAQ,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;AA4B9E,IAAa,iBAAb,cAAoC,gBAAgB;CAClD,YACE,UACA,KACA,SACA;AACA,QAAM,QAAQ;EACd,MAAM,YAAY,gBAAgB,KAAK,IAAI,YAAY,SAAS,CAAC;AACjE,OAAK,QAAQ,CACX,GAAG,KAAK,OACR,IAAI,YAAY;GACd,MAAM;GACN,MAAM,OAAO,UAAkB;AAE7B,YADe,MAAM,UAAU,KAAK,EAAE,OAAO,CAAC,EAChC;;GAEhB,aAAa;GACd,CAAC,CACH;;;;;;;;;;;;;;;;;;;;;;AAuBL,SAAgB,mBACd,KACA,gBACA,MACA;CACA,MAAM,EACJ,SAAS,gBACT,SAAS,gBACT,iBAAiB,CAAC,SAAS,mBAAmB,KAC5C,QAAQ,EAAE;CAEd,MAAM,EAAE,UAAU;CAWlB,MAAM,QAAQ,IAAI,cAAc;EAAE,UALpB,IAAI,SAAS;GACzB,QANa,cAAc,aAAa,OAAO;IAC/C;IACA;IACA;IACD,CAAC;GAGA;GACD,CAAC;EAEiD,cADjC,MAAM,KAAK,SAAS,KAAK,KAAK;EAC4B,CAAC;AAC7E,QAAO,cAAc,kBAAkB;EACrC;EACA;EACA,yBAAyB;EAC1B,CAAC"}

package/dist/agents/toolkits/openapi/prompt.cjs

@@ -1,4 +1,3 @@
-
 //#region src/agents/toolkits/openapi/prompt.ts
 const OPENAPI_PREFIX = `You are an agent designed to answer questions by making web requests to an API given the OpenAPI spec.
 
@@ -29,9 +28,9 @@ Example inputs to this tool:
     'What are the required query parameters for a GET request to the /bar endpoint?'
     'What are the required parameters in the request body for a POST request to the /foo endpoint?'
 Always give this tool a specific question.`;
-
 //#endregion
 exports.JSON_EXPLORER_DESCRIPTION = JSON_EXPLORER_DESCRIPTION;
 exports.OPENAPI_PREFIX = OPENAPI_PREFIX;
 exports.OPENAPI_SUFFIX = OPENAPI_SUFFIX;
+
 //# sourceMappingURL=prompt.cjs.map

package/dist/agents/toolkits/openapi/prompt.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"prompt.cjs","names":[],"sources":["../../../../src/agents/toolkits/openapi/prompt.ts"],"sourcesContent":["export const OPENAPI_PREFIX = `You are an agent designed to answer questions by making web requests to an API given the OpenAPI spec.\n\nIf the question does not seem related to the API, return I don't know. Do not make up an answer.\nOnly use information provided by the tools to construct your response.\n\nTo find information in the OpenAPI spec, use the 'json_explorer' tool. The input to this tool is a question about the API.\n\nTake the following steps:\nFirst, find the base URL needed to make the request.\n\nSecond, find the relevant paths needed to answer the question. Take note that, sometimes, you might need to make more than one request to more than one path to answer the question.\n\nThird, find the required parameters needed to make the request. For GET requests, these are usually URL parameters and for POST requests, these are request body parameters.\n\nFourth, make the requests needed to answer the question. Ensure that you are sending the correct parameters to the request by checking which parameters are required. For parameters with a fixed set of values, please use the spec to look at which values are allowed.\n\nUse the exact parameter names as listed in the spec, do not make up any names or abbreviate the names of parameters.\nIf you get a not found error, ensure that you are using a path that actually exists in the spec.`;\n\nexport const OPENAPI_SUFFIX = `Begin!\"\n\nQuestion: {input}\nThought: I should explore the spec to find the base url for the API.\n{agent_scratchpad}`;\nexport const JSON_EXPLORER_DESCRIPTION = `\nCan be used to answer questions about the openapi spec for the API. Always use this tool before trying to make a request. \nExample inputs to this tool: \n    'What are the required query parameters for a GET request to the /bar endpoint?'\n    'What are the required parameters in the request body for a POST request to the /foo endpoint?'\nAlways give this tool a specific question.`;\n"],"mappings":";;AAAA,MAAa,iBAAiB;;;;;;;;;;;;;;;;;;AAmB9B,MAAa,iBAAiB;;;;;AAK9B,MAAa,4BAA4B"}
+{"version":3,"file":"prompt.cjs","names":[],"sources":["../../../../src/agents/toolkits/openapi/prompt.ts"],"sourcesContent":["export const OPENAPI_PREFIX = `You are an agent designed to answer questions by making web requests to an API given the OpenAPI spec.\n\nIf the question does not seem related to the API, return I don't know. Do not make up an answer.\nOnly use information provided by the tools to construct your response.\n\nTo find information in the OpenAPI spec, use the 'json_explorer' tool. The input to this tool is a question about the API.\n\nTake the following steps:\nFirst, find the base URL needed to make the request.\n\nSecond, find the relevant paths needed to answer the question. Take note that, sometimes, you might need to make more than one request to more than one path to answer the question.\n\nThird, find the required parameters needed to make the request. For GET requests, these are usually URL parameters and for POST requests, these are request body parameters.\n\nFourth, make the requests needed to answer the question. Ensure that you are sending the correct parameters to the request by checking which parameters are required. For parameters with a fixed set of values, please use the spec to look at which values are allowed.\n\nUse the exact parameter names as listed in the spec, do not make up any names or abbreviate the names of parameters.\nIf you get a not found error, ensure that you are using a path that actually exists in the spec.`;\n\nexport const OPENAPI_SUFFIX = `Begin!\"\n\nQuestion: {input}\nThought: I should explore the spec to find the base url for the API.\n{agent_scratchpad}`;\nexport const JSON_EXPLORER_DESCRIPTION = `\nCan be used to answer questions about the openapi spec for the API. Always use this tool before trying to make a request. \nExample inputs to this tool: \n    'What are the required query parameters for a GET request to the /bar endpoint?'\n    'What are the required parameters in the request body for a POST request to the /foo endpoint?'\nAlways give this tool a specific question.`;\n"],"mappings":";AAAA,MAAa,iBAAiB;;;;;;;;;;;;;;;;;;AAmB9B,MAAa,iBAAiB;;;;;AAK9B,MAAa,4BAA4B"}

package/dist/agents/toolkits/openapi/prompt.js

@@ -28,7 +28,7 @@ Example inputs to this tool:
     'What are the required query parameters for a GET request to the /bar endpoint?'
     'What are the required parameters in the request body for a POST request to the /foo endpoint?'
 Always give this tool a specific question.`;
-
 //#endregion
 export { JSON_EXPLORER_DESCRIPTION, OPENAPI_PREFIX, OPENAPI_SUFFIX };
+
 //# sourceMappingURL=prompt.js.map

package/dist/agents/toolkits/sql/index.cjs

@@ -1,8 +1,7 @@
-Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
-const require_runtime = require('../../../_virtual/_rolldown/runtime.cjs');
-const require_prompt = require('./prompt.cjs');
-const require_sql = require('./sql.cjs');
-
+Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
+const require_runtime = require("../../../_virtual/_rolldown/runtime.cjs");
+const require_prompt = require("./prompt.cjs");
+const require_sql = require("./sql.cjs");
 //#region src/agents/toolkits/sql/index.ts
 var sql_exports = /* @__PURE__ */ require_runtime.__exportAll({
 	SQL_PREFIX: () => require_prompt.SQL_PREFIX,
@@ -10,16 +9,16 @@ var sql_exports = /* @__PURE__ */ require_runtime.__exportAll({
 	SqlToolkit: () => require_sql.SqlToolkit,
 	createSqlAgent: () => require_sql.createSqlAgent
 });
-
 //#endregion
 exports.SQL_PREFIX = require_prompt.SQL_PREFIX;
 exports.SQL_SUFFIX = require_prompt.SQL_SUFFIX;
 exports.SqlToolkit = require_sql.SqlToolkit;
 exports.createSqlAgent = require_sql.createSqlAgent;
-Object.defineProperty(exports, 'sql_exports', {
-  enumerable: true,
-  get: function () {
-    return sql_exports;
-  }
+Object.defineProperty(exports, "sql_exports", {
+	enumerable: true,
+	get: function() {
+		return sql_exports;
+	}
 });
+
 //# sourceMappingURL=index.cjs.map

package/dist/agents/toolkits/sql/index.js

@@ -1,7 +1,6 @@
 import { __exportAll } from "../../../_virtual/_rolldown/runtime.js";
 import { SQL_PREFIX, SQL_SUFFIX } from "./prompt.js";
 import { SqlToolkit, createSqlAgent } from "./sql.js";
-
 //#region src/agents/toolkits/sql/index.ts
 var sql_exports = /* @__PURE__ */ __exportAll({
 	SQL_PREFIX: () => SQL_PREFIX,
@@ -9,7 +8,7 @@ var sql_exports = /* @__PURE__ */ __exportAll({
 	SqlToolkit: () => SqlToolkit,
 	createSqlAgent: () => createSqlAgent
 });
-
 //#endregion
 export { SQL_PREFIX, SQL_SUFFIX, SqlToolkit, createSqlAgent, sql_exports };
+
 //# sourceMappingURL=index.js.map

package/dist/agents/toolkits/sql/prompt.cjs

@@ -1,4 +1,3 @@
-
 //#region src/agents/toolkits/sql/prompt.ts
 const SQL_PREFIX = `You are an agent designed to interact with a SQL database.
 Given an input question, create a syntactically correct {dialect} query to run, then look at the results of the query and return the answer.
@@ -17,8 +16,8 @@ const SQL_SUFFIX = `Begin!
 Question: {input}
 Thought: I should look at the tables in the database to see what I can query.
 {agent_scratchpad}`;
-
 //#endregion
 exports.SQL_PREFIX = SQL_PREFIX;
 exports.SQL_SUFFIX = SQL_SUFFIX;
+
 //# sourceMappingURL=prompt.cjs.map

package/dist/agents/toolkits/sql/prompt.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"prompt.cjs","names":[],"sources":["../../../../src/agents/toolkits/sql/prompt.ts"],"sourcesContent":["export const SQL_PREFIX = `You are an agent designed to interact with a SQL database.\nGiven an input question, create a syntactically correct {dialect} query to run, then look at the results of the query and return the answer.\nUnless the user specifies a specific number of examples they wish to obtain, always limit your query to at most {top_k} results using the LIMIT clause.\nYou can order the results by a relevant column to return the most interesting examples in the database.\nNever query for all the columns from a specific table, only ask for a the few relevant columns given the question.\nYou have access to tools for interacting with the database.\nOnly use the below tools. Only use the information returned by the below tools to construct your final answer.\nYou MUST double check your query before executing it. If you get an error while executing a query, rewrite the query and try again.\n\nDO NOT make any DML statements (INSERT, UPDATE, DELETE, DROP etc.) to the database.\n\nIf the question does not seem related to the database, just return \"I don't know\" as the answer.`;\n\nexport const SQL_SUFFIX = `Begin!\n\nQuestion: {input}\nThought: I should look at the tables in the database to see what I can query.\n{agent_scratchpad}`;\n"],"mappings":";;AAAA,MAAa,aAAa;;;;;;;;;;;;AAa1B,MAAa,aAAa"}
+{"version":3,"file":"prompt.cjs","names":[],"sources":["../../../../src/agents/toolkits/sql/prompt.ts"],"sourcesContent":["export const SQL_PREFIX = `You are an agent designed to interact with a SQL database.\nGiven an input question, create a syntactically correct {dialect} query to run, then look at the results of the query and return the answer.\nUnless the user specifies a specific number of examples they wish to obtain, always limit your query to at most {top_k} results using the LIMIT clause.\nYou can order the results by a relevant column to return the most interesting examples in the database.\nNever query for all the columns from a specific table, only ask for a the few relevant columns given the question.\nYou have access to tools for interacting with the database.\nOnly use the below tools. Only use the information returned by the below tools to construct your final answer.\nYou MUST double check your query before executing it. If you get an error while executing a query, rewrite the query and try again.\n\nDO NOT make any DML statements (INSERT, UPDATE, DELETE, DROP etc.) to the database.\n\nIf the question does not seem related to the database, just return \"I don't know\" as the answer.`;\n\nexport const SQL_SUFFIX = `Begin!\n\nQuestion: {input}\nThought: I should look at the tables in the database to see what I can query.\n{agent_scratchpad}`;\n"],"mappings":";AAAA,MAAa,aAAa;;;;;;;;;;;;AAa1B,MAAa,aAAa"}

package/dist/agents/toolkits/sql/prompt.js

@@ -16,7 +16,7 @@ const SQL_SUFFIX = `Begin!
 Question: {input}
 Thought: I should look at the tables in the database to see what I can query.
 {agent_scratchpad}`;
-
 //#endregion
 export { SQL_PREFIX, SQL_SUFFIX };
+
 //# sourceMappingURL=prompt.js.map

package/dist/agents/toolkits/sql/sql.cjs

@@ -1,12 +1,11 @@
-const require_runtime = require('../../../_virtual/_rolldown/runtime.cjs');
-const require_llm_chain = require('../../../chains/llm_chain.cjs');
-const require_index = require('../../mrkl/index.cjs');
-const require_executor = require('../../executor.cjs');
-const require_tools_sql = require('../../../tools/sql.cjs');
-const require_prompt = require('./prompt.cjs');
+require("../../../_virtual/_rolldown/runtime.cjs");
+const require_llm_chain = require("../../../chains/llm_chain.cjs");
+const require_index = require("../../mrkl/index.cjs");
+const require_executor = require("../../executor.cjs");
+const require_tools_sql = require("../../../tools/sql.cjs");
+const require_prompt = require("./prompt.cjs");
 let _langchain_core_prompts = require("@langchain/core/prompts");
 let _langchain_core_tools = require("@langchain/core/tools");
-
 //#region src/agents/toolkits/sql/sql.ts
 /**
 * Class that represents a toolkit for working with SQL databases. It
@@ -59,8 +58,8 @@ function createSqlAgent(llm, toolkit, args) {
 		returnIntermediateSteps: true
 	});
 }
-
 //#endregion
 exports.SqlToolkit = SqlToolkit;
 exports.createSqlAgent = createSqlAgent;
+
 //# sourceMappingURL=sql.cjs.map

package/dist/agents/toolkits/sql/sql.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"sql.cjs","names":["BaseToolkit","QuerySqlTool","InfoSqlTool","ListTablesSqlTool","QueryCheckerTool","SQL_PREFIX","SQL_SUFFIX","ZeroShotAgent","LLMChain","AgentExecutor"],"sources":["../../../../src/agents/toolkits/sql/sql.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { type ToolInterface, BaseToolkit } from \"@langchain/core/tools\";\nimport { renderTemplate } from \"@langchain/core/prompts\";\nimport {\n  InfoSqlTool,\n  ListTablesSqlTool,\n  QueryCheckerTool,\n  QuerySqlTool,\n} from \"../../../tools/sql.js\";\nimport { SQL_PREFIX, SQL_SUFFIX } from \"./prompt.js\";\nimport { LLMChain } from \"../../../chains/llm_chain.js\";\nimport { ZeroShotAgent, ZeroShotCreatePromptArgs } from \"../../mrkl/index.js\";\nimport { AgentExecutor } from \"../../executor.js\";\nimport { SqlDatabase } from \"../../../sql_db.js\";\n\n/**\n * Interface that extends ZeroShotCreatePromptArgs and adds an optional\n * topK parameter for specifying the number of results to return.\n */\nexport interface SqlCreatePromptArgs extends ZeroShotCreatePromptArgs {\n  /** Number of results to return. */\n  topK?: number;\n}\n\n/**\n * Class that represents a toolkit for working with SQL databases. It\n * initializes SQL tools based on the provided SQL database.\n * @example\n * ```typescript\n * const model = new ChatOpenAI({ model: \"gpt-4o-mini\" });\n * const toolkit = new SqlToolkit(sqlDb, model);\n * const executor = createSqlAgent(model, toolkit);\n * const result = await executor.invoke({ input: 'List the total sales per country. Which country's customers spent the most?' });\n * console.log(`Got output ${result.output}`);\n * ```\n */\nexport class SqlToolkit extends BaseToolkit {\n  tools: ToolInterface[];\n\n  db: SqlDatabase;\n\n  dialect = \"sqlite\";\n\n  constructor(db: SqlDatabase, llm: BaseLanguageModelInterface) {\n    super();\n    this.db = db;\n    this.tools = [\n      new QuerySqlTool(db),\n      new InfoSqlTool(db),\n      new ListTablesSqlTool(db),\n      new QueryCheckerTool({ llm }),\n    ];\n  }\n}\n\nexport function createSqlAgent(\n  llm: BaseLanguageModelInterface,\n  toolkit: SqlToolkit,\n  args?: SqlCreatePromptArgs\n) {\n  const {\n    prefix = SQL_PREFIX,\n    suffix = SQL_SUFFIX,\n    inputVariables = [\"input\", \"agent_scratchpad\"],\n    topK = 10,\n  } = args ?? {};\n  const { tools } = toolkit;\n  const formattedPrefix = renderTemplate(prefix, \"f-string\", {\n    dialect: toolkit.dialect,\n    top_k: topK,\n  });\n\n  const prompt = ZeroShotAgent.createPrompt(tools, {\n    prefix: formattedPrefix,\n    suffix,\n    inputVariables,\n  });\n  const chain = new LLMChain({ prompt, llm });\n  const agent = new ZeroShotAgent({\n    llmChain: chain,\n    allowedTools: tools.map((t) => t.name),\n  });\n  return AgentExecutor.fromAgentAndTools({\n    agent,\n    tools,\n    returnIntermediateSteps: true,\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AAoCA,IAAa,aAAb,cAAgCA,kCAAY;CAC1C;CAEA;CAEA,UAAU;CAEV,YAAY,IAAiB,KAAiC;AAC5D,SAAO;AACP,OAAK,KAAK;AACV,OAAK,QAAQ;GACX,IAAIC,+BAAa,GAAG;GACpB,IAAIC,8BAAY,GAAG;GACnB,IAAIC,oCAAkB,GAAG;GACzB,IAAIC,mCAAiB,EAAE,KAAK,CAAC;GAC9B;;;AAIL,SAAgB,eACd,KACA,SACA,MACA;CACA,MAAM,EACJ,SAASC,2BACT,SAASC,2BACT,iBAAiB,CAAC,SAAS,mBAAmB,EAC9C,OAAO,OACL,QAAQ,EAAE;CACd,MAAM,EAAE,UAAU;CAClB,MAAM,8DAAiC,QAAQ,YAAY;EACzD,SAAS,QAAQ;EACjB,OAAO;EACR,CAAC;CAQF,MAAM,QAAQ,IAAIC,4BAAc;EAC9B,UAFY,IAAIC,2BAAS;GAAE,QALdD,4BAAc,aAAa,OAAO;IAC/C,QAAQ;IACR;IACA;IACD,CAAC;GACmC;GAAK,CAAC;EAGzC,cAAc,MAAM,KAAK,MAAM,EAAE,KAAK;EACvC,CAAC;AACF,QAAOE,+BAAc,kBAAkB;EACrC;EACA;EACA,yBAAyB;EAC1B,CAAC"}
+{"version":3,"file":"sql.cjs","names":["BaseToolkit","QuerySqlTool","InfoSqlTool","ListTablesSqlTool","QueryCheckerTool","SQL_PREFIX","SQL_SUFFIX","ZeroShotAgent","LLMChain","AgentExecutor"],"sources":["../../../../src/agents/toolkits/sql/sql.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { type ToolInterface, BaseToolkit } from \"@langchain/core/tools\";\nimport { renderTemplate } from \"@langchain/core/prompts\";\nimport {\n  InfoSqlTool,\n  ListTablesSqlTool,\n  QueryCheckerTool,\n  QuerySqlTool,\n} from \"../../../tools/sql.js\";\nimport { SQL_PREFIX, SQL_SUFFIX } from \"./prompt.js\";\nimport { LLMChain } from \"../../../chains/llm_chain.js\";\nimport { ZeroShotAgent, ZeroShotCreatePromptArgs } from \"../../mrkl/index.js\";\nimport { AgentExecutor } from \"../../executor.js\";\nimport { SqlDatabase } from \"../../../sql_db.js\";\n\n/**\n * Interface that extends ZeroShotCreatePromptArgs and adds an optional\n * topK parameter for specifying the number of results to return.\n */\nexport interface SqlCreatePromptArgs extends ZeroShotCreatePromptArgs {\n  /** Number of results to return. */\n  topK?: number;\n}\n\n/**\n * Class that represents a toolkit for working with SQL databases. It\n * initializes SQL tools based on the provided SQL database.\n * @example\n * ```typescript\n * const model = new ChatOpenAI({ model: \"gpt-4o-mini\" });\n * const toolkit = new SqlToolkit(sqlDb, model);\n * const executor = createSqlAgent(model, toolkit);\n * const result = await executor.invoke({ input: 'List the total sales per country. Which country's customers spent the most?' });\n * console.log(`Got output ${result.output}`);\n * ```\n */\nexport class SqlToolkit extends BaseToolkit {\n  tools: ToolInterface[];\n\n  db: SqlDatabase;\n\n  dialect = \"sqlite\";\n\n  constructor(db: SqlDatabase, llm: BaseLanguageModelInterface) {\n    super();\n    this.db = db;\n    this.tools = [\n      new QuerySqlTool(db),\n      new InfoSqlTool(db),\n      new ListTablesSqlTool(db),\n      new QueryCheckerTool({ llm }),\n    ];\n  }\n}\n\nexport function createSqlAgent(\n  llm: BaseLanguageModelInterface,\n  toolkit: SqlToolkit,\n  args?: SqlCreatePromptArgs\n) {\n  const {\n    prefix = SQL_PREFIX,\n    suffix = SQL_SUFFIX,\n    inputVariables = [\"input\", \"agent_scratchpad\"],\n    topK = 10,\n  } = args ?? {};\n  const { tools } = toolkit;\n  const formattedPrefix = renderTemplate(prefix, \"f-string\", {\n    dialect: toolkit.dialect,\n    top_k: topK,\n  });\n\n  const prompt = ZeroShotAgent.createPrompt(tools, {\n    prefix: formattedPrefix,\n    suffix,\n    inputVariables,\n  });\n  const chain = new LLMChain({ prompt, llm });\n  const agent = new ZeroShotAgent({\n    llmChain: chain,\n    allowedTools: tools.map((t) => t.name),\n  });\n  return AgentExecutor.fromAgentAndTools({\n    agent,\n    tools,\n    returnIntermediateSteps: true,\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;AAoCA,IAAa,aAAb,cAAgCA,sBAAAA,YAAY;CAC1C;CAEA;CAEA,UAAU;CAEV,YAAY,IAAiB,KAAiC;AAC5D,SAAO;AACP,OAAK,KAAK;AACV,OAAK,QAAQ;GACX,IAAIC,kBAAAA,aAAa,GAAG;GACpB,IAAIC,kBAAAA,YAAY,GAAG;GACnB,IAAIC,kBAAAA,kBAAkB,GAAG;GACzB,IAAIC,kBAAAA,iBAAiB,EAAE,KAAK,CAAC;GAC9B;;;AAIL,SAAgB,eACd,KACA,SACA,MACA;CACA,MAAM,EACJ,SAASC,eAAAA,YACT,SAASC,eAAAA,YACT,iBAAiB,CAAC,SAAS,mBAAmB,EAC9C,OAAO,OACL,QAAQ,EAAE;CACd,MAAM,EAAE,UAAU;CAClB,MAAM,mBAAA,GAAA,wBAAA,gBAAiC,QAAQ,YAAY;EACzD,SAAS,QAAQ;EACjB,OAAO;EACR,CAAC;CAQF,MAAM,QAAQ,IAAIC,cAAAA,cAAc;EAC9B,UAFY,IAAIC,kBAAAA,SAAS;GAAE,QALdD,cAAAA,cAAc,aAAa,OAAO;IAC/C,QAAQ;IACR;IACA;IACD,CAAC;GACmC;GAAK,CAAC;EAGzC,cAAc,MAAM,KAAK,MAAM,EAAE,KAAK;EACvC,CAAC;AACF,QAAOE,iBAAAA,cAAc,kBAAkB;EACrC;EACA;EACA,yBAAyB;EAC1B,CAAC"}

package/dist/agents/toolkits/sql/sql.js

@@ -5,7 +5,6 @@ import { InfoSqlTool, ListTablesSqlTool, QueryCheckerTool, QuerySqlTool } from "
 import { SQL_PREFIX, SQL_SUFFIX } from "./prompt.js";
 import { renderTemplate } from "@langchain/core/prompts";
 import { BaseToolkit } from "@langchain/core/tools";
-
 //#region src/agents/toolkits/sql/sql.ts
 /**
 * Class that represents a toolkit for working with SQL databases. It
@@ -58,7 +57,7 @@ function createSqlAgent(llm, toolkit, args) {
 		returnIntermediateSteps: true
 	});
 }
-
 //#endregion
 export { SqlToolkit, createSqlAgent };
+
 //# sourceMappingURL=sql.js.map

package/dist/agents/toolkits/sql/sql.js.map

@@ -1 +1 @@
-{"version":3,"file":"sql.js","names":[],"sources":["../../../../src/agents/toolkits/sql/sql.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { type ToolInterface, BaseToolkit } from \"@langchain/core/tools\";\nimport { renderTemplate } from \"@langchain/core/prompts\";\nimport {\n  InfoSqlTool,\n  ListTablesSqlTool,\n  QueryCheckerTool,\n  QuerySqlTool,\n} from \"../../../tools/sql.js\";\nimport { SQL_PREFIX, SQL_SUFFIX } from \"./prompt.js\";\nimport { LLMChain } from \"../../../chains/llm_chain.js\";\nimport { ZeroShotAgent, ZeroShotCreatePromptArgs } from \"../../mrkl/index.js\";\nimport { AgentExecutor } from \"../../executor.js\";\nimport { SqlDatabase } from \"../../../sql_db.js\";\n\n/**\n * Interface that extends ZeroShotCreatePromptArgs and adds an optional\n * topK parameter for specifying the number of results to return.\n */\nexport interface SqlCreatePromptArgs extends ZeroShotCreatePromptArgs {\n  /** Number of results to return. */\n  topK?: number;\n}\n\n/**\n * Class that represents a toolkit for working with SQL databases. It\n * initializes SQL tools based on the provided SQL database.\n * @example\n * ```typescript\n * const model = new ChatOpenAI({ model: \"gpt-4o-mini\" });\n * const toolkit = new SqlToolkit(sqlDb, model);\n * const executor = createSqlAgent(model, toolkit);\n * const result = await executor.invoke({ input: 'List the total sales per country. Which country's customers spent the most?' });\n * console.log(`Got output ${result.output}`);\n * ```\n */\nexport class SqlToolkit extends BaseToolkit {\n  tools: ToolInterface[];\n\n  db: SqlDatabase;\n\n  dialect = \"sqlite\";\n\n  constructor(db: SqlDatabase, llm: BaseLanguageModelInterface) {\n    super();\n    this.db = db;\n    this.tools = [\n      new QuerySqlTool(db),\n      new InfoSqlTool(db),\n      new ListTablesSqlTool(db),\n      new QueryCheckerTool({ llm }),\n    ];\n  }\n}\n\nexport function createSqlAgent(\n  llm: BaseLanguageModelInterface,\n  toolkit: SqlToolkit,\n  args?: SqlCreatePromptArgs\n) {\n  const {\n    prefix = SQL_PREFIX,\n    suffix = SQL_SUFFIX,\n    inputVariables = [\"input\", \"agent_scratchpad\"],\n    topK = 10,\n  } = args ?? {};\n  const { tools } = toolkit;\n  const formattedPrefix = renderTemplate(prefix, \"f-string\", {\n    dialect: toolkit.dialect,\n    top_k: topK,\n  });\n\n  const prompt = ZeroShotAgent.createPrompt(tools, {\n    prefix: formattedPrefix,\n    suffix,\n    inputVariables,\n  });\n  const chain = new LLMChain({ prompt, llm });\n  const agent = new ZeroShotAgent({\n    llmChain: chain,\n    allowedTools: tools.map((t) => t.name),\n  });\n  return AgentExecutor.fromAgentAndTools({\n    agent,\n    tools,\n    returnIntermediateSteps: true,\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;AAoCA,IAAa,aAAb,cAAgC,YAAY;CAC1C;CAEA;CAEA,UAAU;CAEV,YAAY,IAAiB,KAAiC;AAC5D,SAAO;AACP,OAAK,KAAK;AACV,OAAK,QAAQ;GACX,IAAI,aAAa,GAAG;GACpB,IAAI,YAAY,GAAG;GACnB,IAAI,kBAAkB,GAAG;GACzB,IAAI,iBAAiB,EAAE,KAAK,CAAC;GAC9B;;;AAIL,SAAgB,eACd,KACA,SACA,MACA;CACA,MAAM,EACJ,SAAS,YACT,SAAS,YACT,iBAAiB,CAAC,SAAS,mBAAmB,EAC9C,OAAO,OACL,QAAQ,EAAE;CACd,MAAM,EAAE,UAAU;CAClB,MAAM,kBAAkB,eAAe,QAAQ,YAAY;EACzD,SAAS,QAAQ;EACjB,OAAO;EACR,CAAC;CAQF,MAAM,QAAQ,IAAI,cAAc;EAC9B,UAFY,IAAI,SAAS;GAAE,QALd,cAAc,aAAa,OAAO;IAC/C,QAAQ;IACR;IACA;IACD,CAAC;GACmC;GAAK,CAAC;EAGzC,cAAc,MAAM,KAAK,MAAM,EAAE,KAAK;EACvC,CAAC;AACF,QAAO,cAAc,kBAAkB;EACrC;EACA;EACA,yBAAyB;EAC1B,CAAC"}
+{"version":3,"file":"sql.js","names":[],"sources":["../../../../src/agents/toolkits/sql/sql.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { type ToolInterface, BaseToolkit } from \"@langchain/core/tools\";\nimport { renderTemplate } from \"@langchain/core/prompts\";\nimport {\n  InfoSqlTool,\n  ListTablesSqlTool,\n  QueryCheckerTool,\n  QuerySqlTool,\n} from \"../../../tools/sql.js\";\nimport { SQL_PREFIX, SQL_SUFFIX } from \"./prompt.js\";\nimport { LLMChain } from \"../../../chains/llm_chain.js\";\nimport { ZeroShotAgent, ZeroShotCreatePromptArgs } from \"../../mrkl/index.js\";\nimport { AgentExecutor } from \"../../executor.js\";\nimport { SqlDatabase } from \"../../../sql_db.js\";\n\n/**\n * Interface that extends ZeroShotCreatePromptArgs and adds an optional\n * topK parameter for specifying the number of results to return.\n */\nexport interface SqlCreatePromptArgs extends ZeroShotCreatePromptArgs {\n  /** Number of results to return. */\n  topK?: number;\n}\n\n/**\n * Class that represents a toolkit for working with SQL databases. It\n * initializes SQL tools based on the provided SQL database.\n * @example\n * ```typescript\n * const model = new ChatOpenAI({ model: \"gpt-4o-mini\" });\n * const toolkit = new SqlToolkit(sqlDb, model);\n * const executor = createSqlAgent(model, toolkit);\n * const result = await executor.invoke({ input: 'List the total sales per country. Which country's customers spent the most?' });\n * console.log(`Got output ${result.output}`);\n * ```\n */\nexport class SqlToolkit extends BaseToolkit {\n  tools: ToolInterface[];\n\n  db: SqlDatabase;\n\n  dialect = \"sqlite\";\n\n  constructor(db: SqlDatabase, llm: BaseLanguageModelInterface) {\n    super();\n    this.db = db;\n    this.tools = [\n      new QuerySqlTool(db),\n      new InfoSqlTool(db),\n      new ListTablesSqlTool(db),\n      new QueryCheckerTool({ llm }),\n    ];\n  }\n}\n\nexport function createSqlAgent(\n  llm: BaseLanguageModelInterface,\n  toolkit: SqlToolkit,\n  args?: SqlCreatePromptArgs\n) {\n  const {\n    prefix = SQL_PREFIX,\n    suffix = SQL_SUFFIX,\n    inputVariables = [\"input\", \"agent_scratchpad\"],\n    topK = 10,\n  } = args ?? {};\n  const { tools } = toolkit;\n  const formattedPrefix = renderTemplate(prefix, \"f-string\", {\n    dialect: toolkit.dialect,\n    top_k: topK,\n  });\n\n  const prompt = ZeroShotAgent.createPrompt(tools, {\n    prefix: formattedPrefix,\n    suffix,\n    inputVariables,\n  });\n  const chain = new LLMChain({ prompt, llm });\n  const agent = new ZeroShotAgent({\n    llmChain: chain,\n    allowedTools: tools.map((t) => t.name),\n  });\n  return AgentExecutor.fromAgentAndTools({\n    agent,\n    tools,\n    returnIntermediateSteps: true,\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAoCA,IAAa,aAAb,cAAgC,YAAY;CAC1C;CAEA;CAEA,UAAU;CAEV,YAAY,IAAiB,KAAiC;AAC5D,SAAO;AACP,OAAK,KAAK;AACV,OAAK,QAAQ;GACX,IAAI,aAAa,GAAG;GACpB,IAAI,YAAY,GAAG;GACnB,IAAI,kBAAkB,GAAG;GACzB,IAAI,iBAAiB,EAAE,KAAK,CAAC;GAC9B;;;AAIL,SAAgB,eACd,KACA,SACA,MACA;CACA,MAAM,EACJ,SAAS,YACT,SAAS,YACT,iBAAiB,CAAC,SAAS,mBAAmB,EAC9C,OAAO,OACL,QAAQ,EAAE;CACd,MAAM,EAAE,UAAU;CAClB,MAAM,kBAAkB,eAAe,QAAQ,YAAY;EACzD,SAAS,QAAQ;EACjB,OAAO;EACR,CAAC;CAQF,MAAM,QAAQ,IAAI,cAAc;EAC9B,UAFY,IAAI,SAAS;GAAE,QALd,cAAc,aAAa,OAAO;IAC/C,QAAQ;IACR;IACA;IACD,CAAC;GACmC;GAAK,CAAC;EAGzC,cAAc,MAAM,KAAK,MAAM,EAAE,KAAK;EACvC,CAAC;AACF,QAAO,cAAc,kBAAkB;EACrC;EACA;EACA,yBAAyB;EAC1B,CAAC"}

package/dist/agents/toolkits/vectorstore/prompt.cjs

@@ -1,4 +1,3 @@
-
 //#region src/agents/toolkits/vectorstore/prompt.ts
 const VECTOR_PREFIX = `You are an agent designed to answer questions about sets of documents.
 You have access to tools for interacting with the documents, and the inputs to the tools are questions.
@@ -8,8 +7,8 @@ const VECTOR_ROUTER_PREFIX = `You are an agent designed to answer questions.
 You have access to tools for interacting with different sources, and the inputs to the tools are questions.
 Your main task is to decide which of the tools is relevant for answering question at hand.
 For complex questions, you can break the question down into sub questions and use tools to answers the sub questions.`;
-
 //#endregion
 exports.VECTOR_PREFIX = VECTOR_PREFIX;
 exports.VECTOR_ROUTER_PREFIX = VECTOR_ROUTER_PREFIX;
+
 //# sourceMappingURL=prompt.cjs.map

package/dist/agents/toolkits/vectorstore/prompt.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"prompt.cjs","names":[],"sources":["../../../../src/agents/toolkits/vectorstore/prompt.ts"],"sourcesContent":["export const VECTOR_PREFIX = `You are an agent designed to answer questions about sets of documents.\nYou have access to tools for interacting with the documents, and the inputs to the tools are questions.\nSometimes, you will be asked to provide sources for your questions, in which case you should use the appropriate tool to do so.\nIf the question does not seem relevant to any of the tools provided, just return \"I don't know\" as the answer.`;\n\nexport const VECTOR_ROUTER_PREFIX = `You are an agent designed to answer questions.\nYou have access to tools for interacting with different sources, and the inputs to the tools are questions.\nYour main task is to decide which of the tools is relevant for answering question at hand.\nFor complex questions, you can break the question down into sub questions and use tools to answers the sub questions.`;\n"],"mappings":";;AAAA,MAAa,gBAAgB;;;;AAK7B,MAAa,uBAAuB"}
+{"version":3,"file":"prompt.cjs","names":[],"sources":["../../../../src/agents/toolkits/vectorstore/prompt.ts"],"sourcesContent":["export const VECTOR_PREFIX = `You are an agent designed to answer questions about sets of documents.\nYou have access to tools for interacting with the documents, and the inputs to the tools are questions.\nSometimes, you will be asked to provide sources for your questions, in which case you should use the appropriate tool to do so.\nIf the question does not seem relevant to any of the tools provided, just return \"I don't know\" as the answer.`;\n\nexport const VECTOR_ROUTER_PREFIX = `You are an agent designed to answer questions.\nYou have access to tools for interacting with different sources, and the inputs to the tools are questions.\nYour main task is to decide which of the tools is relevant for answering question at hand.\nFor complex questions, you can break the question down into sub questions and use tools to answers the sub questions.`;\n"],"mappings":";AAAA,MAAa,gBAAgB;;;;AAK7B,MAAa,uBAAuB"}

package/dist/agents/toolkits/vectorstore/prompt.js

@@ -7,7 +7,7 @@ const VECTOR_ROUTER_PREFIX = `You are an agent designed to answer questions.
 You have access to tools for interacting with different sources, and the inputs to the tools are questions.
 Your main task is to decide which of the tools is relevant for answering question at hand.
 For complex questions, you can break the question down into sub questions and use tools to answers the sub questions.`;
-
 //#endregion
 export { VECTOR_PREFIX, VECTOR_ROUTER_PREFIX };
+
 //# sourceMappingURL=prompt.js.map

package/dist/agents/toolkits/vectorstore/vectorstore.cjs

@@ -1,12 +1,11 @@
-const require_runtime = require('../../../_virtual/_rolldown/runtime.cjs');
-const require_llm_chain = require('../../../chains/llm_chain.cjs');
-const require_prompt = require('../../mrkl/prompt.cjs');
-const require_index = require('../../mrkl/index.cjs');
-const require_executor = require('../../executor.cjs');
-const require_vectorstore = require('../../../tools/vectorstore.cjs');
-const require_prompt$1 = require('./prompt.cjs');
+require("../../../_virtual/_rolldown/runtime.cjs");
+const require_llm_chain = require("../../../chains/llm_chain.cjs");
+const require_prompt = require("../../mrkl/prompt.cjs");
+const require_index = require("../../mrkl/index.cjs");
+const require_executor = require("../../executor.cjs");
+const require_vectorstore = require("../../../tools/vectorstore.cjs");
+const require_prompt$1 = require("./prompt.cjs");
 let _langchain_core_tools = require("@langchain/core/tools");
-
 //#region src/agents/toolkits/vectorstore/vectorstore.ts
 /**
 * Class representing a toolkit for working with a single vector store. It
@@ -104,10 +103,10 @@ function createVectorStoreRouterAgent(llm, toolkit, args) {
 		returnIntermediateSteps: true
 	});
 }
-
 //#endregion
 exports.VectorStoreRouterToolkit = VectorStoreRouterToolkit;
 exports.VectorStoreToolkit = VectorStoreToolkit;
 exports.createVectorStoreAgent = createVectorStoreAgent;
 exports.createVectorStoreRouterAgent = createVectorStoreRouterAgent;
+
 //# sourceMappingURL=vectorstore.cjs.map

package/dist/agents/toolkits/vectorstore/vectorstore.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"vectorstore.cjs","names":["BaseToolkit","VectorStoreQATool","VECTOR_PREFIX","SUFFIX","ZeroShotAgent","LLMChain","AgentExecutor","VECTOR_ROUTER_PREFIX"],"sources":["../../../../src/agents/toolkits/vectorstore/vectorstore.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { VectorStoreInterface } from \"@langchain/core/vectorstores\";\nimport { ToolInterface, BaseToolkit } from \"@langchain/core/tools\";\nimport { VectorStoreQATool } from \"../../../tools/vectorstore.js\";\nimport { ZeroShotCreatePromptArgs, ZeroShotAgent } from \"../../mrkl/index.js\";\nimport { VECTOR_PREFIX, VECTOR_ROUTER_PREFIX } from \"./prompt.js\";\nimport { SUFFIX } from \"../../mrkl/prompt.js\";\nimport { LLMChain } from \"../../../chains/llm_chain.js\";\nimport { AgentExecutor } from \"../../executor.js\";\n\n/**\n * Interface that defines the information about a vector store, including\n * the vector store itself, its name, and description.\n */\nexport interface VectorStoreInfo {\n  vectorStore: VectorStoreInterface;\n  name: string;\n  description: string;\n}\n\n/**\n * Class representing a toolkit for working with a single vector store. It\n * initializes the vector store QA tool based on the provided vector store\n * information and language model.\n * @example\n * ```typescript\n * const toolkit = new VectorStoreToolkit(\n *   {\n *     name: \"state_of_union_address\",\n *     description: \"the most recent state of the Union address\",\n *     vectorStore: new HNSWLib(),\n *   },\n *   new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n * );\n * const result = await toolkit.invoke({\n *   input:\n *     \"What did biden say about Ketanji Brown Jackson in the state of the union address?\",\n * });\n * console.log(`Got output ${result.output}`);\n * ```\n */\nexport class VectorStoreToolkit extends BaseToolkit {\n  tools: ToolInterface[];\n\n  llm: BaseLanguageModelInterface;\n\n  constructor(\n    vectorStoreInfo: VectorStoreInfo,\n    llm: BaseLanguageModelInterface\n  ) {\n    super();\n    const description = VectorStoreQATool.getDescription(\n      vectorStoreInfo.name,\n      vectorStoreInfo.description\n    );\n    this.llm = llm;\n    this.tools = [\n      new VectorStoreQATool(vectorStoreInfo.name, description, {\n        vectorStore: vectorStoreInfo.vectorStore,\n        llm: this.llm,\n      }),\n    ];\n  }\n}\n\n/**\n * Class representing a toolkit for working with multiple vector stores.\n * It initializes multiple vector store QA tools based on the provided\n * vector store information and language model.\n */\nexport class VectorStoreRouterToolkit extends BaseToolkit {\n  tools: ToolInterface[];\n\n  vectorStoreInfos: VectorStoreInfo[];\n\n  llm: BaseLanguageModelInterface;\n\n  constructor(\n    vectorStoreInfos: VectorStoreInfo[],\n    llm: BaseLanguageModelInterface\n  ) {\n    super();\n    this.llm = llm;\n    this.vectorStoreInfos = vectorStoreInfos;\n    this.tools = vectorStoreInfos.map((vectorStoreInfo) => {\n      const description = VectorStoreQATool.getDescription(\n        vectorStoreInfo.name,\n        vectorStoreInfo.description\n      );\n      return new VectorStoreQATool(vectorStoreInfo.name, description, {\n        vectorStore: vectorStoreInfo.vectorStore,\n        llm: this.llm,\n      });\n    });\n  }\n}\n\nexport function createVectorStoreAgent(\n  llm: BaseLanguageModelInterface,\n  toolkit: VectorStoreToolkit,\n  args?: ZeroShotCreatePromptArgs\n) {\n  const {\n    prefix = VECTOR_PREFIX,\n    suffix = SUFFIX,\n    inputVariables = [\"input\", \"agent_scratchpad\"],\n  } = args ?? {};\n  const { tools } = toolkit;\n  const prompt = ZeroShotAgent.createPrompt(tools, {\n    prefix,\n    suffix,\n    inputVariables,\n  });\n  const chain = new LLMChain({ prompt, llm });\n  const agent = new ZeroShotAgent({\n    llmChain: chain,\n    allowedTools: tools.map((t) => t.name),\n  });\n  return AgentExecutor.fromAgentAndTools({\n    agent,\n    tools,\n    returnIntermediateSteps: true,\n  });\n}\n\nexport function createVectorStoreRouterAgent(\n  llm: BaseLanguageModelInterface,\n  toolkit: VectorStoreRouterToolkit,\n  args?: ZeroShotCreatePromptArgs\n) {\n  const {\n    prefix = VECTOR_ROUTER_PREFIX,\n    suffix = SUFFIX,\n    inputVariables = [\"input\", \"agent_scratchpad\"],\n  } = args ?? {};\n  const { tools } = toolkit;\n  const prompt = ZeroShotAgent.createPrompt(tools, {\n    prefix,\n    suffix,\n    inputVariables,\n  });\n  const chain = new LLMChain({ prompt, llm });\n  const agent = new ZeroShotAgent({\n    llmChain: chain,\n    allowedTools: tools.map((t) => t.name),\n  });\n  return AgentExecutor.fromAgentAndTools({\n    agent,\n    tools,\n    returnIntermediateSteps: true,\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAyCA,IAAa,qBAAb,cAAwCA,kCAAY;CAClD;CAEA;CAEA,YACE,iBACA,KACA;AACA,SAAO;EACP,MAAM,cAAcC,sCAAkB,eACpC,gBAAgB,MAChB,gBAAgB,YACjB;AACD,OAAK,MAAM;AACX,OAAK,QAAQ,CACX,IAAIA,sCAAkB,gBAAgB,MAAM,aAAa;GACvD,aAAa,gBAAgB;GAC7B,KAAK,KAAK;GACX,CAAC,CACH;;;;;;;;AASL,IAAa,2BAAb,cAA8CD,kCAAY;CACxD;CAEA;CAEA;CAEA,YACE,kBACA,KACA;AACA,SAAO;AACP,OAAK,MAAM;AACX,OAAK,mBAAmB;AACxB,OAAK,QAAQ,iBAAiB,KAAK,oBAAoB;GACrD,MAAM,cAAcC,sCAAkB,eACpC,gBAAgB,MAChB,gBAAgB,YACjB;AACD,UAAO,IAAIA,sCAAkB,gBAAgB,MAAM,aAAa;IAC9D,aAAa,gBAAgB;IAC7B,KAAK,KAAK;IACX,CAAC;IACF;;;AAIN,SAAgB,uBACd,KACA,SACA,MACA;CACA,MAAM,EACJ,SAASC,gCACT,SAASC,uBACT,iBAAiB,CAAC,SAAS,mBAAmB,KAC5C,QAAQ,EAAE;CACd,MAAM,EAAE,UAAU;CAOlB,MAAM,QAAQ,IAAIC,4BAAc;EAC9B,UAFY,IAAIC,2BAAS;GAAE,QALdD,4BAAc,aAAa,OAAO;IAC/C;IACA;IACA;IACD,CAAC;GACmC;GAAK,CAAC;EAGzC,cAAc,MAAM,KAAK,MAAM,EAAE,KAAK;EACvC,CAAC;AACF,QAAOE,+BAAc,kBAAkB;EACrC;EACA;EACA,yBAAyB;EAC1B,CAAC;;AAGJ,SAAgB,6BACd,KACA,SACA,MACA;CACA,MAAM,EACJ,SAASC,uCACT,SAASJ,uBACT,iBAAiB,CAAC,SAAS,mBAAmB,KAC5C,QAAQ,EAAE;CACd,MAAM,EAAE,UAAU;CAOlB,MAAM,QAAQ,IAAIC,4BAAc;EAC9B,UAFY,IAAIC,2BAAS;GAAE,QALdD,4BAAc,aAAa,OAAO;IAC/C;IACA;IACA;IACD,CAAC;GACmC;GAAK,CAAC;EAGzC,cAAc,MAAM,KAAK,MAAM,EAAE,KAAK;EACvC,CAAC;AACF,QAAOE,+BAAc,kBAAkB;EACrC;EACA;EACA,yBAAyB;EAC1B,CAAC"}
+{"version":3,"file":"vectorstore.cjs","names":["BaseToolkit","VectorStoreQATool","VECTOR_PREFIX","SUFFIX","ZeroShotAgent","LLMChain","AgentExecutor","VECTOR_ROUTER_PREFIX"],"sources":["../../../../src/agents/toolkits/vectorstore/vectorstore.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { VectorStoreInterface } from \"@langchain/core/vectorstores\";\nimport { ToolInterface, BaseToolkit } from \"@langchain/core/tools\";\nimport { VectorStoreQATool } from \"../../../tools/vectorstore.js\";\nimport { ZeroShotCreatePromptArgs, ZeroShotAgent } from \"../../mrkl/index.js\";\nimport { VECTOR_PREFIX, VECTOR_ROUTER_PREFIX } from \"./prompt.js\";\nimport { SUFFIX } from \"../../mrkl/prompt.js\";\nimport { LLMChain } from \"../../../chains/llm_chain.js\";\nimport { AgentExecutor } from \"../../executor.js\";\n\n/**\n * Interface that defines the information about a vector store, including\n * the vector store itself, its name, and description.\n */\nexport interface VectorStoreInfo {\n  vectorStore: VectorStoreInterface;\n  name: string;\n  description: string;\n}\n\n/**\n * Class representing a toolkit for working with a single vector store. It\n * initializes the vector store QA tool based on the provided vector store\n * information and language model.\n * @example\n * ```typescript\n * const toolkit = new VectorStoreToolkit(\n *   {\n *     name: \"state_of_union_address\",\n *     description: \"the most recent state of the Union address\",\n *     vectorStore: new HNSWLib(),\n *   },\n *   new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n * );\n * const result = await toolkit.invoke({\n *   input:\n *     \"What did biden say about Ketanji Brown Jackson in the state of the union address?\",\n * });\n * console.log(`Got output ${result.output}`);\n * ```\n */\nexport class VectorStoreToolkit extends BaseToolkit {\n  tools: ToolInterface[];\n\n  llm: BaseLanguageModelInterface;\n\n  constructor(\n    vectorStoreInfo: VectorStoreInfo,\n    llm: BaseLanguageModelInterface\n  ) {\n    super();\n    const description = VectorStoreQATool.getDescription(\n      vectorStoreInfo.name,\n      vectorStoreInfo.description\n    );\n    this.llm = llm;\n    this.tools = [\n      new VectorStoreQATool(vectorStoreInfo.name, description, {\n        vectorStore: vectorStoreInfo.vectorStore,\n        llm: this.llm,\n      }),\n    ];\n  }\n}\n\n/**\n * Class representing a toolkit for working with multiple vector stores.\n * It initializes multiple vector store QA tools based on the provided\n * vector store information and language model.\n */\nexport class VectorStoreRouterToolkit extends BaseToolkit {\n  tools: ToolInterface[];\n\n  vectorStoreInfos: VectorStoreInfo[];\n\n  llm: BaseLanguageModelInterface;\n\n  constructor(\n    vectorStoreInfos: VectorStoreInfo[],\n    llm: BaseLanguageModelInterface\n  ) {\n    super();\n    this.llm = llm;\n    this.vectorStoreInfos = vectorStoreInfos;\n    this.tools = vectorStoreInfos.map((vectorStoreInfo) => {\n      const description = VectorStoreQATool.getDescription(\n        vectorStoreInfo.name,\n        vectorStoreInfo.description\n      );\n      return new VectorStoreQATool(vectorStoreInfo.name, description, {\n        vectorStore: vectorStoreInfo.vectorStore,\n        llm: this.llm,\n      });\n    });\n  }\n}\n\nexport function createVectorStoreAgent(\n  llm: BaseLanguageModelInterface,\n  toolkit: VectorStoreToolkit,\n  args?: ZeroShotCreatePromptArgs\n) {\n  const {\n    prefix = VECTOR_PREFIX,\n    suffix = SUFFIX,\n    inputVariables = [\"input\", \"agent_scratchpad\"],\n  } = args ?? {};\n  const { tools } = toolkit;\n  const prompt = ZeroShotAgent.createPrompt(tools, {\n    prefix,\n    suffix,\n    inputVariables,\n  });\n  const chain = new LLMChain({ prompt, llm });\n  const agent = new ZeroShotAgent({\n    llmChain: chain,\n    allowedTools: tools.map((t) => t.name),\n  });\n  return AgentExecutor.fromAgentAndTools({\n    agent,\n    tools,\n    returnIntermediateSteps: true,\n  });\n}\n\nexport function createVectorStoreRouterAgent(\n  llm: BaseLanguageModelInterface,\n  toolkit: VectorStoreRouterToolkit,\n  args?: ZeroShotCreatePromptArgs\n) {\n  const {\n    prefix = VECTOR_ROUTER_PREFIX,\n    suffix = SUFFIX,\n    inputVariables = [\"input\", \"agent_scratchpad\"],\n  } = args ?? {};\n  const { tools } = toolkit;\n  const prompt = ZeroShotAgent.createPrompt(tools, {\n    prefix,\n    suffix,\n    inputVariables,\n  });\n  const chain = new LLMChain({ prompt, llm });\n  const agent = new ZeroShotAgent({\n    llmChain: chain,\n    allowedTools: tools.map((t) => t.name),\n  });\n  return AgentExecutor.fromAgentAndTools({\n    agent,\n    tools,\n    returnIntermediateSteps: true,\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAyCA,IAAa,qBAAb,cAAwCA,sBAAAA,YAAY;CAClD;CAEA;CAEA,YACE,iBACA,KACA;AACA,SAAO;EACP,MAAM,cAAcC,oBAAAA,kBAAkB,eACpC,gBAAgB,MAChB,gBAAgB,YACjB;AACD,OAAK,MAAM;AACX,OAAK,QAAQ,CACX,IAAIA,oBAAAA,kBAAkB,gBAAgB,MAAM,aAAa;GACvD,aAAa,gBAAgB;GAC7B,KAAK,KAAK;GACX,CAAC,CACH;;;;;;;;AASL,IAAa,2BAAb,cAA8CD,sBAAAA,YAAY;CACxD;CAEA;CAEA;CAEA,YACE,kBACA,KACA;AACA,SAAO;AACP,OAAK,MAAM;AACX,OAAK,mBAAmB;AACxB,OAAK,QAAQ,iBAAiB,KAAK,oBAAoB;GACrD,MAAM,cAAcC,oBAAAA,kBAAkB,eACpC,gBAAgB,MAChB,gBAAgB,YACjB;AACD,UAAO,IAAIA,oBAAAA,kBAAkB,gBAAgB,MAAM,aAAa;IAC9D,aAAa,gBAAgB;IAC7B,KAAK,KAAK;IACX,CAAC;IACF;;;AAIN,SAAgB,uBACd,KACA,SACA,MACA;CACA,MAAM,EACJ,SAASC,iBAAAA,eACT,SAASC,eAAAA,QACT,iBAAiB,CAAC,SAAS,mBAAmB,KAC5C,QAAQ,EAAE;CACd,MAAM,EAAE,UAAU;CAOlB,MAAM,QAAQ,IAAIC,cAAAA,cAAc;EAC9B,UAFY,IAAIC,kBAAAA,SAAS;GAAE,QALdD,cAAAA,cAAc,aAAa,OAAO;IAC/C;IACA;IACA;IACD,CAAC;GACmC;GAAK,CAAC;EAGzC,cAAc,MAAM,KAAK,MAAM,EAAE,KAAK;EACvC,CAAC;AACF,QAAOE,iBAAAA,cAAc,kBAAkB;EACrC;EACA;EACA,yBAAyB;EAC1B,CAAC;;AAGJ,SAAgB,6BACd,KACA,SACA,MACA;CACA,MAAM,EACJ,SAASC,iBAAAA,sBACT,SAASJ,eAAAA,QACT,iBAAiB,CAAC,SAAS,mBAAmB,KAC5C,QAAQ,EAAE;CACd,MAAM,EAAE,UAAU;CAOlB,MAAM,QAAQ,IAAIC,cAAAA,cAAc;EAC9B,UAFY,IAAIC,kBAAAA,SAAS;GAAE,QALdD,cAAAA,cAAc,aAAa,OAAO;IAC/C;IACA;IACA;IACD,CAAC;GACmC;GAAK,CAAC;EAGzC,cAAc,MAAM,KAAK,MAAM,EAAE,KAAK;EACvC,CAAC;AACF,QAAOE,iBAAAA,cAAc,kBAAkB;EACrC;EACA;EACA,yBAAyB;EAC1B,CAAC"}

package/dist/agents/toolkits/vectorstore/vectorstore.js

@@ -5,7 +5,6 @@ import { AgentExecutor } from "../../executor.js";
 import { VectorStoreQATool } from "../../../tools/vectorstore.js";
 import { VECTOR_PREFIX, VECTOR_ROUTER_PREFIX } from "./prompt.js";
 import { BaseToolkit } from "@langchain/core/tools";
-
 //#region src/agents/toolkits/vectorstore/vectorstore.ts
 /**
 * Class representing a toolkit for working with a single vector store. It
@@ -103,7 +102,7 @@ function createVectorStoreRouterAgent(llm, toolkit, args) {
 		returnIntermediateSteps: true
 	});
 }
-
 //#endregion
 export { VectorStoreRouterToolkit, VectorStoreToolkit, createVectorStoreAgent, createVectorStoreRouterAgent };
+
 //# sourceMappingURL=vectorstore.js.map