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.
- package/README.md +33 -0
- package/agents.d.ts +1 -0
- package/agents.js +1 -0
- package/chains.d.ts +1 -0
- package/chains.js +1 -0
- package/dist/{cjs/agents → agents}/agent.d.ts +1 -1
- package/dist/{cjs/agents → agents}/agent.js +0 -0
- package/dist/agents/agent.js.map +1 -0
- package/dist/{cjs/agents → agents}/executor.d.ts +0 -0
- package/dist/{cjs/agents → agents}/executor.js +0 -0
- package/dist/agents/executor.js.map +1 -0
- package/dist/{cjs/agents → agents}/helpers.d.ts +0 -0
- package/dist/{cjs/agents → agents}/helpers.js +0 -0
- package/dist/agents/helpers.js.map +1 -0
- package/dist/{cjs/agents → agents}/index.d.ts +0 -0
- package/dist/{cjs/agents → agents}/index.js +0 -0
- package/dist/agents/index.js.map +1 -0
- package/dist/{cjs/agents → agents}/initialize.d.ts +0 -0
- package/dist/{cjs/agents → agents}/initialize.js +2 -4
- package/dist/agents/initialize.js.map +1 -0
- package/dist/{cjs/agents → agents}/load.d.ts +0 -0
- package/dist/{cjs/agents → agents}/load.js +0 -0
- package/dist/agents/load.js.map +1 -0
- package/dist/{cjs/agents → agents}/mrkl/index.d.ts +1 -1
- package/dist/{cjs/agents → agents}/mrkl/index.js +5 -5
- package/dist/agents/mrkl/index.js.map +1 -0
- package/dist/{cjs/agents → agents}/mrkl/prompt.d.ts +0 -0
- package/dist/{cjs/agents → agents}/mrkl/prompt.js +0 -0
- package/dist/agents/mrkl/prompt.js.map +1 -0
- package/dist/{cjs/agents → agents}/tests/agent.test.d.ts +0 -0
- package/dist/{cjs/agents → agents}/tests/agent.test.js +0 -0
- package/dist/agents/tests/agent.test.js.map +1 -0
- package/dist/{cjs/agents → agents}/tests/calculator.test.d.ts +0 -0
- package/dist/{cjs/agents → agents}/tests/calculator.test.js +0 -0
- package/dist/agents/tests/calculator.test.js.map +1 -0
- package/dist/{cjs/agents → agents}/tools/base.d.ts +0 -0
- package/dist/{cjs/agents → agents}/tools/base.js +0 -0
- package/dist/agents/tools/base.js.map +1 -0
- package/dist/{cjs/agents → agents}/tools/calculator.d.ts +0 -0
- package/dist/{cjs/agents → agents}/tools/calculator.js +0 -0
- package/dist/agents/tools/calculator.js.map +1 -0
- package/dist/{cjs/agents → agents}/tools/index.d.ts +0 -0
- package/dist/{cjs/agents → agents}/tools/index.js +0 -0
- package/dist/agents/tools/index.js.map +1 -0
- package/dist/{cjs/agents → agents}/tools/llm-math.d.ts +0 -0
- package/dist/{cjs/agents → agents}/tools/llm-math.js +0 -0
- package/dist/{cjs/agents → agents}/tools/llm-math.js.map +1 -1
- package/dist/{cjs/agents → agents}/tools/serpapi.d.ts +0 -0
- package/dist/{cjs/agents → agents}/tools/serpapi.js +0 -0
- package/dist/agents/tools/serpapi.js.map +1 -0
- package/dist/{cjs/agents → agents}/types.d.ts +0 -0
- package/dist/{cjs/agents → agents}/types.js +0 -0
- package/dist/{cjs/agents → agents}/types.js.map +1 -1
- package/dist/{cjs/cache.d.ts → cache.d.ts} +0 -0
- package/dist/{cjs/cache.js → cache.js} +0 -0
- package/dist/cache.js.map +1 -0
- package/dist/{cjs/chains → chains}/base.d.ts +0 -0
- package/dist/{cjs/chains → chains}/base.js +8 -2
- package/dist/chains/base.js.map +1 -0
- package/dist/{cjs/chains → chains}/chat_vector_db_chain.d.ts +0 -0
- package/dist/{cjs/chains → chains}/chat_vector_db_chain.js +21 -7
- package/dist/chains/chat_vector_db_chain.js.map +1 -0
- package/dist/{cjs/chains → chains}/combine_docs_chain.d.ts +0 -0
- package/dist/{cjs/chains → chains}/combine_docs_chain.js +0 -0
- package/dist/chains/combine_docs_chain.js.map +1 -0
- package/dist/{esm/chains → chains}/conversation.d.ts +1 -1
- package/dist/{cjs/chains → chains}/conversation.js +2 -2
- package/dist/chains/conversation.js.map +1 -0
- package/dist/{esm/chains → chains}/index.d.ts +1 -1
- package/dist/{cjs/chains → chains}/index.js +0 -0
- package/dist/chains/index.js.map +1 -0
- package/dist/{esm/chains → chains}/llm_chain.d.ts +1 -1
- package/dist/{cjs/chains → chains}/llm_chain.js +3 -3
- package/dist/chains/llm_chain.js.map +1 -0
- package/dist/{cjs/chains → chains}/load.d.ts +0 -0
- package/dist/{cjs/chains → chains}/load.js +0 -0
- package/dist/chains/load.js.map +1 -0
- package/dist/{cjs/chains → chains}/question_answering/load.d.ts +0 -0
- package/dist/{cjs/chains → chains}/question_answering/load.js +0 -0
- package/dist/chains/question_answering/load.js.map +1 -0
- package/dist/{cjs/chains → chains}/question_answering/stuff_prompts.d.ts +1 -1
- package/dist/{cjs/chains → chains}/question_answering/stuff_prompts.js +2 -2
- package/dist/chains/question_answering/stuff_prompts.js.map +1 -0
- package/dist/{cjs/chains → chains}/question_answering/tests/load.test.d.ts +0 -0
- package/dist/{cjs/chains → chains}/question_answering/tests/load.test.js +5 -1
- package/dist/chains/question_answering/tests/load.test.js.map +1 -0
- package/dist/{cjs/chains → chains}/tests/chat_vector_db_qa_chain.test.d.ts +0 -0
- package/dist/{cjs/chains → chains}/tests/chat_vector_db_qa_chain.test.js +2 -2
- package/dist/chains/tests/chat_vector_db_qa_chain.test.js.map +1 -0
- package/dist/{cjs/chains → chains}/tests/combine_docs_chain.test.d.ts +0 -0
- package/dist/{cjs/chains → chains}/tests/combine_docs_chain.test.js +16 -5
- package/dist/chains/tests/combine_docs_chain.test.js.map +1 -0
- package/dist/{cjs/chains → chains}/tests/llm_chain.test.d.ts +0 -0
- package/dist/{cjs/chains → chains}/tests/llm_chain.test.js +2 -2
- package/dist/chains/tests/llm_chain.test.js.map +1 -0
- package/dist/{cjs/chains → chains}/tests/vector_db_qa_chain.test.d.ts +0 -0
- package/dist/{cjs/chains → chains}/tests/vector_db_qa_chain.test.js +2 -2
- package/dist/chains/tests/vector_db_qa_chain.test.js.map +1 -0
- package/dist/{cjs/chains → chains}/vector_db_qa.d.ts +0 -0
- package/dist/{cjs/chains → chains}/vector_db_qa.js +0 -0
- package/dist/chains/vector_db_qa.js.map +1 -0
- package/dist/{cjs/document.d.ts → document.d.ts} +0 -0
- package/dist/{cjs/document.js → document.js} +0 -0
- package/dist/document.js.map +1 -0
- package/dist/{cjs/embeddings → embeddings}/base.d.ts +0 -0
- package/dist/{cjs/embeddings → embeddings}/base.js +0 -0
- package/dist/embeddings/base.js.map +1 -0
- package/dist/{cjs/embeddings → embeddings}/index.d.ts +0 -0
- package/dist/{cjs/embeddings → embeddings}/index.js +0 -0
- package/dist/embeddings/index.js.map +1 -0
- package/dist/{cjs/embeddings → embeddings}/openai.d.ts +0 -0
- package/dist/{cjs/embeddings → embeddings}/openai.js +1 -1
- package/dist/embeddings/openai.js.map +1 -0
- package/dist/{cjs/embeddings → embeddings}/tests/openai.test.d.ts +0 -0
- package/dist/{cjs/embeddings → embeddings}/tests/openai.test.js +0 -0
- package/dist/embeddings/tests/openai.test.js.map +1 -0
- package/dist/{cjs/index.d.ts → index.d.ts} +1 -1
- package/dist/{cjs/index.js → index.js} +4 -4
- package/dist/index.js.map +1 -0
- package/dist/{cjs/llms → llms}/base.d.ts +0 -0
- package/dist/{cjs/llms → llms}/base.js +4 -3
- package/dist/llms/base.js.map +1 -0
- package/dist/llms/cohere.d.ts +24 -0
- package/dist/llms/cohere.js +55 -0
- package/dist/llms/cohere.js.map +1 -0
- package/dist/{cjs/llms → llms}/index.d.ts +5 -3
- package/dist/{cjs/llms → llms}/index.js +3 -1
- package/dist/llms/index.js.map +1 -0
- package/dist/{cjs/llms → llms}/load.d.ts +0 -0
- package/dist/{cjs/llms → llms}/load.js +0 -0
- package/dist/llms/load.js.map +1 -0
- package/dist/{esm/llms → llms}/openai.d.ts +5 -2
- package/dist/{cjs/llms → llms}/openai.js +49 -4
- package/dist/llms/openai.js.map +1 -0
- package/dist/{cjs/llms/tests/openai.test.d.ts → llms/tests/cohere.test.d.ts} +0 -0
- package/dist/llms/tests/cohere.test.js +10 -0
- package/dist/llms/tests/cohere.test.js.map +1 -0
- package/dist/{esm/embeddings → llms}/tests/openai.test.d.ts +0 -0
- package/dist/{cjs/llms → llms}/tests/openai.test.js +0 -0
- package/dist/llms/tests/openai.test.js.map +1 -0
- package/dist/{cjs/memory → memory}/base.d.ts +0 -0
- package/dist/{cjs/memory → memory}/base.js +0 -0
- package/dist/memory/base.js.map +1 -0
- package/dist/{cjs/memory → memory}/buffer_memory.d.ts +0 -0
- package/dist/{cjs/memory → memory}/buffer_memory.js +0 -0
- package/dist/memory/buffer_memory.js.map +1 -0
- package/dist/{cjs/memory → memory}/index.d.ts +0 -0
- package/dist/{cjs/memory → memory}/index.js +0 -0
- package/dist/memory/index.js.map +1 -0
- package/dist/{cjs/memory → memory}/tests/buffer_memory.test.d.ts +0 -0
- package/dist/{cjs/memory → memory}/tests/buffer_memory.test.js +0 -0
- package/dist/memory/tests/buffer_memory.test.js.map +1 -0
- package/dist/{cjs/prompt → prompts}/base.d.ts +0 -0
- package/dist/{cjs/prompt → prompts}/base.js +0 -0
- package/dist/prompts/base.js.map +1 -0
- package/dist/{cjs/prompt → prompts}/few_shot.d.ts +0 -0
- package/dist/{cjs/prompt → prompts}/few_shot.js +0 -0
- package/dist/prompts/few_shot.js.map +1 -0
- package/dist/{cjs/prompt → prompts}/index.d.ts +0 -0
- package/dist/{cjs/prompt → prompts}/index.js +0 -0
- package/dist/prompts/index.js.map +1 -0
- package/dist/{cjs/prompt → prompts}/load.d.ts +2 -2
- package/dist/{cjs/prompt → prompts}/load.js +2 -2
- package/dist/prompts/load.js.map +1 -0
- package/dist/{cjs/prompt → prompts}/parser.d.ts +0 -0
- package/dist/{cjs/prompt → prompts}/parser.js +0 -0
- package/dist/prompts/parser.js.map +1 -0
- package/dist/{cjs/prompt → prompts}/prompt.d.ts +1 -1
- package/dist/{cjs/prompt → prompts}/prompt.js +1 -1
- package/dist/prompts/prompt.js.map +1 -0
- package/dist/{cjs/prompt → prompts}/template.d.ts +0 -0
- package/dist/{cjs/prompt → prompts}/template.js +0 -0
- package/dist/prompts/template.js.map +1 -0
- package/dist/{cjs/prompt → prompts}/tests/load.test.d.ts +0 -0
- package/dist/{cjs/prompt → prompts}/tests/load.test.js +0 -0
- package/dist/prompts/tests/load.test.js.map +1 -0
- package/dist/{cjs/prompt → prompts}/tests/template.test.d.ts +0 -0
- package/dist/{cjs/prompt → prompts}/tests/template.test.js +0 -0
- package/dist/prompts/tests/template.test.js.map +1 -0
- package/dist/{cjs/text_splitter.d.ts → text_splitter.d.ts} +0 -0
- package/dist/{cjs/text_splitter.js → text_splitter.js} +3 -1
- package/dist/text_splitter.js.map +1 -0
- package/dist/{cjs/util → util}/hub.d.ts +0 -0
- package/dist/{cjs/util → util}/hub.js +0 -0
- package/dist/util/hub.js.map +1 -0
- package/dist/{cjs/util → util}/index.d.ts +0 -0
- package/dist/{cjs/util → util}/index.js +0 -0
- package/dist/util/index.js.map +1 -0
- package/dist/{esm/vectorstores → vectorstores}/base.d.ts +1 -1
- package/dist/{cjs/vectorstores → vectorstores}/base.js +1 -1
- package/dist/vectorstores/base.js.map +1 -0
- package/dist/{cjs/vectorstores → vectorstores}/hnswlib.d.ts +0 -0
- package/dist/{cjs/vectorstores → vectorstores}/hnswlib.js +6 -3
- package/dist/vectorstores/hnswlib.js.map +1 -0
- package/dist/{cjs/vectorstores → vectorstores}/index.d.ts +0 -0
- package/dist/{cjs/vectorstores → vectorstores}/index.js +0 -0
- package/dist/vectorstores/index.js.map +1 -0
- package/dist/{cjs/vectorstores → vectorstores}/tests/hnswlib.test.d.ts +0 -0
- package/dist/{cjs/vectorstores → vectorstores}/tests/hnswlib.test.js +6 -6
- package/dist/vectorstores/tests/hnswlib.test.js.map +1 -0
- package/document.d.ts +1 -0
- package/document.js +1 -0
- package/embeddings.d.ts +1 -0
- package/embeddings.js +1 -0
- package/llms.d.ts +1 -0
- package/llms.js +1 -0
- package/memory.d.ts +1 -0
- package/memory.js +1 -0
- package/package.json +48 -65
- package/prompts.d.ts +1 -0
- package/prompts.js +1 -0
- package/text_splitter.d.ts +1 -0
- package/text_splitter.js +1 -0
- package/tools.d.ts +1 -0
- package/tools.js +1 -0
- package/vectorstores.d.ts +1 -0
- package/vectorstores.js +1 -0
- package/dist/cjs/agents/agent.js.map +0 -1
- package/dist/cjs/agents/executor.js.map +0 -1
- package/dist/cjs/agents/helpers.js.map +0 -1
- package/dist/cjs/agents/index.js.map +0 -1
- package/dist/cjs/agents/initialize.js.map +0 -1
- package/dist/cjs/agents/load.js.map +0 -1
- package/dist/cjs/agents/mrkl/index.js.map +0 -1
- package/dist/cjs/agents/mrkl/prompt.js.map +0 -1
- package/dist/cjs/agents/tests/agent.test.js.map +0 -1
- package/dist/cjs/agents/tests/calculator.test.js.map +0 -1
- package/dist/cjs/agents/tools/base.js.map +0 -1
- package/dist/cjs/agents/tools/calculator.js.map +0 -1
- package/dist/cjs/agents/tools/index.js.map +0 -1
- package/dist/cjs/agents/tools/serpapi.js.map +0 -1
- package/dist/cjs/cache.js.map +0 -1
- package/dist/cjs/chains/base.js.map +0 -1
- package/dist/cjs/chains/chat_vector_db_chain.js.map +0 -1
- package/dist/cjs/chains/combine_docs_chain.js.map +0 -1
- package/dist/cjs/chains/conversation.d.ts +0 -12
- package/dist/cjs/chains/conversation.js.map +0 -1
- package/dist/cjs/chains/index.d.ts +0 -7
- package/dist/cjs/chains/index.js.map +0 -1
- package/dist/cjs/chains/llm_chain.d.ts +0 -65
- package/dist/cjs/chains/llm_chain.js.map +0 -1
- package/dist/cjs/chains/llm_math_chain.d.ts +0 -0
- package/dist/cjs/chains/llm_math_chain.js +0 -129
- package/dist/cjs/chains/llm_math_chain.js.map +0 -1
- package/dist/cjs/chains/load.js.map +0 -1
- package/dist/cjs/chains/question_answering/load.js.map +0 -1
- package/dist/cjs/chains/question_answering/stuff_prompts.js.map +0 -1
- package/dist/cjs/chains/question_answering/tests/load.test.js.map +0 -1
- package/dist/cjs/chains/tests/chat_vector_db_qa_chain.test.js.map +0 -1
- package/dist/cjs/chains/tests/combine_docs_chain.test.js.map +0 -1
- package/dist/cjs/chains/tests/llm_chain.test.js.map +0 -1
- package/dist/cjs/chains/tests/vector_db_qa_chain.test.js.map +0 -1
- package/dist/cjs/chains/vector_db_qa.js.map +0 -1
- package/dist/cjs/document.js.map +0 -1
- package/dist/cjs/embeddings/base.js.map +0 -1
- package/dist/cjs/embeddings/index.js.map +0 -1
- package/dist/cjs/embeddings/openai.js.map +0 -1
- package/dist/cjs/embeddings/tests/openai.test.js.map +0 -1
- package/dist/cjs/index.js.map +0 -1
- package/dist/cjs/llms/base.js.map +0 -1
- package/dist/cjs/llms/index.js.map +0 -1
- package/dist/cjs/llms/load.js.map +0 -1
- package/dist/cjs/llms/openai.d.ts +0 -125
- package/dist/cjs/llms/openai.js.map +0 -1
- package/dist/cjs/llms/tests/openai.test.js.map +0 -1
- package/dist/cjs/memory/base.js.map +0 -1
- package/dist/cjs/memory/buffer_memory.js.map +0 -1
- package/dist/cjs/memory/index.js.map +0 -1
- package/dist/cjs/memory/tests/buffer_memory.test.js.map +0 -1
- package/dist/cjs/prompt/base.js.map +0 -1
- package/dist/cjs/prompt/few_shot.js.map +0 -1
- package/dist/cjs/prompt/index.js.map +0 -1
- package/dist/cjs/prompt/load.js.map +0 -1
- package/dist/cjs/prompt/parser.js.map +0 -1
- package/dist/cjs/prompt/prompt.js.map +0 -1
- package/dist/cjs/prompt/template.js.map +0 -1
- package/dist/cjs/prompt/tests/load.test.js.map +0 -1
- package/dist/cjs/prompt/tests/template.test.js.map +0 -1
- package/dist/cjs/text_splitter.js.map +0 -1
- package/dist/cjs/text_splitter.test.d.ts +0 -1
- package/dist/cjs/text_splitter.test.js +0 -87
- package/dist/cjs/text_splitter.test.js.map +0 -1
- package/dist/cjs/util/hub.js.map +0 -1
- package/dist/cjs/util/index.js.map +0 -1
- package/dist/cjs/vectorstores/base.d.ts +0 -18
- package/dist/cjs/vectorstores/base.js.map +0 -1
- package/dist/cjs/vectorstores/hnswlib.js.map +0 -1
- package/dist/cjs/vectorstores/index.js.map +0 -1
- package/dist/cjs/vectorstores/tests/hnswlib.test.js.map +0 -1
- package/dist/esm/agents/agent.d.ts +0 -96
- package/dist/esm/agents/agent.js +0 -146
- package/dist/esm/agents/agent.js.map +0 -1
- package/dist/esm/agents/executor.d.ts +0 -32
- package/dist/esm/agents/executor.js +0 -95
- package/dist/esm/agents/executor.js.map +0 -1
- package/dist/esm/agents/helpers.d.ts +0 -3
- package/dist/esm/agents/helpers.js +0 -17
- package/dist/esm/agents/helpers.js.map +0 -1
- package/dist/esm/agents/index.d.ts +0 -7
- package/dist/esm/agents/index.js +0 -7
- package/dist/esm/agents/index.js.map +0 -1
- package/dist/esm/agents/initialize.d.ts +0 -4
- package/dist/esm/agents/initialize.js +0 -17
- package/dist/esm/agents/initialize.js.map +0 -1
- package/dist/esm/agents/load.d.ts +0 -6
- package/dist/esm/agents/load.js +0 -15
- package/dist/esm/agents/load.js.map +0 -1
- package/dist/esm/agents/mrkl/index.d.ts +0 -50
- package/dist/esm/agents/mrkl/index.js +0 -98
- package/dist/esm/agents/mrkl/index.js.map +0 -1
- package/dist/esm/agents/mrkl/prompt.d.ts +0 -3
- package/dist/esm/agents/mrkl/prompt.js +0 -16
- package/dist/esm/agents/mrkl/prompt.js.map +0 -1
- package/dist/esm/agents/tests/agent.test.d.ts +0 -1
- package/dist/esm/agents/tests/agent.test.js +0 -33
- package/dist/esm/agents/tests/agent.test.js.map +0 -1
- package/dist/esm/agents/tests/calculator.test.d.ts +0 -1
- package/dist/esm/agents/tests/calculator.test.js +0 -28
- package/dist/esm/agents/tests/calculator.test.js.map +0 -1
- package/dist/esm/agents/tools/base.d.ts +0 -6
- package/dist/esm/agents/tools/base.js +0 -11
- package/dist/esm/agents/tools/base.js.map +0 -1
- package/dist/esm/agents/tools/calculator.d.ts +0 -6
- package/dist/esm/agents/tools/calculator.js +0 -29
- package/dist/esm/agents/tools/calculator.js.map +0 -1
- package/dist/esm/agents/tools/index.d.ts +0 -3
- package/dist/esm/agents/tools/index.js +0 -4
- package/dist/esm/agents/tools/index.js.map +0 -1
- package/dist/esm/agents/tools/llm-math.d.ts +0 -0
- package/dist/esm/agents/tools/llm-math.js +0 -2
- package/dist/esm/agents/tools/llm-math.js.map +0 -1
- package/dist/esm/agents/tools/serpapi.d.ts +0 -18
- package/dist/esm/agents/tools/serpapi.js +0 -89
- package/dist/esm/agents/tools/serpapi.js.map +0 -1
- package/dist/esm/agents/types.d.ts +0 -25
- package/dist/esm/agents/types.js +0 -2
- package/dist/esm/agents/types.js.map +0 -1
- package/dist/esm/cache.d.ts +0 -11
- package/dist/esm/cache.js +0 -22
- package/dist/esm/cache.js.map +0 -1
- package/dist/esm/chains/base.d.ts +0 -42
- package/dist/esm/chains/base.js +0 -57
- package/dist/esm/chains/base.js.map +0 -1
- package/dist/esm/chains/chat_vector_db_chain.d.ts +0 -41
- package/dist/esm/chains/chat_vector_db_chain.js +0 -128
- package/dist/esm/chains/chat_vector_db_chain.js.map +0 -1
- package/dist/esm/chains/combine_docs_chain.d.ts +0 -35
- package/dist/esm/chains/combine_docs_chain.js +0 -71
- package/dist/esm/chains/combine_docs_chain.js.map +0 -1
- package/dist/esm/chains/conversation.js +0 -25
- package/dist/esm/chains/conversation.js.map +0 -1
- package/dist/esm/chains/index.js +0 -8
- package/dist/esm/chains/index.js.map +0 -1
- package/dist/esm/chains/llm_chain.js +0 -110
- package/dist/esm/chains/llm_chain.js.map +0 -1
- package/dist/esm/chains/llm_math_chain.d.ts +0 -0
- package/dist/esm/chains/llm_math_chain.js +0 -129
- package/dist/esm/chains/llm_math_chain.js.map +0 -1
- package/dist/esm/chains/load.d.ts +0 -21
- package/dist/esm/chains/load.js +0 -33
- package/dist/esm/chains/load.js.map +0 -1
- package/dist/esm/chains/question_answering/load.d.ts +0 -3
- package/dist/esm/chains/question_answering/load.js +0 -9
- package/dist/esm/chains/question_answering/load.js.map +0 -1
- package/dist/esm/chains/question_answering/stuff_prompts.d.ts +0 -2
- package/dist/esm/chains/question_answering/stuff_prompts.js +0 -7
- package/dist/esm/chains/question_answering/stuff_prompts.js.map +0 -1
- package/dist/esm/chains/question_answering/tests/load.test.d.ts +0 -1
- package/dist/esm/chains/question_answering/tests/load.test.js +0 -12
- package/dist/esm/chains/question_answering/tests/load.test.js.map +0 -1
- package/dist/esm/chains/tests/chat_vector_db_qa_chain.test.d.ts +0 -1
- package/dist/esm/chains/tests/chat_vector_db_qa_chain.test.js +0 -33
- package/dist/esm/chains/tests/chat_vector_db_qa_chain.test.js.map +0 -1
- package/dist/esm/chains/tests/combine_docs_chain.test.d.ts +0 -1
- package/dist/esm/chains/tests/combine_docs_chain.test.js +0 -26
- package/dist/esm/chains/tests/combine_docs_chain.test.js.map +0 -1
- package/dist/esm/chains/tests/llm_chain.test.d.ts +0 -1
- package/dist/esm/chains/tests/llm_chain.test.js +0 -27
- package/dist/esm/chains/tests/llm_chain.test.js.map +0 -1
- package/dist/esm/chains/tests/vector_db_qa_chain.test.d.ts +0 -1
- package/dist/esm/chains/tests/vector_db_qa_chain.test.js +0 -44
- package/dist/esm/chains/tests/vector_db_qa_chain.test.js.map +0 -1
- package/dist/esm/chains/vector_db_qa.d.ts +0 -36
- package/dist/esm/chains/vector_db_qa.js +0 -81
- package/dist/esm/chains/vector_db_qa.js.map +0 -1
- package/dist/esm/document.d.ts +0 -13
- package/dist/esm/document.js +0 -24
- package/dist/esm/document.js.map +0 -1
- package/dist/esm/embeddings/base.d.ts +0 -4
- package/dist/esm/embeddings/base.js +0 -3
- package/dist/esm/embeddings/base.js.map +0 -1
- package/dist/esm/embeddings/index.d.ts +0 -1
- package/dist/esm/embeddings/index.js +0 -2
- package/dist/esm/embeddings/index.js.map +0 -1
- package/dist/esm/embeddings/openai.d.ts +0 -20
- package/dist/esm/embeddings/openai.js +0 -82
- package/dist/esm/embeddings/openai.js.map +0 -1
- package/dist/esm/embeddings/tests/openai.test.js +0 -15
- package/dist/esm/embeddings/tests/openai.test.js.map +0 -1
- package/dist/esm/index.d.ts +0 -3
- package/dist/esm/index.js +0 -4
- package/dist/esm/index.js.map +0 -1
- package/dist/esm/llms/base.d.ts +0 -64
- package/dist/esm/llms/base.js +0 -159
- package/dist/esm/llms/base.js.map +0 -1
- package/dist/esm/llms/index.d.ts +0 -37
- package/dist/esm/llms/index.js +0 -4
- package/dist/esm/llms/index.js.map +0 -1
- package/dist/esm/llms/load.d.ts +0 -11
- package/dist/esm/llms/load.js +0 -13
- package/dist/esm/llms/load.js.map +0 -1
- package/dist/esm/llms/openai.js +0 -236
- package/dist/esm/llms/openai.js.map +0 -1
- package/dist/esm/llms/tests/openai.test.d.ts +0 -1
- package/dist/esm/llms/tests/openai.test.js +0 -8
- package/dist/esm/llms/tests/openai.test.js.map +0 -1
- package/dist/esm/memory/base.d.ts +0 -7
- package/dist/esm/memory/base.js +0 -3
- package/dist/esm/memory/base.js.map +0 -1
- package/dist/esm/memory/buffer_memory.d.ts +0 -15
- package/dist/esm/memory/buffer_memory.js +0 -53
- package/dist/esm/memory/buffer_memory.js.map +0 -1
- package/dist/esm/memory/index.d.ts +0 -2
- package/dist/esm/memory/index.js +0 -3
- package/dist/esm/memory/index.js.map +0 -1
- package/dist/esm/memory/tests/buffer_memory.test.d.ts +0 -1
- package/dist/esm/memory/tests/buffer_memory.test.js +0 -15
- package/dist/esm/memory/tests/buffer_memory.test.js.map +0 -1
- package/dist/esm/prompt/base.d.ts +0 -58
- package/dist/esm/prompt/base.js +0 -49
- package/dist/esm/prompt/base.js.map +0 -1
- package/dist/esm/prompt/few_shot.d.ts +0 -82
- package/dist/esm/prompt/few_shot.js +0 -133
- package/dist/esm/prompt/few_shot.js.map +0 -1
- package/dist/esm/prompt/index.d.ts +0 -4
- package/dist/esm/prompt/index.js +0 -5
- package/dist/esm/prompt/index.js.map +0 -1
- package/dist/esm/prompt/load.d.ts +0 -19
- package/dist/esm/prompt/load.js +0 -29
- package/dist/esm/prompt/load.js.map +0 -1
- package/dist/esm/prompt/parser.d.ts +0 -70
- package/dist/esm/prompt/parser.js +0 -107
- package/dist/esm/prompt/parser.js.map +0 -1
- package/dist/esm/prompt/prompt.d.ts +0 -76
- package/dist/esm/prompt/prompt.js +0 -107
- package/dist/esm/prompt/prompt.js.map +0 -1
- package/dist/esm/prompt/template.d.ts +0 -16
- package/dist/esm/prompt/template.js +0 -80
- package/dist/esm/prompt/template.js.map +0 -1
- package/dist/esm/prompt/tests/load.test.d.ts +0 -1
- package/dist/esm/prompt/tests/load.test.js +0 -16
- package/dist/esm/prompt/tests/load.test.js.map +0 -1
- package/dist/esm/prompt/tests/template.test.d.ts +0 -1
- package/dist/esm/prompt/tests/template.test.js +0 -25
- package/dist/esm/prompt/tests/template.test.js.map +0 -1
- package/dist/esm/text_splitter.d.ts +0 -32
- package/dist/esm/text_splitter.js +0 -160
- package/dist/esm/text_splitter.js.map +0 -1
- package/dist/esm/text_splitter.test.d.ts +0 -1
- package/dist/esm/text_splitter.test.js +0 -85
- package/dist/esm/text_splitter.test.js.map +0 -1
- package/dist/esm/util/hub.d.ts +0 -2
- package/dist/esm/util/hub.js +0 -34
- package/dist/esm/util/hub.js.map +0 -1
- package/dist/esm/util/index.d.ts +0 -13
- package/dist/esm/util/index.js +0 -61
- package/dist/esm/util/index.js.map +0 -1
- package/dist/esm/vectorstores/base.js +0 -33
- package/dist/esm/vectorstores/base.js.map +0 -1
- package/dist/esm/vectorstores/hnswlib.d.ts +0 -19
- package/dist/esm/vectorstores/hnswlib.js +0 -119
- package/dist/esm/vectorstores/hnswlib.js.map +0 -1
- package/dist/esm/vectorstores/index.d.ts +0 -1
- package/dist/esm/vectorstores/index.js +0 -2
- package/dist/esm/vectorstores/index.js.map +0 -1
- package/dist/esm/vectorstores/tests/hnswlib.test.d.ts +0 -1
- package/dist/esm/vectorstores/tests/hnswlib.test.js +0 -39
- 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"}
|
package/dist/cjs/util/hub.js.map
DELETED
|
@@ -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 {};
|
package/dist/esm/agents/agent.js
DELETED
|
@@ -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";
|
package/dist/esm/agents/index.js
DELETED
|
@@ -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,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"}
|