langchain 0.0.36 → 0.0.37

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (71) hide show
  1. package/base_language.d.ts +1 -0
  2. package/base_language.js +1 -0
  3. package/dist/agents/agent.d.ts +3 -3
  4. package/dist/agents/agent.js.map +1 -1
  5. package/dist/agents/agent_toolkits/json/json.d.ts +2 -2
  6. package/dist/agents/agent_toolkits/json/json.js.map +1 -1
  7. package/dist/agents/agent_toolkits/openapi/openapi.d.ts +3 -3
  8. package/dist/agents/agent_toolkits/openapi/openapi.js.map +1 -1
  9. package/dist/agents/agent_toolkits/sql/sql.d.ts +2 -2
  10. package/dist/agents/agent_toolkits/sql/sql.js +2 -2
  11. package/dist/agents/agent_toolkits/sql/sql.js.map +1 -1
  12. package/dist/agents/agent_toolkits/vectorstore/vectorstore.d.ts +7 -7
  13. package/dist/agents/agent_toolkits/vectorstore/vectorstore.js.map +1 -1
  14. package/dist/agents/chat_convo/index.d.ts +3 -4
  15. package/dist/agents/chat_convo/index.js +16 -13
  16. package/dist/agents/chat_convo/index.js.map +1 -1
  17. package/dist/agents/index.d.ts +1 -1
  18. package/dist/agents/index.js +1 -1
  19. package/dist/agents/index.js.map +1 -1
  20. package/dist/agents/load.d.ts +2 -2
  21. package/dist/agents/load.js.map +1 -1
  22. package/dist/agents/tools/vectorstore.d.ts +3 -3
  23. package/dist/agents/tools/vectorstore.js.map +1 -1
  24. package/dist/agents/tools/zapier.d.ts +0 -1
  25. package/dist/agents/tools/zapier.js +4 -10
  26. package/dist/agents/tools/zapier.js.map +1 -1
  27. package/dist/base_language/index.d.ts +17 -0
  28. package/dist/base_language/index.js +33 -0
  29. package/dist/base_language/index.js.map +1 -1
  30. package/dist/chains/chat_vector_db_chain.d.ts +2 -2
  31. package/dist/chains/chat_vector_db_chain.js.map +1 -1
  32. package/dist/chains/llm_chain.js +3 -3
  33. package/dist/chains/llm_chain.js.map +1 -1
  34. package/dist/chains/prompt_selector.d.ts +1 -2
  35. package/dist/chains/prompt_selector.js +2 -2
  36. package/dist/chains/prompt_selector.js.map +1 -1
  37. package/dist/chains/question_answering/load.d.ts +4 -4
  38. package/dist/chains/question_answering/load.js.map +1 -1
  39. package/dist/chains/sql_db/sql_db_chain.d.ts +3 -3
  40. package/dist/chains/sql_db/sql_db_chain.js +2 -2
  41. package/dist/chains/sql_db/sql_db_chain.js.map +1 -1
  42. package/dist/chains/summarization/load.d.ts +2 -2
  43. package/dist/chains/summarization/load.js.map +1 -1
  44. package/dist/chains/tests/llm_chain.int.test.js +14 -0
  45. package/dist/chains/tests/llm_chain.int.test.js.map +1 -1
  46. package/dist/chains/vector_db_qa.d.ts +2 -2
  47. package/dist/chains/vector_db_qa.js.map +1 -1
  48. package/dist/chat_models/base.d.ts +5 -8
  49. package/dist/chat_models/base.js +7 -21
  50. package/dist/chat_models/base.js.map +1 -1
  51. package/dist/chat_models/openai.d.ts +15 -6
  52. package/dist/chat_models/openai.js +20 -1
  53. package/dist/chat_models/openai.js.map +1 -1
  54. package/dist/chat_models/tests/chatopenai.int.test.js +25 -1
  55. package/dist/chat_models/tests/chatopenai.int.test.js.map +1 -1
  56. package/dist/llms/load.d.ts +2 -2
  57. package/dist/llms/load.js +2 -2
  58. package/dist/llms/load.js.map +1 -1
  59. package/dist/llms/openai-chat.d.ts +8 -2
  60. package/dist/llms/openai-chat.js +8 -0
  61. package/dist/llms/openai-chat.js.map +1 -1
  62. package/dist/llms/openai.js +2 -1
  63. package/dist/llms/openai.js.map +1 -1
  64. package/dist/llms/tests/openai-chat.int.test.js +3 -1
  65. package/dist/llms/tests/openai-chat.int.test.js.map +1 -1
  66. package/dist/prompts/index.d.ts +1 -0
  67. package/dist/prompts/index.js +1 -0
  68. package/dist/prompts/index.js.map +1 -1
  69. package/dist/prompts/tests/few_shot.test.js +1 -1
  70. package/dist/prompts/tests/few_shot.test.js.map +1 -1
  71. package/package.json +7 -1
@@ -0,0 +1 @@
1
+ export * from './dist/base_language/index.js'
@@ -0,0 +1 @@
1
+ export * from './dist/base_language/index.js'
@@ -1,4 +1,4 @@
1
- import { BaseLLM } from "../llms/index.js";
1
+ import { BaseLanguageModel } from "../base_language/index.js";
2
2
  import { LLMChain } from "../chains/llm_chain.js";
3
3
  import { BasePromptTemplate } from "../prompts/index.js";
4
4
  import { AgentAction, AgentFinish, AgentStep, ChainValues, BaseChatMessage } from "../schema/index.js";
@@ -54,7 +54,7 @@ export declare abstract class Agent {
54
54
  * */
55
55
  static createPrompt(_tools: Tool[], _fields?: Record<string, any>): BasePromptTemplate;
56
56
  /** Construct an agent from an LLM and a list of tools */
57
- static fromLLMAndTools(_llm: BaseLLM, _tools: Tool[], _args?: Record<string, any>): Agent;
57
+ static fromLLMAndTools(_llm: BaseLanguageModel, _tools: Tool[], _args?: Record<string, any>): Agent;
58
58
  /**
59
59
  * Validate that appropriate tools are passed in
60
60
  */
@@ -86,7 +86,7 @@ export declare abstract class Agent {
86
86
  * Load an agent from a json-like object describing it.
87
87
  */
88
88
  static deserialize(data: SerializedAgent & {
89
- llm?: BaseLLM;
89
+ llm?: BaseLanguageModel;
90
90
  tools?: Tool[];
91
91
  }): Promise<Agent>;
92
92
  }
@@ -1 +1 @@
1
- {"version":3,"file":"agent.js","sourceRoot":"","sources":["../../src/agents/agent.ts"],"names":[],"mappings":"AAaA,MAAM,UAAW,SAAQ,KAAK;IAG5B,YAAY,GAAW,EAAE,MAAc;QACrC,KAAK,CAAC,GAAG,CAAC,CAAC;QAHb;;;;;WAAe;QAIb,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;CACF;AAED;;;;;;GAMG;AACH,MAAM,OAAgB,KAAK;IAOzB,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,kBAAkB,CAAC,CAAC;IACzE,CAAC;IAED,YAAY,KAAiB;QAV7B;;;;;WAAmB;QAEnB;;;;mBAA0B,SAAS;WAAC;QAEpC;;;;mBAAe,CAAC,QAAQ,CAAC;WAAC;QAOxB,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC,YAAY,CAAC;IACzC,CAAC;IAwBD;;OAEG;IACH,iBAAiB,KAAU,CAAC;IAE5B;;OAEG;IACH,KAAK,CAAC,gBAAgB,CACpB,aAA0C,EAC1C,MAAmB;QAEnB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED;;;;;;;SAOK;IACL,MAAM,CAAC,YAAY,CACjB,MAAc;IACd,8DAA8D;IAC9D,OAA6B;QAE7B,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;IAED,yDAAyD;IACzD,MAAM,CAAC,eAAe,CACpB,IAAa,EACb,MAAc;IACd,8DAA8D;IAC9D,KAA2B;QAE3B,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,aAAa,CAAC,MAAc,IAAS,CAAC;IAE7C,KAAK;QACH,OAAO,CAAC,KAAK,IAAI,CAAC,iBAAiB,EAAE,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,cAAc,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,mBAAmB,CAAC,KAAkB;QACpC,OAAO,KAAK,CAAC,MAAM,CACjB,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE,CACpC,QAAQ;YACR;gBACE,MAAM,CAAC,GAAG;gBACV,GAAG,IAAI,CAAC,iBAAiB,EAAE,GAAG,WAAW,EAAE;gBAC3C,IAAI,CAAC,SAAS,EAAE;aACjB,CAAC,IAAI,CAAC,IAAI,CAAC,EACd,EAAE,CACH,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,KAAK,CACjB,KAAkB,EAClB,MAAmB,EACnB,MAAe;QAEf,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;QACjD,MAAM,SAAS,GAAgB;YAC7B,GAAG,MAAM;YACT,gBAAgB,EAAE,MAAM,CAAC,CAAC,CAAC,GAAG,QAAQ,GAAG,MAAM,EAAE,CAAC,CAAC,CAAC,QAAQ;SAC7D,CAAC;QAEF,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE;YAC7B,SAAS,CAAC,IAAI,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;SAC/B;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACtD,MAAM,MAAM,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;QAChD,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,UAAU,CAAC,mBAAmB,MAAM,EAAE,EAAE,MAAM,CAAC,CAAC;SAC3D;QACD,MAAM,MAAM,GAAG;YACb,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,SAAS,EAAE,MAAM,CAAC,KAAK;YACvB,GAAG,EAAE,MAAM;SACZ,CAAC;QACF,IAAI,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC,cAAc,EAAE,EAAE;YACzC,OAAO,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC;SACxE;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;;;OAOG;IACH,IAAI,CACF,KAAkB,EAClB,MAAmB;QAEnB,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,qBAAqB,CACzB,mBAAmC,EACnC,KAAkB,EAClB,MAAmB;QAEnB,IAAI,mBAAmB,KAAK,OAAO,EAAE;YACnC,OAAO;gBACL,YAAY,EAAE,EAAE,MAAM,EAAE,sCAAsC,EAAE;gBAChE,GAAG,EAAE,EAAE;aACR,CAAC;SACH;QAED,IAAI,mBAAmB,KAAK,UAAU,EAAE;YACtC,IAAI;gBACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAC7B,KAAK,EACL,MAAM,EACN,sEAAsE,CACvE,CAAC;gBACF,IAAI,cAAc,IAAI,MAAM,EAAE;oBAC5B,OAAO,MAAM,CAAC;iBACf;gBAED,OAAO,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,GAAG,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC;aAClE;YAAC,OAAO,GAAG,EAAE;gBACZ,IAAI,CAAC,CAAC,GAAG,YAAY,UAAU,CAAC,EAAE;oBAChC,MAAM,GAAG,CAAC;iBACX;gBACD,OAAO,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC;aAClE;SACF;QAED,MAAM,IAAI,KAAK,CAAC,4BAA4B,mBAAmB,EAAE,CAAC,CAAC;IACrE,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,WAAW,CACtB,IAAyD;QAEzD,QAAQ,IAAI,CAAC,KAAK,EAAE;YAClB,KAAK,6BAA6B,CAAC,CAAC;gBAClC,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC;gBAC1D,OAAO,aAAa,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;aACxC;YACD;gBACE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;SACzC;IACH,CAAC;CACF"}
1
+ {"version":3,"file":"agent.js","sourceRoot":"","sources":["../../src/agents/agent.ts"],"names":[],"mappings":"AAaA,MAAM,UAAW,SAAQ,KAAK;IAG5B,YAAY,GAAW,EAAE,MAAc;QACrC,KAAK,CAAC,GAAG,CAAC,CAAC;QAHb;;;;;WAAe;QAIb,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;CACF;AAED;;;;;;GAMG;AACH,MAAM,OAAgB,KAAK;IAOzB,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,kBAAkB,CAAC,CAAC;IACzE,CAAC;IAED,YAAY,KAAiB;QAV7B;;;;;WAAmB;QAEnB;;;;mBAA0B,SAAS;WAAC;QAEpC;;;;mBAAe,CAAC,QAAQ,CAAC;WAAC;QAOxB,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC,YAAY,CAAC;IACzC,CAAC;IAwBD;;OAEG;IACH,iBAAiB,KAAU,CAAC;IAE5B;;OAEG;IACH,KAAK,CAAC,gBAAgB,CACpB,aAA0C,EAC1C,MAAmB;QAEnB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED;;;;;;;SAOK;IACL,MAAM,CAAC,YAAY,CACjB,MAAc;IACd,8DAA8D;IAC9D,OAA6B;QAE7B,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;IAED,yDAAyD;IACzD,MAAM,CAAC,eAAe,CACpB,IAAuB,EACvB,MAAc;IACd,8DAA8D;IAC9D,KAA2B;QAE3B,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,aAAa,CAAC,MAAc,IAAS,CAAC;IAE7C,KAAK;QACH,OAAO,CAAC,KAAK,IAAI,CAAC,iBAAiB,EAAE,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,cAAc,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,mBAAmB,CAAC,KAAkB;QACpC,OAAO,KAAK,CAAC,MAAM,CACjB,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE,CACpC,QAAQ;YACR;gBACE,MAAM,CAAC,GAAG;gBACV,GAAG,IAAI,CAAC,iBAAiB,EAAE,GAAG,WAAW,EAAE;gBAC3C,IAAI,CAAC,SAAS,EAAE;aACjB,CAAC,IAAI,CAAC,IAAI,CAAC,EACd,EAAE,CACH,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,KAAK,CACjB,KAAkB,EAClB,MAAmB,EACnB,MAAe;QAEf,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;QACjD,MAAM,SAAS,GAAgB;YAC7B,GAAG,MAAM;YACT,gBAAgB,EAAE,MAAM,CAAC,CAAC,CAAC,GAAG,QAAQ,GAAG,MAAM,EAAE,CAAC,CAAC,CAAC,QAAQ;SAC7D,CAAC;QAEF,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE;YAC7B,SAAS,CAAC,IAAI,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;SAC/B;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACtD,MAAM,MAAM,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;QAChD,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,UAAU,CAAC,mBAAmB,MAAM,EAAE,EAAE,MAAM,CAAC,CAAC;SAC3D;QACD,MAAM,MAAM,GAAG;YACb,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,SAAS,EAAE,MAAM,CAAC,KAAK;YACvB,GAAG,EAAE,MAAM;SACZ,CAAC;QACF,IAAI,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC,cAAc,EAAE,EAAE;YACzC,OAAO,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC;SACxE;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;;;OAOG;IACH,IAAI,CACF,KAAkB,EAClB,MAAmB;QAEnB,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,qBAAqB,CACzB,mBAAmC,EACnC,KAAkB,EAClB,MAAmB;QAEnB,IAAI,mBAAmB,KAAK,OAAO,EAAE;YACnC,OAAO;gBACL,YAAY,EAAE,EAAE,MAAM,EAAE,sCAAsC,EAAE;gBAChE,GAAG,EAAE,EAAE;aACR,CAAC;SACH;QAED,IAAI,mBAAmB,KAAK,UAAU,EAAE;YACtC,IAAI;gBACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAC7B,KAAK,EACL,MAAM,EACN,sEAAsE,CACvE,CAAC;gBACF,IAAI,cAAc,IAAI,MAAM,EAAE;oBAC5B,OAAO,MAAM,CAAC;iBACf;gBAED,OAAO,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,GAAG,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC;aAClE;YAAC,OAAO,GAAG,EAAE;gBACZ,IAAI,CAAC,CAAC,GAAG,YAAY,UAAU,CAAC,EAAE;oBAChC,MAAM,GAAG,CAAC;iBACX;gBACD,OAAO,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC;aAClE;SACF;QAED,MAAM,IAAI,KAAK,CAAC,4BAA4B,mBAAmB,EAAE,CAAC,CAAC;IACrE,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,WAAW,CACtB,IAAmE;QAEnE,QAAQ,IAAI,CAAC,KAAK,EAAE;YAClB,KAAK,6BAA6B,CAAC,CAAC;gBAClC,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC;gBAC1D,OAAO,aAAa,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;aACxC;YACD;gBACE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;SACzC;IACH,CAAC;CACF"}
@@ -1,4 +1,4 @@
1
- import { BaseLLM } from "../../../llms/index.js";
1
+ import { BaseLanguageModel } from "../../../base_language/index.js";
2
2
  import { JsonSpec, Tool } from "../../tools/index.js";
3
3
  import { CreatePromptArgs } from "../../mrkl/index.js";
4
4
  import { Toolkit } from "../base.js";
@@ -8,4 +8,4 @@ export declare class JsonToolkit extends Toolkit {
8
8
  tools: Tool[];
9
9
  constructor(jsonSpec: JsonSpec);
10
10
  }
11
- export declare function createJsonAgent(llm: BaseLLM, toolkit: JsonToolkit, args?: CreatePromptArgs): AgentExecutor;
11
+ export declare function createJsonAgent(llm: BaseLanguageModel, toolkit: JsonToolkit, args?: CreatePromptArgs): AgentExecutor;
@@ -1 +1 @@
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"}
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,GAAsB,EACtB,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"}
@@ -1,4 +1,4 @@
1
- import { BaseLLM } from "../../../llms/index.js";
1
+ import { BaseLanguageModel } from "../../../base_language/index.js";
2
2
  import { JsonSpec, Tool } from "../../tools/index.js";
3
3
  import { AgentExecutor } from "../../executor.js";
4
4
  import { CreatePromptArgs } from "../../mrkl/index.js";
@@ -9,6 +9,6 @@ export declare class RequestsToolkit extends Toolkit {
9
9
  constructor(headers?: Headers);
10
10
  }
11
11
  export declare class OpenApiToolkit extends RequestsToolkit {
12
- constructor(jsonSpec: JsonSpec, llm: BaseLLM, headers?: Headers);
12
+ constructor(jsonSpec: JsonSpec, llm: BaseLanguageModel, headers?: Headers);
13
13
  }
14
- export declare function createOpenApiAgent(llm: BaseLLM, openApiToolkit: OpenApiToolkit, args?: CreatePromptArgs): AgentExecutor;
14
+ export declare function createOpenApiAgent(llm: BaseLanguageModel, openApiToolkit: OpenApiToolkit, args?: CreatePromptArgs): AgentExecutor;
@@ -1 +1 @@
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"}
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,GAAsB,EAAE,OAAiB;QACvE,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,GAAsB,EACtB,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"}
@@ -1,6 +1,6 @@
1
1
  import { Tool } from "../../tools/index.js";
2
2
  import { Toolkit } from "../base.js";
3
- import { BaseLLM } from "../../../llms/index.js";
3
+ import { BaseLanguageModel } from "../../../base_language/index.js";
4
4
  import { CreatePromptArgs } from "../../mrkl/index.js";
5
5
  import { AgentExecutor } from "../../executor.js";
6
6
  import { SqlDatabase } from "../../../sql_db.js";
@@ -14,5 +14,5 @@ export declare class SqlToolkit extends Toolkit {
14
14
  dialect: string;
15
15
  constructor(db: SqlDatabase);
16
16
  }
17
- export declare function createSqlAgent(llm: BaseLLM, toolkit: SqlToolkit, args?: SqlCreatePromptArgs): AgentExecutor;
17
+ export declare function createSqlAgent(llm: BaseLanguageModel, toolkit: SqlToolkit, args?: SqlCreatePromptArgs): AgentExecutor;
18
18
  export {};
@@ -1,7 +1,7 @@
1
1
  import { InfoSqlTool, ListTablesSqlTool, QueryCheckerTool, QuerySqlTool, } from "../../tools/index.js";
2
2
  import { Toolkit } from "../base.js";
3
3
  import { SQL_PREFIX, SQL_SUFFIX } from "./prompt.js";
4
- import { interpolateFString } from "../../../prompts/template.js";
4
+ import { renderTemplate } from "../../../prompts/template.js";
5
5
  import { LLMChain } from "../../../chains/index.js";
6
6
  import { ZeroShotAgent } from "../../mrkl/index.js";
7
7
  import { AgentExecutor } from "../../executor.js";
@@ -38,7 +38,7 @@ export class SqlToolkit extends Toolkit {
38
38
  export function createSqlAgent(llm, toolkit, args) {
39
39
  const { prefix = SQL_PREFIX, suffix = SQL_SUFFIX, inputVariables = ["input", "agent_scratchpad"], topK = 10, } = args ?? {};
40
40
  const { tools } = toolkit;
41
- const formattedPrefix = interpolateFString(prefix, {
41
+ const formattedPrefix = renderTemplate(prefix, "f-string", {
42
42
  dialect: toolkit.dialect,
43
43
  top_k: topK,
44
44
  });
@@ -1 +1 @@
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,GACb,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;AAQlD,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"}
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,GACb,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAErC,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAC9D,OAAO,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AACpD,OAAO,EAAE,aAAa,EAAoB,MAAM,qBAAqB,CAAC;AACtE,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAQlD,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,GAAsB,EACtB,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,cAAc,CAAC,MAAM,EAAE,UAAU,EAAE;QACzD,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,KAAK,EAAE,IAAI;KACZ,CAAC,CAAC;IAEH,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"}
@@ -1,7 +1,7 @@
1
1
  import { Tool } from "../../tools/index.js";
2
2
  import { VectorStore } from "../../../vectorstores/index.js";
3
3
  import { Toolkit } from "../base.js";
4
- import { BaseLLM } from "../../../llms/index.js";
4
+ import { BaseLanguageModel } from "../../../base_language/index.js";
5
5
  import { CreatePromptArgs } from "../../mrkl/index.js";
6
6
  import { AgentExecutor } from "../../executor.js";
7
7
  export interface VectorStoreInfo {
@@ -11,14 +11,14 @@ export interface VectorStoreInfo {
11
11
  }
12
12
  export declare class VectorStoreToolkit extends Toolkit {
13
13
  tools: Tool[];
14
- llm: BaseLLM;
15
- constructor(vectorStoreInfo: VectorStoreInfo, llm: BaseLLM);
14
+ llm: BaseLanguageModel;
15
+ constructor(vectorStoreInfo: VectorStoreInfo, llm: BaseLanguageModel);
16
16
  }
17
17
  export declare class VectorStoreRouterToolkit extends Toolkit {
18
18
  tools: Tool[];
19
19
  vectorStoreInfos: VectorStoreInfo[];
20
- llm: BaseLLM;
21
- constructor(vectorStoreInfos: VectorStoreInfo[], llm: BaseLLM);
20
+ llm: BaseLanguageModel;
21
+ constructor(vectorStoreInfos: VectorStoreInfo[], llm: BaseLanguageModel);
22
22
  }
23
- export declare function createVectorStoreAgent(llm: BaseLLM, toolkit: VectorStoreToolkit, args?: CreatePromptArgs): AgentExecutor;
24
- export declare function createVectorStoreRouterAgent(llm: BaseLLM, toolkit: VectorStoreRouterToolkit, args?: CreatePromptArgs): AgentExecutor;
23
+ export declare function createVectorStoreAgent(llm: BaseLanguageModel, toolkit: VectorStoreToolkit, args?: CreatePromptArgs): AgentExecutor;
24
+ export declare function createVectorStoreRouterAgent(llm: BaseLanguageModel, toolkit: VectorStoreRouterToolkit, args?: CreatePromptArgs): AgentExecutor;
@@ -1 +1 @@
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"}
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,GAAsB;QAClE,KAAK,EAAE,CAAC;QALV;;;;;WAAc;QAEd;;;;;WAAuB;QAIrB,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,GAAsB;QACrE,KAAK,EAAE,CAAC;QAPV;;;;;WAAc;QAEd;;;;;WAAoC;QAEpC;;;;;WAAuB;QAIrB,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,GAAsB,EACtB,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,GAAsB,EACtB,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"}
@@ -6,7 +6,7 @@ import { BaseOutputParser } from "../../output_parsers/base.js";
6
6
  import { SerializedOutputParser } from "../../output_parsers/serde.js";
7
7
  import { AgentInput } from "../types.js";
8
8
  import { Tool } from "../tools/base.js";
9
- export declare class AgentOutputParser extends BaseOutputParser {
9
+ export declare class ChatConversationalAgentOutputParser extends BaseOutputParser {
10
10
  parse(text: string): unknown;
11
11
  getFormatInstructions(): string;
12
12
  serialize(): SerializedOutputParser;
@@ -21,14 +21,14 @@ export type CreatePromptArgs = {
21
21
  /** Output parser to use for formatting. */
22
22
  outputParser?: BaseOutputParser;
23
23
  };
24
- type ZeroShotAgentInput = AgentInput;
24
+ export type ChatConversationalAgentInput = AgentInput;
25
25
  /**
26
26
  * Agent for the MRKL chain.
27
27
  * @augments Agent
28
28
  */
29
29
  export declare class ChatConversationalAgent extends Agent {
30
30
  outputParser: BaseOutputParser;
31
- constructor(input: ZeroShotAgentInput, outputParser: BaseOutputParser);
31
+ constructor(input: ChatConversationalAgentInput, outputParser?: BaseOutputParser);
32
32
  _agentType(): string;
33
33
  observationPrefix(): string;
34
34
  llmPrefix(): string;
@@ -50,4 +50,3 @@ export declare class ChatConversationalAgent extends Agent {
50
50
  input: string;
51
51
  } | null;
52
52
  }
53
- export {};
@@ -1,22 +1,24 @@
1
1
  import { LLMChain } from "../../chains/index.js";
2
2
  import { Agent } from "../agent.js";
3
3
  import { SystemMessagePromptTemplate, HumanMessagePromptTemplate, ChatPromptTemplate, MessagesPlaceholder, } from "../../prompts/index.js";
4
- import { interpolateFString } from "../../prompts/template.js";
4
+ import { renderTemplate } from "../../prompts/template.js";
5
5
  import { PREFIX, SUFFIX, FORMAT_INSTRUCTIONS, TEMPLATE_TOOL_RESPONSE, } from "./prompt.js";
6
6
  import { AIChatMessage, HumanChatMessage, } from "../../schema/index.js";
7
7
  import { BaseOutputParser } from "../../output_parsers/base.js";
8
- export class AgentOutputParser extends BaseOutputParser {
8
+ export class ChatConversationalAgentOutputParser extends BaseOutputParser {
9
9
  parse(text) {
10
- const cleanedOutput = text.trim();
11
- let jsonOutput = cleanedOutput;
10
+ let jsonOutput = text.trim();
12
11
  if (jsonOutput.includes("```json")) {
13
- jsonOutput = jsonOutput.split("```json")[1].trimLeft();
12
+ jsonOutput = jsonOutput.split("```json")[1].trimStart();
13
+ }
14
+ if (jsonOutput.includes("```")) {
15
+ jsonOutput = jsonOutput.split("```")[0].trimEnd();
14
16
  }
15
17
  if (jsonOutput.startsWith("```")) {
16
- jsonOutput = jsonOutput.slice(3).trimLeft();
18
+ jsonOutput = jsonOutput.slice(3).trimStart();
17
19
  }
18
20
  if (jsonOutput.endsWith("```")) {
19
- jsonOutput = jsonOutput.slice(0, -3).trimRight();
21
+ jsonOutput = jsonOutput.slice(0, -3).trimEnd();
20
22
  }
21
23
  const response = JSON.parse(jsonOutput);
22
24
  return { action: response.action, action_input: response.action_input };
@@ -41,7 +43,8 @@ export class ChatConversationalAgent extends Agent {
41
43
  writable: true,
42
44
  value: void 0
43
45
  });
44
- this.outputParser = outputParser;
46
+ this.outputParser =
47
+ outputParser ?? new ChatConversationalAgentOutputParser();
45
48
  }
46
49
  _agentType() {
47
50
  /** Not turning on serialization until more sure of abstractions. */
@@ -68,7 +71,7 @@ export class ChatConversationalAgent extends Agent {
68
71
  const thoughts = [];
69
72
  for (const step of steps) {
70
73
  thoughts.push(new AIChatMessage(step.action.log));
71
- thoughts.push(new HumanChatMessage(interpolateFString(TEMPLATE_TOOL_RESPONSE, {
74
+ thoughts.push(new HumanChatMessage(renderTemplate(TEMPLATE_TOOL_RESPONSE, "f-string", {
72
75
  observation: step.observation,
73
76
  })));
74
77
  }
@@ -83,15 +86,15 @@ export class ChatConversationalAgent extends Agent {
83
86
  * @param args.prefix - String to put before the list of tools.
84
87
  */
85
88
  static createPrompt(tools, args) {
86
- const { systemMessage = PREFIX, humanMessage = SUFFIX, outputParser = new AgentOutputParser(), } = args ?? {};
89
+ const { systemMessage = PREFIX, humanMessage = SUFFIX, outputParser = new ChatConversationalAgentOutputParser(), } = args ?? {};
87
90
  const toolStrings = tools
88
91
  .map((tool) => `${tool.name}: ${tool.description}`)
89
92
  .join("\n");
90
- const formatInstructions = interpolateFString(humanMessage, {
93
+ const formatInstructions = renderTemplate(humanMessage, "f-string", {
91
94
  format_instructions: outputParser.getFormatInstructions(),
92
95
  });
93
96
  const toolNames = tools.map((tool) => tool.name).join("\n");
94
- const finalPrompt = interpolateFString(formatInstructions, {
97
+ const finalPrompt = renderTemplate(formatInstructions, "f-string", {
95
98
  tools: toolStrings,
96
99
  tool_names: toolNames,
97
100
  });
@@ -107,7 +110,7 @@ export class ChatConversationalAgent extends Agent {
107
110
  ChatConversationalAgent.validateTools(tools);
108
111
  const prompt = ChatConversationalAgent.createPrompt(tools, args);
109
112
  const chain = new LLMChain({ prompt, llm });
110
- const { outputParser = new AgentOutputParser() } = args ?? {};
113
+ const { outputParser = new ChatConversationalAgentOutputParser() } = args ?? {};
111
114
  return new ChatConversationalAgent({
112
115
  llmChain: chain,
113
116
  allowedTools: tools.map((t) => t.name),
@@ -1 +1 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/agents/chat_convo/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AACjD,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACpC,OAAO,EACL,2BAA2B,EAC3B,0BAA0B,EAC1B,kBAAkB,EAClB,mBAAmB,GACpB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EACL,MAAM,EACN,MAAM,EACN,mBAAmB,EACnB,sBAAsB,GACvB,MAAM,aAAa,CAAC;AAErB,OAAO,EAGL,aAAa,EACb,gBAAgB,GACjB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAKhE,MAAM,OAAO,iBAAkB,SAAQ,gBAAgB;IACrD,KAAK,CAAC,IAAY;QAChB,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QAClC,IAAI,UAAU,GAAG,aAAa,CAAC;QAC/B,IAAI,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE;YAClC,UAAU,GAAG,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;SACxD;QACD,IAAI,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE;YAChC,UAAU,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;SAC7C;QACD,IAAI,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE;YAC9B,UAAU,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;SAClD;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QACxC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE,YAAY,EAAE,QAAQ,CAAC,YAAY,EAAE,CAAC;IAC1E,CAAC;IAED,qBAAqB;QACnB,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IAED,SAAS;QACP,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC7C,CAAC;CACF;AAeD;;;GAGG;AACH,MAAM,OAAO,uBAAwB,SAAQ,KAAK;IAGhD,YAAY,KAAyB,EAAE,YAA8B;QACnE,KAAK,CAAC,KAAK,CAAC,CAAC;QAHf;;;;;WAA+B;QAI7B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;IACnC,CAAC;IAED,UAAU;QACR,oEAAoE;QACpE,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC7C,CAAC;IAED,iBAAiB;QACf,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,SAAS;QACP,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,KAAK;QACH,OAAO,CAAC,cAAc,CAAC,CAAC;IAC1B,CAAC;IAED,MAAM,CAAC,aAAa,CAAC,KAAa;QAChC,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC5D,IAAI,WAAW,EAAE;YACf,MAAM,GAAG,GACP,cAAc,WAAW,CAAC,IAAI,yBAAyB;gBACvD,kDAAkD,CAAC;YACrD,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC;SACtB;IACH,CAAC;IAED,mBAAmB,CAAC,KAAkB;QACpC,MAAM,QAAQ,GAAsB,EAAE,CAAC;QACvC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;YACxB,QAAQ,CAAC,IAAI,CAAC,IAAI,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAClD,QAAQ,CAAC,IAAI,CACX,IAAI,gBAAgB,CAClB,kBAAkB,CAAC,sBAAsB,EAAE;gBACzC,WAAW,EAAE,IAAI,CAAC,WAAW;aAC9B,CAAC,CACH,CACF,CAAC;SACH;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;;;;;;OAOG;IACH,MAAM,CAAC,YAAY,CAAC,KAAa,EAAE,IAAuB;QACxD,MAAM,EACJ,aAAa,GAAG,MAAM,EACtB,YAAY,GAAG,MAAM,EACrB,YAAY,GAAG,IAAI,iBAAiB,EAAE,GACvC,GAAG,IAAI,IAAI,EAAE,CAAC;QACf,MAAM,WAAW,GAAG,KAAK;aACtB,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC;aAClD,IAAI,CAAC,IAAI,CAAC,CAAC;QACd,MAAM,kBAAkB,GAAG,kBAAkB,CAAC,YAAY,EAAE;YAC1D,mBAAmB,EAAE,YAAY,CAAC,qBAAqB,EAAE;SAC1D,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5D,MAAM,WAAW,GAAG,kBAAkB,CAAC,kBAAkB,EAAE;YACzD,KAAK,EAAE,WAAW;YAClB,UAAU,EAAE,SAAS;SACtB,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG;YACf,2BAA2B,CAAC,YAAY,CAAC,aAAa,CAAC;YACvD,IAAI,mBAAmB,CAAC,cAAc,CAAC;YACvC,0BAA0B,CAAC,YAAY,CAAC,WAAW,CAAC;YACpD,IAAI,mBAAmB,CAAC,kBAAkB,CAAC;SAC5C,CAAC;QACF,OAAO,kBAAkB,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;IACzD,CAAC;IAED,MAAM,CAAC,eAAe,CACpB,GAAsB,EACtB,KAAa,EACb,IAAuB;QAEvB,uBAAuB,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QAC7C,MAAM,MAAM,GAAG,uBAAuB,CAAC,YAAY,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QACjE,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;QAC5C,MAAM,EAAE,YAAY,GAAG,IAAI,iBAAiB,EAAE,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC;QAC9D,OAAO,IAAI,uBAAuB,CAChC;YACE,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;SACvC,EACD,YAAY,CACb,CAAC;IACJ,CAAC;IAED,mBAAmB,CAAC,IAAY;QAC9B,IAAI;YACF,MAAM,QAAQ,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,CAG5C,CAAC;YACF,OAAO,EAAE,IAAI,EAAE,QAAQ,CAAC,MAAM,EAAE,KAAK,EAAE,QAAQ,CAAC,YAAY,EAAE,CAAC;SAChE;QAAC,MAAM;YACN,MAAM,IAAI,KAAK,CACb,qDAAqD,IAAI,EAAE,CAC5D,CAAC;SACH;IACH,CAAC;CACF"}
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/agents/chat_convo/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AACjD,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACpC,OAAO,EACL,2BAA2B,EAC3B,0BAA0B,EAC1B,kBAAkB,EAClB,mBAAmB,GACpB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAC3D,OAAO,EACL,MAAM,EACN,MAAM,EACN,mBAAmB,EACnB,sBAAsB,GACvB,MAAM,aAAa,CAAC;AAErB,OAAO,EAGL,aAAa,EACb,gBAAgB,GACjB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAKhE,MAAM,OAAO,mCAAoC,SAAQ,gBAAgB;IACvE,KAAK,CAAC,IAAY;QAChB,IAAI,UAAU,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QAC7B,IAAI,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE;YAClC,UAAU,GAAG,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;SACzD;QACD,IAAI,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE;YAC9B,UAAU,GAAG,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;SACnD;QACD,IAAI,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE;YAChC,UAAU,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;SAC9C;QACD,IAAI,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE;YAC9B,UAAU,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;SAChD;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QACxC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE,YAAY,EAAE,QAAQ,CAAC,YAAY,EAAE,CAAC;IAC1E,CAAC;IAED,qBAAqB;QACnB,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IAED,SAAS;QACP,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC7C,CAAC;CACF;AAeD;;;GAGG;AACH,MAAM,OAAO,uBAAwB,SAAQ,KAAK;IAGhD,YACE,KAAmC,EACnC,YAA+B;QAE/B,KAAK,CAAC,KAAK,CAAC,CAAC;QANf;;;;;WAA+B;QAO7B,IAAI,CAAC,YAAY;YACf,YAAY,IAAI,IAAI,mCAAmC,EAAE,CAAC;IAC9D,CAAC;IAED,UAAU;QACR,oEAAoE;QACpE,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC7C,CAAC;IAED,iBAAiB;QACf,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,SAAS;QACP,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,KAAK;QACH,OAAO,CAAC,cAAc,CAAC,CAAC;IAC1B,CAAC;IAED,MAAM,CAAC,aAAa,CAAC,KAAa;QAChC,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC5D,IAAI,WAAW,EAAE;YACf,MAAM,GAAG,GACP,cAAc,WAAW,CAAC,IAAI,yBAAyB;gBACvD,kDAAkD,CAAC;YACrD,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC;SACtB;IACH,CAAC;IAED,mBAAmB,CAAC,KAAkB;QACpC,MAAM,QAAQ,GAAsB,EAAE,CAAC;QACvC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;YACxB,QAAQ,CAAC,IAAI,CAAC,IAAI,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAClD,QAAQ,CAAC,IAAI,CACX,IAAI,gBAAgB,CAClB,cAAc,CAAC,sBAAsB,EAAE,UAAU,EAAE;gBACjD,WAAW,EAAE,IAAI,CAAC,WAAW;aAC9B,CAAC,CACH,CACF,CAAC;SACH;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;;;;;;OAOG;IACH,MAAM,CAAC,YAAY,CAAC,KAAa,EAAE,IAAuB;QACxD,MAAM,EACJ,aAAa,GAAG,MAAM,EACtB,YAAY,GAAG,MAAM,EACrB,YAAY,GAAG,IAAI,mCAAmC,EAAE,GACzD,GAAG,IAAI,IAAI,EAAE,CAAC;QACf,MAAM,WAAW,GAAG,KAAK;aACtB,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC;aAClD,IAAI,CAAC,IAAI,CAAC,CAAC;QACd,MAAM,kBAAkB,GAAG,cAAc,CAAC,YAAY,EAAE,UAAU,EAAE;YAClE,mBAAmB,EAAE,YAAY,CAAC,qBAAqB,EAAE;SAC1D,CAAC,CAAC;QACH,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5D,MAAM,WAAW,GAAG,cAAc,CAAC,kBAAkB,EAAE,UAAU,EAAE;YACjE,KAAK,EAAE,WAAW;YAClB,UAAU,EAAE,SAAS;SACtB,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG;YACf,2BAA2B,CAAC,YAAY,CAAC,aAAa,CAAC;YACvD,IAAI,mBAAmB,CAAC,cAAc,CAAC;YACvC,0BAA0B,CAAC,YAAY,CAAC,WAAW,CAAC;YACpD,IAAI,mBAAmB,CAAC,kBAAkB,CAAC;SAC5C,CAAC;QACF,OAAO,kBAAkB,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;IACzD,CAAC;IAED,MAAM,CAAC,eAAe,CACpB,GAAsB,EACtB,KAAa,EACb,IAAuB;QAEvB,uBAAuB,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QAC7C,MAAM,MAAM,GAAG,uBAAuB,CAAC,YAAY,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QACjE,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;QAC5C,MAAM,EAAE,YAAY,GAAG,IAAI,mCAAmC,EAAE,EAAE,GAChE,IAAI,IAAI,EAAE,CAAC;QACb,OAAO,IAAI,uBAAuB,CAChC;YACE,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;SACvC,EACD,YAAY,CACb,CAAC;IACJ,CAAC;IAED,mBAAmB,CAAC,IAAY;QAC9B,IAAI;YACF,MAAM,QAAQ,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,CAG5C,CAAC;YACF,OAAO,EAAE,IAAI,EAAE,QAAQ,CAAC,MAAM,EAAE,KAAK,EAAE,QAAQ,CAAC,YAAY,EAAE,CAAC;SAChE;QAAC,MAAM;YACN,MAAM,IAAI,KAAK,CACb,qDAAqD,IAAI,EAAE,CAC5D,CAAC;SACH;IACH,CAAC;CACF"}
@@ -3,7 +3,7 @@ export { Agent } from "./agent.js";
3
3
  export { AgentExecutor } from "./executor.js";
4
4
  export { ZeroShotAgent } from "./mrkl/index.js";
5
5
  export { ChatAgent } from "./chat/index.js";
6
- export { ChatConversationalAgent } from "./chat_convo/index.js";
6
+ export { ChatConversationalAgent, ChatConversationalAgentOutputParser, ChatConversationalAgentInput, } from "./chat_convo/index.js";
7
7
  export { Tool } from "./tools/index.js";
8
8
  export { initializeAgentExecutor } from "./initialize.js";
9
9
  export { loadAgent } from "./load.js";
@@ -2,7 +2,7 @@ export { Agent } from "./agent.js";
2
2
  export { AgentExecutor } from "./executor.js";
3
3
  export { ZeroShotAgent } from "./mrkl/index.js";
4
4
  export { ChatAgent } from "./chat/index.js";
5
- export { ChatConversationalAgent } from "./chat_convo/index.js";
5
+ export { ChatConversationalAgent, ChatConversationalAgentOutputParser, } from "./chat_convo/index.js";
6
6
  export { Tool } from "./tools/index.js";
7
7
  export { initializeAgentExecutor } from "./initialize.js";
8
8
  export { loadAgent } from "./load.js";
@@ -1 +1 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/agents/index.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAC9C,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAC;AAChE,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,aAAa,EACb,cAAc,EACd,eAAe,EACf,kBAAkB,EAClB,sBAAsB,GACvB,MAAM,2BAA2B,CAAC"}
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/agents/index.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAC9C,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EACL,uBAAuB,EACvB,mCAAmC,GAEpC,MAAM,uBAAuB,CAAC;AAC/B,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,aAAa,EACb,cAAc,EACd,eAAe,EACf,kBAAkB,EAClB,sBAAsB,GACvB,MAAM,2BAA2B,CAAC"}
@@ -1,7 +1,7 @@
1
1
  import { Agent } from "./agent.js";
2
2
  import { Tool } from "./tools/base.js";
3
- import { BaseLLM } from "../llms/index.js";
3
+ import { BaseLanguageModel } from "../base_language/index.js";
4
4
  export declare const loadAgent: (uri: string, llmAndTools?: {
5
- llm?: BaseLLM;
5
+ llm?: BaseLanguageModel;
6
6
  tools?: Tool[];
7
7
  }) => Promise<Agent>;
@@ -1 +1 @@
1
- {"version":3,"file":"load.js","sourceRoot":"","sources":["../../src/agents/load.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAGnC,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,EAAc,YAAY,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAE7E,MAAM,iBAAiB,GAAsB,KAAK,EAChD,IAAY,EACZ,IAAY,EACZ,WAA+C,EAC/C,EAAE;IACF,MAAM,UAAU,GAAG,eAAe,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAC/C,OAAO,KAAK,CAAC,WAAW,CAAC,EAAE,GAAG,UAAU,EAAE,GAAG,WAAW,EAAE,CAAC,CAAC;AAC9D,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,SAAS,GAAG,KAAK,EAC5B,GAAW,EACX,WAA+C,EAC/B,EAAE;IAClB,MAAM,SAAS,GAAG,MAAM,WAAW,CACjC,GAAG,EACH,iBAAiB,EACjB,QAAQ,EACR,IAAI,GAAG,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,EACzB,WAAW,CACZ,CAAC;IACF,IAAI,SAAS,EAAE;QACb,OAAO,SAAS,CAAC;KAClB;IAED,OAAO,YAAY,CAAC,GAAG,EAAE,iBAAiB,EAAE,WAAW,CAAC,CAAC;AAC3D,CAAC,CAAC"}
1
+ {"version":3,"file":"load.js","sourceRoot":"","sources":["../../src/agents/load.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAGnC,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,EAAc,YAAY,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAE7E,MAAM,iBAAiB,GAAsB,KAAK,EAChD,IAAY,EACZ,IAAY,EACZ,WAAyD,EACzD,EAAE;IACF,MAAM,UAAU,GAAG,eAAe,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAC/C,OAAO,KAAK,CAAC,WAAW,CAAC,EAAE,GAAG,UAAU,EAAE,GAAG,WAAW,EAAE,CAAC,CAAC;AAC9D,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,SAAS,GAAG,KAAK,EAC5B,GAAW,EACX,WAAyD,EACzC,EAAE;IAClB,MAAM,SAAS,GAAG,MAAM,WAAW,CACjC,GAAG,EACH,iBAAiB,EACjB,QAAQ,EACR,IAAI,GAAG,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,EACzB,WAAW,CACZ,CAAC;IACF,IAAI,SAAS,EAAE;QACb,OAAO,SAAS,CAAC;KAClB;IAED,OAAO,YAAY,CAAC,GAAG,EAAE,iBAAiB,EAAE,WAAW,CAAC,CAAC;AAC3D,CAAC,CAAC"}
@@ -1,14 +1,14 @@
1
1
  import { VectorStore } from "../../vectorstores/index.js";
2
- import { BaseLLM } from "../../llms/index.js";
2
+ import { BaseLanguageModel } from "../../base_language/index.js";
3
3
  import { VectorDBQAChain } from "../../chains/index.js";
4
4
  import { Tool } from "./base.js";
5
5
  interface VectorStoreTool {
6
6
  vectorStore: VectorStore;
7
- llm: BaseLLM;
7
+ llm: BaseLanguageModel;
8
8
  }
9
9
  export declare class VectorStoreQATool extends Tool implements VectorStoreTool {
10
10
  vectorStore: VectorStore;
11
- llm: BaseLLM;
11
+ llm: BaseLanguageModel;
12
12
  name: string;
13
13
  description: string;
14
14
  chain: VectorDBQAChain;
@@ -1 +1 @@
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,KAAK,CAAC,KAAa;QACvB,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;CACF"}
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;;;;;WAAuB;QAEvB;;;;;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,KAAK,CAAC,KAAa;QACvB,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;CACF"}
@@ -3,7 +3,6 @@ export type ZapierValues = Record<string, any>;
3
3
  export declare class ZapierNLAWrapper {
4
4
  zapierNlaApiKey: string;
5
5
  zapierNlaApiBase: string;
6
- zapierNlaApiDynamicBase: string;
7
6
  constructor(zapierNlaApiKey?: string);
8
7
  protected _getHeaders(): Record<string, string>;
9
8
  protected _getActionRequest(actionId: string, instructions: string, params?: ZapierValues): Promise<ZapierValues>;
@@ -1,5 +1,5 @@
1
1
  import { Tool } from "./base.js";
2
- import { interpolateFString } from "../../prompts/template.js";
2
+ import { renderTemplate } from "../../prompts/template.js";
3
3
  const zapierNLABaseDescription = "A wrapper around Zapier NLA actions. " +
4
4
  "The input to this tool is a natural language instruction, " +
5
5
  'for example "get the latest email from my bank" or ' +
@@ -26,12 +26,6 @@ export class ZapierNLAWrapper {
26
26
  writable: true,
27
27
  value: "https://nla.zapier.com/api/v1/"
28
28
  });
29
- Object.defineProperty(this, "zapierNlaApiDynamicBase", {
30
- enumerable: true,
31
- configurable: true,
32
- writable: true,
33
- value: "https://nla.zapier.com/api/v1/dynamic/"
34
- });
35
29
  const apiKey = zapierNlaApiKey ?? process.env.ZAPIER_NLA_API_KEY;
36
30
  if (!apiKey) {
37
31
  throw new Error("ZAPIER_NLA_API_KEY not set");
@@ -84,7 +78,7 @@ export class ZapierNLAWrapper {
84
78
  const data = params ?? {};
85
79
  data.preview_only = true;
86
80
  const resp = await this._getActionRequest(actionId, instructions, data);
87
- return resp.params;
81
+ return resp.input_params;
88
82
  }
89
83
  /**
90
84
  * Returns a list of all exposed (enabled) actions associated with
@@ -93,7 +87,7 @@ export class ZapierNLAWrapper {
93
87
  */
94
88
  async listActions() {
95
89
  const headers = this._getHeaders();
96
- const resp = await fetch(`${this.zapierNlaApiDynamicBase}exposed/`, {
90
+ const resp = await fetch(`${this.zapierNlaApiBase}exposed/`, {
97
91
  method: "GET",
98
92
  headers,
99
93
  });
@@ -170,7 +164,7 @@ export class ZapierNLARunAction extends Tool {
170
164
  const paramsSchemaWithoutInstructions = { ...paramsSchema };
171
165
  delete paramsSchemaWithoutInstructions.instructions;
172
166
  const paramsSchemaKeysString = JSON.stringify(Object.keys(paramsSchemaWithoutInstructions));
173
- this.description = interpolateFString(zapierNLABaseDescription, {
167
+ this.description = renderTemplate(zapierNLABaseDescription, "f-string", {
174
168
  zapier_description: zapierDescription,
175
169
  params: paramsSchemaKeysString,
176
170
  });
@@ -1 +1 @@
1
- {"version":3,"file":"zapier.js","sourceRoot":"","sources":["../../../src/agents/tools/zapier.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAE/D,MAAM,wBAAwB,GAC5B,uCAAuC;IACvC,4DAA4D;IAC5D,qDAAqD;IACrD,kDAAkD;IAClD,mJAAmJ;IACnJ,kLAAkL;IAClL,+JAA+J;IAC/J,oFAAoF;IACpF,+IAA+I;IAC/I,kFAAkF;IAClF,yDAAyD;IACzD,0BAA0B,CAAC;AAK7B,MAAM,OAAO,gBAAgB;IAO3B,YAAY,eAAwB;QANpC;;;;;WAAwB;QAExB;;;;mBAAmB,gCAAgC;WAAC;QAEpD;;;;mBAA0B,wCAAwC;WAAC;QAGjE,MAAM,MAAM,GAAG,eAAe,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;QACjE,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;SAC/C;QACD,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC;IAChC,CAAC;IAES,WAAW;QACnB,OAAO;YACL,cAAc,EAAE,kBAAkB;YAClC,MAAM,EAAE,kBAAkB;YAC1B,WAAW,EAAE,IAAI,CAAC,eAAe;SAClC,CAAC;IACJ,CAAC;IAES,KAAK,CAAC,iBAAiB,CAC/B,QAAgB,EAChB,YAAoB,EACpB,MAAqB;QAErB,MAAM,IAAI,GAAG,MAAM,IAAI,EAAE,CAAC;QAC1B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QAEnC,wBAAwB;QACxB,MAAM,IAAI,GAAG,MAAM,KAAK,CACtB,GAAG,IAAI,CAAC,gBAAgB,WAAW,QAAQ,WAAW,EACtD;YACE,MAAM,EAAE,MAAM;YACd,OAAO;YACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CACF,CAAC;QAEF,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE;YACZ,MAAM,IAAI,KAAK,CACb,4BAA4B,QAAQ,sBAAsB,YAAY,EAAE,CACzE,CAAC;SACH;QACD,OAAO,IAAI,CAAC,IAAI,EAAE,CAAC;IACrB,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,SAAS,CACb,QAAgB,EAChB,YAAoB,EACpB,MAAqB;QAErB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;QAC1E,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,aAAa,CACjB,QAAgB,EAChB,YAAoB,EACpB,MAAqB;QAErB,MAAM,IAAI,GAAG,MAAM,IAAI,EAAE,CAAC;QAC1B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,YAAY,EAAE,IAAI,CAAC,CAAC;QACxE,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,WAAW;QACf,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QACnC,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,uBAAuB,UAAU,EAAE;YAClE,MAAM,EAAE,KAAK;YACb,OAAO;SACR,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE;YACZ,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;SAC3C;QACD,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC;IACrC,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,WAAW,CACf,QAAgB,EAChB,YAAoB,EACpB,MAAqB;QAErB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;QACpE,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,eAAe,CACnB,QAAgB,EAChB,YAAoB,EACpB,MAAqB;QAErB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;QACxE,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,mBAAmB;QACvB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QACxC,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;CACF;AAED,MAAM,OAAO,kBAAmB,SAAQ,IAAI;IAW1C,YACE,UAA4B,EAC5B,QAAgB,EAChB,iBAAyB,EACzB,YAA0B,EAC1B,MAAqB;QAErB,KAAK,EAAE,CAAC;QAjBV;;;;;WAA6B;QAE7B;;;;;WAAiB;QAEjB;;;;;WAAsB;QAEtB;;;;;WAAa;QAEb;;;;;WAAoB;QAUlB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,IAAI,GAAG,iBAAiB,CAAC;QAC9B,MAAM,+BAA+B,GAAG,EAAE,GAAG,YAAY,EAAE,CAAC;QAC5D,OAAO,+BAA+B,CAAC,YAAY,CAAC;QACpD,MAAM,sBAAsB,GAAG,IAAI,CAAC,SAAS,CAC3C,MAAM,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAC7C,CAAC;QACF,IAAI,CAAC,WAAW,GAAG,kBAAkB,CAAC,wBAAwB,EAAE;YAC9D,kBAAkB,EAAE,iBAAiB;YACrC,MAAM,EAAE,sBAAsB;SAC/B,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,GAAW;QACrB,OAAO,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IACtE,CAAC;CACF"}
1
+ {"version":3,"file":"zapier.js","sourceRoot":"","sources":["../../../src/agents/tools/zapier.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAE3D,MAAM,wBAAwB,GAC5B,uCAAuC;IACvC,4DAA4D;IAC5D,qDAAqD;IACrD,kDAAkD;IAClD,mJAAmJ;IACnJ,kLAAkL;IAClL,+JAA+J;IAC/J,oFAAoF;IACpF,+IAA+I;IAC/I,kFAAkF;IAClF,yDAAyD;IACzD,0BAA0B,CAAC;AAK7B,MAAM,OAAO,gBAAgB;IAK3B,YAAY,eAAwB;QAJpC;;;;;WAAwB;QAExB;;;;mBAAmB,gCAAgC;WAAC;QAGlD,MAAM,MAAM,GAAG,eAAe,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;QACjE,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;SAC/C;QACD,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC;IAChC,CAAC;IAES,WAAW;QACnB,OAAO;YACL,cAAc,EAAE,kBAAkB;YAClC,MAAM,EAAE,kBAAkB;YAC1B,WAAW,EAAE,IAAI,CAAC,eAAe;SAClC,CAAC;IACJ,CAAC;IAES,KAAK,CAAC,iBAAiB,CAC/B,QAAgB,EAChB,YAAoB,EACpB,MAAqB;QAErB,MAAM,IAAI,GAAG,MAAM,IAAI,EAAE,CAAC;QAC1B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QAEnC,wBAAwB;QACxB,MAAM,IAAI,GAAG,MAAM,KAAK,CACtB,GAAG,IAAI,CAAC,gBAAgB,WAAW,QAAQ,WAAW,EACtD;YACE,MAAM,EAAE,MAAM;YACd,OAAO;YACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CACF,CAAC;QAEF,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE;YACZ,MAAM,IAAI,KAAK,CACb,4BAA4B,QAAQ,sBAAsB,YAAY,EAAE,CACzE,CAAC;SACH;QACD,OAAO,IAAI,CAAC,IAAI,EAAE,CAAC;IACrB,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,SAAS,CACb,QAAgB,EAChB,YAAoB,EACpB,MAAqB;QAErB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;QAC1E,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,aAAa,CACjB,QAAgB,EAChB,YAAoB,EACpB,MAAqB;QAErB,MAAM,IAAI,GAAG,MAAM,IAAI,EAAE,CAAC;QAC1B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,YAAY,EAAE,IAAI,CAAC,CAAC;QACxE,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,WAAW;QACf,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QACnC,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,gBAAgB,UAAU,EAAE;YAC3D,MAAM,EAAE,KAAK;YACb,OAAO;SACR,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE;YACZ,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;SAC3C;QACD,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC;IACrC,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,WAAW,CACf,QAAgB,EAChB,YAAoB,EACpB,MAAqB;QAErB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;QACpE,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,eAAe,CACnB,QAAgB,EAChB,YAAoB,EACpB,MAAqB;QAErB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;QACxE,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,mBAAmB;QACvB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QACxC,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;CACF;AAED,MAAM,OAAO,kBAAmB,SAAQ,IAAI;IAW1C,YACE,UAA4B,EAC5B,QAAgB,EAChB,iBAAyB,EACzB,YAA0B,EAC1B,MAAqB;QAErB,KAAK,EAAE,CAAC;QAjBV;;;;;WAA6B;QAE7B;;;;;WAAiB;QAEjB;;;;;WAAsB;QAEtB;;;;;WAAa;QAEb;;;;;WAAoB;QAUlB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,IAAI,GAAG,iBAAiB,CAAC;QAC9B,MAAM,+BAA+B,GAAG,EAAE,GAAG,YAAY,EAAE,CAAC;QAC5D,OAAO,+BAA+B,CAAC,YAAY,CAAC;QACpD,MAAM,sBAAsB,GAAG,IAAI,CAAC,SAAS,CAC3C,MAAM,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAC7C,CAAC;QACF,IAAI,CAAC,WAAW,GAAG,cAAc,CAAC,wBAAwB,EAAE,UAAU,EAAE;YACtE,kBAAkB,EAAE,iBAAiB;YACrC,MAAM,EAAE,sBAAsB;SAC/B,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,GAAW;QACrB,OAAO,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IACtE,CAAC;CACF"}
@@ -1,5 +1,9 @@
1
1
  import { BasePromptValue, LLMResult } from "../schema/index.js";
2
2
  import { CallbackManager } from "../callbacks/index.js";
3
+ export type SerializedLLM = {
4
+ _model: string;
5
+ _type: string;
6
+ } & Record<string, any>;
3
7
  /**
4
8
  * Base interface for language model parameters.
5
9
  * A subclass of {@link BaseLanguageModel} should have a constructor that
@@ -21,5 +25,18 @@ export declare abstract class BaseLanguageModel implements BaseLanguageModelPara
21
25
  protected constructor(params: BaseLanguageModelParams);
22
26
  abstract generatePrompt(promptValues: BasePromptValue[], stop?: string[]): Promise<LLMResult>;
23
27
  abstract _modelType(): string;
28
+ abstract _llmType(): string;
24
29
  abstract getNumTokens(text: string): number;
30
+ /**
31
+ * Get the identifying parameters of the LLM.
32
+ */
33
+ _identifyingParams(): Record<string, any>;
34
+ /**
35
+ * Return a json-like object representing this LLM.
36
+ */
37
+ serialize(): SerializedLLM;
38
+ /**
39
+ * Load an LLM from a json-like object describing it.
40
+ */
41
+ static deserialize(data: SerializedLLM): Promise<BaseLanguageModel>;
25
42
  }
@@ -23,5 +23,38 @@ export class BaseLanguageModel {
23
23
  this.verbose = params.verbose ?? getVerbosity();
24
24
  this.callbackManager = params.callbackManager ?? getCallbackManager();
25
25
  }
26
+ /**
27
+ * Get the identifying parameters of the LLM.
28
+ */
29
+ // eslint-disable-next-line @typescript-eslint/no-explicit-any
30
+ _identifyingParams() {
31
+ return {};
32
+ }
33
+ /**
34
+ * Return a json-like object representing this LLM.
35
+ */
36
+ serialize() {
37
+ return {
38
+ ...this._identifyingParams(),
39
+ _type: this._llmType(),
40
+ _model: this._modelType(),
41
+ };
42
+ }
43
+ /**
44
+ * Load an LLM from a json-like object describing it.
45
+ */
46
+ static async deserialize(data) {
47
+ const { _type, _model, ...rest } = data;
48
+ if (_model && _model !== "base_chat_model") {
49
+ throw new Error(`Cannot load LLM with model ${_model}`);
50
+ }
51
+ const Cls = {
52
+ openai: (await import("../chat_models/openai.js")).ChatOpenAI,
53
+ }[_type];
54
+ if (Cls === undefined) {
55
+ throw new Error(`Cannot load LLM with type ${_type}`);
56
+ }
57
+ return new Cls(rest);
58
+ }
26
59
  }
27
60
  //# sourceMappingURL=index.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/base_language/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAmB,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAE5E,MAAM,YAAY,GAAG,GAAG,EAAE,CAAC,KAAK,CAAC;AAYjC;;GAEG;AACH,MAAM,OAAgB,iBAAiB;IAQrC,YAAsB,MAA+B;QAPrD;;WAEG;QACH;;;;;WAAiB;QAEjB;;;;;WAAiC;QAG/B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,YAAY,EAAE,CAAC;QAChD,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,IAAI,kBAAkB,EAAE,CAAC;IACxE,CAAC;CAUF"}
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/base_language/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAmB,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAE5E,MAAM,YAAY,GAAG,GAAG,EAAE,CAAC,KAAK,CAAC;AAkBjC;;GAEG;AACH,MAAM,OAAgB,iBAAiB;IAQrC,YAAsB,MAA+B;QAPrD;;WAEG;QACH;;;;;WAAiB;QAEjB;;;;;WAAiC;QAG/B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,YAAY,EAAE,CAAC;QAChD,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,IAAI,kBAAkB,EAAE,CAAC;IACxE,CAAC;IAaD;;OAEG;IACH,8DAA8D;IAC9D,kBAAkB;QAChB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO;YACL,GAAG,IAAI,CAAC,kBAAkB,EAAE;YAC5B,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAE;YACtB,MAAM,EAAE,IAAI,CAAC,UAAU,EAAE;SAC1B,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAmB;QAC1C,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,GAAG,IAAI,CAAC;QACxC,IAAI,MAAM,IAAI,MAAM,KAAK,iBAAiB,EAAE;YAC1C,MAAM,IAAI,KAAK,CAAC,8BAA8B,MAAM,EAAE,CAAC,CAAC;SACzD;QACD,MAAM,GAAG,GAAG;YACV,MAAM,EAAE,CAAC,MAAM,MAAM,CAAC,0BAA0B,CAAC,CAAC,CAAC,UAAU;SAC9D,CAAC,KAAK,CAAC,CAAC;QACT,IAAI,GAAG,KAAK,SAAS,EAAE;YACrB,MAAM,IAAI,KAAK,CAAC,8BAA8B,KAAK,EAAE,CAAC,CAAC;SACxD;QACD,OAAO,IAAI,GAAG,CAAC,IAAI,CAAC,CAAC;IACvB,CAAC;CACF"}
@@ -1,4 +1,4 @@
1
- import { BaseLLM } from "../llms/index.js";
1
+ import { BaseLanguageModel } from "../base_language/index.js";
2
2
  import { VectorStore } from "../vectorstores/base.js";
3
3
  import { SerializedChatVectorDBQAChain } from "./serde.js";
4
4
  import { ChainValues } from "../schema/index.js";
@@ -36,7 +36,7 @@ export declare class ChatVectorDBQAChain extends BaseChain implements ChatVector
36
36
  _chainType(): "chat-vector-db";
37
37
  static deserialize(data: SerializedChatVectorDBQAChain, values: LoadValues): Promise<ChatVectorDBQAChain>;
38
38
  serialize(): SerializedChatVectorDBQAChain;
39
- static fromLLM(llm: BaseLLM, vectorstore: VectorStore, options?: {
39
+ static fromLLM(llm: BaseLanguageModel, vectorstore: VectorStore, options?: {
40
40
  inputKey?: string;
41
41
  outputKey?: string;
42
42
  k?: number;
@@ -1 +1 @@
1
- {"version":3,"file":"chat_vector_db_chain.js","sourceRoot":"","sources":["../../src/chains/chat_vector_db_chain.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAQrD,OAAO,EAAE,qBAAqB,EAAE,MAAM,kBAAkB,CAAC;AAEzD,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACtC,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAKhE,MAAM,2BAA2B,GAAG;;;;;qBAKf,CAAC;AAEtB,MAAM,WAAW,GAAG;;;;;gBAKJ,CAAC;AAWjB,MAAM,OAAO,mBACX,SAAQ,SAAS;IASjB,IAAI,SAAS;QACX,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;IAC9C,CAAC;IAYD,YAAY,MAQX;QACC,KAAK,EAAE,CAAC;QA7BV;;;;mBAAI,CAAC;WAAC;QAEN;;;;mBAAW,UAAU;WAAC;QAEtB;;;;mBAAiB,cAAc;WAAC;QAMhC;;;;mBAAY,QAAQ;WAAC;QAErB;;;;;WAAyB;QAEzB;;;;;WAAiC;QAEjC;;;;;WAAiC;QAEjC;;;;mBAAwB,KAAK;WAAC;QAY5B,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,qBAAqB,GAAG,MAAM,CAAC,qBAAqB,CAAC;QAC1D,IAAI,CAAC,sBAAsB,GAAG,MAAM,CAAC,sBAAsB,CAAC;QAC5D,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,CAAC,GAAG,MAAM,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;QAC5B,IAAI,CAAC,qBAAqB;YACxB,MAAM,CAAC,qBAAqB,IAAI,IAAI,CAAC,qBAAqB,CAAC;IAC/D,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,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,IAAI,MAAM,CAAC,EAAE;YACpC,MAAM,IAAI,KAAK,CAAC,oBAAoB,IAAI,CAAC,QAAQ,aAAa,CAAC,CAAC;SACjE;QACD,MAAM,QAAQ,GAAW,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,WAAW,GAAW,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACxD,IAAI,WAAW,GAAG,QAAQ,CAAC;QAC3B,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE;YAC1B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC;gBACpD,QAAQ;gBACR,YAAY,EAAE,WAAW;aAC1B,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACjC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE;gBACrB,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;aAC/B;iBAAM;gBACL,MAAM,IAAI,KAAK,CACb,0EAA0E,CAC3E,CAAC;aACH;SACF;QACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1E,MAAM,MAAM,GAAG;YACb,QAAQ,EAAE,WAAW;YACrB,eAAe,EAAE,IAAI;YACrB,YAAY,EAAE,WAAW;SAC1B,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7D,IAAI,IAAI,CAAC,qBAAqB,EAAE;YAC9B,OAAO;gBACL,GAAG,MAAM;gBACT,eAAe,EAAE,IAAI;aACtB,CAAC;SACH;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,UAAU;QACR,OAAO,gBAAyB,CAAC;IACnC,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,WAAW,CACtB,IAAmC,EACnC,MAAkB;QAElB,IAAI,CAAC,CAAC,aAAa,IAAI,MAAM,CAAC,EAAE;YAC9B,MAAM,IAAI,KAAK,CACb,8DAA8D,CAC/D,CAAC;SACH;QACD,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC;QAC/B,MAAM,+BAA+B,GAAG,MAAM,qBAAqB,CAGjE,yBAAyB,EAAE,IAAI,CAAC,CAAC;QACnC,MAAM,gCAAgC,GAAG,MAAM,qBAAqB,CAGlE,oBAAoB,EAAE,IAAI,CAAC,CAAC;QAE9B,OAAO,IAAI,mBAAmB,CAAC;YAC7B,qBAAqB,EAAE,MAAM,SAAS,CAAC,WAAW,CAChD,+BAA+B,CAChC;YACD,sBAAsB,EAAE,MAAM,QAAQ,CAAC,WAAW,CAChD,gCAAgC,CACjC;YACD,CAAC,EAAE,IAAI,CAAC,CAAC;YACT,WAAW;SACZ,CAAC,CAAC;IACL,CAAC;IAED,SAAS;QACP,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE;YACxB,uBAAuB,EAAE,IAAI,CAAC,qBAAqB,CAAC,SAAS,EAAE;YAC/D,kBAAkB,EAAE,IAAI,CAAC,sBAAsB,CAAC,SAAS,EAAE;YAC3D,CAAC,EAAE,IAAI,CAAC,CAAC;SACV,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,OAAO,CACZ,GAAY,EACZ,WAAwB,EACxB,UAOI,EAAE;QAEN,MAAM,EAAE,yBAAyB,EAAE,UAAU,EAAE,GAAG,IAAI,EAAE,GAAG,OAAO,CAAC;QACnE,MAAM,yBAAyB,GAAG,cAAc,CAAC,YAAY,CAC3D,yBAAyB,IAAI,2BAA2B,CACzD,CAAC;QACF,MAAM,SAAS,GAAG,cAAc,CAAC,YAAY,CAAC,UAAU,IAAI,WAAW,CAAC,CAAC;QAEzE,MAAM,OAAO,GAAG,gBAAgB,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;QAC7D,MAAM,sBAAsB,GAAG,IAAI,QAAQ,CAAC;YAC1C,MAAM,EAAE,yBAAyB;YACjC,GAAG;SACJ,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,IAAI,IAAI,CAAC;YACxB,WAAW;YACX,qBAAqB,EAAE,OAAO;YAC9B,sBAAsB;YACtB,GAAG,IAAI;SACR,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF"}
1
+ {"version":3,"file":"chat_vector_db_chain.js","sourceRoot":"","sources":["../../src/chains/chat_vector_db_chain.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAQrD,OAAO,EAAE,qBAAqB,EAAE,MAAM,kBAAkB,CAAC;AAEzD,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACtC,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAKhE,MAAM,2BAA2B,GAAG;;;;;qBAKf,CAAC;AAEtB,MAAM,WAAW,GAAG;;;;;gBAKJ,CAAC;AAWjB,MAAM,OAAO,mBACX,SAAQ,SAAS;IASjB,IAAI,SAAS;QACX,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;IAC9C,CAAC;IAYD,YAAY,MAQX;QACC,KAAK,EAAE,CAAC;QA7BV;;;;mBAAI,CAAC;WAAC;QAEN;;;;mBAAW,UAAU;WAAC;QAEtB;;;;mBAAiB,cAAc;WAAC;QAMhC;;;;mBAAY,QAAQ;WAAC;QAErB;;;;;WAAyB;QAEzB;;;;;WAAiC;QAEjC;;;;;WAAiC;QAEjC;;;;mBAAwB,KAAK;WAAC;QAY5B,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,qBAAqB,GAAG,MAAM,CAAC,qBAAqB,CAAC;QAC1D,IAAI,CAAC,sBAAsB,GAAG,MAAM,CAAC,sBAAsB,CAAC;QAC5D,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,CAAC,GAAG,MAAM,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;QAC5B,IAAI,CAAC,qBAAqB;YACxB,MAAM,CAAC,qBAAqB,IAAI,IAAI,CAAC,qBAAqB,CAAC;IAC/D,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,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,IAAI,MAAM,CAAC,EAAE;YACpC,MAAM,IAAI,KAAK,CAAC,oBAAoB,IAAI,CAAC,QAAQ,aAAa,CAAC,CAAC;SACjE;QACD,MAAM,QAAQ,GAAW,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,WAAW,GAAW,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACxD,IAAI,WAAW,GAAG,QAAQ,CAAC;QAC3B,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE;YAC1B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC;gBACpD,QAAQ;gBACR,YAAY,EAAE,WAAW;aAC1B,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACjC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE;gBACrB,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;aAC/B;iBAAM;gBACL,MAAM,IAAI,KAAK,CACb,0EAA0E,CAC3E,CAAC;aACH;SACF;QACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1E,MAAM,MAAM,GAAG;YACb,QAAQ,EAAE,WAAW;YACrB,eAAe,EAAE,IAAI;YACrB,YAAY,EAAE,WAAW;SAC1B,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7D,IAAI,IAAI,CAAC,qBAAqB,EAAE;YAC9B,OAAO;gBACL,GAAG,MAAM;gBACT,eAAe,EAAE,IAAI;aACtB,CAAC;SACH;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,UAAU;QACR,OAAO,gBAAyB,CAAC;IACnC,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,WAAW,CACtB,IAAmC,EACnC,MAAkB;QAElB,IAAI,CAAC,CAAC,aAAa,IAAI,MAAM,CAAC,EAAE;YAC9B,MAAM,IAAI,KAAK,CACb,8DAA8D,CAC/D,CAAC;SACH;QACD,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC;QAC/B,MAAM,+BAA+B,GAAG,MAAM,qBAAqB,CAGjE,yBAAyB,EAAE,IAAI,CAAC,CAAC;QACnC,MAAM,gCAAgC,GAAG,MAAM,qBAAqB,CAGlE,oBAAoB,EAAE,IAAI,CAAC,CAAC;QAE9B,OAAO,IAAI,mBAAmB,CAAC;YAC7B,qBAAqB,EAAE,MAAM,SAAS,CAAC,WAAW,CAChD,+BAA+B,CAChC;YACD,sBAAsB,EAAE,MAAM,QAAQ,CAAC,WAAW,CAChD,gCAAgC,CACjC;YACD,CAAC,EAAE,IAAI,CAAC,CAAC;YACT,WAAW;SACZ,CAAC,CAAC;IACL,CAAC;IAED,SAAS;QACP,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE;YACxB,uBAAuB,EAAE,IAAI,CAAC,qBAAqB,CAAC,SAAS,EAAE;YAC/D,kBAAkB,EAAE,IAAI,CAAC,sBAAsB,CAAC,SAAS,EAAE;YAC3D,CAAC,EAAE,IAAI,CAAC,CAAC;SACV,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,OAAO,CACZ,GAAsB,EACtB,WAAwB,EACxB,UAOI,EAAE;QAEN,MAAM,EAAE,yBAAyB,EAAE,UAAU,EAAE,GAAG,IAAI,EAAE,GAAG,OAAO,CAAC;QACnE,MAAM,yBAAyB,GAAG,cAAc,CAAC,YAAY,CAC3D,yBAAyB,IAAI,2BAA2B,CACzD,CAAC;QACF,MAAM,SAAS,GAAG,cAAc,CAAC,YAAY,CAAC,UAAU,IAAI,WAAW,CAAC,CAAC;QAEzE,MAAM,OAAO,GAAG,gBAAgB,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;QAC7D,MAAM,sBAAsB,GAAG,IAAI,QAAQ,CAAC;YAC1C,MAAM,EAAE,yBAAyB;YACjC,GAAG;SACJ,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,IAAI,IAAI,CAAC;YACxB,WAAW;YACX,qBAAqB,EAAE,OAAO;YAC9B,sBAAsB;YACtB,GAAG,IAAI;SACR,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF"}