langchain 0.0.15 → 0.0.17

package/dist/agents/agent_toolkits/index.d.ts

@@ -1,3 +1,4 @@
-export { JsonToolkit, createJsonAgent } from "./json.js";
-export { SqlToolkit, createSqlAgent } from "./sql.js";
-export { RequestsToolkit, OpenApiToolkit, createOpenApiAgent, } from "./openapi.js";
+export { JsonToolkit, createJsonAgent } from "./json/json.js";
+export { SqlToolkit, createSqlAgent } from "./sql/sql.js";
+export { RequestsToolkit, OpenApiToolkit, createOpenApiAgent, } from "./openapi/openapi.js";
+export { VectorStoreInfo, VectorStoreToolkit, VectorStoreRouterToolkit, createVectorStoreAgent, createVectorStoreRouterAgent, } from "./vectorstore/vectorstore.js";

package/dist/agents/agent_toolkits/index.js

@@ -1,4 +1,5 @@
-export { JsonToolkit, createJsonAgent } from "./json.js";
-export { SqlToolkit, createSqlAgent } from "./sql.js";
-export { RequestsToolkit, OpenApiToolkit, createOpenApiAgent, } from "./openapi.js";
+export { JsonToolkit, createJsonAgent } from "./json/json.js";
+export { SqlToolkit, createSqlAgent } from "./sql/sql.js";
+export { RequestsToolkit, OpenApiToolkit, createOpenApiAgent, } from "./openapi/openapi.js";
+export { VectorStoreToolkit, VectorStoreRouterToolkit, createVectorStoreAgent, createVectorStoreRouterAgent, } from "./vectorstore/vectorstore.js";
 //# sourceMappingURL=index.js.map

package/dist/agents/agent_toolkits/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/agents/agent_toolkits/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AACzD,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AACtD,OAAO,EACL,eAAe,EACf,cAAc,EACd,kBAAkB,GACnB,MAAM,cAAc,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/agents/agent_toolkits/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAC9D,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC1D,OAAO,EACL,eAAe,EACf,cAAc,EACd,kBAAkB,GACnB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAEL,kBAAkB,EAClB,wBAAwB,EACxB,sBAAsB,EACtB,4BAA4B,GAC7B,MAAM,8BAA8B,CAAC"}

package/dist/agents/agent_toolkits/{json.d.ts → json/json.d.ts}

@@ -1,8 +1,8 @@
-import { BaseLLM } from "../../llms/index.js";
-import { JsonSpec, Tool } from "../tools/index.js";
-import { CreatePromptArgs } from "../mrkl/index.js";
-import { Toolkit } from "./base.js";
-import { AgentExecutor } from "../executor.js";
+import { BaseLLM } from "../../../llms/index.js";
+import { JsonSpec, Tool } from "../../tools/index.js";
+import { CreatePromptArgs } from "../../mrkl/index.js";
+import { Toolkit } from "../base.js";
+import { AgentExecutor } from "../../executor.js";
 export declare class JsonToolkit extends Toolkit {
     jsonSpec: JsonSpec;
     tools: Tool[];

package/dist/agents/agent_toolkits/{json.js → json/json.js}

@@ -1,9 +1,9 @@
-import { JsonGetValueTool, JsonListKeysTool, } from "../tools/index.js";
+import { JsonGetValueTool, JsonListKeysTool, } from "../../tools/index.js";
 import { JSON_PREFIX, JSON_SUFFIX } from "./prompt.js";
-import { LLMChain } from "../../chains/index.js";
-import { ZeroShotAgent } from "../mrkl/index.js";
-import { Toolkit } from "./base.js";
-import { AgentExecutor } from "../executor.js";
+import { LLMChain } from "../../../chains/index.js";
+import { ZeroShotAgent } from "../../mrkl/index.js";
+import { Toolkit } from "../base.js";
+import { AgentExecutor } from "../../executor.js";
 export class JsonToolkit extends Toolkit {
     constructor(jsonSpec) {
         super();

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

@@ -0,0 +1 @@
+{"version":3,"file":"json.js","sourceRoot":"","sources":["../../../../src/agents/agent_toolkits/json/json.ts"],"names":[],"mappings":"AACA,OAAO,EACL,gBAAgB,EAChB,gBAAgB,GAGjB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AACpD,OAAO,EAAoB,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACtE,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AACrC,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAElD,MAAM,OAAO,WAAY,SAAQ,OAAO;IAGtC,YAAmB,QAAkB;QACnC,KAAK,EAAE,CAAC;;;;;mBADS;;QAFnB;;;;;WAAc;QAIZ,IAAI,CAAC,KAAK,GAAG;YACX,IAAI,gBAAgB,CAAC,QAAQ,CAAC;YAC9B,IAAI,gBAAgB,CAAC,QAAQ,CAAC;SAC/B,CAAC;IACJ,CAAC;CACF;AAED,MAAM,UAAU,eAAe,CAC7B,GAAY,EACZ,OAAoB,EACpB,IAAuB;IAEvB,MAAM,EACJ,MAAM,GAAG,WAAW,EACpB,MAAM,GAAG,WAAW,EACpB,cAAc,GAAG,CAAC,OAAO,EAAE,kBAAkB,CAAC,GAC/C,GAAG,IAAI,IAAI,EAAE,CAAC;IACf,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC;IAC1B,MAAM,MAAM,GAAG,aAAa,CAAC,YAAY,CAAC,KAAK,EAAE;QAC/C,MAAM;QACN,MAAM;QACN,cAAc;KACf,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5C,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC;QAC9B,QAAQ,EAAE,KAAK;QACf,YAAY,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;KACvC,CAAC,CAAC;IACH,OAAO,aAAa,CAAC,iBAAiB,CAAC;QACrC,KAAK;QACL,KAAK;QACL,uBAAuB,EAAE,IAAI;KAC9B,CAAC,CAAC;AACL,CAAC"}

package/dist/agents/agent_toolkits/json/prompt.d.ts

@@ -0,0 +1,2 @@
+export declare 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.";
+export declare 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}";

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

@@ -0,0 +1,25 @@
+export 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.
+You have access to the following tools which help you learn more about the JSON you are interacting with.
+Only use the below tools. Only use the information returned by the below tools to construct your final answer.
+Do not make up any information that is not contained in the JSON.
+Your input to the tools should be in the form of in json pointer syntax (e.g. /key1/0/key2).
+You must escape a slash in a key with a ~1, and escape a tilde with a ~0.
+For example, to access the key /foo, you would use /~1foo
+You 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'.
+If you have not seen a key in one of those responses, you cannot use it.
+You should only add one key at a time to the path. You cannot add multiple keys at once.
+If you encounter a null or undefined value, go back to the previous key, look at the available keys, and try again.
+
+If the question does not seem to be related to the JSON, just return "I don't know" as the answer.
+Always begin your interaction with the 'json_list_keys' with an empty string as the input to see what keys exist in the JSON.
+
+Note 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".
+In this case, you should ALWAYS follow up by using the 'json_list_keys' tool to see what keys exist at that path.
+Do 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.`;
+export 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}`;
+//# sourceMappingURL=prompt.js.map

package/dist/agents/agent_toolkits/json/prompt.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt.js","sourceRoot":"","sources":["../../../../src/agents/agent_toolkits/json/prompt.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,WAAW,GAAG;;;;;;;;;;;;;;;;;;mKAkBwI,CAAC;AAEpK,MAAM,CAAC,MAAM,WAAW,GAAG;;;;mBAIR,CAAC"}

package/dist/agents/agent_toolkits/{openapi.d.ts → openapi/openapi.d.ts}

@@ -1,9 +1,9 @@
-import { BaseLLM } from "../../llms/index.js";
-import { JsonSpec, Tool } from "../tools/index.js";
-import { AgentExecutor } from "../executor.js";
-import { CreatePromptArgs } from "../mrkl/index.js";
-import { Toolkit } from "./base.js";
-import { Headers } from "../tools/requests.js";
+import { BaseLLM } from "../../../llms/index.js";
+import { JsonSpec, Tool } from "../../tools/index.js";
+import { AgentExecutor } from "../../executor.js";
+import { CreatePromptArgs } from "../../mrkl/index.js";
+import { Toolkit } from "../base.js";
+import { Headers } from "../../tools/requests.js";
 export declare class RequestsToolkit extends Toolkit {
     tools: Tool[];
     constructor(headers?: Headers);

package/dist/agents/agent_toolkits/{openapi.js → openapi/openapi.js}

@@ -1,10 +1,10 @@
-import { DynamicTool, RequestsGetTool, RequestsPostTool, } from "../tools/index.js";
-import { AgentExecutor } from "../executor.js";
+import { DynamicTool, RequestsGetTool, RequestsPostTool, } from "../../tools/index.js";
+import { AgentExecutor } from "../../executor.js";
 import { OPENAPI_PREFIX, OPENAPI_SUFFIX, JSON_EXPLORER_DESCRIPTION, } from "./prompt.js";
-import { LLMChain } from "../../chains/index.js";
-import { ZeroShotAgent } from "../mrkl/index.js";
-import { Toolkit } from "./base.js";
-import { createJsonAgent, JsonToolkit } from "./json.js";
+import { LLMChain } from "../../../chains/index.js";
+import { ZeroShotAgent } from "../../mrkl/index.js";
+import { Toolkit } from "../base.js";
+import { createJsonAgent, JsonToolkit } from "../json/json.js";
 export class RequestsToolkit extends Toolkit {
     constructor(headers) {
         super();

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

@@ -0,0 +1 @@
+{"version":3,"file":"openapi.js","sourceRoot":"","sources":["../../../../src/agents/agent_toolkits/openapi/openapi.ts"],"names":[],"mappings":"AACA,OAAO,EACL,WAAW,EAEX,eAAe,EACf,gBAAgB,GAEjB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAClD,OAAO,EACL,cAAc,EACd,cAAc,EACd,yBAAyB,GAC1B,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AACpD,OAAO,EAAoB,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACtE,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAErC,OAAO,EAAE,eAAe,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAE/D,MAAM,OAAO,eAAgB,SAAQ,OAAO;IAG1C,YAAY,OAAiB;QAC3B,KAAK,EAAE,CAAC;QAHV;;;;;WAAc;QAIZ,IAAI,CAAC,KAAK,GAAG,CAAC,IAAI,eAAe,CAAC,OAAO,CAAC,EAAE,IAAI,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;IAC7E,CAAC;CACF;AAED,MAAM,OAAO,cAAe,SAAQ,eAAe;IACjD,YAAY,QAAkB,EAAE,GAAY,EAAE,OAAiB;QAC7D,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,MAAM,SAAS,GAAG,eAAe,CAAC,GAAG,EAAE,IAAI,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC;QAClE,IAAI,CAAC,KAAK,GAAG;YACX,GAAG,IAAI,CAAC,KAAK;YACb,IAAI,WAAW,CAAC;gBACd,IAAI,EAAE,eAAe;gBACrB,IAAI,EAAE,KAAK,EAAE,KAAa,EAAE,EAAE;oBAC5B,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;oBAC/C,OAAO,MAAM,CAAC,MAAgB,CAAC;gBACjC,CAAC;gBACD,WAAW,EAAE,yBAAyB;aACvC,CAAC;SACH,CAAC;IACJ,CAAC;CACF;AAED,MAAM,UAAU,kBAAkB,CAChC,GAAY,EACZ,cAA8B,EAC9B,IAAuB;IAEvB,MAAM,EACJ,MAAM,GAAG,cAAc,EACvB,MAAM,GAAG,cAAc,EACvB,cAAc,GAAG,CAAC,OAAO,EAAE,kBAAkB,CAAC,GAC/C,GAAG,IAAI,IAAI,EAAE,CAAC;IAEf,MAAM,EAAE,KAAK,EAAE,GAAG,cAAc,CAAC;IACjC,MAAM,MAAM,GAAG,aAAa,CAAC,YAAY,CAAC,KAAK,EAAE;QAC/C,MAAM;QACN,MAAM;QACN,cAAc;KACf,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC;QACzB,MAAM;QACN,GAAG;KACJ,CAAC,CAAC;IACH,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACjD,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC,CAAC;IAC9E,OAAO,aAAa,CAAC,iBAAiB,CAAC;QACrC,KAAK;QACL,KAAK;QACL,uBAAuB,EAAE,IAAI;KAC9B,CAAC,CAAC;AACL,CAAC"}

package/dist/agents/agent_toolkits/openapi/prompt.d.ts

@@ -0,0 +1,3 @@
+export declare 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.";
+export declare const OPENAPI_SUFFIX = "Begin!\"\n\nQuestion: {input}\nThought: I should explore the spec to find the base url for the API.\n{agent_scratchpad}";
+export declare 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.";

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

@@ -0,0 +1,30 @@
+export const OPENAPI_PREFIX = `You are an agent designed to answer questions by making web requests to an API given the OpenAPI spec.
+
+If the question does not seem related to the API, return I don't know. Do not make up an answer.
+Only use information provided by the tools to construct your response.
+
+To find information in the OpenAPI spec, use the 'json_explorer' tool. The input to this tool is a question about the API.
+
+Take the following steps:
+First, find the base URL needed to make the request.
+
+Second, 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.
+
+Third, 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.
+
+Fourth, 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.
+
+Use the exact parameter names as listed in the spec, do not make up any names or abbreviate the names of parameters.
+If you get a not found error, ensure that you are using a path that actually exists in the spec.`;
+export const OPENAPI_SUFFIX = `Begin!"
+
+Question: {input}
+Thought: I should explore the spec to find the base url for the API.
+{agent_scratchpad}`;
+export const JSON_EXPLORER_DESCRIPTION = `
+Can be used to answer questions about the openapi spec for the API. Always use this tool before trying to make a request. 
+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.`;
+//# sourceMappingURL=prompt.js.map

package/dist/agents/agent_toolkits/openapi/prompt.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt.js","sourceRoot":"","sources":["../../../../src/agents/agent_toolkits/openapi/prompt.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,cAAc,GAAG;;;;;;;;;;;;;;;;;iGAiBmE,CAAC;AAElG,MAAM,CAAC,MAAM,cAAc,GAAG;;;;mBAIX,CAAC;AACpB,MAAM,CAAC,MAAM,yBAAyB,GAAG;;;;;2CAKE,CAAC"}

package/dist/agents/agent_toolkits/sql/prompt.d.ts

@@ -0,0 +1,2 @@
+export declare 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.";
+export declare 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}";

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

@@ -0,0 +1,18 @@
+export 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.
+Unless 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.
+You can order the results by a relevant column to return the most interesting examples in the database.
+Never query for all the columns from a specific table, only ask for a the few relevant columns given the question.
+You have access to tools for interacting with the database.
+Only use the below tools. Only use the information returned by the below tools to construct your final answer.
+You MUST double check your query before executing it. If you get an error while executing a query, rewrite the query and try again.
+
+DO NOT make any DML statements (INSERT, UPDATE, DELETE, DROP etc.) to the database.
+
+If the question does not seem related to the database, just return "I don't know" as the answer.`;
+export const SQL_SUFFIX = `Begin!
+
+Question: {input}
+Thought: I should look at the tables in the database to see what I can query.
+{agent_scratchpad}`;
+//# sourceMappingURL=prompt.js.map

package/dist/agents/agent_toolkits/sql/prompt.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt.js","sourceRoot":"","sources":["../../../../src/agents/agent_toolkits/sql/prompt.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,UAAU,GAAG;;;;;;;;;;;iGAWuE,CAAC;AAElG,MAAM,CAAC,MAAM,UAAU,GAAG;;;;mBAIP,CAAC"}

package/dist/agents/agent_toolkits/{sql.d.ts → sql/sql.d.ts}

@@ -1,8 +1,8 @@
-import { Tool, SqlDatabase } from "../tools/index.js";
-import { Toolkit } from "./base.js";
-import { BaseLLM } from "../../llms/index.js";
-import { CreatePromptArgs } from "../mrkl/index.js";
-import { AgentExecutor } from "../executor.js";
+import { Tool, SqlDatabase } from "../../tools/index.js";
+import { Toolkit } from "../base.js";
+import { BaseLLM } from "../../../llms/index.js";
+import { CreatePromptArgs } from "../../mrkl/index.js";
+import { AgentExecutor } from "../../executor.js";
 type SqlCreatePromptArgs = {
     /** Number of results to return. */
     topK?: number;

package/dist/agents/agent_toolkits/{sql.js → sql/sql.js}

@@ -1,10 +1,10 @@
-import { InfoSqlTool, ListTablesSqlTool, QueryCheckerTool, QuerySqlTool, } from "../tools/index.js";
-import { Toolkit } from "./base.js";
+import { InfoSqlTool, ListTablesSqlTool, QueryCheckerTool, QuerySqlTool, } from "../../tools/index.js";
+import { Toolkit } from "../base.js";
 import { SQL_PREFIX, SQL_SUFFIX } from "./prompt.js";
-import { interpolateFString } from "../../prompts/template.js";
-import { LLMChain } from "../../chains/index.js";
-import { ZeroShotAgent } from "../mrkl/index.js";
-import { AgentExecutor } from "../executor.js";
+import { interpolateFString } from "../../../prompts/template.js";
+import { LLMChain } from "../../../chains/index.js";
+import { ZeroShotAgent } from "../../mrkl/index.js";
+import { AgentExecutor } from "../../executor.js";
 export class SqlToolkit extends Toolkit {
     constructor(db) {
         super();

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

@@ -0,0 +1 @@
+{"version":3,"file":"sql.js","sourceRoot":"","sources":["../../../../src/agents/agent_toolkits/sql/sql.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,WAAW,EACX,iBAAiB,EACjB,gBAAgB,EAChB,YAAY,GAEb,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAErC,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACrD,OAAO,EAAE,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AAClE,OAAO,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AACpD,OAAO,EAAE,aAAa,EAAoB,MAAM,qBAAqB,CAAC;AACtE,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAOlD,MAAM,OAAO,UAAW,SAAQ,OAAO;IAOrC,YAAY,EAAe;QACzB,KAAK,EAAE,CAAC;QAPV;;;;;WAAc;QAEd;;;;;WAAgB;QAEhB;;;;mBAAU,QAAQ;WAAC;QAIjB,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,KAAK,GAAG;YACX,IAAI,YAAY,CAAC,EAAE,CAAC;YACpB,IAAI,WAAW,CAAC,EAAE,CAAC;YACnB,IAAI,iBAAiB,CAAC,EAAE,CAAC;YACzB,IAAI,gBAAgB,EAAE;SACvB,CAAC;IACJ,CAAC;CACF;AAED,MAAM,UAAU,cAAc,CAC5B,GAAY,EACZ,OAAmB,EACnB,IAA0B;IAE1B,MAAM,EACJ,MAAM,GAAG,UAAU,EACnB,MAAM,GAAG,UAAU,EACnB,cAAc,GAAG,CAAC,OAAO,EAAE,kBAAkB,CAAC,EAC9C,IAAI,GAAG,EAAE,GACV,GAAG,IAAI,IAAI,EAAE,CAAC;IACf,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC;IAC1B,MAAM,eAAe,GAAG,kBAAkB,CAAC,MAAM,EAAE;QACjD,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,KAAK,EAAE,IAAI;KACZ,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,aAAa,CAAC,YAAY,CAAC,KAAK,EAAE;QAC/C,MAAM,EAAE,eAAe;QACvB,MAAM;QACN,cAAc;KACf,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5C,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC;QAC9B,QAAQ,EAAE,KAAK;QACf,YAAY,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;KACvC,CAAC,CAAC;IACH,OAAO,aAAa,CAAC,iBAAiB,CAAC;QACrC,KAAK;QACL,KAAK;QACL,uBAAuB,EAAE,IAAI;KAC9B,CAAC,CAAC;AACL,CAAC"}

package/dist/agents/agent_toolkits/vectorstore/prompt.d.ts

@@ -0,0 +1,2 @@
+export declare 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.";
+export declare 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.";

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

@@ -0,0 +1,9 @@
+export 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.
+Sometimes, you will be asked to provide sources for your questions, in which case you should use the appropriate tool to do so.
+If the question does not seem relevant to any of the tools provided, just return "I don't know" as the answer.`;
+export 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.`;
+//# sourceMappingURL=prompt.js.map

package/dist/agents/agent_toolkits/vectorstore/prompt.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt.js","sourceRoot":"","sources":["../../../../src/agents/agent_toolkits/vectorstore/prompt.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,aAAa,GAAG;;;+GAGkF,CAAC;AAEhH,MAAM,CAAC,MAAM,oBAAoB,GAAG;;;sHAGkF,CAAC"}

package/dist/agents/agent_toolkits/vectorstore/vectorstore.d.ts

@@ -0,0 +1,24 @@
+import { Tool } from "../../tools/index.js";
+import { VectorStore } from "../../../vectorstores/index.js";
+import { Toolkit } from "../base.js";
+import { BaseLLM } from "../../../llms/index.js";
+import { CreatePromptArgs } from "../../mrkl/index.js";
+import { AgentExecutor } from "../../executor.js";
+export interface VectorStoreInfo {
+    vectorStore: VectorStore;
+    name: string;
+    description: string;
+}
+export declare class VectorStoreToolkit extends Toolkit {
+    tools: Tool[];
+    llm: BaseLLM;
+    constructor(vectorStoreInfo: VectorStoreInfo, llm: BaseLLM);
+}
+export declare class VectorStoreRouterToolkit extends Toolkit {
+    tools: Tool[];
+    vectorStoreInfos: VectorStoreInfo[];
+    llm: BaseLLM;
+    constructor(vectorStoreInfos: VectorStoreInfo[], llm: BaseLLM);
+}
+export declare function createVectorStoreAgent(llm: BaseLLM, toolkit: VectorStoreToolkit, args?: CreatePromptArgs): AgentExecutor;
+export declare function createVectorStoreRouterAgent(llm: BaseLLM, toolkit: VectorStoreRouterToolkit, args?: CreatePromptArgs): AgentExecutor;

package/dist/agents/agent_toolkits/vectorstore/vectorstore.js

@@ -0,0 +1,103 @@
+import { VectorStoreQATool } from "../../tools/index.js";
+import { Toolkit } from "../base.js";
+import { ZeroShotAgent } from "../../mrkl/index.js";
+import { VECTOR_PREFIX, VECTOR_ROUTER_PREFIX } from "./prompt.js";
+import { SUFFIX } from "../../mrkl/prompt.js";
+import { LLMChain } from "../../../chains/index.js";
+import { AgentExecutor } from "../../executor.js";
+export class VectorStoreToolkit extends Toolkit {
+    constructor(vectorStoreInfo, llm) {
+        super();
+        Object.defineProperty(this, "tools", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "llm", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        const description = VectorStoreQATool.getDescription(vectorStoreInfo.name, vectorStoreInfo.description);
+        this.llm = llm;
+        this.tools = [
+            new VectorStoreQATool(vectorStoreInfo.name, description, {
+                vectorStore: vectorStoreInfo.vectorStore,
+                llm: this.llm,
+            }),
+        ];
+    }
+}
+export class VectorStoreRouterToolkit extends Toolkit {
+    constructor(vectorStoreInfos, llm) {
+        super();
+        Object.defineProperty(this, "tools", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "vectorStoreInfos", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "llm", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.llm = llm;
+        this.vectorStoreInfos = vectorStoreInfos;
+        this.tools = vectorStoreInfos.map((vectorStoreInfo) => {
+            const description = VectorStoreQATool.getDescription(vectorStoreInfo.name, vectorStoreInfo.description);
+            return new VectorStoreQATool(vectorStoreInfo.name, description, {
+                vectorStore: vectorStoreInfo.vectorStore,
+                llm: this.llm,
+            });
+        });
+    }
+}
+export function createVectorStoreAgent(llm, toolkit, args) {
+    const { prefix = VECTOR_PREFIX, suffix = SUFFIX, inputVariables = ["input", "agent_scratchpad"], } = args ?? {};
+    const { tools } = toolkit;
+    const prompt = ZeroShotAgent.createPrompt(tools, {
+        prefix,
+        suffix,
+        inputVariables,
+    });
+    const chain = new LLMChain({ prompt, llm });
+    const agent = new ZeroShotAgent({
+        llmChain: chain,
+        allowedTools: tools.map((t) => t.name),
+    });
+    return AgentExecutor.fromAgentAndTools({
+        agent,
+        tools,
+        returnIntermediateSteps: true,
+    });
+}
+export function createVectorStoreRouterAgent(llm, toolkit, args) {
+    const { prefix = VECTOR_ROUTER_PREFIX, suffix = SUFFIX, inputVariables = ["input", "agent_scratchpad"], } = args ?? {};
+    const { tools } = toolkit;
+    const prompt = ZeroShotAgent.createPrompt(tools, {
+        prefix,
+        suffix,
+        inputVariables,
+    });
+    const chain = new LLMChain({ prompt, llm });
+    const agent = new ZeroShotAgent({
+        llmChain: chain,
+        allowedTools: tools.map((t) => t.name),
+    });
+    return AgentExecutor.fromAgentAndTools({
+        agent,
+        tools,
+        returnIntermediateSteps: true,
+    });
+}
+//# sourceMappingURL=vectorstore.js.map

package/dist/agents/agent_toolkits/vectorstore/vectorstore.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vectorstore.js","sourceRoot":"","sources":["../../../../src/agents/agent_toolkits/vectorstore/vectorstore.ts"],"names":[],"mappings":"AAAA,OAAO,EAAQ,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAE/D,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAErC,OAAO,EAAoB,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACtE,OAAO,EAAE,aAAa,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAClE,OAAO,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AACpD,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAQlD,MAAM,OAAO,kBAAmB,SAAQ,OAAO;IAK7C,YAAY,eAAgC,EAAE,GAAY;QACxD,KAAK,EAAE,CAAC;QALV;;;;;WAAc;QAEd;;;;;WAAa;QAIX,MAAM,WAAW,GAAG,iBAAiB,CAAC,cAAc,CAClD,eAAe,CAAC,IAAI,EACpB,eAAe,CAAC,WAAW,CAC5B,CAAC;QACF,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,KAAK,GAAG;YACX,IAAI,iBAAiB,CAAC,eAAe,CAAC,IAAI,EAAE,WAAW,EAAE;gBACvD,WAAW,EAAE,eAAe,CAAC,WAAW;gBACxC,GAAG,EAAE,IAAI,CAAC,GAAG;aACd,CAAC;SACH,CAAC;IACJ,CAAC;CACF;AAED,MAAM,OAAO,wBAAyB,SAAQ,OAAO;IAOnD,YAAY,gBAAmC,EAAE,GAAY;QAC3D,KAAK,EAAE,CAAC;QAPV;;;;;WAAc;QAEd;;;;;WAAoC;QAEpC;;;;;WAAa;QAIX,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,KAAK,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC,eAAe,EAAE,EAAE;YACpD,MAAM,WAAW,GAAG,iBAAiB,CAAC,cAAc,CAClD,eAAe,CAAC,IAAI,EACpB,eAAe,CAAC,WAAW,CAC5B,CAAC;YACF,OAAO,IAAI,iBAAiB,CAAC,eAAe,CAAC,IAAI,EAAE,WAAW,EAAE;gBAC9D,WAAW,EAAE,eAAe,CAAC,WAAW;gBACxC,GAAG,EAAE,IAAI,CAAC,GAAG;aACd,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAED,MAAM,UAAU,sBAAsB,CACpC,GAAY,EACZ,OAA2B,EAC3B,IAAuB;IAEvB,MAAM,EACJ,MAAM,GAAG,aAAa,EACtB,MAAM,GAAG,MAAM,EACf,cAAc,GAAG,CAAC,OAAO,EAAE,kBAAkB,CAAC,GAC/C,GAAG,IAAI,IAAI,EAAE,CAAC;IACf,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC;IAC1B,MAAM,MAAM,GAAG,aAAa,CAAC,YAAY,CAAC,KAAK,EAAE;QAC/C,MAAM;QACN,MAAM;QACN,cAAc;KACf,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5C,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC;QAC9B,QAAQ,EAAE,KAAK;QACf,YAAY,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;KACvC,CAAC,CAAC;IACH,OAAO,aAAa,CAAC,iBAAiB,CAAC;QACrC,KAAK;QACL,KAAK;QACL,uBAAuB,EAAE,IAAI;KAC9B,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,4BAA4B,CAC1C,GAAY,EACZ,OAAiC,EACjC,IAAuB;IAEvB,MAAM,EACJ,MAAM,GAAG,oBAAoB,EAC7B,MAAM,GAAG,MAAM,EACf,cAAc,GAAG,CAAC,OAAO,EAAE,kBAAkB,CAAC,GAC/C,GAAG,IAAI,IAAI,EAAE,CAAC;IACf,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC;IAC1B,MAAM,MAAM,GAAG,aAAa,CAAC,YAAY,CAAC,KAAK,EAAE;QAC/C,MAAM;QACN,MAAM;QACN,cAAc;KACf,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5C,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC;QAC9B,QAAQ,EAAE,KAAK;QACf,YAAY,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;KACvC,CAAC,CAAC;IACH,OAAO,aAAa,CAAC,iBAAiB,CAAC;QACrC,KAAK;QACL,KAAK;QACL,uBAAuB,EAAE,IAAI;KAC9B,CAAC,CAAC;AACL,CAAC"}

package/dist/agents/index.d.ts

@@ -5,4 +5,4 @@ export { ZeroShotAgent, SerializedZeroShotAgent } from "./mrkl/index.js";
 export { Tool } from "./tools/index.js";
 export { initializeAgentExecutor } from "./initialize.js";
 export { loadAgent } from "./load.js";
-export { SqlToolkit, JsonToolkit, RequestsToolkit, OpenApiToolkit, createSqlAgent, createJsonAgent, createOpenApiAgent, } from "./agent_toolkits/index.js";
+export { SqlToolkit, JsonToolkit, RequestsToolkit, OpenApiToolkit, VectorStoreInfo, VectorStoreToolkit, VectorStoreRouterToolkit, createSqlAgent, createJsonAgent, createOpenApiAgent, createVectorStoreAgent, } from "./agent_toolkits/index.js";

package/dist/agents/index.js

@@ -4,5 +4,5 @@ export { ZeroShotAgent } from "./mrkl/index.js";
 export { Tool } from "./tools/index.js";
 export { initializeAgentExecutor } from "./initialize.js";
 export { loadAgent } from "./load.js";
-export { SqlToolkit, JsonToolkit, RequestsToolkit, OpenApiToolkit, createSqlAgent, createJsonAgent, createOpenApiAgent, } from "./agent_toolkits/index.js";
+export { SqlToolkit, JsonToolkit, RequestsToolkit, OpenApiToolkit, VectorStoreToolkit, VectorStoreRouterToolkit, createSqlAgent, createJsonAgent, createOpenApiAgent, createVectorStoreAgent, } from "./agent_toolkits/index.js";
 //# sourceMappingURL=index.js.map

package/dist/agents/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/agents/index.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,KAAK,EAAe,gBAAgB,EAAc,MAAM,YAAY,CAAC;AAC9E,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAC9C,OAAO,EAAE,aAAa,EAA2B,MAAM,iBAAiB,CAAC;AACzE,OAAO,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AACxC,OAAO,EAAE,uBAAuB,EAAE,MAAM,iBAAiB,CAAC;AAE1D,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAEtC,OAAO,EACL,UAAU,EACV,WAAW,EACX,eAAe,EACf,cAAc,EACd,cAAc,EACd,eAAe,EACf,kBAAkB,GACnB,MAAM,2BAA2B,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/agents/index.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,KAAK,EAAe,gBAAgB,EAAc,MAAM,YAAY,CAAC;AAC9E,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAC9C,OAAO,EAAE,aAAa,EAA2B,MAAM,iBAAiB,CAAC;AACzE,OAAO,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AACxC,OAAO,EAAE,uBAAuB,EAAE,MAAM,iBAAiB,CAAC;AAE1D,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAEtC,OAAO,EACL,UAAU,EACV,WAAW,EACX,eAAe,EACf,cAAc,EAEd,kBAAkB,EAClB,wBAAwB,EACxB,cAAc,EACd,eAAe,EACf,kBAAkB,EAClB,sBAAsB,GACvB,MAAM,2BAA2B,CAAC"}

package/dist/agents/tools/index.d.ts

@@ -9,3 +9,4 @@ export { ChainTool } from "./chain.js";
 export { SqlDatabase, QuerySqlTool, InfoSqlTool, ListTablesSqlTool, QueryCheckerTool, } from "./sql.js";
 export { JsonSpec, JsonListKeysTool, JsonGetValueTool, JsonObject, Json, } from "./json.js";
 export { RequestsGetTool, RequestsPostTool } from "./requests.js";
+export { VectorStoreQATool } from "./vectorstore.js";

package/dist/agents/tools/index.js

@@ -9,4 +9,5 @@ export { ChainTool } from "./chain.js";
 export { SqlDatabase, QuerySqlTool, InfoSqlTool, ListTablesSqlTool, QueryCheckerTool, } from "./sql.js";
 export { JsonSpec, JsonListKeysTool, JsonGetValueTool, } from "./json.js";
 export { RequestsGetTool, RequestsPostTool } from "./requests.js";
+export { VectorStoreQATool } from "./vectorstore.js";
 //# sourceMappingURL=index.js.map

package/dist/agents/tools/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/agents/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AACvC,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAC3C,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AACvC,OAAO,EACL,WAAW,EACX,YAAY,EACZ,WAAW,EACX,iBAAiB,EACjB,gBAAgB,GACjB,MAAM,UAAU,CAAC;AAClB,OAAO,EACL,QAAQ,EACR,gBAAgB,EAChB,gBAAgB,GAGjB,MAAM,WAAW,CAAC;AACnB,OAAO,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/agents/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AACvC,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAC3C,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AACvC,OAAO,EACL,WAAW,EACX,YAAY,EACZ,WAAW,EACX,iBAAiB,EACjB,gBAAgB,GACjB,MAAM,UAAU,CAAC;AAClB,OAAO,EACL,QAAQ,EACR,gBAAgB,EAChB,gBAAgB,GAGjB,MAAM,WAAW,CAAC;AACnB,OAAO,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AAClE,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC"}

package/dist/agents/tools/vectorstore.d.ts

@@ -0,0 +1,19 @@
+import { VectorStore } from "../../vectorstores/index.js";
+import { BaseLLM } from "../../llms/index.js";
+import { VectorDBQAChain } from "../../chains/index.js";
+import { Tool } from "./base.js";
+interface VectorStoreTool {
+    vectorStore: VectorStore;
+    llm: BaseLLM;
+}
+export declare class VectorStoreQATool extends Tool implements VectorStoreTool {
+    vectorStore: VectorStore;
+    llm: BaseLLM;
+    name: string;
+    description: string;
+    chain: VectorDBQAChain;
+    constructor(name: string, description: string, fields: VectorStoreTool);
+    static getDescription(name: string, description: string): string;
+    call(input: string): Promise<string>;
+}
+export {};

package/dist/agents/tools/vectorstore.js

@@ -0,0 +1,49 @@
+import { VectorDBQAChain } from "../../chains/index.js";
+import { Tool } from "./base.js";
+export class VectorStoreQATool extends Tool {
+    constructor(name, description, fields) {
+        super();
+        Object.defineProperty(this, "vectorStore", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "llm", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "name", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "description", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "chain", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.name = name;
+        this.description = description;
+        this.vectorStore = fields.vectorStore;
+        this.llm = fields.llm;
+        this.chain = VectorDBQAChain.fromLLM(this.llm, this.vectorStore);
+    }
+    static getDescription(name, description) {
+        return `Useful for when you need to answer questions about ${name}. Whenever you need information about ${description} you should ALWAYS use this. Input should be a fully formed question.`;
+    }
+    async call(input) {
+        return this.chain.run(input);
+    }
+}
+//# sourceMappingURL=vectorstore.js.map

package/dist/agents/tools/vectorstore.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vectorstore.js","sourceRoot":"","sources":["../../../src/agents/tools/vectorstore.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAOjC,MAAM,OAAO,iBAAkB,SAAQ,IAAI;IAWzC,YAAY,IAAY,EAAE,WAAmB,EAAE,MAAuB;QACpE,KAAK,EAAE,CAAC;QAXV;;;;;WAAyB;QAEzB;;;;;WAAa;QAEb;;;;;WAAa;QAEb;;;;;WAAoB;QAEpB;;;;;WAAuB;QAIrB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;QACtB,IAAI,CAAC,KAAK,GAAG,eAAe,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;IACnE,CAAC;IAED,MAAM,CAAC,cAAc,CAAC,IAAY,EAAE,WAAmB;QACrD,OAAO,sDAAsD,IAAI,yCAAyC,WAAW,uEAAuE,CAAC;IAC/L,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,KAAa;QACtB,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;CACF"}

package/dist/chains/combine_docs_chain.js

@@ -143,9 +143,11 @@ export class MapReduceDocumentsChain extends BaseChain {
                 [this.documentVariableName]: d.pageContent,
                 ...rest,
             }));
-            const length = inputs
-                .map((i) => this.llmChain.llm.getNumTokens(this.llmChain.prompt.format(i)))
-                .reduce((a, b) => a + b, 0);
+            const promises = inputs.map(async (i) => {
+                const prompt = await this.llmChain.prompt.format(i);
+                return this.llmChain.llm.getNumTokens(prompt);
+            });
+            const length = await Promise.all(promises).then((results) => results.reduce((a, b) => a + b, 0));
             if (length < this.maxTokens) {
                 break;
             }

package/dist/chains/combine_docs_chain.js.map

@@ -1 +1 @@
-{"version":3,"file":"combine_docs_chain.js","sourceRoot":"","sources":["../../src/chains/combine_docs_chain.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,SAAS,EAET,QAAQ,GAGT,MAAM,YAAY,CAAC;AAIpB,OAAO,EAAE,qBAAqB,EAAE,MAAM,kBAAkB,CAAC;AAiBzD;;;;GAIG;AACH,MAAM,OAAO,mBACX,SAAQ,SAAS;IAWjB,IAAI,SAAS;QACX,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IACrD,CAAC;IAED,YAAY,MAKX;QACC,KAAK,EAAE,CAAC;QAlBV;;;;;WAAmB;QAEnB;;;;mBAAW,iBAAiB;WAAC;QAE7B;;;;mBAAY,aAAa;WAAC;QAE1B;;;;mBAAuB,SAAS;WAAC;QAa/B,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,oBAAoB;YACvB,MAAM,CAAC,oBAAoB,IAAI,IAAI,CAAC,oBAAoB,CAAC;QAC3D,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC;QACjD,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAmB;QAC7B,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,IAAI,MAAM,CAAC,EAAE;YAC9B,MAAM,IAAI,KAAK,CAAC,gBAAgB,IAAI,CAAC,QAAQ,aAAa,CAAC,CAAC;SAC7D;QACD,MAAM,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QAClD,MAAM,KAAK,GAAI,IAAmB,CAAC,GAAG,CAAC,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC;QACzE,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAChC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACtC,GAAG,IAAI;YACP,CAAC,IAAI,CAAC,oBAAoB,CAAC,EAAE,IAAI;SAClC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,UAAU;QACR,OAAO,uBAAgC,CAAC;IAC1C,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAmC;QAC1D,MAAM,kBAAkB,GAAG,MAAM,qBAAqB,CAGpD,WAAW,EAAE,IAAI,CAAC,CAAC;QAErB,OAAO,IAAI,mBAAmB,CAAC;YAC7B,QAAQ,EAAE,MAAM,QAAQ,CAAC,WAAW,CAAC,kBAAkB,CAAC;SACzD,CAAC,CAAC;IACL,CAAC;IAED,SAAS;QACP,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE;YACxB,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE;SACrC,CAAC;IACJ,CAAC;CACF;AAgBD;;;;GAIG;AACH,MAAM,OAAO,uBACX,SAAQ,SAAS;IAWjB,IAAI,SAAS;QACX,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;IACjE,CAAC;IAQD,YAAY,MAQX;QACC,KAAK,EAAE,CAAC;QA3BV;;;;;WAAmB;QAEnB;;;;mBAAW,iBAAiB;WAAC;QAE7B;;;;mBAAY,aAAa;WAAC;QAE1B;;;;mBAAuB,SAAS;WAAC;QAMjC;;;;mBAAY,IAAI;WAAC;QAEjB;;;;mBAAgB,EAAE;WAAC;QAEnB;;;;;WAAgC;QAY9B,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,oBAAoB,GAAG,MAAM,CAAC,oBAAoB,CAAC;QACxD,IAAI,CAAC,oBAAoB;YACvB,MAAM,CAAC,oBAAoB,IAAI,IAAI,CAAC,oBAAoB,CAAC;QAC3D,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC;QACjD,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACpD,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACpD,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,aAAa,IAAI,IAAI,CAAC,aAAa,CAAC;IAClE,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAmB;QAC7B,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,IAAI,MAAM,CAAC,EAAE;YAC9B,MAAM,IAAI,KAAK,CAAC,gBAAgB,IAAI,CAAC,QAAQ,aAAa,CAAC,CAAC;SAC7D;QACD,MAAM,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QAElD,IAAI,WAAW,GAAG,IAAkB,CAAC;QAErC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC,IAAI,CAAC,EAAE;YAC9C,MAAM,MAAM,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACrC,CAAC,IAAI,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC,WAAW;gBAC1C,GAAG,IAAI;aACR,CAAC,CAAC,CAAC;YACJ,MAAM,MAAM,GAAG,MAAM;iBAClB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CACT,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAC/D;iBACA,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YAE9B,IAAI,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE;gBAC3B,MAAM;aACP;YAED,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAClD,MAAM,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC;YAEpC,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAc,EAAE,EAAE,CAAC,CAAC;gBAC7C,WAAW,EAAE,CAAC,CAAC,SAAS,CAAC;aAC1B,CAAC,CAAC,CAAC;SACL;QACD,MAAM,SAAS,GAAG,EAAE,eAAe,EAAE,WAAW,EAAE,GAAG,IAAI,EAAE,CAAC;QAC5D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/D,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,UAAU;QACR,OAAO,4BAAqC,CAAC;IAC/C,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAuC;QAC9D,MAAM,kBAAkB,GAAG,MAAM,qBAAqB,CAGpD,WAAW,EAAE,IAAI,CAAC,CAAC;QAErB,MAAM,8BAA8B,GAAG,MAAM,qBAAqB,CAGhE,wBAAwB,EAAE,IAAI,CAAC,CAAC;QAElC,OAAO,IAAI,uBAAuB,CAAC;YACjC,QAAQ,EAAE,MAAM,QAAQ,CAAC,WAAW,CAAC,kBAAkB,CAAC;YACxD,oBAAoB,EAAE,MAAM,SAAS,CAAC,WAAW,CAC/C,8BAA8B,CAC/B;SACF,CAAC,CAAC;IACL,CAAC;IAED,SAAS;QACP,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE;YACxB,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE;YACpC,sBAAsB,EAAE,IAAI,CAAC,oBAAoB,CAAC,SAAS,EAAE;SAC9D,CAAC;IACJ,CAAC;CACF"}
+{"version":3,"file":"combine_docs_chain.js","sourceRoot":"","sources":["../../src/chains/combine_docs_chain.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,SAAS,EAET,QAAQ,GAGT,MAAM,YAAY,CAAC;AAIpB,OAAO,EAAE,qBAAqB,EAAE,MAAM,kBAAkB,CAAC;AAiBzD;;;;GAIG;AACH,MAAM,OAAO,mBACX,SAAQ,SAAS;IAWjB,IAAI,SAAS;QACX,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IACrD,CAAC;IAED,YAAY,MAKX;QACC,KAAK,EAAE,CAAC;QAlBV;;;;;WAAmB;QAEnB;;;;mBAAW,iBAAiB;WAAC;QAE7B;;;;mBAAY,aAAa;WAAC;QAE1B;;;;mBAAuB,SAAS;WAAC;QAa/B,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,oBAAoB;YACvB,MAAM,CAAC,oBAAoB,IAAI,IAAI,CAAC,oBAAoB,CAAC;QAC3D,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC;QACjD,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAmB;QAC7B,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,IAAI,MAAM,CAAC,EAAE;YAC9B,MAAM,IAAI,KAAK,CAAC,gBAAgB,IAAI,CAAC,QAAQ,aAAa,CAAC,CAAC;SAC7D;QACD,MAAM,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QAClD,MAAM,KAAK,GAAI,IAAmB,CAAC,GAAG,CAAC,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC;QACzE,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAChC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACtC,GAAG,IAAI;YACP,CAAC,IAAI,CAAC,oBAAoB,CAAC,EAAE,IAAI;SAClC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,UAAU;QACR,OAAO,uBAAgC,CAAC;IAC1C,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAmC;QAC1D,MAAM,kBAAkB,GAAG,MAAM,qBAAqB,CAGpD,WAAW,EAAE,IAAI,CAAC,CAAC;QAErB,OAAO,IAAI,mBAAmB,CAAC;YAC7B,QAAQ,EAAE,MAAM,QAAQ,CAAC,WAAW,CAAC,kBAAkB,CAAC;SACzD,CAAC,CAAC;IACL,CAAC;IAED,SAAS;QACP,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE;YACxB,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE;SACrC,CAAC;IACJ,CAAC;CACF;AAgBD;;;;GAIG;AACH,MAAM,OAAO,uBACX,SAAQ,SAAS;IAWjB,IAAI,SAAS;QACX,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;IACjE,CAAC;IAQD,YAAY,MAQX;QACC,KAAK,EAAE,CAAC;QA3BV;;;;;WAAmB;QAEnB;;;;mBAAW,iBAAiB;WAAC;QAE7B;;;;mBAAY,aAAa;WAAC;QAE1B;;;;mBAAuB,SAAS;WAAC;QAMjC;;;;mBAAY,IAAI;WAAC;QAEjB;;;;mBAAgB,EAAE;WAAC;QAEnB;;;;;WAAgC;QAY9B,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,oBAAoB,GAAG,MAAM,CAAC,oBAAoB,CAAC;QACxD,IAAI,CAAC,oBAAoB;YACvB,MAAM,CAAC,oBAAoB,IAAI,IAAI,CAAC,oBAAoB,CAAC;QAC3D,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC;QACjD,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACpD,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACpD,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,aAAa,IAAI,IAAI,CAAC,aAAa,CAAC;IAClE,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAmB;QAC7B,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,IAAI,MAAM,CAAC,EAAE;YAC9B,MAAM,IAAI,KAAK,CAAC,gBAAgB,IAAI,CAAC,QAAQ,aAAa,CAAC,CAAC;SAC7D;QACD,MAAM,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QAElD,IAAI,WAAW,GAAG,IAAkB,CAAC;QAErC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC,IAAI,CAAC,EAAE;YAC9C,MAAM,MAAM,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACrC,CAAC,IAAI,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC,WAAW;gBAC1C,GAAG,IAAI;aACR,CAAC,CAAC,CAAC;YACJ,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE;gBACtC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBACpD,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YAChD,CAAC,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAC1D,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CACnC,CAAC;YAEF,IAAI,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE;gBAC3B,MAAM;aACP;YAED,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAClD,MAAM,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC;YAEpC,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAc,EAAE,EAAE,CAAC,CAAC;gBAC7C,WAAW,EAAE,CAAC,CAAC,SAAS,CAAC;aAC1B,CAAC,CAAC,CAAC;SACL;QACD,MAAM,SAAS,GAAG,EAAE,eAAe,EAAE,WAAW,EAAE,GAAG,IAAI,EAAE,CAAC;QAC5D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/D,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,UAAU;QACR,OAAO,4BAAqC,CAAC;IAC/C,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAuC;QAC9D,MAAM,kBAAkB,GAAG,MAAM,qBAAqB,CAGpD,WAAW,EAAE,IAAI,CAAC,CAAC;QAErB,MAAM,8BAA8B,GAAG,MAAM,qBAAqB,CAGhE,wBAAwB,EAAE,IAAI,CAAC,CAAC;QAElC,OAAO,IAAI,uBAAuB,CAAC;YACjC,QAAQ,EAAE,MAAM,QAAQ,CAAC,WAAW,CAAC,kBAAkB,CAAC;YACxD,oBAAoB,EAAE,MAAM,SAAS,CAAC,WAAW,CAC/C,8BAA8B,CAC/B;SACF,CAAC,CAAC;IACL,CAAC;IAED,SAAS;QACP,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE;YACxB,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE;YACpC,sBAAsB,EAAE,IAAI,CAAC,oBAAoB,CAAC,SAAS,EAAE;SAC9D,CAAC;IACJ,CAAC;CACF"}

package/dist/chains/llm_chain.js

@@ -48,7 +48,7 @@ export class LLMChain extends BaseChain {
         if ("stop" in values && Array.isArray(values.stop)) {
             stop = values.stop;
         }
-        const formattedString = this.prompt.format(values);
+        const formattedString = await this.prompt.format(values);
         const llmResult = await this.llm.call(formattedString, stop);
         const result = { [this.outputKey]: llmResult };
         return result;

package/dist/chains/llm_chain.js.map

@@ -1 +1 @@
-{"version":3,"file":"llm_chain.js","sourceRoot":"","sources":["../../src/chains/llm_chain.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAA4B,MAAM,YAAY,CAAC;AAEjE,OAAO,EAAE,OAAO,EAAiB,MAAM,kBAAkB,CAAC;AAE1D,OAAO,EAAc,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAC9D,OAAO,EACL,kBAAkB,EAElB,cAAc,GACf,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAE,qBAAqB,EAAE,MAAM,kBAAkB,CAAC;AAoBzD;;;;;;;;;;;GAWG;AACH,MAAM,OAAO,QAAS,SAAQ,SAAS;IAOrC,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC;IACpC,CAAC;IAED,YAAY,MAKX;QACC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAhBvB;;;;;WAA2B;QAE3B;;;;;WAAa;QAEb;;;;mBAAY,MAAM;WAAC;QAajB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAmB;QAC7B,IAAI,IAAI,CAAC;QACT,IAAI,MAAM,IAAI,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE;YAClD,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;SACpB;QACD,MAAM,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACnD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;QAC7D,MAAM,MAAM,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,CAAC;QAC/C,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;;;;;;OAUG;IACH,KAAK,CAAC,OAAO,CAAC,MAAmB;QAC/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvC,OAAO,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAChC,CAAC;IAED,UAAU;QACR,OAAO,WAAoB,CAAC;IAC9B,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAwB;QAC/C,MAAM,aAAa,GAAG,MAAM,qBAAqB,CAC/C,KAAK,EACL,IAAI,CACL,CAAC;QACF,MAAM,gBAAgB,GAAG,MAAM,qBAAqB,CAGlD,QAAQ,EAAE,IAAI,CAAC,CAAC;QAElB,OAAO,IAAI,QAAQ,CAAC;YAClB,GAAG,EAAE,MAAM,OAAO,CAAC,WAAW,CAAC,aAAa,CAAC;YAC7C,MAAM,EAAE,MAAM,kBAAkB,CAAC,WAAW,CAAC,gBAAgB,CAAC;SAC/D,CAAC,CAAC;IACL,CAAC;IAED,SAAS;QACP,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE;YACxB,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE;SAChC,CAAC;IACJ,CAAC;CACF;AAED,mCAAmC;AACnC,MAAM,eAAe,GAAG;;;;;IAKpB,CAAC;AAEL,MAAM,aAAa,GAAG,IAAI,cAAc,CAAC;IACvC,QAAQ,EAAE,eAAe;IACzB,cAAc,EAAE,CAAC,SAAS,EAAE,OAAO,CAAC;CACrC,CAAC,CAAC;AAEH,MAAM,OAAO,iBAAkB,SAAQ,QAAQ;IAC7C,YAAY,MAKX;QACC,KAAK,CAAC;YACJ,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,aAAa;YACtC,GAAG,EAAE,MAAM,CAAC,GAAG;YACf,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,UAAU;SAC1C,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;IACpD,CAAC;CACF"}
+{"version":3,"file":"llm_chain.js","sourceRoot":"","sources":["../../src/chains/llm_chain.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAA4B,MAAM,YAAY,CAAC;AAEjE,OAAO,EAAE,OAAO,EAAiB,MAAM,kBAAkB,CAAC;AAE1D,OAAO,EAAc,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAC9D,OAAO,EACL,kBAAkB,EAElB,cAAc,GACf,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAE,qBAAqB,EAAE,MAAM,kBAAkB,CAAC;AAoBzD;;;;;;;;;;;GAWG;AACH,MAAM,OAAO,QAAS,SAAQ,SAAS;IAOrC,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC;IACpC,CAAC;IAED,YAAY,MAKX;QACC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAhBvB;;;;;WAA2B;QAE3B;;;;;WAAa;QAEb;;;;mBAAY,MAAM;WAAC;QAajB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAmB;QAC7B,IAAI,IAAI,CAAC;QACT,IAAI,MAAM,IAAI,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE;YAClD,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;SACpB;QACD,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACzD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;QAC7D,MAAM,MAAM,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,CAAC;QAC/C,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;;;;;;OAUG;IACH,KAAK,CAAC,OAAO,CAAC,MAAmB;QAC/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvC,OAAO,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAChC,CAAC;IAED,UAAU;QACR,OAAO,WAAoB,CAAC;IAC9B,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAwB;QAC/C,MAAM,aAAa,GAAG,MAAM,qBAAqB,CAC/C,KAAK,EACL,IAAI,CACL,CAAC;QACF,MAAM,gBAAgB,GAAG,MAAM,qBAAqB,CAGlD,QAAQ,EAAE,IAAI,CAAC,CAAC;QAElB,OAAO,IAAI,QAAQ,CAAC;YAClB,GAAG,EAAE,MAAM,OAAO,CAAC,WAAW,CAAC,aAAa,CAAC;YAC7C,MAAM,EAAE,MAAM,kBAAkB,CAAC,WAAW,CAAC,gBAAgB,CAAC;SAC/D,CAAC,CAAC;IACL,CAAC;IAED,SAAS;QACP,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE;YACxB,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE;SAChC,CAAC;IACJ,CAAC;CACF;AAED,mCAAmC;AACnC,MAAM,eAAe,GAAG;;;;;IAKpB,CAAC;AAEL,MAAM,aAAa,GAAG,IAAI,cAAc,CAAC;IACvC,QAAQ,EAAE,eAAe;IACzB,cAAc,EAAE,CAAC,SAAS,EAAE,OAAO,CAAC;CACrC,CAAC,CAAC;AAEH,MAAM,OAAO,iBAAkB,SAAQ,QAAQ;IAC7C,YAAY,MAKX;QACC,KAAK,CAAC;YACJ,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,aAAa;YACtC,GAAG,EAAE,MAAM,CAAC,GAAG;YACf,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,UAAU;SAC1C,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;IACpD,CAAC;CACF"}

package/dist/document_loaders/tests/{college_confidential.test.js → college_confidential.int.test.js}

@@ -4,4 +4,4 @@ test("Test College confidential loader", async () => {
     const loader = new CollegeConfidentialLoader("https://www.collegeconfidential.com/colleges/brown-university/");
     await loader.load();
 }, 10000);
-//# sourceMappingURL=college_confidential.test.js.map
+//# sourceMappingURL=college_confidential.int.test.js.map

package/dist/document_loaders/tests/college_confidential.int.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"college_confidential.int.test.js","sourceRoot":"","sources":["../../../src/document_loaders/tests/college_confidential.int.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,EAAE,yBAAyB,EAAE,MAAM,4BAA4B,CAAC;AAEvE,IAAI,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;IAClD,MAAM,MAAM,GAAG,IAAI,yBAAyB,CAC1C,gEAAgE,CACjE,CAAC;IACF,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;AACtB,CAAC,EAAE,KAAK,CAAC,CAAC"}