langchain 0.0.5 → 0.0.7

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 (478) hide show
  1. package/README.md +33 -0
  2. package/agents.d.ts +1 -0
  3. package/agents.js +1 -0
  4. package/chains.d.ts +1 -0
  5. package/chains.js +1 -0
  6. package/dist/{cjs/agents → agents}/agent.d.ts +1 -1
  7. package/dist/{cjs/agents → agents}/agent.js +0 -0
  8. package/dist/agents/agent.js.map +1 -0
  9. package/dist/{cjs/agents → agents}/executor.d.ts +0 -0
  10. package/dist/{cjs/agents → agents}/executor.js +0 -0
  11. package/dist/agents/executor.js.map +1 -0
  12. package/dist/{cjs/agents → agents}/helpers.d.ts +0 -0
  13. package/dist/{cjs/agents → agents}/helpers.js +0 -0
  14. package/dist/agents/helpers.js.map +1 -0
  15. package/dist/{cjs/agents → agents}/index.d.ts +0 -0
  16. package/dist/{cjs/agents → agents}/index.js +0 -0
  17. package/dist/agents/index.js.map +1 -0
  18. package/dist/{cjs/agents → agents}/initialize.d.ts +0 -0
  19. package/dist/{cjs/agents → agents}/initialize.js +2 -4
  20. package/dist/agents/initialize.js.map +1 -0
  21. package/dist/{cjs/agents → agents}/load.d.ts +0 -0
  22. package/dist/{cjs/agents → agents}/load.js +0 -0
  23. package/dist/agents/load.js.map +1 -0
  24. package/dist/{cjs/agents → agents}/mrkl/index.d.ts +1 -1
  25. package/dist/{cjs/agents → agents}/mrkl/index.js +5 -5
  26. package/dist/agents/mrkl/index.js.map +1 -0
  27. package/dist/{cjs/agents → agents}/mrkl/prompt.d.ts +0 -0
  28. package/dist/{cjs/agents → agents}/mrkl/prompt.js +0 -0
  29. package/dist/agents/mrkl/prompt.js.map +1 -0
  30. package/dist/{cjs/agents → agents}/tests/agent.test.d.ts +0 -0
  31. package/dist/{cjs/agents → agents}/tests/agent.test.js +0 -0
  32. package/dist/agents/tests/agent.test.js.map +1 -0
  33. package/dist/{cjs/agents → agents}/tests/calculator.test.d.ts +0 -0
  34. package/dist/{cjs/agents → agents}/tests/calculator.test.js +0 -0
  35. package/dist/agents/tests/calculator.test.js.map +1 -0
  36. package/dist/{cjs/agents → agents}/tools/base.d.ts +0 -0
  37. package/dist/{cjs/agents → agents}/tools/base.js +0 -0
  38. package/dist/agents/tools/base.js.map +1 -0
  39. package/dist/{cjs/agents → agents}/tools/calculator.d.ts +0 -0
  40. package/dist/{cjs/agents → agents}/tools/calculator.js +0 -0
  41. package/dist/agents/tools/calculator.js.map +1 -0
  42. package/dist/{cjs/agents → agents}/tools/index.d.ts +0 -0
  43. package/dist/{cjs/agents → agents}/tools/index.js +0 -0
  44. package/dist/agents/tools/index.js.map +1 -0
  45. package/dist/{cjs/agents → agents}/tools/llm-math.d.ts +0 -0
  46. package/dist/{cjs/agents → agents}/tools/llm-math.js +0 -0
  47. package/dist/{cjs/agents → agents}/tools/llm-math.js.map +1 -1
  48. package/dist/{cjs/agents → agents}/tools/serpapi.d.ts +0 -0
  49. package/dist/{cjs/agents → agents}/tools/serpapi.js +0 -0
  50. package/dist/agents/tools/serpapi.js.map +1 -0
  51. package/dist/{cjs/agents → agents}/types.d.ts +0 -0
  52. package/dist/{cjs/agents → agents}/types.js +0 -0
  53. package/dist/{cjs/agents → agents}/types.js.map +1 -1
  54. package/dist/{cjs/cache.d.ts → cache.d.ts} +0 -0
  55. package/dist/{cjs/cache.js → cache.js} +0 -0
  56. package/dist/cache.js.map +1 -0
  57. package/dist/{cjs/chains → chains}/base.d.ts +0 -0
  58. package/dist/{cjs/chains → chains}/base.js +8 -2
  59. package/dist/chains/base.js.map +1 -0
  60. package/dist/{cjs/chains → chains}/chat_vector_db_chain.d.ts +0 -0
  61. package/dist/{cjs/chains → chains}/chat_vector_db_chain.js +21 -7
  62. package/dist/chains/chat_vector_db_chain.js.map +1 -0
  63. package/dist/{cjs/chains → chains}/combine_docs_chain.d.ts +0 -0
  64. package/dist/{cjs/chains → chains}/combine_docs_chain.js +0 -0
  65. package/dist/chains/combine_docs_chain.js.map +1 -0
  66. package/dist/{esm/chains → chains}/conversation.d.ts +1 -1
  67. package/dist/{cjs/chains → chains}/conversation.js +2 -2
  68. package/dist/chains/conversation.js.map +1 -0
  69. package/dist/{esm/chains → chains}/index.d.ts +1 -1
  70. package/dist/{cjs/chains → chains}/index.js +0 -0
  71. package/dist/chains/index.js.map +1 -0
  72. package/dist/{esm/chains → chains}/llm_chain.d.ts +1 -1
  73. package/dist/{cjs/chains → chains}/llm_chain.js +3 -3
  74. package/dist/chains/llm_chain.js.map +1 -0
  75. package/dist/{cjs/chains → chains}/load.d.ts +0 -0
  76. package/dist/{cjs/chains → chains}/load.js +0 -0
  77. package/dist/chains/load.js.map +1 -0
  78. package/dist/{cjs/chains → chains}/question_answering/load.d.ts +0 -0
  79. package/dist/{cjs/chains → chains}/question_answering/load.js +0 -0
  80. package/dist/chains/question_answering/load.js.map +1 -0
  81. package/dist/{cjs/chains → chains}/question_answering/stuff_prompts.d.ts +1 -1
  82. package/dist/{cjs/chains → chains}/question_answering/stuff_prompts.js +2 -2
  83. package/dist/chains/question_answering/stuff_prompts.js.map +1 -0
  84. package/dist/{cjs/chains → chains}/question_answering/tests/load.test.d.ts +0 -0
  85. package/dist/{cjs/chains → chains}/question_answering/tests/load.test.js +5 -1
  86. package/dist/chains/question_answering/tests/load.test.js.map +1 -0
  87. package/dist/{cjs/chains → chains}/tests/chat_vector_db_qa_chain.test.d.ts +0 -0
  88. package/dist/{cjs/chains → chains}/tests/chat_vector_db_qa_chain.test.js +2 -2
  89. package/dist/chains/tests/chat_vector_db_qa_chain.test.js.map +1 -0
  90. package/dist/{cjs/chains → chains}/tests/combine_docs_chain.test.d.ts +0 -0
  91. package/dist/{cjs/chains → chains}/tests/combine_docs_chain.test.js +16 -5
  92. package/dist/chains/tests/combine_docs_chain.test.js.map +1 -0
  93. package/dist/{cjs/chains → chains}/tests/llm_chain.test.d.ts +0 -0
  94. package/dist/{cjs/chains → chains}/tests/llm_chain.test.js +2 -2
  95. package/dist/chains/tests/llm_chain.test.js.map +1 -0
  96. package/dist/{cjs/chains → chains}/tests/vector_db_qa_chain.test.d.ts +0 -0
  97. package/dist/{cjs/chains → chains}/tests/vector_db_qa_chain.test.js +2 -2
  98. package/dist/chains/tests/vector_db_qa_chain.test.js.map +1 -0
  99. package/dist/{cjs/chains → chains}/vector_db_qa.d.ts +0 -0
  100. package/dist/{cjs/chains → chains}/vector_db_qa.js +0 -0
  101. package/dist/chains/vector_db_qa.js.map +1 -0
  102. package/dist/{cjs/document.d.ts → document.d.ts} +0 -0
  103. package/dist/{cjs/document.js → document.js} +0 -0
  104. package/dist/document.js.map +1 -0
  105. package/dist/{cjs/embeddings → embeddings}/base.d.ts +0 -0
  106. package/dist/{cjs/embeddings → embeddings}/base.js +0 -0
  107. package/dist/embeddings/base.js.map +1 -0
  108. package/dist/{cjs/embeddings → embeddings}/index.d.ts +0 -0
  109. package/dist/{cjs/embeddings → embeddings}/index.js +0 -0
  110. package/dist/embeddings/index.js.map +1 -0
  111. package/dist/{cjs/embeddings → embeddings}/openai.d.ts +0 -0
  112. package/dist/{cjs/embeddings → embeddings}/openai.js +1 -1
  113. package/dist/embeddings/openai.js.map +1 -0
  114. package/dist/{cjs/embeddings → embeddings}/tests/openai.test.d.ts +0 -0
  115. package/dist/{cjs/embeddings → embeddings}/tests/openai.test.js +0 -0
  116. package/dist/embeddings/tests/openai.test.js.map +1 -0
  117. package/dist/{cjs/index.d.ts → index.d.ts} +1 -1
  118. package/dist/{cjs/index.js → index.js} +4 -4
  119. package/dist/index.js.map +1 -0
  120. package/dist/{cjs/llms → llms}/base.d.ts +0 -0
  121. package/dist/{cjs/llms → llms}/base.js +4 -3
  122. package/dist/llms/base.js.map +1 -0
  123. package/dist/llms/cohere.d.ts +24 -0
  124. package/dist/llms/cohere.js +55 -0
  125. package/dist/llms/cohere.js.map +1 -0
  126. package/dist/{cjs/llms → llms}/index.d.ts +5 -3
  127. package/dist/{cjs/llms → llms}/index.js +3 -1
  128. package/dist/llms/index.js.map +1 -0
  129. package/dist/{cjs/llms → llms}/load.d.ts +0 -0
  130. package/dist/{cjs/llms → llms}/load.js +0 -0
  131. package/dist/llms/load.js.map +1 -0
  132. package/dist/{esm/llms → llms}/openai.d.ts +5 -2
  133. package/dist/{cjs/llms → llms}/openai.js +49 -4
  134. package/dist/llms/openai.js.map +1 -0
  135. package/dist/{cjs/llms/tests/openai.test.d.ts → llms/tests/cohere.test.d.ts} +0 -0
  136. package/dist/llms/tests/cohere.test.js +10 -0
  137. package/dist/llms/tests/cohere.test.js.map +1 -0
  138. package/dist/{esm/embeddings → llms}/tests/openai.test.d.ts +0 -0
  139. package/dist/{cjs/llms → llms}/tests/openai.test.js +0 -0
  140. package/dist/llms/tests/openai.test.js.map +1 -0
  141. package/dist/{cjs/memory → memory}/base.d.ts +0 -0
  142. package/dist/{cjs/memory → memory}/base.js +0 -0
  143. package/dist/memory/base.js.map +1 -0
  144. package/dist/{cjs/memory → memory}/buffer_memory.d.ts +0 -0
  145. package/dist/{cjs/memory → memory}/buffer_memory.js +0 -0
  146. package/dist/memory/buffer_memory.js.map +1 -0
  147. package/dist/{cjs/memory → memory}/index.d.ts +0 -0
  148. package/dist/{cjs/memory → memory}/index.js +0 -0
  149. package/dist/memory/index.js.map +1 -0
  150. package/dist/{cjs/memory → memory}/tests/buffer_memory.test.d.ts +0 -0
  151. package/dist/{cjs/memory → memory}/tests/buffer_memory.test.js +0 -0
  152. package/dist/memory/tests/buffer_memory.test.js.map +1 -0
  153. package/dist/{cjs/prompt → prompts}/base.d.ts +0 -0
  154. package/dist/{cjs/prompt → prompts}/base.js +0 -0
  155. package/dist/prompts/base.js.map +1 -0
  156. package/dist/{cjs/prompt → prompts}/few_shot.d.ts +0 -0
  157. package/dist/{cjs/prompt → prompts}/few_shot.js +0 -0
  158. package/dist/prompts/few_shot.js.map +1 -0
  159. package/dist/{cjs/prompt → prompts}/index.d.ts +0 -0
  160. package/dist/{cjs/prompt → prompts}/index.js +0 -0
  161. package/dist/prompts/index.js.map +1 -0
  162. package/dist/{cjs/prompt → prompts}/load.d.ts +2 -2
  163. package/dist/{cjs/prompt → prompts}/load.js +2 -2
  164. package/dist/prompts/load.js.map +1 -0
  165. package/dist/{cjs/prompt → prompts}/parser.d.ts +0 -0
  166. package/dist/{cjs/prompt → prompts}/parser.js +0 -0
  167. package/dist/prompts/parser.js.map +1 -0
  168. package/dist/{cjs/prompt → prompts}/prompt.d.ts +1 -1
  169. package/dist/{cjs/prompt → prompts}/prompt.js +1 -1
  170. package/dist/prompts/prompt.js.map +1 -0
  171. package/dist/{cjs/prompt → prompts}/template.d.ts +0 -0
  172. package/dist/{cjs/prompt → prompts}/template.js +0 -0
  173. package/dist/prompts/template.js.map +1 -0
  174. package/dist/{cjs/prompt → prompts}/tests/load.test.d.ts +0 -0
  175. package/dist/{cjs/prompt → prompts}/tests/load.test.js +0 -0
  176. package/dist/prompts/tests/load.test.js.map +1 -0
  177. package/dist/{cjs/prompt → prompts}/tests/template.test.d.ts +0 -0
  178. package/dist/{cjs/prompt → prompts}/tests/template.test.js +0 -0
  179. package/dist/prompts/tests/template.test.js.map +1 -0
  180. package/dist/{cjs/text_splitter.d.ts → text_splitter.d.ts} +0 -0
  181. package/dist/{cjs/text_splitter.js → text_splitter.js} +3 -1
  182. package/dist/text_splitter.js.map +1 -0
  183. package/dist/{cjs/util → util}/hub.d.ts +0 -0
  184. package/dist/{cjs/util → util}/hub.js +0 -0
  185. package/dist/util/hub.js.map +1 -0
  186. package/dist/{cjs/util → util}/index.d.ts +0 -0
  187. package/dist/{cjs/util → util}/index.js +0 -0
  188. package/dist/util/index.js.map +1 -0
  189. package/dist/{esm/vectorstores → vectorstores}/base.d.ts +1 -1
  190. package/dist/{cjs/vectorstores → vectorstores}/base.js +1 -1
  191. package/dist/vectorstores/base.js.map +1 -0
  192. package/dist/{cjs/vectorstores → vectorstores}/hnswlib.d.ts +0 -0
  193. package/dist/{cjs/vectorstores → vectorstores}/hnswlib.js +6 -3
  194. package/dist/vectorstores/hnswlib.js.map +1 -0
  195. package/dist/{cjs/vectorstores → vectorstores}/index.d.ts +0 -0
  196. package/dist/{cjs/vectorstores → vectorstores}/index.js +0 -0
  197. package/dist/vectorstores/index.js.map +1 -0
  198. package/dist/{cjs/vectorstores → vectorstores}/tests/hnswlib.test.d.ts +0 -0
  199. package/dist/{cjs/vectorstores → vectorstores}/tests/hnswlib.test.js +6 -6
  200. package/dist/vectorstores/tests/hnswlib.test.js.map +1 -0
  201. package/document.d.ts +1 -0
  202. package/document.js +1 -0
  203. package/embeddings.d.ts +1 -0
  204. package/embeddings.js +1 -0
  205. package/llms.d.ts +1 -0
  206. package/llms.js +1 -0
  207. package/memory.d.ts +1 -0
  208. package/memory.js +1 -0
  209. package/package.json +48 -65
  210. package/prompts.d.ts +1 -0
  211. package/prompts.js +1 -0
  212. package/text_splitter.d.ts +1 -0
  213. package/text_splitter.js +1 -0
  214. package/tools.d.ts +1 -0
  215. package/tools.js +1 -0
  216. package/vectorstores.d.ts +1 -0
  217. package/vectorstores.js +1 -0
  218. package/dist/cjs/agents/agent.js.map +0 -1
  219. package/dist/cjs/agents/executor.js.map +0 -1
  220. package/dist/cjs/agents/helpers.js.map +0 -1
  221. package/dist/cjs/agents/index.js.map +0 -1
  222. package/dist/cjs/agents/initialize.js.map +0 -1
  223. package/dist/cjs/agents/load.js.map +0 -1
  224. package/dist/cjs/agents/mrkl/index.js.map +0 -1
  225. package/dist/cjs/agents/mrkl/prompt.js.map +0 -1
  226. package/dist/cjs/agents/tests/agent.test.js.map +0 -1
  227. package/dist/cjs/agents/tests/calculator.test.js.map +0 -1
  228. package/dist/cjs/agents/tools/base.js.map +0 -1
  229. package/dist/cjs/agents/tools/calculator.js.map +0 -1
  230. package/dist/cjs/agents/tools/index.js.map +0 -1
  231. package/dist/cjs/agents/tools/serpapi.js.map +0 -1
  232. package/dist/cjs/cache.js.map +0 -1
  233. package/dist/cjs/chains/base.js.map +0 -1
  234. package/dist/cjs/chains/chat_vector_db_chain.js.map +0 -1
  235. package/dist/cjs/chains/combine_docs_chain.js.map +0 -1
  236. package/dist/cjs/chains/conversation.d.ts +0 -12
  237. package/dist/cjs/chains/conversation.js.map +0 -1
  238. package/dist/cjs/chains/index.d.ts +0 -7
  239. package/dist/cjs/chains/index.js.map +0 -1
  240. package/dist/cjs/chains/llm_chain.d.ts +0 -65
  241. package/dist/cjs/chains/llm_chain.js.map +0 -1
  242. package/dist/cjs/chains/llm_math_chain.d.ts +0 -0
  243. package/dist/cjs/chains/llm_math_chain.js +0 -129
  244. package/dist/cjs/chains/llm_math_chain.js.map +0 -1
  245. package/dist/cjs/chains/load.js.map +0 -1
  246. package/dist/cjs/chains/question_answering/load.js.map +0 -1
  247. package/dist/cjs/chains/question_answering/stuff_prompts.js.map +0 -1
  248. package/dist/cjs/chains/question_answering/tests/load.test.js.map +0 -1
  249. package/dist/cjs/chains/tests/chat_vector_db_qa_chain.test.js.map +0 -1
  250. package/dist/cjs/chains/tests/combine_docs_chain.test.js.map +0 -1
  251. package/dist/cjs/chains/tests/llm_chain.test.js.map +0 -1
  252. package/dist/cjs/chains/tests/vector_db_qa_chain.test.js.map +0 -1
  253. package/dist/cjs/chains/vector_db_qa.js.map +0 -1
  254. package/dist/cjs/document.js.map +0 -1
  255. package/dist/cjs/embeddings/base.js.map +0 -1
  256. package/dist/cjs/embeddings/index.js.map +0 -1
  257. package/dist/cjs/embeddings/openai.js.map +0 -1
  258. package/dist/cjs/embeddings/tests/openai.test.js.map +0 -1
  259. package/dist/cjs/index.js.map +0 -1
  260. package/dist/cjs/llms/base.js.map +0 -1
  261. package/dist/cjs/llms/index.js.map +0 -1
  262. package/dist/cjs/llms/load.js.map +0 -1
  263. package/dist/cjs/llms/openai.d.ts +0 -125
  264. package/dist/cjs/llms/openai.js.map +0 -1
  265. package/dist/cjs/llms/tests/openai.test.js.map +0 -1
  266. package/dist/cjs/memory/base.js.map +0 -1
  267. package/dist/cjs/memory/buffer_memory.js.map +0 -1
  268. package/dist/cjs/memory/index.js.map +0 -1
  269. package/dist/cjs/memory/tests/buffer_memory.test.js.map +0 -1
  270. package/dist/cjs/prompt/base.js.map +0 -1
  271. package/dist/cjs/prompt/few_shot.js.map +0 -1
  272. package/dist/cjs/prompt/index.js.map +0 -1
  273. package/dist/cjs/prompt/load.js.map +0 -1
  274. package/dist/cjs/prompt/parser.js.map +0 -1
  275. package/dist/cjs/prompt/prompt.js.map +0 -1
  276. package/dist/cjs/prompt/template.js.map +0 -1
  277. package/dist/cjs/prompt/tests/load.test.js.map +0 -1
  278. package/dist/cjs/prompt/tests/template.test.js.map +0 -1
  279. package/dist/cjs/text_splitter.js.map +0 -1
  280. package/dist/cjs/text_splitter.test.d.ts +0 -1
  281. package/dist/cjs/text_splitter.test.js +0 -87
  282. package/dist/cjs/text_splitter.test.js.map +0 -1
  283. package/dist/cjs/util/hub.js.map +0 -1
  284. package/dist/cjs/util/index.js.map +0 -1
  285. package/dist/cjs/vectorstores/base.d.ts +0 -18
  286. package/dist/cjs/vectorstores/base.js.map +0 -1
  287. package/dist/cjs/vectorstores/hnswlib.js.map +0 -1
  288. package/dist/cjs/vectorstores/index.js.map +0 -1
  289. package/dist/cjs/vectorstores/tests/hnswlib.test.js.map +0 -1
  290. package/dist/esm/agents/agent.d.ts +0 -96
  291. package/dist/esm/agents/agent.js +0 -146
  292. package/dist/esm/agents/agent.js.map +0 -1
  293. package/dist/esm/agents/executor.d.ts +0 -32
  294. package/dist/esm/agents/executor.js +0 -95
  295. package/dist/esm/agents/executor.js.map +0 -1
  296. package/dist/esm/agents/helpers.d.ts +0 -3
  297. package/dist/esm/agents/helpers.js +0 -17
  298. package/dist/esm/agents/helpers.js.map +0 -1
  299. package/dist/esm/agents/index.d.ts +0 -7
  300. package/dist/esm/agents/index.js +0 -7
  301. package/dist/esm/agents/index.js.map +0 -1
  302. package/dist/esm/agents/initialize.d.ts +0 -4
  303. package/dist/esm/agents/initialize.js +0 -17
  304. package/dist/esm/agents/initialize.js.map +0 -1
  305. package/dist/esm/agents/load.d.ts +0 -6
  306. package/dist/esm/agents/load.js +0 -15
  307. package/dist/esm/agents/load.js.map +0 -1
  308. package/dist/esm/agents/mrkl/index.d.ts +0 -50
  309. package/dist/esm/agents/mrkl/index.js +0 -98
  310. package/dist/esm/agents/mrkl/index.js.map +0 -1
  311. package/dist/esm/agents/mrkl/prompt.d.ts +0 -3
  312. package/dist/esm/agents/mrkl/prompt.js +0 -16
  313. package/dist/esm/agents/mrkl/prompt.js.map +0 -1
  314. package/dist/esm/agents/tests/agent.test.d.ts +0 -1
  315. package/dist/esm/agents/tests/agent.test.js +0 -33
  316. package/dist/esm/agents/tests/agent.test.js.map +0 -1
  317. package/dist/esm/agents/tests/calculator.test.d.ts +0 -1
  318. package/dist/esm/agents/tests/calculator.test.js +0 -28
  319. package/dist/esm/agents/tests/calculator.test.js.map +0 -1
  320. package/dist/esm/agents/tools/base.d.ts +0 -6
  321. package/dist/esm/agents/tools/base.js +0 -11
  322. package/dist/esm/agents/tools/base.js.map +0 -1
  323. package/dist/esm/agents/tools/calculator.d.ts +0 -6
  324. package/dist/esm/agents/tools/calculator.js +0 -29
  325. package/dist/esm/agents/tools/calculator.js.map +0 -1
  326. package/dist/esm/agents/tools/index.d.ts +0 -3
  327. package/dist/esm/agents/tools/index.js +0 -4
  328. package/dist/esm/agents/tools/index.js.map +0 -1
  329. package/dist/esm/agents/tools/llm-math.d.ts +0 -0
  330. package/dist/esm/agents/tools/llm-math.js +0 -2
  331. package/dist/esm/agents/tools/llm-math.js.map +0 -1
  332. package/dist/esm/agents/tools/serpapi.d.ts +0 -18
  333. package/dist/esm/agents/tools/serpapi.js +0 -89
  334. package/dist/esm/agents/tools/serpapi.js.map +0 -1
  335. package/dist/esm/agents/types.d.ts +0 -25
  336. package/dist/esm/agents/types.js +0 -2
  337. package/dist/esm/agents/types.js.map +0 -1
  338. package/dist/esm/cache.d.ts +0 -11
  339. package/dist/esm/cache.js +0 -22
  340. package/dist/esm/cache.js.map +0 -1
  341. package/dist/esm/chains/base.d.ts +0 -42
  342. package/dist/esm/chains/base.js +0 -57
  343. package/dist/esm/chains/base.js.map +0 -1
  344. package/dist/esm/chains/chat_vector_db_chain.d.ts +0 -41
  345. package/dist/esm/chains/chat_vector_db_chain.js +0 -128
  346. package/dist/esm/chains/chat_vector_db_chain.js.map +0 -1
  347. package/dist/esm/chains/combine_docs_chain.d.ts +0 -35
  348. package/dist/esm/chains/combine_docs_chain.js +0 -71
  349. package/dist/esm/chains/combine_docs_chain.js.map +0 -1
  350. package/dist/esm/chains/conversation.js +0 -25
  351. package/dist/esm/chains/conversation.js.map +0 -1
  352. package/dist/esm/chains/index.js +0 -8
  353. package/dist/esm/chains/index.js.map +0 -1
  354. package/dist/esm/chains/llm_chain.js +0 -110
  355. package/dist/esm/chains/llm_chain.js.map +0 -1
  356. package/dist/esm/chains/llm_math_chain.d.ts +0 -0
  357. package/dist/esm/chains/llm_math_chain.js +0 -129
  358. package/dist/esm/chains/llm_math_chain.js.map +0 -1
  359. package/dist/esm/chains/load.d.ts +0 -21
  360. package/dist/esm/chains/load.js +0 -33
  361. package/dist/esm/chains/load.js.map +0 -1
  362. package/dist/esm/chains/question_answering/load.d.ts +0 -3
  363. package/dist/esm/chains/question_answering/load.js +0 -9
  364. package/dist/esm/chains/question_answering/load.js.map +0 -1
  365. package/dist/esm/chains/question_answering/stuff_prompts.d.ts +0 -2
  366. package/dist/esm/chains/question_answering/stuff_prompts.js +0 -7
  367. package/dist/esm/chains/question_answering/stuff_prompts.js.map +0 -1
  368. package/dist/esm/chains/question_answering/tests/load.test.d.ts +0 -1
  369. package/dist/esm/chains/question_answering/tests/load.test.js +0 -12
  370. package/dist/esm/chains/question_answering/tests/load.test.js.map +0 -1
  371. package/dist/esm/chains/tests/chat_vector_db_qa_chain.test.d.ts +0 -1
  372. package/dist/esm/chains/tests/chat_vector_db_qa_chain.test.js +0 -33
  373. package/dist/esm/chains/tests/chat_vector_db_qa_chain.test.js.map +0 -1
  374. package/dist/esm/chains/tests/combine_docs_chain.test.d.ts +0 -1
  375. package/dist/esm/chains/tests/combine_docs_chain.test.js +0 -26
  376. package/dist/esm/chains/tests/combine_docs_chain.test.js.map +0 -1
  377. package/dist/esm/chains/tests/llm_chain.test.d.ts +0 -1
  378. package/dist/esm/chains/tests/llm_chain.test.js +0 -27
  379. package/dist/esm/chains/tests/llm_chain.test.js.map +0 -1
  380. package/dist/esm/chains/tests/vector_db_qa_chain.test.d.ts +0 -1
  381. package/dist/esm/chains/tests/vector_db_qa_chain.test.js +0 -44
  382. package/dist/esm/chains/tests/vector_db_qa_chain.test.js.map +0 -1
  383. package/dist/esm/chains/vector_db_qa.d.ts +0 -36
  384. package/dist/esm/chains/vector_db_qa.js +0 -81
  385. package/dist/esm/chains/vector_db_qa.js.map +0 -1
  386. package/dist/esm/document.d.ts +0 -13
  387. package/dist/esm/document.js +0 -24
  388. package/dist/esm/document.js.map +0 -1
  389. package/dist/esm/embeddings/base.d.ts +0 -4
  390. package/dist/esm/embeddings/base.js +0 -3
  391. package/dist/esm/embeddings/base.js.map +0 -1
  392. package/dist/esm/embeddings/index.d.ts +0 -1
  393. package/dist/esm/embeddings/index.js +0 -2
  394. package/dist/esm/embeddings/index.js.map +0 -1
  395. package/dist/esm/embeddings/openai.d.ts +0 -20
  396. package/dist/esm/embeddings/openai.js +0 -82
  397. package/dist/esm/embeddings/openai.js.map +0 -1
  398. package/dist/esm/embeddings/tests/openai.test.js +0 -15
  399. package/dist/esm/embeddings/tests/openai.test.js.map +0 -1
  400. package/dist/esm/index.d.ts +0 -3
  401. package/dist/esm/index.js +0 -4
  402. package/dist/esm/index.js.map +0 -1
  403. package/dist/esm/llms/base.d.ts +0 -64
  404. package/dist/esm/llms/base.js +0 -159
  405. package/dist/esm/llms/base.js.map +0 -1
  406. package/dist/esm/llms/index.d.ts +0 -37
  407. package/dist/esm/llms/index.js +0 -4
  408. package/dist/esm/llms/index.js.map +0 -1
  409. package/dist/esm/llms/load.d.ts +0 -11
  410. package/dist/esm/llms/load.js +0 -13
  411. package/dist/esm/llms/load.js.map +0 -1
  412. package/dist/esm/llms/openai.js +0 -236
  413. package/dist/esm/llms/openai.js.map +0 -1
  414. package/dist/esm/llms/tests/openai.test.d.ts +0 -1
  415. package/dist/esm/llms/tests/openai.test.js +0 -8
  416. package/dist/esm/llms/tests/openai.test.js.map +0 -1
  417. package/dist/esm/memory/base.d.ts +0 -7
  418. package/dist/esm/memory/base.js +0 -3
  419. package/dist/esm/memory/base.js.map +0 -1
  420. package/dist/esm/memory/buffer_memory.d.ts +0 -15
  421. package/dist/esm/memory/buffer_memory.js +0 -53
  422. package/dist/esm/memory/buffer_memory.js.map +0 -1
  423. package/dist/esm/memory/index.d.ts +0 -2
  424. package/dist/esm/memory/index.js +0 -3
  425. package/dist/esm/memory/index.js.map +0 -1
  426. package/dist/esm/memory/tests/buffer_memory.test.d.ts +0 -1
  427. package/dist/esm/memory/tests/buffer_memory.test.js +0 -15
  428. package/dist/esm/memory/tests/buffer_memory.test.js.map +0 -1
  429. package/dist/esm/prompt/base.d.ts +0 -58
  430. package/dist/esm/prompt/base.js +0 -49
  431. package/dist/esm/prompt/base.js.map +0 -1
  432. package/dist/esm/prompt/few_shot.d.ts +0 -82
  433. package/dist/esm/prompt/few_shot.js +0 -133
  434. package/dist/esm/prompt/few_shot.js.map +0 -1
  435. package/dist/esm/prompt/index.d.ts +0 -4
  436. package/dist/esm/prompt/index.js +0 -5
  437. package/dist/esm/prompt/index.js.map +0 -1
  438. package/dist/esm/prompt/load.d.ts +0 -19
  439. package/dist/esm/prompt/load.js +0 -29
  440. package/dist/esm/prompt/load.js.map +0 -1
  441. package/dist/esm/prompt/parser.d.ts +0 -70
  442. package/dist/esm/prompt/parser.js +0 -107
  443. package/dist/esm/prompt/parser.js.map +0 -1
  444. package/dist/esm/prompt/prompt.d.ts +0 -76
  445. package/dist/esm/prompt/prompt.js +0 -107
  446. package/dist/esm/prompt/prompt.js.map +0 -1
  447. package/dist/esm/prompt/template.d.ts +0 -16
  448. package/dist/esm/prompt/template.js +0 -80
  449. package/dist/esm/prompt/template.js.map +0 -1
  450. package/dist/esm/prompt/tests/load.test.d.ts +0 -1
  451. package/dist/esm/prompt/tests/load.test.js +0 -16
  452. package/dist/esm/prompt/tests/load.test.js.map +0 -1
  453. package/dist/esm/prompt/tests/template.test.d.ts +0 -1
  454. package/dist/esm/prompt/tests/template.test.js +0 -25
  455. package/dist/esm/prompt/tests/template.test.js.map +0 -1
  456. package/dist/esm/text_splitter.d.ts +0 -32
  457. package/dist/esm/text_splitter.js +0 -160
  458. package/dist/esm/text_splitter.js.map +0 -1
  459. package/dist/esm/text_splitter.test.d.ts +0 -1
  460. package/dist/esm/text_splitter.test.js +0 -85
  461. package/dist/esm/text_splitter.test.js.map +0 -1
  462. package/dist/esm/util/hub.d.ts +0 -2
  463. package/dist/esm/util/hub.js +0 -34
  464. package/dist/esm/util/hub.js.map +0 -1
  465. package/dist/esm/util/index.d.ts +0 -13
  466. package/dist/esm/util/index.js +0 -61
  467. package/dist/esm/util/index.js.map +0 -1
  468. package/dist/esm/vectorstores/base.js +0 -33
  469. package/dist/esm/vectorstores/base.js.map +0 -1
  470. package/dist/esm/vectorstores/hnswlib.d.ts +0 -19
  471. package/dist/esm/vectorstores/hnswlib.js +0 -119
  472. package/dist/esm/vectorstores/hnswlib.js.map +0 -1
  473. package/dist/esm/vectorstores/index.d.ts +0 -1
  474. package/dist/esm/vectorstores/index.js +0 -2
  475. package/dist/esm/vectorstores/index.js.map +0 -1
  476. package/dist/esm/vectorstores/tests/hnswlib.test.d.ts +0 -1
  477. package/dist/esm/vectorstores/tests/hnswlib.test.js +0 -39
  478. package/dist/esm/vectorstores/tests/hnswlib.test.js.map +0 -1
@@ -1,87 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- const globals_1 = require("@jest/globals");
4
- const document_1 = require("./document");
5
- const text_splitter_1 = require("./text_splitter");
6
- (0, globals_1.test)('Test splitting by character count.', () => {
7
- const text = 'foo bar baz 123';
8
- const splitter = new text_splitter_1.CharacterTextSplitter({ separator: ' ', chunkSize: 7, chunkOverlap: 3 });
9
- const output = splitter.splitText(text);
10
- const expectedOutput = ['foo bar', 'bar baz', 'baz 123'];
11
- (0, globals_1.expect)(output).toEqual(expectedOutput);
12
- });
13
- (0, globals_1.test)('Test splitting by character count doesn\'t create empty documents.', () => {
14
- const text = 'foo bar';
15
- const splitter = new text_splitter_1.CharacterTextSplitter({ separator: ' ', chunkSize: 2, chunkOverlap: 0 });
16
- const output = splitter.splitText(text);
17
- const expectedOutput = ['foo', 'bar'];
18
- (0, globals_1.expect)(output).toEqual(expectedOutput);
19
- });
20
- (0, globals_1.test)('Test splitting by character count on long words.', () => {
21
- const text = 'foo bar baz a a';
22
- const splitter = new text_splitter_1.CharacterTextSplitter({ separator: ' ', chunkSize: 3, chunkOverlap: 1 });
23
- const output = splitter.splitText(text);
24
- const expectedOutput = ['foo', 'bar', 'baz', 'a a'];
25
- (0, globals_1.expect)(output).toEqual(expectedOutput);
26
- });
27
- (0, globals_1.test)('Test splitting by character count when shorter words are first.', () => {
28
- const text = 'a a foo bar baz';
29
- const splitter = new text_splitter_1.CharacterTextSplitter({ separator: ' ', chunkSize: 3, chunkOverlap: 1 });
30
- const output = splitter.splitText(text);
31
- const expectedOutput = ['a a', 'foo', 'bar', 'baz'];
32
- (0, globals_1.expect)(output).toEqual(expectedOutput);
33
- });
34
- (0, globals_1.test)('Test splitting by characters when splits not found easily.', () => {
35
- const text = 'foo bar baz 123';
36
- const splitter = new text_splitter_1.CharacterTextSplitter({ separator: ' ', chunkSize: 1, chunkOverlap: 0 });
37
- const output = splitter.splitText(text);
38
- const expectedOutput = ['foo', 'bar', 'baz', '123'];
39
- (0, globals_1.expect)(output).toEqual(expectedOutput);
40
- });
41
- (0, globals_1.test)('Test invalid arguments.', () => {
42
- (0, globals_1.expect)(() => {
43
- new text_splitter_1.CharacterTextSplitter({ chunkSize: 2, chunkOverlap: 4 });
44
- }).toThrow();
45
- });
46
- (0, globals_1.test)('Test create documents method.', () => {
47
- const texts = ['foo bar', 'baz'];
48
- const splitter = new text_splitter_1.CharacterTextSplitter({ separator: ' ', chunkSize: 3, chunkOverlap: 0 });
49
- const docs = splitter.createDocuments(texts);
50
- const expectedDocs = [new document_1.Document({ pageContent: 'foo' }), new document_1.Document({ pageContent: 'bar' }), new document_1.Document({ pageContent: 'baz' }),];
51
- (0, globals_1.expect)(docs).toEqual(expectedDocs);
52
- });
53
- (0, globals_1.test)('Test create documents with metadata method.', () => {
54
- const texts = ['foo bar', 'baz'];
55
- const splitter = new text_splitter_1.CharacterTextSplitter({ separator: ' ', chunkSize: 3, chunkOverlap: 0 });
56
- const docs = splitter.createDocuments(texts, [{ source: '1' }, { source: '2' }]);
57
- const expectedDocs = [new document_1.Document({ pageContent: 'foo', metadata: { source: '1' } }), new document_1.Document({ pageContent: 'bar', metadata: { source: '1' } }), new document_1.Document({ pageContent: 'baz', metadata: { source: '2' } }),];
58
- (0, globals_1.expect)(docs).toEqual(expectedDocs);
59
- });
60
- (0, globals_1.test)('Test iterative text splitter.', () => {
61
- const text = `Hi.\n\nI'm Harrison.\n\nHow? Are? You?\nOkay then f f f f.
62
- This is a weird text to write, but gotta test the splittingggg some how.\n\n
63
- Bye!\n\n-H.`;
64
- const splitter = new text_splitter_1.RecursiveCharacterTextSplitter({ chunkSize: 10, chunkOverlap: 1 });
65
- const output = splitter.splitText(text);
66
- const expectedOutput = [
67
- "Hi.",
68
- "I'm",
69
- "Harrison.",
70
- "How? Are?",
71
- "You?",
72
- "Okay then f",
73
- "f f f f.",
74
- "This is a",
75
- "a weird",
76
- "text to",
77
- "write, but",
78
- "gotta test",
79
- "the",
80
- "splitting",
81
- "gggg",
82
- "some how.",
83
- "Bye!\n\n-H.",
84
- ];
85
- (0, globals_1.expect)(output).toEqual(expectedOutput);
86
- });
87
- //# sourceMappingURL=text_splitter.test.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"text_splitter.test.js","sourceRoot":"","sources":["../../text_splitter.test.ts"],"names":[],"mappings":";;AAAA,2CAA6C;AAC7C,yCAAsC;AACtC,mDAAwF;AAExF,IAAA,cAAI,EAAC,oCAAoC,EAAE,GAAG,EAAE;IAC9C,MAAM,IAAI,GAAG,iBAAiB,CAAC;IAC/B,MAAM,QAAQ,GAAG,IAAI,qCAAqB,CAAC,EAAC,SAAS,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;IAC7F,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IACxC,MAAM,cAAc,GAAG,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;IACzD,IAAA,gBAAM,EAAC,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;AACzC,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,oEAAoE,EAAE,GAAG,EAAE;IAC9E,MAAM,IAAI,GAAG,UAAU,CAAC;IACxB,MAAM,QAAQ,GAAG,IAAI,qCAAqB,CAAC,EAAC,SAAS,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;IAC7F,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IACxC,MAAM,cAAc,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IACtC,IAAA,gBAAM,EAAC,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;AACzC,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,kDAAkD,EAAE,GAAG,EAAE;IAC5D,MAAM,IAAI,GAAG,iBAAiB,CAAC;IAC/B,MAAM,QAAQ,GAAG,IAAI,qCAAqB,CAAC,EAAC,SAAS,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;IAC7F,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IACxC,MAAM,cAAc,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;IACpD,IAAA,gBAAM,EAAC,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;AACzC,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,iEAAiE,EAAE,GAAG,EAAE;IAC3E,MAAM,IAAI,GAAG,iBAAiB,CAAC;IAC/B,MAAM,QAAQ,GAAG,IAAI,qCAAqB,CAAC,EAAC,SAAS,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;IAC7F,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IACxC,MAAM,cAAc,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;IACpD,IAAA,gBAAM,EAAC,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;AACzC,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,4DAA4D,EAAE,GAAG,EAAE;IACtE,MAAM,IAAI,GAAG,iBAAiB,CAAC;IAC/B,MAAM,QAAQ,GAAG,IAAI,qCAAqB,CAAC,EAAC,SAAS,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;IAC7F,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IACxC,MAAM,cAAc,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;IACpD,IAAA,gBAAM,EAAC,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;AACzC,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,yBAAyB,EAAE,GAAG,EAAE;IACnC,IAAA,gBAAM,EAAC,GAAG,EAAE;QACV,IAAI,qCAAqB,CAAC,EAAC,SAAS,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;AACf,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,+BAA+B,EAAE,GAAG,EAAE;IACzC,MAAM,KAAK,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IACjC,MAAM,QAAQ,GAAG,IAAI,qCAAqB,CAAC,EAAC,SAAS,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;IAC7F,MAAM,IAAI,GAAG,QAAQ,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAC7C,MAAM,YAAY,GAAG,CAAK,IAAI,mBAAQ,CAAC,EAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAK,IAAI,mBAAQ,CAAC,EAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAK,IAAI,mBAAQ,CAAC,EAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAI,CAAC;IAClJ,IAAA,gBAAM,EAAC,IAAI,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;AACrC,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,6CAA6C,EAAE,GAAG,EAAE;IACvD,MAAM,KAAK,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IACjC,MAAM,QAAQ,GAAG,IAAI,qCAAqB,CAAC,EAAC,SAAS,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;IAC7F,MAAM,IAAI,GAAG,QAAQ,CAAC,eAAe,CAAC,KAAK,EAAE,CAAE,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;IAClF,MAAM,YAAY,GAAG,CAAK,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,EAAK,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,EAAK,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,EAAI,CAAC;IACtO,IAAA,gBAAM,EAAC,IAAI,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;AACrC,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,+BAA+B,EAAE,GAAG,EAAE;IACvC,MAAM,IAAI,GAAG;;YAEL,CAAC;IACT,MAAM,QAAQ,GAAG,IAAI,8CAA8B,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;IACxF,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IACxC,MAAM,cAAc,GAAG;QACnB,KAAK;QACL,KAAK;QACL,WAAW;QACX,WAAW;QACX,MAAM;QACN,aAAa;QACb,UAAU;QACV,WAAW;QACX,SAAS;QACT,SAAS;QACT,YAAY;QACZ,YAAY;QACZ,KAAK;QACL,WAAW;QACX,MAAM;QACN,WAAW;QACX,aAAa;KAChB,CAAC;IACF,IAAA,gBAAM,EAAC,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;AAC3C,CAAC,CAAC,CAAC"}
@@ -1 +0,0 @@
1
- {"version":3,"file":"hub.js","sourceRoot":"","sources":["../../../util/hub.ts"],"names":[],"mappings":";;;;;;;AAAA,gDAAwB;AACxB,4CAAoB;AACpB,4CAAoB;AACpB,mCAA2C;AAE3C,MAAM,cAAc,GAAG,sBAAsB,CAAC;AAC9C,MAAM,WAAW,GAAG,MAAA,OAAO,CAAC,GAAG,CAAC,yBAAyB,mCAAI,QAAQ,CAAC;AACtE,MAAM,QAAQ,GACZ,MAAA,OAAO,CAAC,GAAG,CAAC,sBAAsB,mCAClC,4DAA4D,CAAC;AAIxD,MAAM,WAAW,GAAG,KAAK,EAC9B,GAAW,EACX,MAA4C,EAC5C,WAAmB,EACnB,aAA0B,EAC1B,SAAqB,EAAE,EACC,EAAE;IAC1B,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IACxC,IAAI,CAAC,KAAK,EAAE;QACV,OAAO,SAAS,CAAC;KAClB;IACD,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAC5C,MAAM,GAAG,GAAG,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC;IACnD,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,cAAI,CAAC,GAAG,CAAC,CAAC;IACzC,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,WAAW,EAAE;QAC5B,OAAO,SAAS,CAAC;KAClB;IAED,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,cAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE;QACzD,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;KAC3C;IAED,MAAM,GAAG,GAAG,CAAC,QAAQ,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAClD,MAAM,GAAG,GAAG,MAAM,IAAA,wBAAgB,EAAC,GAAG,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3D,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE;QACtB,MAAM,IAAI,KAAK,CAAC,0BAA0B,GAAG,EAAE,CAAC,CAAC;KAClD;IAED,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;IAC9B,MAAM,MAAM,GAAG,YAAE,CAAC,WAAW,CAAC,cAAI,CAAC,IAAI,CAAC,YAAE,CAAC,MAAM,EAAE,EAAE,WAAW,CAAC,CAAC,CAAC;IACnE,MAAM,IAAI,GAAG,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,cAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC;IAC1D,YAAE,CAAC,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAC7B,OAAO,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;AAC9B,CAAC,CAAC;AAjCW,QAAA,WAAW,eAiCtB"}
@@ -1 +0,0 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../util/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,gDAAwB;AACxB,4DAAgD;AAChD,4CAAoB;AACpB,2CAA6B;AAEtB,MAAM,gBAAgB,GAAG,KAAK,EACnC,GAAW,EACX,IAAuD,EACvD,EAAE;IACF,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,EAAE,GAAG,IAAI,CAAC;IAClC,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,OAAO,CAAC,CAAC;IAChE,8DAA8D;IAC9D,MAAM,GAAG,GAAG,MAAM,IAAA,oBAAK,EAAC,GAAG,EAAE,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,UAAU,CAAC,MAAa,EAAE,CAAC,CAAC;IAC5E,YAAY,CAAC,SAAS,CAAC,CAAC;IACxB,OAAO,GAAG,CAAC;AACb,CAAC,CAAC;AAXW,QAAA,gBAAgB,oBAW3B;AAEF,MAAM,gBAAgB,GAAG,CAAC,QAAgB,EAAE,MAAc,EAAE,EAAE;IAC5D,QAAQ,MAAM,EAAE;QACd,KAAK,OAAO;YACV,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAC9B,KAAK,MAAM,CAAC;QACZ,KAAK,OAAO;YACV,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAC9B;YACE,MAAM,IAAI,KAAK,CAAC,wBAAwB,MAAM,EAAE,CAAC,CAAC;KACrD;AACH,CAAC,CAAC;AAiBF,MAAM,oBAAoB,GAAG,CAC3B,SAAY,EACZ,MAAsB,EACtB,IAA6C,EAC7C,eAA0B,EACvB,EAAE;IACL,MAAM,SAAS,GAAG,MAAM,CAAC,GAAG,SAAS,OAAO,CAAuB,CAAC;IACpE,MAAM,KAAK,GAAG,MAAM,CAAC,SAAS,CAAkB,CAAC;IACjD,IAAI,SAAS,KAAK,SAAS,IAAI,KAAK,KAAK,SAAS,EAAE;QAClD,MAAM,IAAI,KAAK,CACb,SAAS,SAAS,eAAe,SAAS,uBAAuB,CAClE,CAAC;KACH;IAED,IAAI,KAAK,KAAK,SAAS,EAAE;QACvB,OAAO,KAAK,CAAC;KACd;IAED,IAAI,SAAS,KAAK,SAAS,EAAE;QAC3B,MAAM,MAAM,GAAG,cAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACvC,IAAI,eAAe,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE;YACxD,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;SACtC;QAED,OAAO,IAAI,CAAC,YAAE,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,QAAQ,EAAE,EAAE,MAAM,CAAC,CAAC;KAC5D;IAED,MAAM,IAAI,KAAK,CACb,WAAW,SAAS,eAAe,SAAS,qBAAqB,CAClE,CAAC;AACJ,CAAC,CAAC;AAEK,MAAM,uBAAuB,GAAG,CACrC,SAAY,EACZ,MAA2B,EAC3B,EAAE,CAAC,oBAAoB,CAAC,SAAS,EAAE,MAAM,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;AAHlE,QAAA,uBAAuB,2BAG2C;AAExE,MAAM,qBAAqB,GAAG,CACnC,SAAY,EACZ,MAAsB,EACnB,EAAE,CACL,oBAAoB,CAAC,SAAS,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;AAJnE,QAAA,qBAAqB,yBAI8C;AAEzE,MAAM,eAAe,GAAG,CAAC,IAAY,EAAE,cAAyB,EAAE,EAAE;IACzE,MAAM,MAAM,GAAG,cAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAElC,IACE,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC;QACpC,CAAC,cAAc,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,EACpD;QACA,MAAM,IAAI,KAAK,CAAC,wBAAwB,MAAM,EAAE,CAAC,CAAC;KACnD;IAED,MAAM,QAAQ,GAAG,YAAE,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;IAClD,OAAO,gBAAgB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;AAC5C,CAAC,CAAC;AAZW,QAAA,eAAe,mBAY1B;AAEK,MAAM,UAAU,GAAG,CAAI,GAAQ,EAAE,SAAiB,EAAE,EAAE,CAC3D,GAAG,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE;IACjC,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC,CAAC;IACjD,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;IACvC,6CAA6C;IAC7C,MAAM,CAAC,UAAU,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAC1C,OAAO,MAAM,CAAC;AAChB,CAAC,EAAE,EAAW,CAAC,CAAC;AAPL,QAAA,UAAU,cAOL"}
@@ -1,18 +0,0 @@
1
- import { Embeddings } from "embeddings/base";
2
- import { Document } from "../document";
3
- export interface DocStore {
4
- [key: number]: object;
5
- }
6
- export declare abstract class VectorStore {
7
- embeddings: Embeddings;
8
- docstore: DocStore;
9
- abstract addVectors(vectors: number[][], documents: Document[]): Promise<void>;
10
- abstract similaritySearchVectorWithScore(query: number[], k: number): Promise<[Document, number][]>;
11
- addDocuments(documents: Document[]): Promise<void>;
12
- similaritySearch(query: string, k?: number): Promise<Document[]>;
13
- similaritySearchWithScore(query: string, k?: number): Promise<[object, number][]>;
14
- }
15
- export declare abstract class SaveableVectorStore extends VectorStore {
16
- abstract save(directory: string): Promise<void>;
17
- static load(_directory: string, _embeddings: Embeddings): Promise<SaveableVectorStore>;
18
- }
@@ -1 +0,0 @@
1
- {"version":3,"file":"base.js","sourceRoot":"","sources":["../../../vectorstores/base.ts"],"names":[],"mappings":";;;AAQA,MAAsB,WAAW;IAAjC;QACE;;;;;WAAuB;QAEvB;;;;;WAAmB;IAmCrB,CAAC;IA1BC,KAAK,CAAC,YAAY,CAAC,SAAqB;QACtC,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAE,CAAC,EAAC,WAAW,EAAC,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC;QAC/D,IAAI,CAAC,UAAU,CACb,MAAM,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,KAAK,CAAC,EAC3C,SAAS,CACV,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,KAAa,EAAE,CAAC,GAAG,CAAC;QACzC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,+BAA+B,CACxD,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,EACvC,CAAC,CACF,CAAC;QAEF,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,yBAAyB,CAC7B,KAAa,EACb,CAAC,GAAG,CAAC;QAEL,OAAO,IAAI,CAAC,+BAA+B,CACzC,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,EACvC,CAAC,CACF,CAAC;IACJ,CAAC;CACF;AAtCD,kCAsCC;AAED,MAAsB,mBAAoB,SAAQ,WAAW;IAG3D,MAAM,CAAC,IAAI,CACT,UAAkB,EAClB,WAAuB;QAEvB,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;CACF;AATD,kDASC"}
@@ -1 +0,0 @@
1
- {"version":3,"file":"hnswlib.js","sourceRoot":"","sources":["../../../vectorstores/hnswlib.ts"],"names":[],"mappings":";;;;;;AAAA,2DAA6B;AAC7B,gDAAwB;AAQxB,iCAAuD;AACvD,0CAAuC;AAEvC,IAAI,eAAe,GAAmC,IAAI,CAAC;AAE3D,IAAI;IACF,0CAA0C;IAC1C,CAAC,EAAE,eAAe,EAAE,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC;CACjD;AAAC,WAAM;IACN,eAAe;CAChB;AAOD,MAAa,OAAQ,SAAQ,0BAAmB;IAK9C,YACE,IAAiB,EACjB,UAAsB,EACtB,QAAkB,EAClB,KAAwB;QAExB,KAAK,EAAE,CAAC;QAVV;;;;;WAAyB;QAEzB;;;;;WAAkB;QAShB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,OAAmB,EAAE,SAAqB;QACzD,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE;YACxB,OAAO;SACR;QACD,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE;YACf,IAAI,IAAI,CAAC,IAAI,CAAC,aAAa,KAAK,SAAS,EAAE;gBACzC,IAAI,CAAC,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;aAC7C;YACD,IAAI,eAAe,KAAK,IAAI,EAAE;gBAC5B,MAAM,IAAI,KAAK,CACb,sFAAsF,CACvF,CAAC;aACH;YACD,IAAI,CAAC,KAAK,GAAG,IAAI,eAAe,CAC9B,IAAI,CAAC,IAAI,CAAC,KAAK,EACf,IAAI,CAAC,IAAI,CAAC,aAAa,CACxB,CAAC;YACF,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;SACtC;QACD,qEAAqE;QACrE,oEAAoE;QACpE,uEAAuE;QACvE,+DAA+D;QAC/D,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS,CAAC,MAAM,EAAE;YACvC,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;SACpE;QACD,IAAI,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,KAAK,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE;YACjD,MAAM,IAAI,KAAK,CACb,kEAAkE,IAAI,CAAC,IAAI,CAAC,aAAa,GAAG,CAC7F,CAAC;SACH;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,CAAC;QAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,eAAe,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7D,IAAI,MAAM,GAAG,QAAQ,EAAE;YACrB,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,GAAG,QAAQ,CAAC,CAAC;SAC3C;QACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;YAC1C,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YACnC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;SACjC;IACH,CAAC;IAED,KAAK,CAAC,+BAA+B,CAAC,KAAe,EAAE,CAAS;QAC9D,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE;YACf,MAAM,IAAI,KAAK,CACb,iEAAiE,CAClE,CAAC;SACH;QACD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAC9C,OAAO,MAAM,CAAC,SAAS,CAAC,GAAG,CACzB,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,CACxB,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAGtD,CACJ,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,SAAiB;QAC1B,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE;YACf,MAAM,IAAI,KAAK,CACb,iEAAiE,CAClE,CAAC;SACH;QACD,MAAM,kBAAE,CAAC,KAAK,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/C,MAAM,OAAO,CAAC,GAAG,CAAC;YAChB,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;YAC5D,MAAM,kBAAE,CAAC,SAAS,CAChB,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,WAAW,CAAC,EACjC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAC1B;YACD,MAAM,kBAAE,CAAC,SAAS,CAChB,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,eAAe,CAAC,EACrC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAC9B;SACF,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,SAAiB,EAAE,UAAsB;QACzD,IAAI,eAAe,KAAK,IAAI,EAAE;YAC5B,MAAM,IAAI,KAAK,CACb,sFAAsF,CACvF,CAAC;SACH;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CACrB,MAAM,kBAAE,CAAC,QAAQ,CAAC,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,WAAW,CAAC,EAAE,MAAM,CAAC,CAC7D,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QAClE,MAAM,CAAC,QAAQ,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACnC,kBAAE;iBACC,QAAQ,CAAC,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,eAAe,CAAC,EAAE,MAAM,CAAC;iBACvD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC;YACnB,KAAK,CAAC,SAAS,CAAC,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;SACvD,CAAC,CAAC;QAEH,OAAO,IAAI,OAAO,CAAC,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;IACxD,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,SAAS,CACpB,KAAe,EACf,SAAmB,EACnB,UAAsB;QAEtB,MAAM,IAAI,GAAG,EAAE,CAAC;QAChB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;YACnC,MAAM,MAAM,GAAG,IAAI,mBAAQ,CAAC,EAAC,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC,EAAC,CAAC,CAAC;YAC7E,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;SACrB;QACD,OAAO,OAAO,CAAC,aAAa,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;IACjD,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,aAAa,CACxB,IAAgB,EAChB,UAAsB;QAEtB,IAAI,eAAe,KAAK,IAAI,EAAE;YAC5B,MAAM,IAAI,KAAK,CACb,sFAAsF,CACvF,CAAC;SACH;QACD,MAAM,IAAI,GAAgB;YACxB,KAAK,EAAE,IAAI,EAAE,cAAc;SAC5B,CAAC;QACF,MAAM,QAAQ,GAAG,IAAI,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC;QAChD,MAAM,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QAClC,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF;AAjJD,0BAiJC"}
@@ -1 +0,0 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../vectorstores/index.ts"],"names":[],"mappings":";;;AAAA,qCAAoC;AAA3B,kGAAA,OAAO,OAAA"}
@@ -1 +0,0 @@
1
- {"version":3,"file":"hnswlib.test.js","sourceRoot":"","sources":["../../../../vectorstores/tests/hnswlib.test.ts"],"names":[],"mappings":";;;;;AAAA,2CAA6C;AAC7C,2DAA6B;AAC7B,gDAAwB;AACxB,4CAAoB;AAEpB,wCAAqC;AACrC,iDAAoD;AAEpD,IAAA,cAAI,EAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;;IACxC,MAAM,WAAW,GAAG,MAAM,iBAAO,CAAC,SAAS,CACzC,CAAC,aAAa,EAAE,SAAS,EAAE,kBAAkB,CAAC,EAC9C,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EACjC,IAAI,6BAAgB,EAAE,CACvB,CAAC;IACF,IAAA,gBAAM,EAAC,MAAA,WAAW,CAAC,KAAK,0CAAE,eAAe,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAErD,MAAM,SAAS,GAAG,MAAM,WAAW,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;IACvE,MAAM,kBAAkB,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,EAAC,QAAQ,EAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;IACrE,IAAA,gBAAM,EAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAEhD,MAAM,SAAS,GAAG,MAAM,WAAW,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;IACvE,MAAM,kBAAkB,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,EAAC,QAAQ,EAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;IACrE,IAAA,gBAAM,EAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;AACxE,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;;IACpD,MAAM,WAAW,GAAG,MAAM,iBAAO,CAAC,SAAS,CACzC,CAAC,aAAa,EAAE,SAAS,EAAE,kBAAkB,CAAC,EAC9C,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EACjC,IAAI,6BAAgB,EAAE,CACvB,CAAC;IACF,IAAA,gBAAM,EAAC,MAAA,WAAW,CAAC,KAAK,0CAAE,eAAe,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAErD,MAAM,SAAS,GAAG,MAAM,WAAW,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;IACvE,MAAM,kBAAkB,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,EAAC,QAAQ,EAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;IACrE,IAAA,gBAAM,EAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAEhD,MAAM,SAAS,GAAG,MAAM,WAAW,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;IACvE,MAAM,kBAAkB,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,EAAC,QAAQ,EAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;IACrE,IAAA,gBAAM,EAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAEtE,MAAM,aAAa,GAAG,MAAM,kBAAE,CAAC,OAAO,CAAC,cAAI,CAAC,IAAI,CAAC,YAAE,CAAC,MAAM,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC;IAExE,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAE3B,MAAM,WAAW,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAEtC,MAAM,iBAAiB,GAAG,MAAM,iBAAO,CAAC,IAAI,CAC1C,aAAa,EACb,IAAI,6BAAgB,EAAE,CACvB,CAAC;IAEF,MAAM,WAAW,GAAG,MAAM,iBAAiB,CAAC,gBAAgB,CAC1D,aAAa,EACb,CAAC,CACF,CAAC;IAEF,MAAM,oBAAoB,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,EAAC,QAAQ,EAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;IACzE,IAAA,gBAAM,EAAC,oBAAoB,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAElD,MAAM,UAAU,GAAG,MAAM,iBAAiB,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;IAE9E,MAAM,mBAAmB,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,EAAC,QAAQ,EAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;IACvE,IAAA,gBAAM,EAAC,mBAAmB,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;AACzE,CAAC,CAAC,CAAC"}
@@ -1,96 +0,0 @@
1
- import { ChainValues } from "../chains";
2
- import { SerializedZeroShotAgent, AgentAction, AgentFinish, AgentStep, StoppingMethod, Tool } from "./index";
3
- import { BaseLLM } from "../llms";
4
- import { LLMChain } from "../chains/llm_chain";
5
- import { BasePromptTemplate } from "../prompt";
6
- export interface StaticAgent {
7
- /**
8
- * Create a prompt for this class
9
- *
10
- * @param tools - List of tools the agent will have access to, used to format the prompt.
11
- * @param fields - Additional fields used to format the prompt.
12
- *
13
- * @returns A PromptTemplate assembled from the given tools and fields.
14
- * */
15
- createPrompt(tools: Tool[], fields?: Record<string, any>): BasePromptTemplate;
16
- /** Construct an agent from an LLM and a list of tools */
17
- fromLLMAndTools(llm: BaseLLM, tools: Tool[], args?: Record<string, any>): Agent;
18
- validateTools(_: Tool[]): void;
19
- }
20
- export declare const staticImplements: <T>(_: T) => void;
21
- type SerializedAgent = SerializedZeroShotAgent;
22
- export interface AgentInput {
23
- llmChain: LLMChain;
24
- allowedTools?: string[];
25
- }
26
- /**
27
- * Class responsible for calling a language model and deciding an action.
28
- *
29
- * @remarks This is driven by an LLMChain. The prompt in the LLMChain *must*
30
- * include a variable called "agent_scratchpad" where the agent can put its
31
- * intermediary work.
32
- */
33
- export declare abstract class Agent {
34
- llmChain: LLMChain;
35
- allowedTools?: string[];
36
- returnValues: string[];
37
- constructor(input: AgentInput);
38
- /**
39
- * Extract tool and tool input from LLM output.
40
- */
41
- abstract extractToolAndInput(input: string): {
42
- tool: string;
43
- input: string;
44
- } | null;
45
- /**
46
- * Prefix to append the observation with.
47
- */
48
- abstract observationPrefix(): string;
49
- /**
50
- * Prefix to append the LLM call with.
51
- */
52
- abstract llmPrefix(): string;
53
- /**
54
- * Return the string type key uniquely identifying this class of agent.
55
- */
56
- abstract _agentType(): string;
57
- /**
58
- * Prepare the agent for a new call, if needed
59
- */
60
- prepareForNewCall(): void;
61
- /**
62
- * Validate that appropriate tools are passed in
63
- */
64
- static validateTools(_: Tool[]): void;
65
- _stop(): string[];
66
- /**
67
- * Name of tool to use to terminate the chain.
68
- */
69
- finishToolName(): string;
70
- /**
71
- * Construct a scratchpad to let the agent continue its thought process
72
- */
73
- private constructScratchPad;
74
- private _plan;
75
- /**
76
- * Decide what to do given some input.
77
- *
78
- * @param steps - Steps the LLM has taken so far, along with observations from each.
79
- * @param inputs - User inputs.
80
- *
81
- * @returns Action specifying what tool to use.
82
- */
83
- plan(steps: AgentStep[], inputs: ChainValues): Promise<AgentAction | AgentFinish>;
84
- /**
85
- * Return response when agent has been stopped due to max iterations
86
- */
87
- returnStoppedResponse(earlyStoppingMethod: StoppingMethod, steps: AgentStep[], inputs: ChainValues): Promise<AgentFinish>;
88
- /**
89
- * Load an agent from a json-like object describing it.
90
- */
91
- static deserialize(data: SerializedAgent & {
92
- llm?: BaseLLM;
93
- tools?: Tool[];
94
- }): Promise<Agent>;
95
- }
96
- export {};
@@ -1,146 +0,0 @@
1
- import { ZeroShotAgent, } from "./index";
2
- class ParseError extends Error {
3
- constructor(msg, output) {
4
- super(msg);
5
- Object.defineProperty(this, "output", {
6
- enumerable: true,
7
- configurable: true,
8
- writable: true,
9
- value: void 0
10
- });
11
- this.output = output;
12
- }
13
- }
14
- export const staticImplements = (_) => { };
15
- /**
16
- * Class responsible for calling a language model and deciding an action.
17
- *
18
- * @remarks This is driven by an LLMChain. The prompt in the LLMChain *must*
19
- * include a variable called "agent_scratchpad" where the agent can put its
20
- * intermediary work.
21
- */
22
- export class Agent {
23
- constructor(input) {
24
- Object.defineProperty(this, "llmChain", {
25
- enumerable: true,
26
- configurable: true,
27
- writable: true,
28
- value: void 0
29
- });
30
- Object.defineProperty(this, "allowedTools", {
31
- enumerable: true,
32
- configurable: true,
33
- writable: true,
34
- value: undefined
35
- });
36
- Object.defineProperty(this, "returnValues", {
37
- enumerable: true,
38
- configurable: true,
39
- writable: true,
40
- value: ["output"]
41
- });
42
- this.llmChain = input.llmChain;
43
- this.allowedTools = input.allowedTools;
44
- }
45
- /**
46
- * Prepare the agent for a new call, if needed
47
- */
48
- prepareForNewCall() { }
49
- /**
50
- * Validate that appropriate tools are passed in
51
- */
52
- // eslint-disable-next-line no-unused-vars
53
- static validateTools(_) { }
54
- _stop() {
55
- return [`\n${this.observationPrefix()}`];
56
- }
57
- /**
58
- * Name of tool to use to terminate the chain.
59
- */
60
- finishToolName() {
61
- return "Final Answer";
62
- }
63
- /**
64
- * Construct a scratchpad to let the agent continue its thought process
65
- */
66
- constructScratchPad(steps) {
67
- return steps.reduce((thoughts, { action, observation }) => thoughts +
68
- [
69
- action.log,
70
- `${this.observationPrefix()}${observation}`,
71
- this.llmPrefix(),
72
- ].join("\n"), "");
73
- }
74
- async _plan(steps, inputs, suffix) {
75
- const thoughts = this.constructScratchPad(steps);
76
- const newInputs = {
77
- ...inputs,
78
- agent_scratchpad: suffix ? `${thoughts}${suffix}` : thoughts,
79
- stop: this._stop(),
80
- };
81
- const output = await this.llmChain.predict(newInputs);
82
- const parsed = this.extractToolAndInput(output);
83
- if (!parsed) {
84
- throw new ParseError(`Invalid output: ${output}`, output);
85
- }
86
- const action = {
87
- tool: parsed.tool,
88
- toolInput: parsed.input,
89
- log: output,
90
- };
91
- if (action.tool === this.finishToolName()) {
92
- return { returnValues: { output: action.toolInput }, log: action.log };
93
- }
94
- return action;
95
- }
96
- /**
97
- * Decide what to do given some input.
98
- *
99
- * @param steps - Steps the LLM has taken so far, along with observations from each.
100
- * @param inputs - User inputs.
101
- *
102
- * @returns Action specifying what tool to use.
103
- */
104
- plan(steps, inputs) {
105
- return this._plan(steps, inputs);
106
- }
107
- /**
108
- * Return response when agent has been stopped due to max iterations
109
- */
110
- async returnStoppedResponse(earlyStoppingMethod, steps, inputs) {
111
- if (earlyStoppingMethod === "force") {
112
- return {
113
- returnValues: { output: "Agent stopped due to max iterations." },
114
- log: "",
115
- };
116
- }
117
- if (earlyStoppingMethod === "generate") {
118
- try {
119
- const action = await this._plan(steps, inputs, "\n\nI now need to return a final answer based on the previous steps:");
120
- if ("returnValues" in action) {
121
- return action;
122
- }
123
- return { returnValues: { output: action.log }, log: action.log };
124
- }
125
- catch (err) {
126
- if (!(err instanceof ParseError)) {
127
- throw err;
128
- }
129
- return { returnValues: { output: err.output }, log: err.output };
130
- }
131
- }
132
- throw new Error(`Invalid stopping method: ${earlyStoppingMethod}`);
133
- }
134
- /**
135
- * Load an agent from a json-like object describing it.
136
- */
137
- static async deserialize(data) {
138
- switch (data._type) {
139
- case "zero-shot-react-description":
140
- return ZeroShotAgent.deserialize(data);
141
- default:
142
- throw new Error("Unknown agent type");
143
- }
144
- }
145
- }
146
- //# sourceMappingURL=agent.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"agent.js","sourceRoot":"","sources":["../../../agents/agent.ts"],"names":[],"mappings":"AACA,OAAO,EACL,aAAa,GAOd,MAAM,SAAS,CAAC;AAKjB,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;AAyBD,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAI,CAAI,EAAE,EAAE,GAAE,CAAC,CAAC;AAShD;;;;;;GAMG;AACH,MAAM,OAAgB,KAAK;IAOzB,YAAY,KAAiB;QAN7B;;;;;WAAmB;QAEnB;;;;mBAA0B,SAAS;WAAC;QAEpC;;;;mBAAe,CAAC,QAAQ,CAAC;WAAC;QAGxB,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,0CAA0C;IAC1C,MAAM,CAAC,aAAa,CAAC,CAAS,IAAS,CAAC;IAExC,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;IACK,mBAAmB,CAAC,KAAkB;QAC5C,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;YAC5D,IAAI,EAAE,IAAI,CAAC,KAAK,EAAE;SACnB,CAAC;QACF,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;gBAChC,OAAO,aAAa,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACzC;gBACE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;SACzC;IACH,CAAC;CACF"}
@@ -1,32 +0,0 @@
1
- import { ChainValues, BaseChain } from "../chains";
2
- import { Agent, Tool, StoppingMethod } from "./index";
3
- import { SerializedLLMChain } from "../chains/llm_chain";
4
- type AgentExecutorInput = {
5
- agent: Agent;
6
- tools: Tool[];
7
- returnIntermediateSteps?: boolean;
8
- maxIterations?: number;
9
- earlyStoppingMethod?: StoppingMethod;
10
- };
11
- /**
12
- * A chain managing an agent using tools.
13
- * @augments BaseChain
14
- */
15
- export declare class AgentExecutor extends BaseChain {
16
- agent: Agent;
17
- tools: Tool[];
18
- returnIntermediateSteps: boolean;
19
- maxIterations?: number;
20
- earlyStoppingMethod: StoppingMethod;
21
- constructor(input: AgentExecutorInput);
22
- /** Create from agent and a list of tools. */
23
- static fromAgentAndTools(fields: {
24
- agent: Agent;
25
- tools: Tool[];
26
- } & Record<string, any>): AgentExecutor;
27
- private shouldContinue;
28
- _call(inputs: ChainValues): Promise<ChainValues>;
29
- _chainType(): "agent_executor";
30
- serialize(): SerializedLLMChain;
31
- }
32
- export {};
@@ -1,95 +0,0 @@
1
- import { BaseChain } from "../chains";
2
- /**
3
- * A chain managing an agent using tools.
4
- * @augments BaseChain
5
- */
6
- export class AgentExecutor extends BaseChain {
7
- constructor(input) {
8
- var _a, _b, _c;
9
- super();
10
- Object.defineProperty(this, "agent", {
11
- enumerable: true,
12
- configurable: true,
13
- writable: true,
14
- value: void 0
15
- });
16
- Object.defineProperty(this, "tools", {
17
- enumerable: true,
18
- configurable: true,
19
- writable: true,
20
- value: void 0
21
- });
22
- Object.defineProperty(this, "returnIntermediateSteps", {
23
- enumerable: true,
24
- configurable: true,
25
- writable: true,
26
- value: false
27
- });
28
- Object.defineProperty(this, "maxIterations", {
29
- enumerable: true,
30
- configurable: true,
31
- writable: true,
32
- value: 15
33
- });
34
- Object.defineProperty(this, "earlyStoppingMethod", {
35
- enumerable: true,
36
- configurable: true,
37
- writable: true,
38
- value: "force"
39
- });
40
- this.agent = input.agent;
41
- this.tools = input.tools;
42
- this.returnIntermediateSteps =
43
- (_a = input.returnIntermediateSteps) !== null && _a !== void 0 ? _a : this.returnIntermediateSteps;
44
- this.maxIterations = (_b = input.maxIterations) !== null && _b !== void 0 ? _b : this.maxIterations;
45
- this.earlyStoppingMethod =
46
- (_c = input.earlyStoppingMethod) !== null && _c !== void 0 ? _c : this.earlyStoppingMethod;
47
- }
48
- /** Create from agent and a list of tools. */
49
- static fromAgentAndTools(fields) {
50
- return new AgentExecutor(fields);
51
- }
52
- shouldContinue(iterations) {
53
- return this.maxIterations === undefined || iterations < this.maxIterations;
54
- }
55
- async _call(inputs) {
56
- this.agent.prepareForNewCall();
57
- const toolsByName = Object.fromEntries(this.tools.map((t) => [t.name.toLowerCase(), t]));
58
- const steps = [];
59
- let iterations = 0;
60
- const getOutput = (finishStep) => {
61
- const { returnValues } = finishStep;
62
- if (this.returnIntermediateSteps) {
63
- return { ...returnValues, intermediateSteps: steps };
64
- }
65
- return returnValues;
66
- };
67
- while (this.shouldContinue(iterations)) {
68
- const action = await this.agent.plan(steps, inputs);
69
- if ("returnValues" in action) {
70
- return getOutput(action);
71
- }
72
- const tool = toolsByName[action.tool.toLowerCase()];
73
- const observation = tool
74
- ? await tool.call(action.toolInput)
75
- : `${action.tool} is not a valid tool, try another one.`;
76
- steps.push({ action, observation });
77
- if (tool === null || tool === void 0 ? void 0 : tool.returnDirect) {
78
- return getOutput({
79
- returnValues: { [this.agent.returnValues[0]]: observation },
80
- log: "",
81
- });
82
- }
83
- iterations += 1;
84
- }
85
- const finish = await this.agent.returnStoppedResponse(this.earlyStoppingMethod, steps, inputs);
86
- return getOutput(finish);
87
- }
88
- _chainType() {
89
- return "agent_executor";
90
- }
91
- serialize() {
92
- throw new Error("Cannot serialize an AgentExecutor");
93
- }
94
- }
95
- //# sourceMappingURL=executor.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"executor.js","sourceRoot":"","sources":["../../../agents/executor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAe,SAAS,EAAE,MAAM,WAAW,CAAC;AAYnD;;;GAGG;AACH,MAAM,OAAO,aAAc,SAAQ,SAAS;IAW1C,YAAY,KAAyB;;QACnC,KAAK,EAAE,CAAC;QAXV;;;;;WAAa;QAEb;;;;;WAAc;QAEd;;;;mBAA0B,KAAK;WAAC;QAEhC;;;;mBAAyB,EAAE;WAAC;QAE5B;;;;mBAAsC,OAAO;WAAC;QAI5C,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QACzB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QACzB,IAAI,CAAC,uBAAuB;YAC1B,MAAA,KAAK,CAAC,uBAAuB,mCAAI,IAAI,CAAC,uBAAuB,CAAC;QAChE,IAAI,CAAC,aAAa,GAAG,MAAA,KAAK,CAAC,aAAa,mCAAI,IAAI,CAAC,aAAa,CAAC;QAC/D,IAAI,CAAC,mBAAmB;YACtB,MAAA,KAAK,CAAC,mBAAmB,mCAAI,IAAI,CAAC,mBAAmB,CAAC;IAC1D,CAAC;IAED,6CAA6C;IAC7C,MAAM,CAAC,iBAAiB,CACtB,MAIuB;QAEvB,OAAO,IAAI,aAAa,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC;IAEO,cAAc,CAAC,UAAkB;QACvC,OAAO,IAAI,CAAC,aAAa,KAAK,SAAS,IAAI,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC;IAC7E,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAmB;QAC7B,IAAI,CAAC,KAAK,CAAC,iBAAiB,EAAE,CAAC;QAC/B,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,CACpC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC,CACjD,CAAC;QACF,MAAM,KAAK,GAAgB,EAAE,CAAC;QAC9B,IAAI,UAAU,GAAG,CAAC,CAAC;QAEnB,MAAM,SAAS,GAAG,CAAC,UAAuB,EAAE,EAAE;YAC5C,MAAM,EAAE,YAAY,EAAE,GAAG,UAAU,CAAC;YACpC,IAAI,IAAI,CAAC,uBAAuB,EAAE;gBAChC,OAAO,EAAE,GAAG,YAAY,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC;aACtD;YACD,OAAO,YAAY,CAAC;QACtB,CAAC,CAAC;QAEF,OAAO,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE;YACtC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YACpD,IAAI,cAAc,IAAI,MAAM,EAAE;gBAC5B,OAAO,SAAS,CAAC,MAAM,CAAC,CAAC;aAC1B;YAED,MAAM,IAAI,GAAG,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;YACpD,MAAM,WAAW,GAAG,IAAI;gBACtB,CAAC,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;gBACnC,CAAC,CAAC,GAAG,MAAM,CAAC,IAAI,wCAAwC,CAAC;YAC3D,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;YACpC,IAAI,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,YAAY,EAAE;gBACtB,OAAO,SAAS,CAAC;oBACf,YAAY,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE;oBAC3D,GAAG,EAAE,EAAE;iBACR,CAAC,CAAC;aACJ;YACD,UAAU,IAAI,CAAC,CAAC;SACjB;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,qBAAqB,CACnD,IAAI,CAAC,mBAAmB,EACxB,KAAK,EACL,MAAM,CACP,CAAC;QAEF,OAAO,SAAS,CAAC,MAAM,CAAC,CAAC;IAC3B,CAAC;IAED,UAAU;QACR,OAAO,gBAAyB,CAAC;IACnC,CAAC;IAED,SAAS;QACP,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;IACvD,CAAC;CACF"}
@@ -1,3 +0,0 @@
1
- import { SerializedAgentT, Tool, AgentInput } from "./index";
2
- import { BaseLLM } from "../llms";
3
- export declare const deserializeHelper: <T extends string, U, V extends AgentInput, Z>(llm: BaseLLM | undefined, tools: Tool[] | undefined, data: SerializedAgentT<T, U, V>, fromLLMAndTools: (llm: BaseLLM, tools: Tool[], args: U) => Z, fromConstructor: (args: V) => Z) => Promise<Z>;
@@ -1,17 +0,0 @@
1
- import { LLMChain } from "../chains";
2
- import { resolveConfigFromFile } from "../util";
3
- export const deserializeHelper = async (llm, tools, data, fromLLMAndTools, fromConstructor) => {
4
- if (data.load_from_llm_and_tools) {
5
- if (!llm) {
6
- throw new Error("Loading from llm and tools, llm must be provided.");
7
- }
8
- if (!tools) {
9
- throw new Error("Loading from llm and tools, tools must be provided.");
10
- }
11
- return fromLLMAndTools(llm, tools, data);
12
- }
13
- const serializedLLMChain = resolveConfigFromFile("llm_chain", data);
14
- const llmChain = await LLMChain.deserialize(serializedLLMChain);
15
- return fromConstructor({ ...data, llmChain });
16
- };
17
- //# sourceMappingURL=helpers.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"helpers.js","sourceRoot":"","sources":["../../../agents/helpers.ts"],"names":[],"mappings":"AAEA,OAAO,EAAsB,QAAQ,EAAE,MAAM,WAAW,CAAC;AACzD,OAAO,EAAE,qBAAqB,EAAE,MAAM,SAAS,CAAC;AAEhD,MAAM,CAAC,MAAM,iBAAiB,GAAG,KAAK,EAMpC,GAAwB,EACxB,KAAyB,EACzB,IAA+B,EAC/B,eAA4D,EAC5D,eAA+B,EACnB,EAAE;IACd,IAAI,IAAI,CAAC,uBAAuB,EAAE;QAChC,IAAI,CAAC,GAAG,EAAE;YACR,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;SACtE;QAED,IAAI,CAAC,KAAK,EAAE;YACV,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;SACxE;QAED,OAAO,eAAe,CAAC,GAAG,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;KAC1C;IAED,MAAM,kBAAkB,GAAG,qBAAqB,CAG9C,WAAW,EAAE,IAAI,CAAC,CAAC;IACrB,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC,kBAAkB,CAAC,CAAC;IAChE,OAAO,eAAe,CAAC,EAAE,GAAG,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;AAChD,CAAC,CAAC"}
@@ -1,7 +0,0 @@
1
- export { AgentAction, AgentFinish, AgentStep, StoppingMethod, SerializedAgentT, } from "./types";
2
- export { Agent, StaticAgent, staticImplements, AgentInput } from "./agent";
3
- export { AgentExecutor } from "./executor";
4
- export { ZeroShotAgent, SerializedZeroShotAgent } from "./mrkl";
5
- export { Tool } from "./tools";
6
- export { initializeAgentExecutor } from "./initialize";
7
- export { loadAgent } from "./load";
@@ -1,7 +0,0 @@
1
- export { Agent, staticImplements } from "./agent";
2
- export { AgentExecutor } from "./executor";
3
- export { ZeroShotAgent } from "./mrkl";
4
- export { Tool } from "./tools";
5
- export { initializeAgentExecutor } from "./initialize";
6
- export { loadAgent } from "./load";
7
- //# sourceMappingURL=index.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../agents/index.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,KAAK,EAAe,gBAAgB,EAAc,MAAM,SAAS,CAAC;AAC3E,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAC3C,OAAO,EAAE,aAAa,EAA2B,MAAM,QAAQ,CAAC;AAChE,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAC/B,OAAO,EAAC,uBAAuB,EAAC,MAAM,cAAc,CAAC;AAErD,OAAO,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC"}
@@ -1,4 +0,0 @@
1
- import { Tool } from "./tools";
2
- import { BaseLLM } from "../llms";
3
- import { AgentExecutor } from "./executor";
4
- export declare const initializeAgentExecutor: (tools: Tool[], llm: BaseLLM, agentType?: string) => Promise<AgentExecutor>;
@@ -1,17 +0,0 @@
1
- import { AgentExecutor } from "./executor";
2
- import { ZeroShotAgent } from "./mrkl";
3
- export const initializeAgentExecutor = async (tools, llm, agentType = "zero-shot-react-description") => {
4
- switch (agentType) {
5
- case "zero-shot-react-description":
6
- const agent = ZeroShotAgent.fromLLMAndTools(llm, tools);
7
- const executor = AgentExecutor.fromAgentAndTools({
8
- agent,
9
- tools,
10
- returnIntermediateSteps: true,
11
- });
12
- return executor;
13
- default:
14
- throw new Error("Unknown agent type");
15
- }
16
- };
17
- //# sourceMappingURL=initialize.js.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"initialize.js","sourceRoot":"","sources":["../../../agents/initialize.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAC3C,OAAO,EAAE,aAAa,EAAE,MAAM,QAAQ,CAAC;AAEvC,MAAM,CAAC,MAAM,uBAAuB,GAAG,KAAK,EACxC,KAAa,EACb,GAAY,EACZ,SAAS,GAAE,6BAA6B,EAChB,EAAE;IAC1B,QAAQ,SAAS,EAAE;QACf,KAAK,6BAA6B;YAC9B,MAAM,KAAK,GAAG,aAAa,CAAC,eAAe,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YACxD,MAAM,QAAQ,GAAG,aAAa,CAAC,iBAAiB,CAAC;gBAC7C,KAAK;gBACL,KAAK;gBACL,uBAAuB,EAAE,IAAI;aAC5B,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC;QACxB;YACI,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;KAC7C;AACH,CAAC,CAAC"}
@@ -1,6 +0,0 @@
1
- import { Agent, Tool } from ".";
2
- import { BaseLLM } from "../llms";
3
- export declare const loadAgent: (uri: string, llmAndTools?: {
4
- llm?: BaseLLM;
5
- tools?: Tool[];
6
- }) => Promise<Agent>;