langchain 1.0.0-alpha.5 → 1.0.0-alpha.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/dist/agents/ReactAgent.cjs +5 -5
- package/dist/agents/ReactAgent.cjs.map +1 -1
- package/dist/agents/ReactAgent.d.cts +2 -5
- package/dist/agents/ReactAgent.d.cts.map +1 -1
- package/dist/agents/ReactAgent.d.ts +2 -5
- package/dist/agents/ReactAgent.d.ts.map +1 -1
- package/dist/agents/ReactAgent.js +6 -6
- package/dist/agents/ReactAgent.js.map +1 -1
- package/dist/agents/annotation.cjs.map +1 -1
- package/dist/agents/annotation.d.cts +13 -8
- package/dist/agents/annotation.d.cts.map +1 -1
- package/dist/agents/annotation.d.ts +13 -8
- package/dist/agents/annotation.d.ts.map +1 -1
- package/dist/agents/annotation.js.map +1 -1
- package/dist/agents/createAgent.cjs.map +1 -1
- package/dist/agents/createAgent.js.map +1 -1
- package/dist/agents/index.cjs +2 -2
- package/dist/agents/index.cjs.map +1 -1
- package/dist/agents/index.d.cts +50 -50
- package/dist/agents/index.d.cts.map +1 -1
- package/dist/agents/index.d.ts +50 -50
- package/dist/agents/index.d.ts.map +1 -1
- package/dist/agents/index.js +2 -2
- package/dist/agents/index.js.map +1 -1
- package/dist/agents/interrupt.d.cts +14 -1
- package/dist/agents/interrupt.d.cts.map +1 -1
- package/dist/agents/interrupt.d.ts +14 -1
- package/dist/agents/interrupt.d.ts.map +1 -1
- package/dist/agents/middlewareAgent/ReactAgent.cjs +273 -66
- package/dist/agents/middlewareAgent/ReactAgent.cjs.map +1 -1
- package/dist/agents/middlewareAgent/ReactAgent.d.cts +102 -16
- package/dist/agents/middlewareAgent/ReactAgent.d.cts.map +1 -1
- package/dist/agents/middlewareAgent/ReactAgent.d.ts +102 -16
- package/dist/agents/middlewareAgent/ReactAgent.d.ts.map +1 -1
- package/dist/agents/middlewareAgent/ReactAgent.js +275 -68
- package/dist/agents/middlewareAgent/ReactAgent.js.map +1 -1
- package/dist/agents/middlewareAgent/annotation.cjs +12 -6
- package/dist/agents/middlewareAgent/annotation.cjs.map +1 -1
- package/dist/agents/middlewareAgent/annotation.js +12 -6
- package/dist/agents/middlewareAgent/annotation.js.map +1 -1
- package/dist/agents/middlewareAgent/constants.d.cts +5 -0
- package/dist/agents/middlewareAgent/constants.d.cts.map +1 -0
- package/dist/agents/middlewareAgent/constants.d.ts +5 -0
- package/dist/agents/middlewareAgent/constants.d.ts.map +1 -0
- package/dist/agents/middlewareAgent/index.cjs.map +1 -1
- package/dist/agents/middlewareAgent/index.js.map +1 -1
- package/dist/agents/middlewareAgent/middleware/bigTool.cjs +162 -0
- package/dist/agents/middlewareAgent/middleware/bigTool.cjs.map +1 -0
- package/dist/agents/middlewareAgent/middleware/bigTool.d.cts +113 -0
- package/dist/agents/middlewareAgent/middleware/bigTool.d.cts.map +1 -0
- package/dist/agents/middlewareAgent/middleware/bigTool.d.ts +113 -0
- package/dist/agents/middlewareAgent/middleware/bigTool.d.ts.map +1 -0
- package/dist/agents/middlewareAgent/middleware/bigTool.js +161 -0
- package/dist/agents/middlewareAgent/middleware/bigTool.js.map +1 -0
- package/dist/agents/middlewareAgent/middleware/dynamicSystemPrompt.cjs +58 -0
- package/dist/agents/middlewareAgent/middleware/dynamicSystemPrompt.cjs.map +1 -0
- package/dist/agents/middlewareAgent/middleware/dynamicSystemPrompt.d.cts +46 -0
- package/dist/agents/middlewareAgent/middleware/dynamicSystemPrompt.d.cts.map +1 -0
- package/dist/agents/middlewareAgent/middleware/dynamicSystemPrompt.d.ts +46 -0
- package/dist/agents/middlewareAgent/middleware/dynamicSystemPrompt.d.ts.map +1 -0
- package/dist/agents/middlewareAgent/middleware/dynamicSystemPrompt.js +58 -0
- package/dist/agents/middlewareAgent/middleware/dynamicSystemPrompt.js.map +1 -0
- package/dist/agents/middlewareAgent/middleware/hitl.cjs +311 -0
- package/dist/agents/middlewareAgent/middleware/hitl.cjs.map +1 -0
- package/dist/agents/middlewareAgent/middleware/hitl.d.cts +419 -0
- package/dist/agents/middlewareAgent/middleware/hitl.d.cts.map +1 -0
- package/dist/agents/middlewareAgent/middleware/hitl.d.ts +419 -0
- package/dist/agents/middlewareAgent/middleware/hitl.d.ts.map +1 -0
- package/dist/agents/middlewareAgent/middleware/hitl.js +310 -0
- package/dist/agents/middlewareAgent/middleware/hitl.js.map +1 -0
- package/dist/agents/middlewareAgent/middleware/index.cjs +32 -0
- package/dist/agents/middlewareAgent/middleware/index.cjs.map +1 -0
- package/dist/agents/middlewareAgent/middleware/index.d.cts +6 -0
- package/dist/agents/middlewareAgent/middleware/index.d.ts +6 -0
- package/dist/agents/middlewareAgent/middleware/index.js +21 -0
- package/dist/agents/middlewareAgent/middleware/index.js.map +1 -0
- package/dist/agents/middlewareAgent/{middlewares → middleware}/promptCaching.cjs +72 -27
- package/dist/agents/middlewareAgent/middleware/promptCaching.cjs.map +1 -0
- package/dist/agents/middlewareAgent/{middlewares → middleware}/promptCaching.d.cts +60 -15
- package/dist/agents/middlewareAgent/middleware/promptCaching.d.cts.map +1 -0
- package/dist/agents/middlewareAgent/{middlewares → middleware}/promptCaching.d.ts +60 -15
- package/dist/agents/middlewareAgent/middleware/promptCaching.d.ts.map +1 -0
- package/dist/agents/middlewareAgent/{middlewares → middleware}/promptCaching.js +71 -26
- package/dist/agents/middlewareAgent/middleware/promptCaching.js.map +1 -0
- package/dist/agents/middlewareAgent/{middlewares → middleware}/summarization.cjs +35 -23
- package/dist/agents/middlewareAgent/middleware/summarization.cjs.map +1 -0
- package/dist/agents/middlewareAgent/{middlewares → middleware}/summarization.d.cts +7 -5
- package/dist/agents/middlewareAgent/middleware/summarization.d.cts.map +1 -0
- package/dist/agents/middlewareAgent/{middlewares → middleware}/summarization.d.ts +7 -5
- package/dist/agents/middlewareAgent/middleware/summarization.d.ts.map +1 -0
- package/dist/agents/middlewareAgent/{middlewares → middleware}/summarization.js +29 -17
- package/dist/agents/middlewareAgent/middleware/summarization.js.map +1 -0
- package/dist/agents/middlewareAgent/middleware.cjs +8 -5
- package/dist/agents/middlewareAgent/middleware.cjs.map +1 -1
- package/dist/agents/middlewareAgent/middleware.d.cts +67 -7
- package/dist/agents/middlewareAgent/middleware.d.cts.map +1 -1
- package/dist/agents/middlewareAgent/middleware.d.ts +67 -7
- package/dist/agents/middlewareAgent/middleware.d.ts.map +1 -1
- package/dist/agents/middlewareAgent/middleware.js +8 -5
- package/dist/agents/middlewareAgent/middleware.js.map +1 -1
- package/dist/agents/middlewareAgent/nodes/AfterModalNode.cjs +2 -2
- package/dist/agents/middlewareAgent/nodes/AfterModalNode.cjs.map +1 -1
- package/dist/agents/middlewareAgent/nodes/AfterModalNode.js +2 -2
- package/dist/agents/middlewareAgent/nodes/AfterModalNode.js.map +1 -1
- package/dist/agents/middlewareAgent/nodes/AgentNode.cjs +125 -83
- package/dist/agents/middlewareAgent/nodes/AgentNode.cjs.map +1 -1
- package/dist/agents/middlewareAgent/nodes/AgentNode.js +128 -86
- package/dist/agents/middlewareAgent/nodes/AgentNode.js.map +1 -1
- package/dist/agents/middlewareAgent/nodes/BeforeModalNode.cjs +2 -2
- package/dist/agents/middlewareAgent/nodes/BeforeModalNode.cjs.map +1 -1
- package/dist/agents/middlewareAgent/nodes/BeforeModalNode.js +2 -2
- package/dist/agents/middlewareAgent/nodes/BeforeModalNode.js.map +1 -1
- package/dist/agents/middlewareAgent/nodes/middleware.cjs +42 -17
- package/dist/agents/middlewareAgent/nodes/middleware.cjs.map +1 -1
- package/dist/agents/middlewareAgent/nodes/middleware.js +42 -18
- package/dist/agents/middlewareAgent/nodes/middleware.js.map +1 -1
- package/dist/agents/middlewareAgent/nodes/utils.cjs +30 -16
- package/dist/agents/middlewareAgent/nodes/utils.cjs.map +1 -1
- package/dist/agents/middlewareAgent/nodes/utils.js +28 -14
- package/dist/agents/middlewareAgent/nodes/utils.js.map +1 -1
- package/dist/agents/middlewareAgent/types.d.cts +162 -88
- package/dist/agents/middlewareAgent/types.d.cts.map +1 -1
- package/dist/agents/middlewareAgent/types.d.ts +162 -88
- package/dist/agents/middlewareAgent/types.d.ts.map +1 -1
- package/dist/agents/model.cjs +13 -0
- package/dist/agents/model.cjs.map +1 -0
- package/dist/agents/model.js +11 -0
- package/dist/agents/model.js.map +1 -0
- package/dist/agents/nodes/AgentNode.cjs +60 -32
- package/dist/agents/nodes/AgentNode.cjs.map +1 -1
- package/dist/agents/nodes/AgentNode.js +61 -33
- package/dist/agents/nodes/AgentNode.js.map +1 -1
- package/dist/agents/nodes/ToolNode.cjs +4 -4
- package/dist/agents/nodes/ToolNode.cjs.map +1 -1
- package/dist/agents/nodes/ToolNode.d.cts +2 -3
- package/dist/agents/nodes/ToolNode.d.cts.map +1 -1
- package/dist/agents/nodes/ToolNode.d.ts +1 -2
- package/dist/agents/nodes/ToolNode.d.ts.map +1 -1
- package/dist/agents/nodes/ToolNode.js +5 -5
- package/dist/agents/nodes/ToolNode.js.map +1 -1
- package/dist/agents/responses.cjs +53 -11
- package/dist/agents/responses.cjs.map +1 -1
- package/dist/agents/responses.d.cts +12 -20
- package/dist/agents/responses.d.cts.map +1 -1
- package/dist/agents/responses.d.ts +12 -20
- package/dist/agents/responses.d.ts.map +1 -1
- package/dist/agents/responses.js +53 -12
- package/dist/agents/responses.js.map +1 -1
- package/dist/agents/types.d.cts +3 -6
- package/dist/agents/types.d.cts.map +1 -1
- package/dist/agents/types.d.ts +3 -6
- package/dist/agents/types.d.ts.map +1 -1
- package/dist/agents/utils.cjs +11 -29
- package/dist/agents/utils.cjs.map +1 -1
- package/dist/agents/utils.js +9 -26
- package/dist/agents/utils.js.map +1 -1
- package/dist/agents/withAgentName.cjs.map +1 -1
- package/dist/agents/withAgentName.js.map +1 -1
- package/dist/chains/api/prompts.cjs.map +1 -1
- package/dist/chains/api/prompts.js.map +1 -1
- package/dist/chains/constitutional_ai/constitutional_chain.cjs.map +1 -1
- package/dist/chains/constitutional_ai/constitutional_chain.js.map +1 -1
- package/dist/chains/index.cjs +0 -3
- package/dist/chains/index.cjs.map +1 -1
- package/dist/chains/index.d.cts +1 -2
- package/dist/chains/index.d.ts +1 -2
- package/dist/chains/index.js +1 -3
- package/dist/chains/index.js.map +1 -1
- package/dist/chains/openai_functions/extraction.cjs.map +1 -1
- package/dist/chains/openai_functions/extraction.d.cts +1 -3
- package/dist/chains/openai_functions/extraction.d.cts.map +1 -1
- package/dist/chains/openai_functions/extraction.d.ts +1 -3
- package/dist/chains/openai_functions/extraction.d.ts.map +1 -1
- package/dist/chains/openai_functions/extraction.js.map +1 -1
- package/dist/chains/openai_functions/index.cjs +0 -5
- package/dist/chains/openai_functions/index.cjs.map +1 -1
- package/dist/chains/openai_functions/index.d.cts +1 -2
- package/dist/chains/openai_functions/index.d.ts +1 -2
- package/dist/chains/openai_functions/index.js +1 -4
- package/dist/chains/openai_functions/index.js.map +1 -1
- package/dist/chains/openai_functions/openapi.cjs +4 -4
- package/dist/chains/openai_functions/openapi.cjs.map +1 -1
- package/dist/chains/openai_functions/openapi.d.cts +1 -1
- package/dist/chains/openai_functions/openapi.js +4 -4
- package/dist/chains/openai_functions/openapi.js.map +1 -1
- package/dist/chains/openai_functions/tagging.cjs.map +1 -1
- package/dist/chains/openai_functions/tagging.d.cts +1 -3
- package/dist/chains/openai_functions/tagging.d.cts.map +1 -1
- package/dist/chains/openai_functions/tagging.d.ts +1 -3
- package/dist/chains/openai_functions/tagging.d.ts.map +1 -1
- package/dist/chains/openai_functions/tagging.js.map +1 -1
- package/dist/chains/query_constructor/index.cjs +4 -4
- package/dist/chains/query_constructor/index.cjs.map +1 -1
- package/dist/chains/query_constructor/index.d.cts +4 -2
- package/dist/chains/query_constructor/index.d.cts.map +1 -1
- package/dist/chains/query_constructor/index.d.ts +4 -2
- package/dist/chains/query_constructor/index.d.ts.map +1 -1
- package/dist/chains/query_constructor/index.js +1 -1
- package/dist/chains/query_constructor/index.js.map +1 -1
- package/dist/chains/question_answering/load.d.ts +2 -2
- package/dist/chains/question_answering/load.d.ts.map +1 -1
- package/dist/chains/question_answering/map_reduce_prompts.cjs.map +1 -1
- package/dist/chains/question_answering/map_reduce_prompts.js.map +1 -1
- package/dist/chains/question_answering/refine_prompts.cjs.map +1 -1
- package/dist/chains/question_answering/refine_prompts.js.map +1 -1
- package/dist/chains/question_answering/stuff_prompts.cjs.map +1 -1
- package/dist/chains/question_answering/stuff_prompts.js.map +1 -1
- package/dist/chains/router/multi_prompt.cjs +4 -4
- package/dist/chains/router/multi_prompt.cjs.map +1 -1
- package/dist/chains/router/multi_prompt.js +1 -1
- package/dist/chains/router/multi_prompt.js.map +1 -1
- package/dist/chains/router/multi_retrieval_qa.cjs +4 -4
- package/dist/chains/router/multi_retrieval_qa.cjs.map +1 -1
- package/dist/chains/router/multi_retrieval_qa.js +1 -1
- package/dist/chains/router/multi_retrieval_qa.js.map +1 -1
- package/dist/chains/sql_db/sql_db_prompt.cjs.map +1 -1
- package/dist/chains/sql_db/sql_db_prompt.d.cts.map +1 -1
- package/dist/chains/sql_db/sql_db_prompt.d.ts.map +1 -1
- package/dist/chains/sql_db/sql_db_prompt.js.map +1 -1
- package/dist/chains/summarization/stuff_prompts.cjs.map +1 -1
- package/dist/chains/summarization/stuff_prompts.js.map +1 -1
- package/dist/chat_models/universal.cjs +8 -5
- package/dist/chat_models/universal.cjs.map +1 -1
- package/dist/chat_models/universal.d.cts +2 -2
- package/dist/chat_models/universal.d.cts.map +1 -1
- package/dist/chat_models/universal.d.ts +2 -2
- package/dist/chat_models/universal.d.ts.map +1 -1
- package/dist/chat_models/universal.js +8 -5
- package/dist/chat_models/universal.js.map +1 -1
- package/dist/document_loaders/fs/directory.cjs.map +1 -1
- package/dist/document_loaders/fs/directory.d.cts +0 -1
- package/dist/document_loaders/fs/directory.d.cts.map +1 -1
- package/dist/document_loaders/fs/directory.d.ts +0 -1
- package/dist/document_loaders/fs/directory.d.ts.map +1 -1
- package/dist/document_loaders/fs/directory.js.map +1 -1
- package/dist/document_loaders/fs/json.cjs +7 -1
- package/dist/document_loaders/fs/json.cjs.map +1 -1
- package/dist/document_loaders/fs/json.js +7 -1
- package/dist/document_loaders/fs/json.js.map +1 -1
- package/dist/embeddings/cache_backed.cjs +1 -1
- package/dist/embeddings/cache_backed.cjs.map +1 -1
- package/dist/embeddings/cache_backed.d.cts +1 -1
- package/dist/embeddings/cache_backed.d.ts +2 -2
- package/dist/embeddings/cache_backed.js +2 -2
- package/dist/embeddings/cache_backed.js.map +1 -1
- package/dist/evaluation/agents/trajectory.d.cts.map +1 -1
- package/dist/evaluation/comparison/pairwise.d.cts.map +1 -1
- package/dist/evaluation/criteria/criteria.d.cts.map +1 -1
- package/dist/evaluation/embedding_distance/base.cjs +2 -4
- package/dist/evaluation/embedding_distance/base.cjs.map +1 -1
- package/dist/evaluation/embedding_distance/base.js +2 -3
- package/dist/evaluation/embedding_distance/base.js.map +1 -1
- package/dist/evaluation/loader.cjs +7 -12
- package/dist/evaluation/loader.cjs.map +1 -1
- package/dist/evaluation/loader.d.cts +8 -2
- package/dist/evaluation/loader.d.cts.map +1 -1
- package/dist/evaluation/loader.d.ts +8 -2
- package/dist/evaluation/loader.d.ts.map +1 -1
- package/dist/evaluation/loader.js +7 -12
- package/dist/evaluation/loader.js.map +1 -1
- package/dist/hub/base.cjs.map +1 -1
- package/dist/hub/base.js.map +1 -1
- package/dist/index.cjs +38 -6
- package/dist/index.cjs.map +1 -1
- package/dist/index.d.cts +11 -4
- package/dist/index.d.ts +11 -4
- package/dist/index.js +18 -4
- package/dist/index.js.map +1 -1
- package/dist/langchain-core/dist/load/serializable.d.cts.map +1 -1
- package/dist/langchain-core/dist/messages/base.d.cts +24 -33
- package/dist/langchain-core/dist/messages/base.d.cts.map +1 -1
- package/dist/langchain-core/dist/messages/content/index.d.cts +21 -2
- package/dist/langchain-core/dist/messages/content/index.d.cts.map +1 -1
- package/dist/langchain-core/dist/messages/content/tools.d.cts +67 -6
- package/dist/langchain-core/dist/messages/content/tools.d.cts.map +1 -1
- package/dist/langchain-core/dist/messages/message.d.cts +598 -0
- package/dist/langchain-core/dist/messages/message.d.cts.map +1 -0
- package/dist/langchain-core/dist/messages/metadata.d.cts +97 -0
- package/dist/langchain-core/dist/messages/metadata.d.cts.map +1 -0
- package/dist/langchain-core/dist/messages/utils.d.cts +75 -0
- package/dist/langchain-core/dist/messages/utils.d.cts.map +1 -0
- package/dist/langchain-core/dist/prompt_values.d.cts.map +1 -1
- package/dist/langchain-core/dist/utils/types/index.d.cts.map +1 -1
- package/dist/libs/langchain-core/dist/load/serializable.d.ts.map +1 -1
- package/dist/libs/langchain-core/dist/messages/base.d.ts +24 -33
- package/dist/libs/langchain-core/dist/messages/base.d.ts.map +1 -1
- package/dist/libs/langchain-core/dist/messages/content/index.d.ts +21 -2
- package/dist/libs/langchain-core/dist/messages/content/index.d.ts.map +1 -1
- package/dist/libs/langchain-core/dist/messages/content/tools.d.ts +67 -6
- package/dist/libs/langchain-core/dist/messages/content/tools.d.ts.map +1 -1
- package/dist/libs/langchain-core/dist/messages/message.d.ts +598 -0
- package/dist/libs/langchain-core/dist/messages/message.d.ts.map +1 -0
- package/dist/libs/langchain-core/dist/messages/metadata.d.ts +97 -0
- package/dist/libs/langchain-core/dist/messages/metadata.d.ts.map +1 -0
- package/dist/libs/langchain-core/dist/messages/utils.d.ts +75 -0
- package/dist/libs/langchain-core/dist/messages/utils.d.ts.map +1 -0
- package/dist/libs/langchain-core/dist/prompt_values.d.ts.map +1 -1
- package/dist/libs/langchain-core/dist/utils/types/index.d.ts +2 -0
- package/dist/libs/langchain-core/dist/utils/types/index.d.ts.map +1 -1
- package/dist/libs/langchain-core/dist/utils/types/zod.d.ts +1 -0
- package/dist/load/import_map.cjs +3 -14
- package/dist/load/import_map.cjs.map +1 -1
- package/dist/load/import_map.js +3 -14
- package/dist/load/import_map.js.map +1 -1
- package/dist/memory/prompt.cjs.map +1 -1
- package/dist/memory/prompt.d.cts.map +1 -1
- package/dist/memory/prompt.d.ts.map +1 -1
- package/dist/memory/prompt.js.map +1 -1
- package/dist/output_parsers/combining.cjs +1 -1
- package/dist/output_parsers/combining.cjs.map +1 -1
- package/dist/output_parsers/combining.js +1 -1
- package/dist/output_parsers/combining.js.map +1 -1
- package/dist/output_parsers/expression_type_handlers/array_literal_expression_handler.cjs.map +1 -1
- package/dist/output_parsers/expression_type_handlers/array_literal_expression_handler.js.map +1 -1
- package/dist/output_parsers/expression_type_handlers/base.cjs +1 -1
- package/dist/output_parsers/expression_type_handlers/base.cjs.map +1 -1
- package/dist/output_parsers/expression_type_handlers/base.js +1 -1
- package/dist/output_parsers/expression_type_handlers/base.js.map +1 -1
- package/dist/output_parsers/regex.cjs.map +1 -1
- package/dist/output_parsers/regex.js.map +1 -1
- package/dist/output_parsers/structured.cjs +4 -4
- package/dist/output_parsers/structured.cjs.map +1 -1
- package/dist/output_parsers/structured.d.cts +1 -1
- package/dist/output_parsers/structured.d.cts.map +1 -1
- package/dist/output_parsers/structured.d.ts +1 -1
- package/dist/output_parsers/structured.d.ts.map +1 -1
- package/dist/output_parsers/structured.js +2 -2
- package/dist/output_parsers/structured.js.map +1 -1
- package/dist/retrievers/ensemble.cjs.map +1 -1
- package/dist/retrievers/ensemble.js.map +1 -1
- package/dist/storage/file_system.cjs +1 -1
- package/dist/storage/file_system.cjs.map +1 -1
- package/dist/storage/file_system.js +1 -1
- package/dist/storage/file_system.js.map +1 -1
- package/dist/tools/fs.cjs +5 -5
- package/dist/tools/fs.cjs.map +1 -1
- package/dist/tools/fs.d.cts +1 -1
- package/dist/tools/fs.d.cts.map +1 -1
- package/dist/tools/fs.d.ts +1 -1
- package/dist/tools/fs.d.ts.map +1 -1
- package/dist/tools/fs.js +1 -1
- package/dist/tools/fs.js.map +1 -1
- package/dist/tools/retriever.cjs +2 -2
- package/dist/tools/retriever.cjs.map +1 -1
- package/dist/tools/retriever.d.cts +1 -1
- package/dist/tools/retriever.d.cts.map +1 -1
- package/dist/tools/retriever.d.ts +1 -1
- package/dist/tools/retriever.d.ts.map +1 -1
- package/dist/tools/retriever.js +1 -1
- package/dist/tools/retriever.js.map +1 -1
- package/dist/tools/sql.cjs +1 -2
- package/dist/tools/sql.cjs.map +1 -1
- package/dist/tools/sql.d.cts +1 -1
- package/dist/tools/sql.d.cts.map +1 -1
- package/dist/tools/sql.d.ts +1 -1
- package/dist/tools/sql.d.ts.map +1 -1
- package/dist/tools/sql.js +1 -2
- package/dist/tools/sql.js.map +1 -1
- package/dist/types/expression-parser.d.cts +2 -0
- package/dist/types/expression-parser.d.cts.map +1 -1
- package/dist/types/expression-parser.d.ts +2 -0
- package/dist/types/expression-parser.d.ts.map +1 -1
- package/dist/util/hub.cjs +1 -1
- package/dist/util/hub.js +1 -1
- package/dist/util/openapi.cjs +1 -1
- package/dist/util/openapi.cjs.map +1 -1
- package/dist/util/openapi.js +1 -1
- package/dist/util/openapi.js.map +1 -1
- package/package.json +15 -21
- package/dist/agents/middlewareAgent/middlewares/hitl.cjs +0 -235
- package/dist/agents/middlewareAgent/middlewares/hitl.cjs.map +0 -1
- package/dist/agents/middlewareAgent/middlewares/hitl.d.cts +0 -199
- package/dist/agents/middlewareAgent/middlewares/hitl.d.cts.map +0 -1
- package/dist/agents/middlewareAgent/middlewares/hitl.d.ts +0 -199
- package/dist/agents/middlewareAgent/middlewares/hitl.d.ts.map +0 -1
- package/dist/agents/middlewareAgent/middlewares/hitl.js +0 -234
- package/dist/agents/middlewareAgent/middlewares/hitl.js.map +0 -1
- package/dist/agents/middlewareAgent/middlewares/index.cjs +0 -8
- package/dist/agents/middlewareAgent/middlewares/index.d.cts +0 -4
- package/dist/agents/middlewareAgent/middlewares/index.d.ts +0 -4
- package/dist/agents/middlewareAgent/middlewares/index.js +0 -5
- package/dist/agents/middlewareAgent/middlewares/promptCaching.cjs.map +0 -1
- package/dist/agents/middlewareAgent/middlewares/promptCaching.d.cts.map +0 -1
- package/dist/agents/middlewareAgent/middlewares/promptCaching.d.ts.map +0 -1
- package/dist/agents/middlewareAgent/middlewares/promptCaching.js.map +0 -1
- package/dist/agents/middlewareAgent/middlewares/summarization.cjs.map +0 -1
- package/dist/agents/middlewareAgent/middlewares/summarization.d.cts.map +0 -1
- package/dist/agents/middlewareAgent/middlewares/summarization.d.ts.map +0 -1
- package/dist/agents/middlewareAgent/middlewares/summarization.js.map +0 -1
- package/dist/chains/openai_functions/structured_output.cjs +0 -107
- package/dist/chains/openai_functions/structured_output.cjs.map +0 -1
- package/dist/chains/openai_functions/structured_output.d.cts +0 -38
- package/dist/chains/openai_functions/structured_output.d.cts.map +0 -1
- package/dist/chains/openai_functions/structured_output.d.ts +0 -38
- package/dist/chains/openai_functions/structured_output.d.ts.map +0 -1
- package/dist/chains/openai_functions/structured_output.js +0 -105
- package/dist/chains/openai_functions/structured_output.js.map +0 -1
- package/dist/chains/openai_moderation.cjs +0 -107
- package/dist/chains/openai_moderation.cjs.map +0 -1
- package/dist/chains/openai_moderation.d.cts +0 -74
- package/dist/chains/openai_moderation.d.cts.map +0 -1
- package/dist/chains/openai_moderation.d.ts +0 -74
- package/dist/chains/openai_moderation.d.ts.map +0 -1
- package/dist/chains/openai_moderation.js +0 -106
- package/dist/chains/openai_moderation.js.map +0 -1
|
@@ -3,7 +3,7 @@ import { DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMP
|
|
|
3
3
|
import { AsymmetricStructuredOutputParser } from "../../output_parsers/structured.js";
|
|
4
4
|
import { Runnable, RunnableConfig } from "@langchain/core/runnables";
|
|
5
5
|
import { InferInteropZodOutput, InputValues } from "@langchain/core/utils/types";
|
|
6
|
-
import { z } from "zod";
|
|
6
|
+
import { z } from "zod/v3";
|
|
7
7
|
import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
|
|
8
8
|
import { Comparator, Operator, StructuredQuery } from "@langchain/core/structured_query";
|
|
9
9
|
|
|
@@ -74,7 +74,9 @@ type QueryConstructorRunnableOptions = {
|
|
|
74
74
|
};
|
|
75
75
|
/** @deprecated */
|
|
76
76
|
type QueryConstructorChainOptions = QueryConstructorRunnableOptions;
|
|
77
|
-
declare function loadQueryConstructorRunnable(opts: QueryConstructorRunnableOptions
|
|
77
|
+
declare function loadQueryConstructorRunnable(opts: QueryConstructorRunnableOptions
|
|
78
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
79
|
+
): Runnable<any, StructuredQuery, RunnableConfig<Record<string, any>>>;
|
|
78
80
|
//#endregion
|
|
79
81
|
export { AttributeInfo, DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMPLE_PROMPT, QueryConstructorChainOptions, QueryConstructorRunnableOptions, QueryTransformer, StructuredQueryOutputParser, type TraverseType, formatAttributeInfo, loadQueryConstructorRunnable };
|
|
80
82
|
//# sourceMappingURL=index.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","names":["z","BaseLanguageModelInterface","InferInteropZodOutput","InputValues","Comparator","Operator","StructuredQuery","Runnable","RunnableConfig","QueryTransformer","TraverseType","DEFAULT_EXAMPLES","DEFAULT_PREFIX","DEFAULT_SCHEMA","DEFAULT_SUFFIX","EXAMPLE_PROMPT","AsymmetricStructuredOutputParser","AttributeInfo","queryInputSchema","ZodString","ZodOptional","ZodTypeAny","ZodObject","StructuredQueryOutputParser","query","filter","Promise","formatAttributeInfo","QueryConstructorRunnableOptions","QueryConstructorChainOptions","loadQueryConstructorRunnable","Record"],"sources":["../../../src/chains/query_constructor/index.d.ts"],"sourcesContent":["import { z } from \"zod\";\nimport type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { InferInteropZodOutput, InputValues } from \"@langchain/core/utils/types\";\nimport { Comparator, Operator, StructuredQuery } from \"@langchain/core/structured_query\";\nimport { Runnable, RunnableConfig } from \"@langchain/core/runnables\";\nimport { QueryTransformer, TraverseType } from \"./parser.js\";\nimport { DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMPLE_PROMPT } from \"./prompt.js\";\nimport { AsymmetricStructuredOutputParser } from \"../../output_parsers/structured.js\";\n/**\n * A simple data structure that holds information about an attribute. It\n * is typically used to provide metadata about attributes in other classes\n * or data structures within the LangChain framework.\n */\nexport declare class AttributeInfo {\n name: string;\n type: string;\n description: string;\n constructor(name: string, type: string, description: string);\n}\nexport { QueryTransformer, type TraverseType };\nexport { DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMPLE_PROMPT, };\ndeclare const queryInputSchema: z.ZodObject<{\n query: z.ZodString;\n filter: z.ZodOptional<z.ZodString>;\n}, \"strip\", z.ZodTypeAny, {\n query: string;\n filter?: string | undefined;\n}, {\n query: string;\n filter?: string | undefined;\n}>;\n/**\n * A class that extends AsymmetricStructuredOutputParser to parse\n * structured query output.\n */\nexport declare class StructuredQueryOutputParser extends AsymmetricStructuredOutputParser<typeof queryInputSchema, StructuredQuery> {\n lc_namespace: string[];\n queryTransformer: QueryTransformer;\n constructor(fields: {\n allowedComparators: Comparator[];\n allowedOperators: Operator[];\n });\n /**\n * Processes the output of a structured query.\n * @param query The query string.\n * @param filter The filter condition.\n * @returns A Promise that resolves to a StructuredQuery instance.\n */\n outputProcessor({ query, filter }: InferInteropZodOutput<typeof queryInputSchema>): Promise<StructuredQuery>;\n /**\n * Creates a new StructuredQueryOutputParser instance from the provided\n * components.\n * @param allowedComparators An array of allowed Comparator instances.\n * @param allowedOperators An array of allowed Operator instances.\n * @returns A new StructuredQueryOutputParser instance.\n */\n static fromComponents(allowedComparators?: Comparator[], allowedOperators?: Operator[]): StructuredQueryOutputParser;\n}\nexport declare function formatAttributeInfo(info: AttributeInfo[]): string;\n/**\n * A type that represents options for the query constructor chain.\n */\nexport type QueryConstructorRunnableOptions = {\n llm: BaseLanguageModelInterface;\n documentContents: string;\n attributeInfo: AttributeInfo[];\n examples?: InputValues[];\n allowedComparators?: Comparator[];\n allowedOperators?: Operator[];\n};\n/** @deprecated */\nexport type QueryConstructorChainOptions = QueryConstructorRunnableOptions;\nexport declare function loadQueryConstructorRunnable(opts: QueryConstructorRunnableOptions): Runnable<any, StructuredQuery, RunnableConfig<Record<string, any>>>;\n"],"mappings":";;;;;;;;;;;;;AAaA;AAO6F;;AAElFA,cATUiB,aAAAA,CASRE;EAAS,IACInB,EAAEmB,MAAAA;EAAS,IAAzBnB,EAAEoB,MAAAA;EAAW,WACXC,EAAAA,MAAAA;EAAU,WAHUC,CAAAA,IAAAA,EAAAA,MAAAA,EAAAA,IAAAA,EAAAA,MAAAA,EAAAA,WAAAA,EAAAA,MAAAA;AAAS;cAA7BJ,gBAcmFA,EAdjElB,CAAAA,CAAEsB,SAc+DJ,CAAAA;EAAgB,KAAEZ,EAbxGN,CAAAA,CAAEmB,SAasGb;EAAe,MAE5GG,EAdVT,CAAAA,CAAEoB,WAcQX,CAdIT,CAAAA,CAAEmB,SAcNV,CAAAA;CAAgB,EAAA,OAEVL,EAfhBJ,CAAAA,CAAEqB,UAecjB,EAAAA;EAAU,KACZC,EAAAA,MAAAA;EAAQ,MAQZmB,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;CAAK,EAAA;EAAQ,KAAiCN,EAAAA,MAAAA;EAAgB,MAA7ChB,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;CAAqB,CAAA;;;;;AAbHc,cAApCO,2BAAAA,SAAoCP,gCAAAA,CAAAA,OAAwCE,gBAAxCF,EAA0DV,eAA1DU,CAAAA,CAAAA;EAAgC,YAAA,EAAA,MAAA,EAAA;EAuBjEW,gBAAAA,EArBFlB,gBAqB4BQ;EAItCW,WAAAA,CAAAA,MAAAA,EAAAA;IAA+B,kBAAA,EAvBfxB,UAuBe,EAAA;IAClCH,gBAAAA,EAvBiBI,QAuBjBJ,EAAAA;EAA0B,CAAA;EAEH;;;AAGD;AAG/B;AACA;EAAoD,eAAA,CAAA;IAAA,KAAA;IAAA;EAAA,CAAA,EAxBbC,qBAwBa,CAAA,OAxBgBgB,gBAwBhB,CAAA,CAAA,EAxBoCQ,OAwBpC,CAxB4CpB,eAwB5C,CAAA;EAAA;;;;;
|
|
1
|
+
{"version":3,"file":"index.d.ts","names":["z","BaseLanguageModelInterface","InferInteropZodOutput","InputValues","Comparator","Operator","StructuredQuery","Runnable","RunnableConfig","QueryTransformer","TraverseType","DEFAULT_EXAMPLES","DEFAULT_PREFIX","DEFAULT_SCHEMA","DEFAULT_SUFFIX","EXAMPLE_PROMPT","AsymmetricStructuredOutputParser","AttributeInfo","queryInputSchema","ZodString","ZodOptional","ZodTypeAny","ZodObject","StructuredQueryOutputParser","query","filter","Promise","formatAttributeInfo","QueryConstructorRunnableOptions","QueryConstructorChainOptions","loadQueryConstructorRunnable","Record"],"sources":["../../../src/chains/query_constructor/index.d.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { InferInteropZodOutput, InputValues } from \"@langchain/core/utils/types\";\nimport { Comparator, Operator, StructuredQuery } from \"@langchain/core/structured_query\";\nimport { Runnable, RunnableConfig } from \"@langchain/core/runnables\";\nimport { QueryTransformer, TraverseType } from \"./parser.js\";\nimport { DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMPLE_PROMPT } from \"./prompt.js\";\nimport { AsymmetricStructuredOutputParser } from \"../../output_parsers/structured.js\";\n/**\n * A simple data structure that holds information about an attribute. It\n * is typically used to provide metadata about attributes in other classes\n * or data structures within the LangChain framework.\n */\nexport declare class AttributeInfo {\n name: string;\n type: string;\n description: string;\n constructor(name: string, type: string, description: string);\n}\nexport { QueryTransformer, type TraverseType };\nexport { DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMPLE_PROMPT, };\ndeclare const queryInputSchema: z.ZodObject<{\n query: z.ZodString;\n filter: z.ZodOptional<z.ZodString>;\n}, \"strip\", z.ZodTypeAny, {\n query: string;\n filter?: string | undefined;\n}, {\n query: string;\n filter?: string | undefined;\n}>;\n/**\n * A class that extends AsymmetricStructuredOutputParser to parse\n * structured query output.\n */\nexport declare class StructuredQueryOutputParser extends AsymmetricStructuredOutputParser<typeof queryInputSchema, StructuredQuery> {\n lc_namespace: string[];\n queryTransformer: QueryTransformer;\n constructor(fields: {\n allowedComparators: Comparator[];\n allowedOperators: Operator[];\n });\n /**\n * Processes the output of a structured query.\n * @param query The query string.\n * @param filter The filter condition.\n * @returns A Promise that resolves to a StructuredQuery instance.\n */\n outputProcessor({ query, filter }: InferInteropZodOutput<typeof queryInputSchema>): Promise<StructuredQuery>;\n /**\n * Creates a new StructuredQueryOutputParser instance from the provided\n * components.\n * @param allowedComparators An array of allowed Comparator instances.\n * @param allowedOperators An array of allowed Operator instances.\n * @returns A new StructuredQueryOutputParser instance.\n */\n static fromComponents(allowedComparators?: Comparator[], allowedOperators?: Operator[]): StructuredQueryOutputParser;\n}\nexport declare function formatAttributeInfo(info: AttributeInfo[]): string;\n/**\n * A type that represents options for the query constructor chain.\n */\nexport type QueryConstructorRunnableOptions = {\n llm: BaseLanguageModelInterface;\n documentContents: string;\n attributeInfo: AttributeInfo[];\n examples?: InputValues[];\n allowedComparators?: Comparator[];\n allowedOperators?: Operator[];\n};\n/** @deprecated */\nexport type QueryConstructorChainOptions = QueryConstructorRunnableOptions;\nexport declare function loadQueryConstructorRunnable(opts: QueryConstructorRunnableOptions\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\n): Runnable<any, StructuredQuery, RunnableConfig<Record<string, any>>>;\n"],"mappings":";;;;;;;;;;;;;AAaA;AAO6F;;AAElFA,cATUiB,aAAAA,CASRE;EAAS,IACInB,EAAEmB,MAAAA;EAAS,IAAzBnB,EAAEoB,MAAAA;EAAW,WACXC,EAAAA,MAAAA;EAAU,WAHUC,CAAAA,IAAAA,EAAAA,MAAAA,EAAAA,IAAAA,EAAAA,MAAAA,EAAAA,WAAAA,EAAAA,MAAAA;AAAS;cAA7BJ,gBAcmFA,EAdjElB,CAAAA,CAAEsB,SAc+DJ,CAAAA;EAAgB,KAAEZ,EAbxGN,CAAAA,CAAEmB,SAasGb;EAAe,MAE5GG,EAdVT,CAAAA,CAAEoB,WAcQX,CAdIT,CAAAA,CAAEmB,SAcNV,CAAAA;CAAgB,EAAA,OAEVL,EAfhBJ,CAAAA,CAAEqB,UAecjB,EAAAA;EAAU,KACZC,EAAAA,MAAAA;EAAQ,MAQZmB,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;CAAK,EAAA;EAAQ,KAAiCN,EAAAA,MAAAA;EAAgB,MAA7ChB,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;CAAqB,CAAA;;;;;AAbHc,cAApCO,2BAAAA,SAAoCP,gCAAAA,CAAAA,OAAwCE,gBAAxCF,EAA0DV,eAA1DU,CAAAA,CAAAA;EAAgC,YAAA,EAAA,MAAA,EAAA;EAuBjEW,gBAAAA,EArBFlB,gBAqB4BQ;EAItCW,WAAAA,CAAAA,MAAAA,EAAAA;IAA+B,kBAAA,EAvBfxB,UAuBe,EAAA;IAClCH,gBAAAA,EAvBiBI,QAuBjBJ,EAAAA;EAA0B,CAAA;EAEH;;;AAGD;AAG/B;AACA;EAAoD,eAAA,CAAA;IAAA,KAAA;IAAA;EAAA,CAAA,EAxBbC,qBAwBa,CAAA,OAxBgBgB,gBAwBhB,CAAA,CAAA,EAxBoCQ,OAwBpC,CAxB4CpB,eAwB5C,CAAA;EAAA;;;;;AAEzC;;6CAlBoCF,iCAAiCC,aAAakB;;iBAErEI,mBAAAA,OAA0BV;;;;KAItCW,+BAAAA;OACH3B;;iBAEUgB;aACJd;uBACUC;qBACFC;;;KAGXwB,4BAAAA,GAA+BD;iBACnBE,4BAAAA,OAAmCF;;GAExDrB,cAAcD,iBAAiBE,eAAeuB"}
|
|
@@ -2,7 +2,7 @@ import { __export } from "../../_virtual/rolldown_runtime.js";
|
|
|
2
2
|
import { AsymmetricStructuredOutputParser } from "../../output_parsers/structured.js";
|
|
3
3
|
import { QueryTransformer } from "./parser.js";
|
|
4
4
|
import { DEFAULT_EXAMPLES, DEFAULT_PREFIX, DEFAULT_SCHEMA, DEFAULT_SUFFIX, EXAMPLE_PROMPT } from "./prompt.js";
|
|
5
|
-
import { z } from "zod";
|
|
5
|
+
import { z } from "zod/v3";
|
|
6
6
|
import { FewShotPromptTemplate, interpolateFString } from "@langchain/core/prompts";
|
|
7
7
|
import { Comparators, Operators, StructuredQuery } from "@langchain/core/structured_query";
|
|
8
8
|
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","names":["name: string","type: string","description: string","fields: {\n allowedComparators: Comparator[];\n allowedOperators: Operator[];\n }","allowedComparators: Comparator[]","allowedOperators: Operator[]","info: AttributeInfo[]","documentContents: string","attributeInfo: AttributeInfo[]","allowedComparators?: Comparator[]","allowedOperators?: Operator[]","examples: InputValues[]","myAllowedComparators: Comparator[]","myAllowedOperators: Operator[]","opts: QueryConstructorRunnableOptions"],"sources":["../../../src/chains/query_constructor/index.ts"],"sourcesContent":["import { z } from \"zod\";\nimport type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport {\n Example,\n interpolateFString,\n FewShotPromptTemplate,\n} from \"@langchain/core/prompts\";\nimport {\n InferInteropZodOutput,\n InputValues,\n} from \"@langchain/core/utils/types\";\nimport {\n Comparator,\n Comparators,\n Operator,\n Operators,\n StructuredQuery,\n} from \"@langchain/core/structured_query\";\nimport { Runnable, RunnableConfig } from \"@langchain/core/runnables\";\nimport { QueryTransformer, TraverseType } from \"./parser.js\";\nimport {\n DEFAULT_EXAMPLES,\n DEFAULT_PREFIX,\n DEFAULT_SCHEMA,\n DEFAULT_SUFFIX,\n EXAMPLE_PROMPT,\n} from \"./prompt.js\";\nimport { AsymmetricStructuredOutputParser } from \"../../output_parsers/structured.js\";\n\n/**\n * A simple data structure that holds information about an attribute. It\n * is typically used to provide metadata about attributes in other classes\n * or data structures within the LangChain framework.\n */\nexport class AttributeInfo {\n constructor(\n public name: string,\n public type: string,\n public description: string\n ) {}\n}\n\nexport { QueryTransformer, type TraverseType };\nexport {\n DEFAULT_EXAMPLES,\n DEFAULT_PREFIX,\n DEFAULT_SCHEMA,\n DEFAULT_SUFFIX,\n EXAMPLE_PROMPT,\n};\n\nconst queryInputSchema = /* #__PURE__ */ z.object({\n query: /* #__PURE__ */ z\n .string()\n .describe(\"text string to compare to document contents\"),\n filter: /* #__PURE__ */ z\n .string()\n .optional()\n .describe(\"logical condition statement for filtering documents\"),\n});\n\n/**\n * A class that extends AsymmetricStructuredOutputParser to parse\n * structured query output.\n */\nexport class StructuredQueryOutputParser extends AsymmetricStructuredOutputParser<\n typeof queryInputSchema,\n StructuredQuery\n> {\n lc_namespace = [\"langchain\", \"chains\", \"query_constructor\"];\n\n queryTransformer: QueryTransformer;\n\n constructor(fields: {\n allowedComparators: Comparator[];\n allowedOperators: Operator[];\n }) {\n super({ ...fields, inputSchema: queryInputSchema });\n\n const { allowedComparators, allowedOperators } = fields;\n this.queryTransformer = new QueryTransformer(\n allowedComparators,\n allowedOperators\n );\n }\n\n /**\n * Processes the output of a structured query.\n * @param query The query string.\n * @param filter The filter condition.\n * @returns A Promise that resolves to a StructuredQuery instance.\n */\n async outputProcessor({\n query,\n filter,\n }: InferInteropZodOutput<typeof queryInputSchema>): Promise<StructuredQuery> {\n let myQuery = query;\n if (myQuery.length === 0) {\n myQuery = \" \";\n }\n if (filter === \"NO_FILTER\" || filter === undefined) {\n return new StructuredQuery(query);\n } else {\n const parsedFilter = await this.queryTransformer.parse(filter);\n return new StructuredQuery(query, parsedFilter);\n }\n }\n\n /**\n * Creates a new StructuredQueryOutputParser instance from the provided\n * components.\n * @param allowedComparators An array of allowed Comparator instances.\n * @param allowedOperators An array of allowed Operator instances.\n * @returns A new StructuredQueryOutputParser instance.\n */\n static fromComponents(\n allowedComparators: Comparator[] = [],\n allowedOperators: Operator[] = []\n ) {\n return new StructuredQueryOutputParser({\n allowedComparators,\n allowedOperators,\n });\n }\n}\n\nexport function formatAttributeInfo(info: AttributeInfo[]) {\n const infoObj = info.reduce((acc, attr) => {\n acc[attr.name] = {\n type: attr.type,\n description: attr.description,\n };\n return acc;\n }, {} as { [name: string]: { type: string; description: string } });\n\n return JSON.stringify(infoObj, null, 2)\n .replaceAll(\"{\", \"{{\")\n .replaceAll(\"}\", \"}}\");\n}\n\nconst defaultExample = DEFAULT_EXAMPLES.map((EXAMPLE) => EXAMPLE as Example);\n\nfunction _getPrompt(\n documentContents: string,\n attributeInfo: AttributeInfo[],\n allowedComparators?: Comparator[],\n allowedOperators?: Operator[],\n examples: InputValues[] = defaultExample\n) {\n const myAllowedComparators: Comparator[] =\n allowedComparators ?? Object.values(Comparators);\n const myAllowedOperators: Operator[] =\n allowedOperators ?? Object.values(Operators);\n const attributeJSON = formatAttributeInfo(attributeInfo);\n const schema = interpolateFString(DEFAULT_SCHEMA, {\n allowed_comparators: myAllowedComparators.join(\" | \"),\n allowed_operators: myAllowedOperators.join(\" | \"),\n });\n const prefix = interpolateFString(DEFAULT_PREFIX, {\n schema,\n });\n const suffix = interpolateFString(DEFAULT_SUFFIX, {\n i: examples.length + 1,\n content: documentContents,\n attributes: attributeJSON,\n });\n\n const outputParser = StructuredQueryOutputParser.fromComponents(\n allowedComparators,\n allowedOperators\n );\n\n return new FewShotPromptTemplate({\n examples,\n examplePrompt: EXAMPLE_PROMPT,\n inputVariables: [\"query\"],\n suffix,\n prefix,\n outputParser,\n });\n}\n\n/**\n * A type that represents options for the query constructor chain.\n */\nexport type QueryConstructorRunnableOptions = {\n llm: BaseLanguageModelInterface;\n documentContents: string;\n attributeInfo: AttributeInfo[];\n examples?: InputValues[];\n allowedComparators?: Comparator[];\n allowedOperators?: Operator[];\n};\n\n/** @deprecated */\nexport type QueryConstructorChainOptions = QueryConstructorRunnableOptions;\n\nexport function loadQueryConstructorRunnable(\n opts: QueryConstructorRunnableOptions\n): Runnable<any, StructuredQuery, RunnableConfig<Record<string, any>>> {\n const prompt = _getPrompt(\n opts.documentContents,\n opts.attributeInfo,\n opts.allowedComparators,\n opts.allowedOperators,\n opts.examples\n );\n const outputParser = StructuredQueryOutputParser.fromComponents(\n opts.allowedComparators,\n opts.allowedOperators\n );\n return prompt.pipe(opts.llm).pipe(outputParser);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;AAkCA,IAAa,gBAAb,MAA2B;CACzB,YACSA,MACAC,MACAC,aACP;EAHO;EACA;EACA;CACL;AACL;AAWD,MAAM,mCAAmC,EAAE,OAAO;CAChD,uBAAuB,EACpB,QAAQ,CACR,SAAS,8CAA8C;CAC1D,wBAAwB,EACrB,QAAQ,CACR,UAAU,CACV,SAAS,sDAAsD;AACnE,EAAC;;;;;AAMF,IAAa,8BAAb,MAAa,oCAAoC,iCAG/C;CACA,eAAe;EAAC;EAAa;EAAU;CAAoB;CAE3D;CAEA,YAAYC,QAGT;EACD,MAAM;GAAE,GAAG;GAAQ,aAAa;EAAkB,EAAC;EAEnD,MAAM,EAAE,oBAAoB,kBAAkB,GAAG;EACjD,KAAK,mBAAmB,IAAI,iBAC1B,oBACA;CAEH;;;;;;;CAQD,MAAM,gBAAgB,EACpB,OACA,QAC+C,EAA4B;EAC3E,IAAI,UAAU;AACd,MAAI,QAAQ,WAAW,GACrB,UAAU;AAEZ,MAAI,WAAW,eAAe,WAAW,OACvC,QAAO,IAAI,gBAAgB;OACtB;GACL,MAAM,eAAe,MAAM,KAAK,iBAAiB,MAAM,OAAO;AAC9D,UAAO,IAAI,gBAAgB,OAAO;EACnC;CACF;;;;;;;;CASD,OAAO,eACLC,qBAAmC,CAAE,GACrCC,mBAA+B,CAAE,GACjC;AACA,SAAO,IAAI,4BAA4B;GACrC;GACA;EACD;CACF;AACF;AAED,SAAgB,oBAAoBC,MAAuB;CACzD,MAAM,UAAU,KAAK,OAAO,CAAC,KAAK,SAAS;EACzC,IAAI,KAAK,QAAQ;GACf,MAAM,KAAK;GACX,aAAa,KAAK;EACnB;AACD,SAAO;CACR,GAAE,CAAE,EAA8D;AAEnE,QAAO,KAAK,UAAU,SAAS,MAAM,EAAE,CACpC,WAAW,KAAK,KAAK,CACrB,WAAW,KAAK,KAAK;AACzB;AAED,MAAM,iBAAiB,iBAAiB,IAAI,CAAC,YAAY,QAAmB;AAE5E,SAAS,WACPC,kBACAC,eACAC,oBACAC,kBACAC,WAA0B,gBAC1B;CACA,MAAMC,uBACJ,sBAAsB,OAAO,OAAO,YAAY;CAClD,MAAMC,qBACJ,oBAAoB,OAAO,OAAO,UAAU;CAC9C,MAAM,gBAAgB,oBAAoB,cAAc;CACxD,MAAM,SAAS,mBAAmB,gBAAgB;EAChD,qBAAqB,qBAAqB,KAAK,MAAM;EACrD,mBAAmB,mBAAmB,KAAK,MAAM;CAClD,EAAC;CACF,MAAM,SAAS,mBAAmB,gBAAgB,EAChD,OACD,EAAC;CACF,MAAM,SAAS,mBAAmB,gBAAgB;EAChD,GAAG,SAAS,SAAS;EACrB,SAAS;EACT,YAAY;CACb,EAAC;CAEF,MAAM,eAAe,4BAA4B,eAC/C,oBACA,iBACD;AAED,QAAO,IAAI,sBAAsB;EAC/B;EACA,eAAe;EACf,gBAAgB,CAAC,OAAQ;EACzB;EACA;EACA;CACD;AACF;AAiBD,SAAgB,6BACdC,
|
|
1
|
+
{"version":3,"file":"index.js","names":["name: string","type: string","description: string","fields: {\n allowedComparators: Comparator[];\n allowedOperators: Operator[];\n }","allowedComparators: Comparator[]","allowedOperators: Operator[]","info: AttributeInfo[]","documentContents: string","attributeInfo: AttributeInfo[]","allowedComparators?: Comparator[]","allowedOperators?: Operator[]","examples: InputValues[]","myAllowedComparators: Comparator[]","myAllowedOperators: Operator[]","opts: QueryConstructorRunnableOptions"],"sources":["../../../src/chains/query_constructor/index.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport {\n Example,\n interpolateFString,\n FewShotPromptTemplate,\n} from \"@langchain/core/prompts\";\nimport {\n InferInteropZodOutput,\n InputValues,\n} from \"@langchain/core/utils/types\";\nimport {\n Comparator,\n Comparators,\n Operator,\n Operators,\n StructuredQuery,\n} from \"@langchain/core/structured_query\";\nimport { Runnable, RunnableConfig } from \"@langchain/core/runnables\";\nimport { QueryTransformer, TraverseType } from \"./parser.js\";\nimport {\n DEFAULT_EXAMPLES,\n DEFAULT_PREFIX,\n DEFAULT_SCHEMA,\n DEFAULT_SUFFIX,\n EXAMPLE_PROMPT,\n} from \"./prompt.js\";\nimport { AsymmetricStructuredOutputParser } from \"../../output_parsers/structured.js\";\n\n/**\n * A simple data structure that holds information about an attribute. It\n * is typically used to provide metadata about attributes in other classes\n * or data structures within the LangChain framework.\n */\nexport class AttributeInfo {\n constructor(\n public name: string,\n public type: string,\n public description: string\n ) {}\n}\n\nexport { QueryTransformer, type TraverseType };\nexport {\n DEFAULT_EXAMPLES,\n DEFAULT_PREFIX,\n DEFAULT_SCHEMA,\n DEFAULT_SUFFIX,\n EXAMPLE_PROMPT,\n};\n\nconst queryInputSchema = /* #__PURE__ */ z.object({\n query: /* #__PURE__ */ z\n .string()\n .describe(\"text string to compare to document contents\"),\n filter: /* #__PURE__ */ z\n .string()\n .optional()\n .describe(\"logical condition statement for filtering documents\"),\n});\n\n/**\n * A class that extends AsymmetricStructuredOutputParser to parse\n * structured query output.\n */\nexport class StructuredQueryOutputParser extends AsymmetricStructuredOutputParser<\n typeof queryInputSchema,\n StructuredQuery\n> {\n lc_namespace = [\"langchain\", \"chains\", \"query_constructor\"];\n\n queryTransformer: QueryTransformer;\n\n constructor(fields: {\n allowedComparators: Comparator[];\n allowedOperators: Operator[];\n }) {\n super({ ...fields, inputSchema: queryInputSchema });\n\n const { allowedComparators, allowedOperators } = fields;\n this.queryTransformer = new QueryTransformer(\n allowedComparators,\n allowedOperators\n );\n }\n\n /**\n * Processes the output of a structured query.\n * @param query The query string.\n * @param filter The filter condition.\n * @returns A Promise that resolves to a StructuredQuery instance.\n */\n async outputProcessor({\n query,\n filter,\n }: InferInteropZodOutput<typeof queryInputSchema>): Promise<StructuredQuery> {\n let myQuery = query;\n if (myQuery.length === 0) {\n myQuery = \" \";\n }\n if (filter === \"NO_FILTER\" || filter === undefined) {\n return new StructuredQuery(query);\n } else {\n const parsedFilter = await this.queryTransformer.parse(filter);\n return new StructuredQuery(query, parsedFilter);\n }\n }\n\n /**\n * Creates a new StructuredQueryOutputParser instance from the provided\n * components.\n * @param allowedComparators An array of allowed Comparator instances.\n * @param allowedOperators An array of allowed Operator instances.\n * @returns A new StructuredQueryOutputParser instance.\n */\n static fromComponents(\n allowedComparators: Comparator[] = [],\n allowedOperators: Operator[] = []\n ) {\n return new StructuredQueryOutputParser({\n allowedComparators,\n allowedOperators,\n });\n }\n}\n\nexport function formatAttributeInfo(info: AttributeInfo[]) {\n const infoObj = info.reduce((acc, attr) => {\n acc[attr.name] = {\n type: attr.type,\n description: attr.description,\n };\n return acc;\n }, {} as { [name: string]: { type: string; description: string } });\n\n return JSON.stringify(infoObj, null, 2)\n .replaceAll(\"{\", \"{{\")\n .replaceAll(\"}\", \"}}\");\n}\n\nconst defaultExample = DEFAULT_EXAMPLES.map((EXAMPLE) => EXAMPLE as Example);\n\nfunction _getPrompt(\n documentContents: string,\n attributeInfo: AttributeInfo[],\n allowedComparators?: Comparator[],\n allowedOperators?: Operator[],\n examples: InputValues[] = defaultExample\n) {\n const myAllowedComparators: Comparator[] =\n allowedComparators ?? Object.values(Comparators);\n const myAllowedOperators: Operator[] =\n allowedOperators ?? Object.values(Operators);\n const attributeJSON = formatAttributeInfo(attributeInfo);\n const schema = interpolateFString(DEFAULT_SCHEMA, {\n allowed_comparators: myAllowedComparators.join(\" | \"),\n allowed_operators: myAllowedOperators.join(\" | \"),\n });\n const prefix = interpolateFString(DEFAULT_PREFIX, {\n schema,\n });\n const suffix = interpolateFString(DEFAULT_SUFFIX, {\n i: examples.length + 1,\n content: documentContents,\n attributes: attributeJSON,\n });\n\n const outputParser = StructuredQueryOutputParser.fromComponents(\n allowedComparators,\n allowedOperators\n );\n\n return new FewShotPromptTemplate({\n examples,\n examplePrompt: EXAMPLE_PROMPT,\n inputVariables: [\"query\"],\n suffix,\n prefix,\n outputParser,\n });\n}\n\n/**\n * A type that represents options for the query constructor chain.\n */\nexport type QueryConstructorRunnableOptions = {\n llm: BaseLanguageModelInterface;\n documentContents: string;\n attributeInfo: AttributeInfo[];\n examples?: InputValues[];\n allowedComparators?: Comparator[];\n allowedOperators?: Operator[];\n};\n\n/** @deprecated */\nexport type QueryConstructorChainOptions = QueryConstructorRunnableOptions;\n\nexport function loadQueryConstructorRunnable(\n opts: QueryConstructorRunnableOptions\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n): Runnable<any, StructuredQuery, RunnableConfig<Record<string, any>>> {\n const prompt = _getPrompt(\n opts.documentContents,\n opts.attributeInfo,\n opts.allowedComparators,\n opts.allowedOperators,\n opts.examples\n );\n const outputParser = StructuredQueryOutputParser.fromComponents(\n opts.allowedComparators,\n opts.allowedOperators\n );\n return prompt.pipe(opts.llm).pipe(outputParser);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;AAkCA,IAAa,gBAAb,MAA2B;CACzB,YACSA,MACAC,MACAC,aACP;EAHO;EACA;EACA;CACL;AACL;AAWD,MAAM,mCAAmC,EAAE,OAAO;CAChD,uBAAuB,EACpB,QAAQ,CACR,SAAS,8CAA8C;CAC1D,wBAAwB,EACrB,QAAQ,CACR,UAAU,CACV,SAAS,sDAAsD;AACnE,EAAC;;;;;AAMF,IAAa,8BAAb,MAAa,oCAAoC,iCAG/C;CACA,eAAe;EAAC;EAAa;EAAU;CAAoB;CAE3D;CAEA,YAAYC,QAGT;EACD,MAAM;GAAE,GAAG;GAAQ,aAAa;EAAkB,EAAC;EAEnD,MAAM,EAAE,oBAAoB,kBAAkB,GAAG;EACjD,KAAK,mBAAmB,IAAI,iBAC1B,oBACA;CAEH;;;;;;;CAQD,MAAM,gBAAgB,EACpB,OACA,QAC+C,EAA4B;EAC3E,IAAI,UAAU;AACd,MAAI,QAAQ,WAAW,GACrB,UAAU;AAEZ,MAAI,WAAW,eAAe,WAAW,OACvC,QAAO,IAAI,gBAAgB;OACtB;GACL,MAAM,eAAe,MAAM,KAAK,iBAAiB,MAAM,OAAO;AAC9D,UAAO,IAAI,gBAAgB,OAAO;EACnC;CACF;;;;;;;;CASD,OAAO,eACLC,qBAAmC,CAAE,GACrCC,mBAA+B,CAAE,GACjC;AACA,SAAO,IAAI,4BAA4B;GACrC;GACA;EACD;CACF;AACF;AAED,SAAgB,oBAAoBC,MAAuB;CACzD,MAAM,UAAU,KAAK,OAAO,CAAC,KAAK,SAAS;EACzC,IAAI,KAAK,QAAQ;GACf,MAAM,KAAK;GACX,aAAa,KAAK;EACnB;AACD,SAAO;CACR,GAAE,CAAE,EAA8D;AAEnE,QAAO,KAAK,UAAU,SAAS,MAAM,EAAE,CACpC,WAAW,KAAK,KAAK,CACrB,WAAW,KAAK,KAAK;AACzB;AAED,MAAM,iBAAiB,iBAAiB,IAAI,CAAC,YAAY,QAAmB;AAE5E,SAAS,WACPC,kBACAC,eACAC,oBACAC,kBACAC,WAA0B,gBAC1B;CACA,MAAMC,uBACJ,sBAAsB,OAAO,OAAO,YAAY;CAClD,MAAMC,qBACJ,oBAAoB,OAAO,OAAO,UAAU;CAC9C,MAAM,gBAAgB,oBAAoB,cAAc;CACxD,MAAM,SAAS,mBAAmB,gBAAgB;EAChD,qBAAqB,qBAAqB,KAAK,MAAM;EACrD,mBAAmB,mBAAmB,KAAK,MAAM;CAClD,EAAC;CACF,MAAM,SAAS,mBAAmB,gBAAgB,EAChD,OACD,EAAC;CACF,MAAM,SAAS,mBAAmB,gBAAgB;EAChD,GAAG,SAAS,SAAS;EACrB,SAAS;EACT,YAAY;CACb,EAAC;CAEF,MAAM,eAAe,4BAA4B,eAC/C,oBACA,iBACD;AAED,QAAO,IAAI,sBAAsB;EAC/B;EACA,eAAe;EACf,gBAAgB,CAAC,OAAQ;EACzB;EACA;EACA;CACD;AACF;AAiBD,SAAgB,6BACdC,MAEqE;CACrE,MAAM,SAAS,WACb,KAAK,kBACL,KAAK,eACL,KAAK,oBACL,KAAK,kBACL,KAAK,SACN;CACD,MAAM,eAAe,4BAA4B,eAC/C,KAAK,oBACL,KAAK,iBACN;AACD,QAAO,OAAO,KAAK,KAAK,IAAI,CAAC,KAAK,aAAa;AAChD"}
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import { MapReduceDocumentsChain, MapReduceDocumentsChainInput, RefineDocumentsChain, StuffDocumentsChain } from "../combine_docs_chain.js";
|
|
2
2
|
import { BasePromptTemplate } from "@langchain/core/prompts";
|
|
3
|
-
import * as
|
|
3
|
+
import * as _langchain_core_language_models_base0 from "@langchain/core/language_models/base";
|
|
4
4
|
import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
|
|
5
5
|
|
|
6
6
|
//#region src/chains/question_answering/load.d.ts
|
|
@@ -16,7 +16,7 @@ type QAChainParams = ({
|
|
|
16
16
|
} & MapReduceQAChainParams) | ({
|
|
17
17
|
type?: "refine";
|
|
18
18
|
} & RefineQAChainParams);
|
|
19
|
-
declare const loadQAChain: (llm: BaseLanguageModelInterface<any,
|
|
19
|
+
declare const loadQAChain: (llm: BaseLanguageModelInterface<any, _langchain_core_language_models_base0.BaseLanguageModelCallOptions>, params?: QAChainParams) => MapReduceDocumentsChain | RefineDocumentsChain | StuffDocumentsChain;
|
|
20
20
|
/**
|
|
21
21
|
* Represents the parameters for creating a StuffQAChain.
|
|
22
22
|
*/
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"load.d.ts","names":["BaseLanguageModelInterface","BasePromptTemplate","StuffDocumentsChain","MapReduceDocumentsChain","RefineDocumentsChain","MapReduceDocumentsChainInput","QAChainParams","StuffQAChainParams","MapReduceQAChainParams","RefineQAChainParams","loadQAChain","
|
|
1
|
+
{"version":3,"file":"load.d.ts","names":["BaseLanguageModelInterface","BasePromptTemplate","StuffDocumentsChain","MapReduceDocumentsChain","RefineDocumentsChain","MapReduceDocumentsChainInput","QAChainParams","StuffQAChainParams","MapReduceQAChainParams","RefineQAChainParams","loadQAChain","_langchain_core_language_models_base0","BaseLanguageModelCallOptions","loadQAStuffChain","loadQAMapReduceChain","loadQARefineChain"],"sources":["../../../src/chains/question_answering/load.d.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { StuffDocumentsChain, MapReduceDocumentsChain, RefineDocumentsChain, MapReduceDocumentsChainInput } from \"../combine_docs_chain.js\";\n/**\n * Represents the parameters for creating a QAChain. It can be of three\n * types: \"stuff\", \"map_reduce\", or \"refine\".\n */\nexport type QAChainParams = ({\n type?: \"stuff\";\n} & StuffQAChainParams) | ({\n type?: \"map_reduce\";\n} & MapReduceQAChainParams) | ({\n type?: \"refine\";\n} & RefineQAChainParams);\nexport declare const loadQAChain: (llm: BaseLanguageModelInterface<any, import(\"@langchain/core/language_models/base\").BaseLanguageModelCallOptions>, params?: QAChainParams) => MapReduceDocumentsChain | RefineDocumentsChain | StuffDocumentsChain;\n/**\n * Represents the parameters for creating a StuffQAChain.\n */\nexport interface StuffQAChainParams {\n prompt?: BasePromptTemplate;\n verbose?: boolean;\n}\n/**\n * Loads a StuffQAChain based on the provided parameters. It takes an LLM\n * instance and StuffQAChainParams as parameters.\n * @param llm An instance of BaseLanguageModel.\n * @param params Parameters for creating a StuffQAChain.\n * @returns A StuffQAChain instance.\n */\nexport declare function loadQAStuffChain(llm: BaseLanguageModelInterface, params?: StuffQAChainParams): StuffDocumentsChain;\n/**\n * Represents the parameters for creating a MapReduceQAChain.\n */\nexport interface MapReduceQAChainParams {\n returnIntermediateSteps?: MapReduceDocumentsChainInput[\"returnIntermediateSteps\"];\n combineMapPrompt?: BasePromptTemplate;\n combinePrompt?: BasePromptTemplate;\n combineLLM?: BaseLanguageModelInterface;\n verbose?: boolean;\n}\n/**\n * Loads a MapReduceQAChain based on the provided parameters. It takes an\n * LLM instance and MapReduceQAChainParams as parameters.\n * @param llm An instance of BaseLanguageModel.\n * @param params Parameters for creating a MapReduceQAChain.\n * @returns A MapReduceQAChain instance.\n */\nexport declare function loadQAMapReduceChain(llm: BaseLanguageModelInterface, params?: MapReduceQAChainParams): MapReduceDocumentsChain;\n/**\n * Represents the parameters for creating a RefineQAChain.\n */\nexport interface RefineQAChainParams {\n questionPrompt?: BasePromptTemplate;\n refinePrompt?: BasePromptTemplate;\n refineLLM?: BaseLanguageModelInterface;\n verbose?: boolean;\n}\n/**\n * Loads a RefineQAChain based on the provided parameters. It takes an LLM\n * instance and RefineQAChainParams as parameters.\n * @param llm An instance of BaseLanguageModel.\n * @param params Parameters for creating a RefineQAChain.\n * @returns A RefineQAChain instance.\n */\nexport declare function loadQARefineChain(llm: BaseLanguageModelInterface, params?: RefineQAChainParams): RefineDocumentsChain;\n"],"mappings":";;;;;;;;;;AAOA;AAAyB,KAAbM,aAAAA,GAAa,CAAA;EAAA,IAErBC,CAAAA,EAAAA,OAAAA;CAAkB,GAAlBA,kBAEAC,CAAAA,GAAAA,CAAAA;EAAsB,IAEtBC,CAAAA,EAAAA,YAAAA;AAAmB,CAAA,GAFnBD,sBAEmB,CAAA,GAAA,CAAA;EACFE,IAAAA,CAAAA,EAAAA,QAAgO;CAAA,GADjPD,mBACiP,CAAA;AAAAE,cAAhOD,WAAgOC,EAAAA,CAAAA,GAAAA,EAA7MX,0BAA+EY,CAAAA,GAAAA,EAA8HD,qCAAAA,CAA9HC,4BAAAA,CAAAA,EAAAA,MAAAA,CAAAA,EAAwCN,aAAxCM,EAAAA,GAA0DT,uBAA1DS,GAAoFR,oBAApFQ,GAA2GV,mBAA3GU;;;;AAAoFR,UAI1LG,kBAAAA,CAJ0LH;EAAoB,MAAGF,CAAAA,EAKrND,kBALqNC;EAAmB,OAAA,CAAA,EAAA,OAAA;AAIrP;AAWA;;;;;AAA2H;AAI3H;AAAuC,iBAJfW,gBAAAA,CAIe,GAAA,EAJOb,0BAIP,EAAA,MAAA,CAAA,EAJ4CO,kBAI5C,CAAA,EAJiEL,mBAIjE;;;;AAItBF,UAJAQ,sBAAAA,CAIAR;EAA0B,uBAAA,CAAA,EAHbK,4BAGa,CAAA,yBAAA,CAAA;EAUnBS,gBAAAA,CAAAA,EAZDb,kBAYqB;EAAA,aAAA,CAAA,EAXxBA,kBAWwB;EAAA,UAAMD,CAAAA,EAVjCA,0BAUiCA;EAA0B,OAAWQ,CAAAA,EAAAA,OAAAA;;AAAgD;AAIvI;;;;;AAG0C;AAUlBO,iBAjBAD,oBAAAA,CAiBiB,GAAA,EAjBSd,0BAiBT,EAAA,MAAA,CAAA,EAjB8CQ,sBAiB9C,CAAA,EAjBuEL,uBAiBvE;;;;AAAiEC,UAbzFK,mBAAAA,CAayFL;EAAoB,cAAA,CAAA,EAZzGH,kBAYyG;iBAX3GA;cACHD;;;;;;;;;;iBAUQe,iBAAAA,MAAuBf,qCAAqCS,sBAAsBL"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"map_reduce_prompts.cjs","names":["PromptTemplate","SystemMessagePromptTemplate","HumanMessagePromptTemplate","ChatPromptTemplate","ConditionalPromptSelector","isChatModel"],"sources":["../../../src/chains/question_answering/map_reduce_prompts.ts"],"sourcesContent":["
|
|
1
|
+
{"version":3,"file":"map_reduce_prompts.cjs","names":["PromptTemplate","SystemMessagePromptTemplate","HumanMessagePromptTemplate","ChatPromptTemplate","ConditionalPromptSelector","isChatModel"],"sources":["../../../src/chains/question_answering/map_reduce_prompts.ts"],"sourcesContent":["import {\n ChatPromptTemplate,\n SystemMessagePromptTemplate,\n HumanMessagePromptTemplate,\n PromptTemplate,\n} from \"@langchain/core/prompts\";\nimport {\n ConditionalPromptSelector,\n isChatModel,\n} from \"@langchain/core/example_selectors\";\n\nconst qa_template = `Use the following portion of a long document to see if any of the text is relevant to answer the question. \nReturn any relevant text verbatim.\n{context}\nQuestion: {question}\nRelevant text, if any:`;\nexport const DEFAULT_COMBINE_QA_PROMPT =\n /*#__PURE__*/\n PromptTemplate.fromTemplate(qa_template);\n\nconst system_template = `Use the following portion of a long document to see if any of the text is relevant to answer the question. \nReturn any relevant text verbatim.\n----------------\n{context}`;\nconst messages = [\n /*#__PURE__*/ SystemMessagePromptTemplate.fromTemplate(system_template),\n /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate(\"{question}\"),\n];\nconst CHAT_QA_PROMPT = /*#__PURE__*/ ChatPromptTemplate.fromMessages(messages);\n\nexport const COMBINE_QA_PROMPT_SELECTOR =\n /*#__PURE__*/ new ConditionalPromptSelector(DEFAULT_COMBINE_QA_PROMPT, [\n [isChatModel, CHAT_QA_PROMPT],\n ]);\n\nconst combine_prompt = `Given the following extracted parts of a long document and a question, create a final answer. \nIf you don't know the answer, just say that you don't know. Don't try to make up an answer.\n\nQUESTION: Which state/country's law governs the interpretation of the contract?\n=========\nContent: This Agreement is governed by English law and the parties submit to the exclusive jurisdiction of the English courts in relation to any dispute (contractual or non-contractual) concerning this Agreement save that either party may apply to any court for an injunction or other relief to protect its Intellectual Property Rights.\n\nContent: No Waiver. Failure or delay in exercising any right or remedy under this Agreement shall not constitute a waiver of such (or any other) right or remedy.\\n\\n11.7 Severability. The invalidity, illegality or unenforceability of any term (or part of a term) of this Agreement shall not affect the continuation in force of the remainder of the term (if any) and this Agreement.\\n\\n11.8 No Agency. Except as expressly stated otherwise, nothing in this Agreement shall create an agency, partnership or joint venture of any kind between the parties.\\n\\n11.9 No Third-Party Beneficiaries.\n\nContent: (b) if Google believes, in good faith, that the Distributor has violated or caused Google to violate any Anti-Bribery Laws (as defined in Clause 8.5) or that such a violation is reasonably likely to occur,\n=========\nFINAL ANSWER: This Agreement is governed by English law.\n\nQUESTION: What did the president say about Michael Jackson?\n=========\nContent: Madam Speaker, Madam Vice President, our First Lady and Second Gentleman. Members of Congress and the Cabinet. Justices of the Supreme Court. My fellow Americans. \\n\\nLast year COVID-19 kept us apart. This year we are finally together again. \\n\\nTonight, we meet as Democrats Republicans and Independents. But most importantly as Americans. \\n\\nWith a duty to one another to the American people to the Constitution. \\n\\nAnd with an unwavering resolve that freedom will always triumph over tyranny. \\n\\nSix days ago, Russia’s Vladimir Putin sought to shake the foundations of the free world thinking he could make it bend to his menacing ways. But he badly miscalculated. \\n\\nHe thought he could roll into Ukraine and the world would roll over. Instead he met a wall of strength he never imagined. \\n\\nHe met the Ukrainian people. \\n\\nFrom President Zelenskyy to every Ukrainian, their fearlessness, their courage, their determination, inspires the world. \\n\\nGroups of citizens blocking tanks with their bodies. Everyone from students to retirees teachers turned soldiers defending their homeland.\n\nContent: And we won’t stop. \\n\\nWe have lost so much to COVID-19. Time with one another. And worst of all, so much loss of life. \\n\\nLet’s use this moment to reset. Let’s stop looking at COVID-19 as a partisan dividing line and see it for what it is: A God-awful disease. \\n\\nLet’s stop seeing each other as enemies, and start seeing each other for who we really are: Fellow Americans. \\n\\nWe can’t change how divided we’ve been. But we can change how we move forward—on COVID-19 and other issues we must face together. \\n\\nI recently visited the New York City Police Department days after the funerals of Officer Wilbert Mora and his partner, Officer Jason Rivera. \\n\\nThey were responding to a 9-1-1 call when a man shot and killed them with a stolen gun. \\n\\nOfficer Mora was 27 years old. \\n\\nOfficer Rivera was 22. \\n\\nBoth Dominican Americans who’d grown up on the same streets they later chose to patrol as police officers. \\n\\nI spoke with their families and told them that we are forever in debt for their sacrifice, and we will carry on their mission to restore the trust and safety every community deserves.\n\nContent: And a proud Ukrainian people, who have known 30 years of independence, have repeatedly shown that they will not tolerate anyone who tries to take their country backwards. \\n\\nTo all Americans, I will be honest with you, as I’ve always promised. A Russian dictator, invading a foreign country, has costs around the world. \\n\\nAnd I’m taking robust action to make sure the pain of our sanctions is targeted at Russia’s economy. And I will use every tool at our disposal to protect American businesses and consumers. \\n\\nTonight, I can announce that the United States has worked with 30 other countries to release 60 Million barrels of oil from reserves around the world. \\n\\nAmerica will lead that effort, releasing 30 Million barrels from our own Strategic Petroleum Reserve. And we stand ready to do more if necessary, unified with our allies. \\n\\nThese steps will help blunt gas prices here at home. And I know the news about what’s happening can seem alarming. \\n\\nBut I want you to know that we are going to be okay.\n\nContent: More support for patients and families. \\n\\nTo get there, I call on Congress to fund ARPA-H, the Advanced Research Projects Agency for Health. \\n\\nIt’s based on DARPA—the Defense Department project that led to the Internet, GPS, and so much more. \\n\\nARPA-H will have a singular purpose—to drive breakthroughs in cancer, Alzheimer’s, diabetes, and more. \\n\\nA unity agenda for the nation. \\n\\nWe can do this. \\n\\nMy fellow Americans—tonight , we have gathered in a sacred space—the citadel of our democracy. \\n\\nIn this Capitol, generation after generation, Americans have debated great questions amid great strife, and have done great things. \\n\\nWe have fought for freedom, expanded liberty, defeated totalitarianism and terror. \\n\\nAnd built the strongest, freest, and most prosperous nation the world has ever known. \\n\\nNow is the hour. \\n\\nOur moment of responsibility. \\n\\nOur test of resolve and conscience, of history itself. \\n\\nIt is in this moment that our character is formed. Our purpose is found. Our future is forged. \\n\\nWell I know this nation.\n=========\nFINAL ANSWER: The president did not mention Michael Jackson.\n\nQUESTION: {question}\n=========\n{summaries}\n=========\nFINAL ANSWER:`;\nexport const COMBINE_PROMPT =\n /*#__PURE__*/ PromptTemplate.fromTemplate(combine_prompt);\n\nconst system_combine_template = `Given the following extracted parts of a long document and a question, create a final answer. \nIf you don't know the answer, just say that you don't know. Don't try to make up an answer.\n----------------\n{summaries}`;\nconst combine_messages = [\n /*#__PURE__*/ SystemMessagePromptTemplate.fromTemplate(\n system_combine_template\n ),\n /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate(\"{question}\"),\n];\nconst CHAT_COMBINE_PROMPT =\n /*#__PURE__*/ ChatPromptTemplate.fromMessages(combine_messages);\n\nexport const COMBINE_PROMPT_SELECTOR =\n /*#__PURE__*/ new ConditionalPromptSelector(COMBINE_PROMPT, [\n [isChatModel, CHAT_COMBINE_PROMPT],\n ]);\n"],"mappings":";;;;;AAWA,MAAM,cAAc,CAAC;;;;sBAIC,CAAC;AACvB,MAAa,4CAEXA,wCAAe,aAAa,YAAY;AAE1C,MAAM,kBAAkB,CAAC;;;SAGhB,CAAC;AACV,MAAM,WAAW,iBACDC,qDAA4B,aAAa,gBAAgB,kBACzDC,oDAA2B,aAAa,aAAa,AACpE;AACD,MAAM,iCAA+BC,4CAAmB,aAAa,SAAS;AAE9E,MAAa,6CACG,IAAIC,6DAA0B,2BAA2B,CACrE,CAACC,gDAAa,cAAe,CAC9B;AAEH,MAAM,iBAAiB,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;aA6BX,CAAC;AACd,MAAa,iCACGL,wCAAe,aAAa,eAAe;AAE3D,MAAM,0BAA0B,CAAC;;;WAGtB,CAAC;AACZ,MAAM,mBAAmB,iBACTC,qDAA4B,aACxC,wBACD,kBACaC,oDAA2B,aAAa,aAAa,AACpE;AACD,MAAM,sCACUC,4CAAmB,aAAa,iBAAiB;AAEjE,MAAa,0CACG,IAAIC,6DAA0B,gBAAgB,CAC1D,CAACC,gDAAa,mBAAoB,CACnC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"map_reduce_prompts.js","names":[],"sources":["../../../src/chains/question_answering/map_reduce_prompts.ts"],"sourcesContent":["
|
|
1
|
+
{"version":3,"file":"map_reduce_prompts.js","names":[],"sources":["../../../src/chains/question_answering/map_reduce_prompts.ts"],"sourcesContent":["import {\n ChatPromptTemplate,\n SystemMessagePromptTemplate,\n HumanMessagePromptTemplate,\n PromptTemplate,\n} from \"@langchain/core/prompts\";\nimport {\n ConditionalPromptSelector,\n isChatModel,\n} from \"@langchain/core/example_selectors\";\n\nconst qa_template = `Use the following portion of a long document to see if any of the text is relevant to answer the question. \nReturn any relevant text verbatim.\n{context}\nQuestion: {question}\nRelevant text, if any:`;\nexport const DEFAULT_COMBINE_QA_PROMPT =\n /*#__PURE__*/\n PromptTemplate.fromTemplate(qa_template);\n\nconst system_template = `Use the following portion of a long document to see if any of the text is relevant to answer the question. \nReturn any relevant text verbatim.\n----------------\n{context}`;\nconst messages = [\n /*#__PURE__*/ SystemMessagePromptTemplate.fromTemplate(system_template),\n /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate(\"{question}\"),\n];\nconst CHAT_QA_PROMPT = /*#__PURE__*/ ChatPromptTemplate.fromMessages(messages);\n\nexport const COMBINE_QA_PROMPT_SELECTOR =\n /*#__PURE__*/ new ConditionalPromptSelector(DEFAULT_COMBINE_QA_PROMPT, [\n [isChatModel, CHAT_QA_PROMPT],\n ]);\n\nconst combine_prompt = `Given the following extracted parts of a long document and a question, create a final answer. \nIf you don't know the answer, just say that you don't know. Don't try to make up an answer.\n\nQUESTION: Which state/country's law governs the interpretation of the contract?\n=========\nContent: This Agreement is governed by English law and the parties submit to the exclusive jurisdiction of the English courts in relation to any dispute (contractual or non-contractual) concerning this Agreement save that either party may apply to any court for an injunction or other relief to protect its Intellectual Property Rights.\n\nContent: No Waiver. Failure or delay in exercising any right or remedy under this Agreement shall not constitute a waiver of such (or any other) right or remedy.\\n\\n11.7 Severability. The invalidity, illegality or unenforceability of any term (or part of a term) of this Agreement shall not affect the continuation in force of the remainder of the term (if any) and this Agreement.\\n\\n11.8 No Agency. Except as expressly stated otherwise, nothing in this Agreement shall create an agency, partnership or joint venture of any kind between the parties.\\n\\n11.9 No Third-Party Beneficiaries.\n\nContent: (b) if Google believes, in good faith, that the Distributor has violated or caused Google to violate any Anti-Bribery Laws (as defined in Clause 8.5) or that such a violation is reasonably likely to occur,\n=========\nFINAL ANSWER: This Agreement is governed by English law.\n\nQUESTION: What did the president say about Michael Jackson?\n=========\nContent: Madam Speaker, Madam Vice President, our First Lady and Second Gentleman. Members of Congress and the Cabinet. Justices of the Supreme Court. My fellow Americans. \\n\\nLast year COVID-19 kept us apart. This year we are finally together again. \\n\\nTonight, we meet as Democrats Republicans and Independents. But most importantly as Americans. \\n\\nWith a duty to one another to the American people to the Constitution. \\n\\nAnd with an unwavering resolve that freedom will always triumph over tyranny. \\n\\nSix days ago, Russia’s Vladimir Putin sought to shake the foundations of the free world thinking he could make it bend to his menacing ways. But he badly miscalculated. \\n\\nHe thought he could roll into Ukraine and the world would roll over. Instead he met a wall of strength he never imagined. \\n\\nHe met the Ukrainian people. \\n\\nFrom President Zelenskyy to every Ukrainian, their fearlessness, their courage, their determination, inspires the world. \\n\\nGroups of citizens blocking tanks with their bodies. Everyone from students to retirees teachers turned soldiers defending their homeland.\n\nContent: And we won’t stop. \\n\\nWe have lost so much to COVID-19. Time with one another. And worst of all, so much loss of life. \\n\\nLet’s use this moment to reset. Let’s stop looking at COVID-19 as a partisan dividing line and see it for what it is: A God-awful disease. \\n\\nLet’s stop seeing each other as enemies, and start seeing each other for who we really are: Fellow Americans. \\n\\nWe can’t change how divided we’ve been. But we can change how we move forward—on COVID-19 and other issues we must face together. \\n\\nI recently visited the New York City Police Department days after the funerals of Officer Wilbert Mora and his partner, Officer Jason Rivera. \\n\\nThey were responding to a 9-1-1 call when a man shot and killed them with a stolen gun. \\n\\nOfficer Mora was 27 years old. \\n\\nOfficer Rivera was 22. \\n\\nBoth Dominican Americans who’d grown up on the same streets they later chose to patrol as police officers. \\n\\nI spoke with their families and told them that we are forever in debt for their sacrifice, and we will carry on their mission to restore the trust and safety every community deserves.\n\nContent: And a proud Ukrainian people, who have known 30 years of independence, have repeatedly shown that they will not tolerate anyone who tries to take their country backwards. \\n\\nTo all Americans, I will be honest with you, as I’ve always promised. A Russian dictator, invading a foreign country, has costs around the world. \\n\\nAnd I’m taking robust action to make sure the pain of our sanctions is targeted at Russia’s economy. And I will use every tool at our disposal to protect American businesses and consumers. \\n\\nTonight, I can announce that the United States has worked with 30 other countries to release 60 Million barrels of oil from reserves around the world. \\n\\nAmerica will lead that effort, releasing 30 Million barrels from our own Strategic Petroleum Reserve. And we stand ready to do more if necessary, unified with our allies. \\n\\nThese steps will help blunt gas prices here at home. And I know the news about what’s happening can seem alarming. \\n\\nBut I want you to know that we are going to be okay.\n\nContent: More support for patients and families. \\n\\nTo get there, I call on Congress to fund ARPA-H, the Advanced Research Projects Agency for Health. \\n\\nIt’s based on DARPA—the Defense Department project that led to the Internet, GPS, and so much more. \\n\\nARPA-H will have a singular purpose—to drive breakthroughs in cancer, Alzheimer’s, diabetes, and more. \\n\\nA unity agenda for the nation. \\n\\nWe can do this. \\n\\nMy fellow Americans—tonight , we have gathered in a sacred space—the citadel of our democracy. \\n\\nIn this Capitol, generation after generation, Americans have debated great questions amid great strife, and have done great things. \\n\\nWe have fought for freedom, expanded liberty, defeated totalitarianism and terror. \\n\\nAnd built the strongest, freest, and most prosperous nation the world has ever known. \\n\\nNow is the hour. \\n\\nOur moment of responsibility. \\n\\nOur test of resolve and conscience, of history itself. \\n\\nIt is in this moment that our character is formed. Our purpose is found. Our future is forged. \\n\\nWell I know this nation.\n=========\nFINAL ANSWER: The president did not mention Michael Jackson.\n\nQUESTION: {question}\n=========\n{summaries}\n=========\nFINAL ANSWER:`;\nexport const COMBINE_PROMPT =\n /*#__PURE__*/ PromptTemplate.fromTemplate(combine_prompt);\n\nconst system_combine_template = `Given the following extracted parts of a long document and a question, create a final answer. \nIf you don't know the answer, just say that you don't know. Don't try to make up an answer.\n----------------\n{summaries}`;\nconst combine_messages = [\n /*#__PURE__*/ SystemMessagePromptTemplate.fromTemplate(\n system_combine_template\n ),\n /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate(\"{question}\"),\n];\nconst CHAT_COMBINE_PROMPT =\n /*#__PURE__*/ ChatPromptTemplate.fromMessages(combine_messages);\n\nexport const COMBINE_PROMPT_SELECTOR =\n /*#__PURE__*/ new ConditionalPromptSelector(COMBINE_PROMPT, [\n [isChatModel, CHAT_COMBINE_PROMPT],\n ]);\n"],"mappings":";;;;AAWA,MAAM,cAAc,CAAC;;;;sBAIC,CAAC;AACvB,MAAa,4CAEX,eAAe,aAAa,YAAY;AAE1C,MAAM,kBAAkB,CAAC;;;SAGhB,CAAC;AACV,MAAM,WAAW,iBACD,4BAA4B,aAAa,gBAAgB,kBACzD,2BAA2B,aAAa,aAAa,AACpE;AACD,MAAM,iCAA+B,mBAAmB,aAAa,SAAS;AAE9E,MAAa,6CACG,IAAI,0BAA0B,2BAA2B,CACrE,CAAC,aAAa,cAAe,CAC9B;AAEH,MAAM,iBAAiB,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;aA6BX,CAAC;AACd,MAAa,iCACG,eAAe,aAAa,eAAe;AAE3D,MAAM,0BAA0B,CAAC;;;WAGtB,CAAC;AACZ,MAAM,mBAAmB,iBACT,4BAA4B,aACxC,wBACD,kBACa,2BAA2B,aAAa,aAAa,AACpE;AACD,MAAM,sCACU,mBAAmB,aAAa,iBAAiB;AAEjE,MAAa,0CACG,IAAI,0BAA0B,gBAAgB,CAC1D,CAAC,aAAa,mBAAoB,CACnC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"refine_prompts.cjs","names":["PromptTemplate","HumanMessagePromptTemplate","AIMessagePromptTemplate","ChatPromptTemplate","ConditionalPromptSelector","isChatModel","SystemMessagePromptTemplate"],"sources":["../../../src/chains/question_answering/refine_prompts.ts"],"sourcesContent":["
|
|
1
|
+
{"version":3,"file":"refine_prompts.cjs","names":["PromptTemplate","HumanMessagePromptTemplate","AIMessagePromptTemplate","ChatPromptTemplate","ConditionalPromptSelector","isChatModel","SystemMessagePromptTemplate"],"sources":["../../../src/chains/question_answering/refine_prompts.ts"],"sourcesContent":["import {\n PromptTemplate,\n ChatPromptTemplate,\n SystemMessagePromptTemplate,\n HumanMessagePromptTemplate,\n AIMessagePromptTemplate,\n} from \"@langchain/core/prompts\";\nimport {\n ConditionalPromptSelector,\n isChatModel,\n} from \"@langchain/core/example_selectors\";\n\nexport const DEFAULT_REFINE_PROMPT_TMPL = `The original question is as follows: {question}\nWe have provided an existing answer: {existing_answer}\nWe have the opportunity to refine the existing answer\n(only if needed) with some more context below.\n------------\n{context}\n------------\nGiven the new context, refine the original answer to better answer the question. \nIf the context isn't useful, return the original answer.`;\nexport const DEFAULT_REFINE_PROMPT = /*#__PURE__*/ new PromptTemplate({\n inputVariables: [\"question\", \"existing_answer\", \"context\"],\n template: DEFAULT_REFINE_PROMPT_TMPL,\n});\n\nconst refineTemplate = `The original question is as follows: {question}\nWe have provided an existing answer: {existing_answer}\nWe have the opportunity to refine the existing answer\n(only if needed) with some more context below.\n------------\n{context}\n------------\nGiven the new context, refine the original answer to better answer the question. \nIf the context isn't useful, return the original answer.`;\n\nconst messages = [\n /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate(\"{question}\"),\n /*#__PURE__*/ AIMessagePromptTemplate.fromTemplate(\"{existing_answer}\"),\n /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate(refineTemplate),\n];\n\nexport const CHAT_REFINE_PROMPT =\n /*#__PURE__*/ ChatPromptTemplate.fromMessages(messages);\n\nexport const REFINE_PROMPT_SELECTOR =\n /*#__PURE__*/ new ConditionalPromptSelector(DEFAULT_REFINE_PROMPT, [\n [isChatModel, CHAT_REFINE_PROMPT],\n ]);\n\nexport const DEFAULT_TEXT_QA_PROMPT_TMPL = `Context information is below. \n---------------------\n{context}\n---------------------\nGiven the context information and no prior knowledge, answer the question: {question}`;\nexport const DEFAULT_TEXT_QA_PROMPT = /*#__PURE__*/ new PromptTemplate({\n inputVariables: [\"context\", \"question\"],\n template: DEFAULT_TEXT_QA_PROMPT_TMPL,\n});\n\nconst chat_qa_prompt_template = `Context information is below. \n---------------------\n{context}\n---------------------\nGiven the context information and no prior knowledge, answer any questions`;\nconst chat_messages = [\n /*#__PURE__*/ SystemMessagePromptTemplate.fromTemplate(\n chat_qa_prompt_template\n ),\n /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate(\"{question}\"),\n];\nexport const CHAT_QUESTION_PROMPT =\n /*#__PURE__*/ ChatPromptTemplate.fromMessages(chat_messages);\nexport const QUESTION_PROMPT_SELECTOR =\n /*#__PURE__*/ new ConditionalPromptSelector(DEFAULT_TEXT_QA_PROMPT, [\n [isChatModel, CHAT_QUESTION_PROMPT],\n ]);\n"],"mappings":";;;;;AAYA,MAAa,6BAA6B,CAAC;;;;;;;;wDAQa,CAAC;AACzD,MAAa,wCAAsC,IAAIA,wCAAe;CACpE,gBAAgB;EAAC;EAAY;EAAmB;CAAU;CAC1D,UAAU;AACX;AAED,MAAM,iBAAiB,CAAC;;;;;;;;wDAQgC,CAAC;AAEzD,MAAM,WAAW;iBACDC,oDAA2B,aAAa,aAAa;iBACrDC,iDAAwB,aAAa,oBAAoB;iBACzDD,oDAA2B,aAAa,eAAe;AACtE;AAED,MAAa,qCACGE,4CAAmB,aAAa,SAAS;AAEzD,MAAa,yCACG,IAAIC,6DAA0B,uBAAuB,CACjE,CAACC,gDAAa,kBAAmB,CAClC;AAEH,MAAa,8BAA8B,CAAC;;;;qFAIyC,CAAC;AACtF,MAAa,yCAAuC,IAAIL,wCAAe;CACrE,gBAAgB,CAAC,WAAW,UAAW;CACvC,UAAU;AACX;AAED,MAAM,0BAA0B,CAAC;;;;0EAIyC,CAAC;AAC3E,MAAM,gBAAgB,iBACNM,qDAA4B,aACxC,wBACD,kBACaL,oDAA2B,aAAa,aAAa,AACpE;AACD,MAAa,uCACGE,4CAAmB,aAAa,cAAc;AAC9D,MAAa,2CACG,IAAIC,6DAA0B,wBAAwB,CAClE,CAACC,gDAAa,oBAAqB,CACpC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"refine_prompts.js","names":[],"sources":["../../../src/chains/question_answering/refine_prompts.ts"],"sourcesContent":["
|
|
1
|
+
{"version":3,"file":"refine_prompts.js","names":[],"sources":["../../../src/chains/question_answering/refine_prompts.ts"],"sourcesContent":["import {\n PromptTemplate,\n ChatPromptTemplate,\n SystemMessagePromptTemplate,\n HumanMessagePromptTemplate,\n AIMessagePromptTemplate,\n} from \"@langchain/core/prompts\";\nimport {\n ConditionalPromptSelector,\n isChatModel,\n} from \"@langchain/core/example_selectors\";\n\nexport const DEFAULT_REFINE_PROMPT_TMPL = `The original question is as follows: {question}\nWe have provided an existing answer: {existing_answer}\nWe have the opportunity to refine the existing answer\n(only if needed) with some more context below.\n------------\n{context}\n------------\nGiven the new context, refine the original answer to better answer the question. \nIf the context isn't useful, return the original answer.`;\nexport const DEFAULT_REFINE_PROMPT = /*#__PURE__*/ new PromptTemplate({\n inputVariables: [\"question\", \"existing_answer\", \"context\"],\n template: DEFAULT_REFINE_PROMPT_TMPL,\n});\n\nconst refineTemplate = `The original question is as follows: {question}\nWe have provided an existing answer: {existing_answer}\nWe have the opportunity to refine the existing answer\n(only if needed) with some more context below.\n------------\n{context}\n------------\nGiven the new context, refine the original answer to better answer the question. \nIf the context isn't useful, return the original answer.`;\n\nconst messages = [\n /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate(\"{question}\"),\n /*#__PURE__*/ AIMessagePromptTemplate.fromTemplate(\"{existing_answer}\"),\n /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate(refineTemplate),\n];\n\nexport const CHAT_REFINE_PROMPT =\n /*#__PURE__*/ ChatPromptTemplate.fromMessages(messages);\n\nexport const REFINE_PROMPT_SELECTOR =\n /*#__PURE__*/ new ConditionalPromptSelector(DEFAULT_REFINE_PROMPT, [\n [isChatModel, CHAT_REFINE_PROMPT],\n ]);\n\nexport const DEFAULT_TEXT_QA_PROMPT_TMPL = `Context information is below. \n---------------------\n{context}\n---------------------\nGiven the context information and no prior knowledge, answer the question: {question}`;\nexport const DEFAULT_TEXT_QA_PROMPT = /*#__PURE__*/ new PromptTemplate({\n inputVariables: [\"context\", \"question\"],\n template: DEFAULT_TEXT_QA_PROMPT_TMPL,\n});\n\nconst chat_qa_prompt_template = `Context information is below. \n---------------------\n{context}\n---------------------\nGiven the context information and no prior knowledge, answer any questions`;\nconst chat_messages = [\n /*#__PURE__*/ SystemMessagePromptTemplate.fromTemplate(\n chat_qa_prompt_template\n ),\n /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate(\"{question}\"),\n];\nexport const CHAT_QUESTION_PROMPT =\n /*#__PURE__*/ ChatPromptTemplate.fromMessages(chat_messages);\nexport const QUESTION_PROMPT_SELECTOR =\n /*#__PURE__*/ new ConditionalPromptSelector(DEFAULT_TEXT_QA_PROMPT, [\n [isChatModel, CHAT_QUESTION_PROMPT],\n ]);\n"],"mappings":";;;;AAYA,MAAa,6BAA6B,CAAC;;;;;;;;wDAQa,CAAC;AACzD,MAAa,wCAAsC,IAAI,eAAe;CACpE,gBAAgB;EAAC;EAAY;EAAmB;CAAU;CAC1D,UAAU;AACX;AAED,MAAM,iBAAiB,CAAC;;;;;;;;wDAQgC,CAAC;AAEzD,MAAM,WAAW;iBACD,2BAA2B,aAAa,aAAa;iBACrD,wBAAwB,aAAa,oBAAoB;iBACzD,2BAA2B,aAAa,eAAe;AACtE;AAED,MAAa,qCACG,mBAAmB,aAAa,SAAS;AAEzD,MAAa,yCACG,IAAI,0BAA0B,uBAAuB,CACjE,CAAC,aAAa,kBAAmB,CAClC;AAEH,MAAa,8BAA8B,CAAC;;;;qFAIyC,CAAC;AACtF,MAAa,yCAAuC,IAAI,eAAe;CACrE,gBAAgB,CAAC,WAAW,UAAW;CACvC,UAAU;AACX;AAED,MAAM,0BAA0B,CAAC;;;;0EAIyC,CAAC;AAC3E,MAAM,gBAAgB,iBACN,4BAA4B,aACxC,wBACD,kBACa,2BAA2B,aAAa,aAAa,AACpE;AACD,MAAa,uCACG,mBAAmB,aAAa,cAAc;AAC9D,MAAa,2CACG,IAAI,0BAA0B,wBAAwB,CAClE,CAAC,aAAa,oBAAqB,CACpC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"stuff_prompts.cjs","names":["PromptTemplate","SystemMessagePromptTemplate","HumanMessagePromptTemplate","ChatPromptTemplate","ConditionalPromptSelector","isChatModel"],"sources":["../../../src/chains/question_answering/stuff_prompts.ts"],"sourcesContent":["
|
|
1
|
+
{"version":3,"file":"stuff_prompts.cjs","names":["PromptTemplate","SystemMessagePromptTemplate","HumanMessagePromptTemplate","ChatPromptTemplate","ConditionalPromptSelector","isChatModel"],"sources":["../../../src/chains/question_answering/stuff_prompts.ts"],"sourcesContent":["import {\n ChatPromptTemplate,\n SystemMessagePromptTemplate,\n HumanMessagePromptTemplate,\n PromptTemplate,\n} from \"@langchain/core/prompts\";\nimport {\n ConditionalPromptSelector,\n isChatModel,\n} from \"@langchain/core/example_selectors\";\n\nexport const DEFAULT_QA_PROMPT = /*#__PURE__*/ new PromptTemplate({\n template:\n \"Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.\\n\\n{context}\\n\\nQuestion: {question}\\nHelpful Answer:\",\n inputVariables: [\"context\", \"question\"],\n});\n\nconst system_template = `Use the following pieces of context to answer the users question. \nIf you don't know the answer, just say that you don't know, don't try to make up an answer.\n----------------\n{context}`;\nconst messages = [\n /*#__PURE__*/ SystemMessagePromptTemplate.fromTemplate(system_template),\n /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate(\"{question}\"),\n];\nconst CHAT_PROMPT = /*#__PURE__*/ ChatPromptTemplate.fromMessages(messages);\n\nexport const QA_PROMPT_SELECTOR = /*#__PURE__*/ new ConditionalPromptSelector(\n DEFAULT_QA_PROMPT,\n [[isChatModel, CHAT_PROMPT]]\n);\n"],"mappings":";;;;;AAWA,MAAa,oCAAkC,IAAIA,wCAAe;CAChE,UACE;CACF,gBAAgB,CAAC,WAAW,UAAW;AACxC;AAED,MAAM,kBAAkB,CAAC;;;SAGhB,CAAC;AACV,MAAM,WAAW,iBACDC,qDAA4B,aAAa,gBAAgB,kBACzDC,oDAA2B,aAAa,aAAa,AACpE;AACD,MAAM,8BAA4BC,4CAAmB,aAAa,SAAS;AAE3E,MAAa,qCAAmC,IAAIC,6DAClD,mBACA,CAAC,CAACC,gDAAa,WAAY,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"stuff_prompts.js","names":[],"sources":["../../../src/chains/question_answering/stuff_prompts.ts"],"sourcesContent":["
|
|
1
|
+
{"version":3,"file":"stuff_prompts.js","names":[],"sources":["../../../src/chains/question_answering/stuff_prompts.ts"],"sourcesContent":["import {\n ChatPromptTemplate,\n SystemMessagePromptTemplate,\n HumanMessagePromptTemplate,\n PromptTemplate,\n} from \"@langchain/core/prompts\";\nimport {\n ConditionalPromptSelector,\n isChatModel,\n} from \"@langchain/core/example_selectors\";\n\nexport const DEFAULT_QA_PROMPT = /*#__PURE__*/ new PromptTemplate({\n template:\n \"Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.\\n\\n{context}\\n\\nQuestion: {question}\\nHelpful Answer:\",\n inputVariables: [\"context\", \"question\"],\n});\n\nconst system_template = `Use the following pieces of context to answer the users question. \nIf you don't know the answer, just say that you don't know, don't try to make up an answer.\n----------------\n{context}`;\nconst messages = [\n /*#__PURE__*/ SystemMessagePromptTemplate.fromTemplate(system_template),\n /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate(\"{question}\"),\n];\nconst CHAT_PROMPT = /*#__PURE__*/ ChatPromptTemplate.fromMessages(messages);\n\nexport const QA_PROMPT_SELECTOR = /*#__PURE__*/ new ConditionalPromptSelector(\n DEFAULT_QA_PROMPT,\n [[isChatModel, CHAT_PROMPT]]\n);\n"],"mappings":";;;;AAWA,MAAa,oCAAkC,IAAI,eAAe;CAChE,UACE;CACF,gBAAgB,CAAC,WAAW,UAAW;AACxC;AAED,MAAM,kBAAkB,CAAC;;;SAGhB,CAAC;AACV,MAAM,WAAW,iBACD,4BAA4B,aAAa,gBAAgB,kBACzD,2BAA2B,aAAa,aAAa,AACpE;AACD,MAAM,8BAA4B,mBAAmB,aAAa,SAAS;AAE3E,MAAa,qCAAmC,IAAI,0BAClD,mBACA,CAAC,CAAC,aAAa,WAAY,CAAC"}
|
|
@@ -6,7 +6,7 @@ const require_llm_router = require('./llm_router.cjs');
|
|
|
6
6
|
const require_multi_prompt_prompt = require('./multi_prompt_prompt.cjs');
|
|
7
7
|
const require_utils = require('./utils.cjs');
|
|
8
8
|
const require_router = require('../../output_parsers/router.cjs');
|
|
9
|
-
const
|
|
9
|
+
const zod_v3 = require_rolldown_runtime.__toESM(require("zod/v3"));
|
|
10
10
|
const __langchain_core_prompts = require_rolldown_runtime.__toESM(require("@langchain/core/prompts"));
|
|
11
11
|
|
|
12
12
|
//#region src/chains/router/multi_prompt.ts
|
|
@@ -66,9 +66,9 @@ var MultiPromptChain = class MultiPromptChain extends require_multi_route.MultiR
|
|
|
66
66
|
*/
|
|
67
67
|
static fromLLMAndPrompts(llm, { promptNames, promptDescriptions, promptTemplates, defaultChain, llmChainOpts, conversationChainOpts, multiRouteChainOpts }) {
|
|
68
68
|
const destinations = require_utils.zipEntries(promptNames, promptDescriptions).map(([name, desc]) => `${name}: ${desc}`);
|
|
69
|
-
const structuredOutputParserSchema =
|
|
70
|
-
destination:
|
|
71
|
-
next_inputs:
|
|
69
|
+
const structuredOutputParserSchema = zod_v3.z.object({
|
|
70
|
+
destination: zod_v3.z.string().optional().describe("name of the question answering system to use or \"DEFAULT\""),
|
|
71
|
+
next_inputs: zod_v3.z.object({ input: zod_v3.z.string().describe("a potentially modified version of the original input") }).describe("input to be fed to the next model")
|
|
72
72
|
});
|
|
73
73
|
const outputParser = new require_router.RouterOutputParser(structuredOutputParserSchema);
|
|
74
74
|
const destinationsStr = destinations.join("\n");
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"multi_prompt.cjs","names":["MultiRouteChain","llm: BaseLanguageModelInterface","promptNames: string[]","promptDescriptions: string[]","promptTemplates: string[] | PromptTemplate[]","defaultChain?: BaseChain","options?: Omit<MultiRouteChainInput, \"defaultChain\">","zipEntries","z","RouterOutputParser","STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE","PromptTemplate","LLMRouterChain","myPrompt: string | PromptTemplate","LLMChain","ConversationChain"],"sources":["../../../src/chains/router/multi_prompt.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { z } from \"zod\";\nimport { interpolateFString, PromptTemplate } from \"@langchain/core/prompts\";\nimport { MultiRouteChain, MultiRouteChainInput } from \"./multi_route.js\";\nimport { STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE } from \"./multi_prompt_prompt.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMChain, LLMChainInput } from \"../../chains/llm_chain.js\";\nimport { LLMRouterChain } from \"./llm_router.js\";\nimport { ConversationChain } from \"../../chains/conversation.js\";\nimport { zipEntries } from \"./utils.js\";\nimport { RouterOutputParser } from \"../../output_parsers/router.js\";\n\n/**\n * A class that represents a multi-prompt chain in the LangChain\n * framework. It extends the MultiRouteChain class and provides additional\n * functionality specific to multi-prompt chains.\n * @example\n * ```typescript\n * const multiPromptChain = MultiPromptChain.fromLLMAndPrompts(\n * new ChatOpenAI({ model: \"gpt-4o-mini\" }),\n * {\n * promptNames: [\"physics\", \"math\", \"history\"],\n * promptDescriptions: [\n * \"Good for answering questions about physics\",\n * \"Good for answering math questions\",\n * \"Good for answering questions about history\",\n * ],\n * promptTemplates: [\n * `You are a very smart physics professor. Here is a question:\\n{input}\\n`,\n * `You are a very good mathematician. Here is a question:\\n{input}\\n`,\n * `You are a very smart history professor. Here is a question:\\n{input}\\n`,\n * ],\n * }\n * );\n * const result = await multiPromptChain.call({\n * input: \"What is the speed of light?\",\n * });\n * ```\n */\nexport class MultiPromptChain extends MultiRouteChain {\n /**\n * @deprecated Use `fromLLMAndPrompts` instead\n */\n static fromPrompts(\n llm: BaseLanguageModelInterface,\n promptNames: string[],\n promptDescriptions: string[],\n promptTemplates: string[] | PromptTemplate[],\n defaultChain?: BaseChain,\n options?: Omit<MultiRouteChainInput, \"defaultChain\">\n ) {\n return MultiPromptChain.fromLLMAndPrompts(llm, {\n promptNames,\n promptDescriptions,\n promptTemplates,\n defaultChain,\n multiRouteChainOpts: options,\n });\n }\n\n /**\n * A static method that creates an instance of MultiPromptChain from a\n * BaseLanguageModel and a set of prompts. It takes in optional parameters\n * for the default chain and additional options.\n * @param llm A BaseLanguageModel instance.\n * @param promptNames An array of prompt names.\n * @param promptDescriptions An array of prompt descriptions.\n * @param promptTemplates An array of prompt templates.\n * @param defaultChain An optional BaseChain instance to be used as the default chain.\n * @param llmChainOpts Optional parameters for the LLMChainInput, excluding 'llm' and 'prompt'.\n * @param conversationChainOpts Optional parameters for the LLMChainInput, excluding 'llm' and 'outputKey'.\n * @param multiRouteChainOpts Optional parameters for the MultiRouteChainInput, excluding 'defaultChain'.\n * @returns An instance of MultiPromptChain.\n */\n static fromLLMAndPrompts(\n llm: BaseLanguageModelInterface,\n {\n promptNames,\n promptDescriptions,\n promptTemplates,\n defaultChain,\n llmChainOpts,\n conversationChainOpts,\n multiRouteChainOpts,\n }: {\n promptNames: string[];\n promptDescriptions: string[];\n promptTemplates: string[] | PromptTemplate[];\n defaultChain?: BaseChain;\n llmChainOpts?: Omit<LLMChainInput, \"llm\" | \"prompt\">;\n conversationChainOpts?: Omit<LLMChainInput, \"llm\" | \"outputKey\">;\n multiRouteChainOpts?: Omit<MultiRouteChainInput, \"defaultChain\">;\n }\n ): MultiPromptChain {\n const destinations = zipEntries(promptNames, promptDescriptions).map(\n ([name, desc]) => `${name}: ${desc}`\n );\n\n const structuredOutputParserSchema = z.object({\n destination: z\n .string()\n .optional()\n .describe('name of the question answering system to use or \"DEFAULT\"'),\n next_inputs: z\n .object({\n input: z\n .string()\n .describe(\"a potentially modified version of the original input\"),\n })\n .describe(\"input to be fed to the next model\"),\n });\n\n const outputParser = new RouterOutputParser(structuredOutputParserSchema);\n\n const destinationsStr = destinations.join(\"\\n\");\n const routerTemplate = interpolateFString(\n STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE(\n outputParser.getFormatInstructions({ interpolationDepth: 4 })\n ),\n {\n destinations: destinationsStr,\n }\n );\n\n const routerPrompt = new PromptTemplate({\n template: routerTemplate,\n inputVariables: [\"input\"],\n outputParser,\n });\n\n const routerChain = LLMRouterChain.fromLLM(llm, routerPrompt);\n const destinationChains = zipEntries<[string, string | PromptTemplate]>(\n promptNames,\n promptTemplates\n ).reduce((acc, [name, template]) => {\n let myPrompt: string | PromptTemplate;\n if (typeof template === \"object\") {\n myPrompt = template;\n } else if (typeof template === \"string\") {\n myPrompt = new PromptTemplate({\n template: template as string,\n inputVariables: [\"input\"],\n });\n } else {\n throw new Error(\"Invalid prompt template\");\n }\n acc[name as string] = new LLMChain({\n ...llmChainOpts,\n llm,\n prompt: myPrompt,\n });\n return acc;\n }, {} as { [name: string]: LLMChain });\n\n const convChain = new ConversationChain({\n ...conversationChainOpts,\n llm,\n outputKey: \"text\",\n });\n\n return new MultiPromptChain({\n ...multiRouteChainOpts,\n routerChain,\n destinationChains,\n defaultChain: defaultChain ?? convChain,\n });\n }\n\n _chainType(): string {\n return \"multi_prompt_chain\";\n }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAuCA,IAAa,mBAAb,MAAa,yBAAyBA,oCAAgB;;;;CAIpD,OAAO,YACLC,KACAC,aACAC,oBACAC,iBACAC,cACAC,SACA;AACA,SAAO,iBAAiB,kBAAkB,KAAK;GAC7C;GACA;GACA;GACA;GACA,qBAAqB;EACtB,EAAC;CACH;;;;;;;;;;;;;;;CAgBD,OAAO,kBACLL,KACA,EACE,aACA,oBACA,iBACA,cACA,cACA,uBACA,qBASD,EACiB;EAClB,MAAM,eAAeM,yBAAW,aAAa,mBAAmB,CAAC,IAC/D,CAAC,CAAC,MAAM,KAAK,KAAK,GAAG,KAAK,EAAE,EAAE,MAAM,CACrC;EAED,MAAM,+BAA+BC,
|
|
1
|
+
{"version":3,"file":"multi_prompt.cjs","names":["MultiRouteChain","llm: BaseLanguageModelInterface","promptNames: string[]","promptDescriptions: string[]","promptTemplates: string[] | PromptTemplate[]","defaultChain?: BaseChain","options?: Omit<MultiRouteChainInput, \"defaultChain\">","zipEntries","z","RouterOutputParser","STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE","PromptTemplate","LLMRouterChain","myPrompt: string | PromptTemplate","LLMChain","ConversationChain"],"sources":["../../../src/chains/router/multi_prompt.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { z } from \"zod/v3\";\nimport { interpolateFString, PromptTemplate } from \"@langchain/core/prompts\";\nimport { MultiRouteChain, MultiRouteChainInput } from \"./multi_route.js\";\nimport { STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE } from \"./multi_prompt_prompt.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMChain, LLMChainInput } from \"../../chains/llm_chain.js\";\nimport { LLMRouterChain } from \"./llm_router.js\";\nimport { ConversationChain } from \"../../chains/conversation.js\";\nimport { zipEntries } from \"./utils.js\";\nimport { RouterOutputParser } from \"../../output_parsers/router.js\";\n\n/**\n * A class that represents a multi-prompt chain in the LangChain\n * framework. It extends the MultiRouteChain class and provides additional\n * functionality specific to multi-prompt chains.\n * @example\n * ```typescript\n * const multiPromptChain = MultiPromptChain.fromLLMAndPrompts(\n * new ChatOpenAI({ model: \"gpt-4o-mini\" }),\n * {\n * promptNames: [\"physics\", \"math\", \"history\"],\n * promptDescriptions: [\n * \"Good for answering questions about physics\",\n * \"Good for answering math questions\",\n * \"Good for answering questions about history\",\n * ],\n * promptTemplates: [\n * `You are a very smart physics professor. Here is a question:\\n{input}\\n`,\n * `You are a very good mathematician. Here is a question:\\n{input}\\n`,\n * `You are a very smart history professor. Here is a question:\\n{input}\\n`,\n * ],\n * }\n * );\n * const result = await multiPromptChain.call({\n * input: \"What is the speed of light?\",\n * });\n * ```\n */\nexport class MultiPromptChain extends MultiRouteChain {\n /**\n * @deprecated Use `fromLLMAndPrompts` instead\n */\n static fromPrompts(\n llm: BaseLanguageModelInterface,\n promptNames: string[],\n promptDescriptions: string[],\n promptTemplates: string[] | PromptTemplate[],\n defaultChain?: BaseChain,\n options?: Omit<MultiRouteChainInput, \"defaultChain\">\n ) {\n return MultiPromptChain.fromLLMAndPrompts(llm, {\n promptNames,\n promptDescriptions,\n promptTemplates,\n defaultChain,\n multiRouteChainOpts: options,\n });\n }\n\n /**\n * A static method that creates an instance of MultiPromptChain from a\n * BaseLanguageModel and a set of prompts. It takes in optional parameters\n * for the default chain and additional options.\n * @param llm A BaseLanguageModel instance.\n * @param promptNames An array of prompt names.\n * @param promptDescriptions An array of prompt descriptions.\n * @param promptTemplates An array of prompt templates.\n * @param defaultChain An optional BaseChain instance to be used as the default chain.\n * @param llmChainOpts Optional parameters for the LLMChainInput, excluding 'llm' and 'prompt'.\n * @param conversationChainOpts Optional parameters for the LLMChainInput, excluding 'llm' and 'outputKey'.\n * @param multiRouteChainOpts Optional parameters for the MultiRouteChainInput, excluding 'defaultChain'.\n * @returns An instance of MultiPromptChain.\n */\n static fromLLMAndPrompts(\n llm: BaseLanguageModelInterface,\n {\n promptNames,\n promptDescriptions,\n promptTemplates,\n defaultChain,\n llmChainOpts,\n conversationChainOpts,\n multiRouteChainOpts,\n }: {\n promptNames: string[];\n promptDescriptions: string[];\n promptTemplates: string[] | PromptTemplate[];\n defaultChain?: BaseChain;\n llmChainOpts?: Omit<LLMChainInput, \"llm\" | \"prompt\">;\n conversationChainOpts?: Omit<LLMChainInput, \"llm\" | \"outputKey\">;\n multiRouteChainOpts?: Omit<MultiRouteChainInput, \"defaultChain\">;\n }\n ): MultiPromptChain {\n const destinations = zipEntries(promptNames, promptDescriptions).map(\n ([name, desc]) => `${name}: ${desc}`\n );\n\n const structuredOutputParserSchema = z.object({\n destination: z\n .string()\n .optional()\n .describe('name of the question answering system to use or \"DEFAULT\"'),\n next_inputs: z\n .object({\n input: z\n .string()\n .describe(\"a potentially modified version of the original input\"),\n })\n .describe(\"input to be fed to the next model\"),\n });\n\n const outputParser = new RouterOutputParser(structuredOutputParserSchema);\n\n const destinationsStr = destinations.join(\"\\n\");\n const routerTemplate = interpolateFString(\n STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE(\n outputParser.getFormatInstructions({ interpolationDepth: 4 })\n ),\n {\n destinations: destinationsStr,\n }\n );\n\n const routerPrompt = new PromptTemplate({\n template: routerTemplate,\n inputVariables: [\"input\"],\n outputParser,\n });\n\n const routerChain = LLMRouterChain.fromLLM(llm, routerPrompt);\n const destinationChains = zipEntries<[string, string | PromptTemplate]>(\n promptNames,\n promptTemplates\n ).reduce((acc, [name, template]) => {\n let myPrompt: string | PromptTemplate;\n if (typeof template === \"object\") {\n myPrompt = template;\n } else if (typeof template === \"string\") {\n myPrompt = new PromptTemplate({\n template: template as string,\n inputVariables: [\"input\"],\n });\n } else {\n throw new Error(\"Invalid prompt template\");\n }\n acc[name as string] = new LLMChain({\n ...llmChainOpts,\n llm,\n prompt: myPrompt,\n });\n return acc;\n }, {} as { [name: string]: LLMChain });\n\n const convChain = new ConversationChain({\n ...conversationChainOpts,\n llm,\n outputKey: \"text\",\n });\n\n return new MultiPromptChain({\n ...multiRouteChainOpts,\n routerChain,\n destinationChains,\n defaultChain: defaultChain ?? convChain,\n });\n }\n\n _chainType(): string {\n return \"multi_prompt_chain\";\n }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAuCA,IAAa,mBAAb,MAAa,yBAAyBA,oCAAgB;;;;CAIpD,OAAO,YACLC,KACAC,aACAC,oBACAC,iBACAC,cACAC,SACA;AACA,SAAO,iBAAiB,kBAAkB,KAAK;GAC7C;GACA;GACA;GACA;GACA,qBAAqB;EACtB,EAAC;CACH;;;;;;;;;;;;;;;CAgBD,OAAO,kBACLL,KACA,EACE,aACA,oBACA,iBACA,cACA,cACA,uBACA,qBASD,EACiB;EAClB,MAAM,eAAeM,yBAAW,aAAa,mBAAmB,CAAC,IAC/D,CAAC,CAAC,MAAM,KAAK,KAAK,GAAG,KAAK,EAAE,EAAE,MAAM,CACrC;EAED,MAAM,+BAA+BC,SAAE,OAAO;GAC5C,aAAaA,SACV,QAAQ,CACR,UAAU,CACV,SAAS,8DAA4D;GACxE,aAAaA,SACV,OAAO,EACN,OAAOA,SACJ,QAAQ,CACR,SAAS,uDAAuD,CACpE,EAAC,CACD,SAAS,oCAAoC;EACjD,EAAC;EAEF,MAAM,eAAe,IAAIC,kCAAmB;EAE5C,MAAM,kBAAkB,aAAa,KAAK,KAAK;EAC/C,MAAM,kEACJC,oEACE,aAAa,sBAAsB,EAAE,oBAAoB,EAAG,EAAC,CAC9D,EACD,EACE,cAAc,gBACf,EACF;EAED,MAAM,eAAe,IAAIC,wCAAe;GACtC,UAAU;GACV,gBAAgB,CAAC,OAAQ;GACzB;EACD;EAED,MAAM,cAAcC,kCAAe,QAAQ,KAAK,aAAa;EAC7D,MAAM,oBAAoBL,yBACxB,aACA,gBACD,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,SAAS,KAAK;GAClC,IAAIM;AACJ,OAAI,OAAO,aAAa,UACtB,WAAW;YACF,OAAO,aAAa,UAC7B,WAAW,IAAIF,wCAAe;IAClB;IACV,gBAAgB,CAAC,OAAQ;GAC1B;OAED,OAAM,IAAI,MAAM;GAElB,IAAI,QAAkB,IAAIG,2BAAS;IACjC,GAAG;IACH;IACA,QAAQ;GACT;AACD,UAAO;EACR,GAAE,CAAE,EAAiC;EAEtC,MAAM,YAAY,IAAIC,uCAAkB;GACtC,GAAG;GACH;GACA,WAAW;EACZ;AAED,SAAO,IAAI,iBAAiB;GAC1B,GAAG;GACH;GACA;GACA,cAAc,gBAAgB;EAC/B;CACF;CAED,aAAqB;AACnB,SAAO;CACR;AACF"}
|
|
@@ -5,7 +5,7 @@ import { LLMRouterChain } from "./llm_router.js";
|
|
|
5
5
|
import { STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE } from "./multi_prompt_prompt.js";
|
|
6
6
|
import { zipEntries } from "./utils.js";
|
|
7
7
|
import { RouterOutputParser } from "../../output_parsers/router.js";
|
|
8
|
-
import { z } from "zod";
|
|
8
|
+
import { z } from "zod/v3";
|
|
9
9
|
import { PromptTemplate, interpolateFString } from "@langchain/core/prompts";
|
|
10
10
|
|
|
11
11
|
//#region src/chains/router/multi_prompt.ts
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"multi_prompt.js","names":["llm: BaseLanguageModelInterface","promptNames: string[]","promptDescriptions: string[]","promptTemplates: string[] | PromptTemplate[]","defaultChain?: BaseChain","options?: Omit<MultiRouteChainInput, \"defaultChain\">","myPrompt: string | PromptTemplate"],"sources":["../../../src/chains/router/multi_prompt.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { z } from \"zod\";\nimport { interpolateFString, PromptTemplate } from \"@langchain/core/prompts\";\nimport { MultiRouteChain, MultiRouteChainInput } from \"./multi_route.js\";\nimport { STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE } from \"./multi_prompt_prompt.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMChain, LLMChainInput } from \"../../chains/llm_chain.js\";\nimport { LLMRouterChain } from \"./llm_router.js\";\nimport { ConversationChain } from \"../../chains/conversation.js\";\nimport { zipEntries } from \"./utils.js\";\nimport { RouterOutputParser } from \"../../output_parsers/router.js\";\n\n/**\n * A class that represents a multi-prompt chain in the LangChain\n * framework. It extends the MultiRouteChain class and provides additional\n * functionality specific to multi-prompt chains.\n * @example\n * ```typescript\n * const multiPromptChain = MultiPromptChain.fromLLMAndPrompts(\n * new ChatOpenAI({ model: \"gpt-4o-mini\" }),\n * {\n * promptNames: [\"physics\", \"math\", \"history\"],\n * promptDescriptions: [\n * \"Good for answering questions about physics\",\n * \"Good for answering math questions\",\n * \"Good for answering questions about history\",\n * ],\n * promptTemplates: [\n * `You are a very smart physics professor. Here is a question:\\n{input}\\n`,\n * `You are a very good mathematician. Here is a question:\\n{input}\\n`,\n * `You are a very smart history professor. Here is a question:\\n{input}\\n`,\n * ],\n * }\n * );\n * const result = await multiPromptChain.call({\n * input: \"What is the speed of light?\",\n * });\n * ```\n */\nexport class MultiPromptChain extends MultiRouteChain {\n /**\n * @deprecated Use `fromLLMAndPrompts` instead\n */\n static fromPrompts(\n llm: BaseLanguageModelInterface,\n promptNames: string[],\n promptDescriptions: string[],\n promptTemplates: string[] | PromptTemplate[],\n defaultChain?: BaseChain,\n options?: Omit<MultiRouteChainInput, \"defaultChain\">\n ) {\n return MultiPromptChain.fromLLMAndPrompts(llm, {\n promptNames,\n promptDescriptions,\n promptTemplates,\n defaultChain,\n multiRouteChainOpts: options,\n });\n }\n\n /**\n * A static method that creates an instance of MultiPromptChain from a\n * BaseLanguageModel and a set of prompts. It takes in optional parameters\n * for the default chain and additional options.\n * @param llm A BaseLanguageModel instance.\n * @param promptNames An array of prompt names.\n * @param promptDescriptions An array of prompt descriptions.\n * @param promptTemplates An array of prompt templates.\n * @param defaultChain An optional BaseChain instance to be used as the default chain.\n * @param llmChainOpts Optional parameters for the LLMChainInput, excluding 'llm' and 'prompt'.\n * @param conversationChainOpts Optional parameters for the LLMChainInput, excluding 'llm' and 'outputKey'.\n * @param multiRouteChainOpts Optional parameters for the MultiRouteChainInput, excluding 'defaultChain'.\n * @returns An instance of MultiPromptChain.\n */\n static fromLLMAndPrompts(\n llm: BaseLanguageModelInterface,\n {\n promptNames,\n promptDescriptions,\n promptTemplates,\n defaultChain,\n llmChainOpts,\n conversationChainOpts,\n multiRouteChainOpts,\n }: {\n promptNames: string[];\n promptDescriptions: string[];\n promptTemplates: string[] | PromptTemplate[];\n defaultChain?: BaseChain;\n llmChainOpts?: Omit<LLMChainInput, \"llm\" | \"prompt\">;\n conversationChainOpts?: Omit<LLMChainInput, \"llm\" | \"outputKey\">;\n multiRouteChainOpts?: Omit<MultiRouteChainInput, \"defaultChain\">;\n }\n ): MultiPromptChain {\n const destinations = zipEntries(promptNames, promptDescriptions).map(\n ([name, desc]) => `${name}: ${desc}`\n );\n\n const structuredOutputParserSchema = z.object({\n destination: z\n .string()\n .optional()\n .describe('name of the question answering system to use or \"DEFAULT\"'),\n next_inputs: z\n .object({\n input: z\n .string()\n .describe(\"a potentially modified version of the original input\"),\n })\n .describe(\"input to be fed to the next model\"),\n });\n\n const outputParser = new RouterOutputParser(structuredOutputParserSchema);\n\n const destinationsStr = destinations.join(\"\\n\");\n const routerTemplate = interpolateFString(\n STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE(\n outputParser.getFormatInstructions({ interpolationDepth: 4 })\n ),\n {\n destinations: destinationsStr,\n }\n );\n\n const routerPrompt = new PromptTemplate({\n template: routerTemplate,\n inputVariables: [\"input\"],\n outputParser,\n });\n\n const routerChain = LLMRouterChain.fromLLM(llm, routerPrompt);\n const destinationChains = zipEntries<[string, string | PromptTemplate]>(\n promptNames,\n promptTemplates\n ).reduce((acc, [name, template]) => {\n let myPrompt: string | PromptTemplate;\n if (typeof template === \"object\") {\n myPrompt = template;\n } else if (typeof template === \"string\") {\n myPrompt = new PromptTemplate({\n template: template as string,\n inputVariables: [\"input\"],\n });\n } else {\n throw new Error(\"Invalid prompt template\");\n }\n acc[name as string] = new LLMChain({\n ...llmChainOpts,\n llm,\n prompt: myPrompt,\n });\n return acc;\n }, {} as { [name: string]: LLMChain });\n\n const convChain = new ConversationChain({\n ...conversationChainOpts,\n llm,\n outputKey: \"text\",\n });\n\n return new MultiPromptChain({\n ...multiRouteChainOpts,\n routerChain,\n destinationChains,\n defaultChain: defaultChain ?? convChain,\n });\n }\n\n _chainType(): string {\n return \"multi_prompt_chain\";\n }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAuCA,IAAa,mBAAb,MAAa,yBAAyB,gBAAgB;;;;CAIpD,OAAO,YACLA,KACAC,aACAC,oBACAC,iBACAC,cACAC,SACA;AACA,SAAO,iBAAiB,kBAAkB,KAAK;GAC7C;GACA;GACA;GACA;GACA,qBAAqB;EACtB,EAAC;CACH;;;;;;;;;;;;;;;CAgBD,OAAO,kBACLL,KACA,EACE,aACA,oBACA,iBACA,cACA,cACA,uBACA,qBASD,EACiB;EAClB,MAAM,eAAe,WAAW,aAAa,mBAAmB,CAAC,IAC/D,CAAC,CAAC,MAAM,KAAK,KAAK,GAAG,KAAK,EAAE,EAAE,MAAM,CACrC;EAED,MAAM,+BAA+B,EAAE,OAAO;GAC5C,aAAa,EACV,QAAQ,CACR,UAAU,CACV,SAAS,8DAA4D;GACxE,aAAa,EACV,OAAO,EACN,OAAO,EACJ,QAAQ,CACR,SAAS,uDAAuD,CACpE,EAAC,CACD,SAAS,oCAAoC;EACjD,EAAC;EAEF,MAAM,eAAe,IAAI,mBAAmB;EAE5C,MAAM,kBAAkB,aAAa,KAAK,KAAK;EAC/C,MAAM,iBAAiB,mBACrB,wCACE,aAAa,sBAAsB,EAAE,oBAAoB,EAAG,EAAC,CAC9D,EACD,EACE,cAAc,gBACf,EACF;EAED,MAAM,eAAe,IAAI,eAAe;GACtC,UAAU;GACV,gBAAgB,CAAC,OAAQ;GACzB;EACD;EAED,MAAM,cAAc,eAAe,QAAQ,KAAK,aAAa;EAC7D,MAAM,oBAAoB,WACxB,aACA,gBACD,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,SAAS,KAAK;GAClC,IAAIM;AACJ,OAAI,OAAO,aAAa,UACtB,WAAW;YACF,OAAO,aAAa,UAC7B,WAAW,IAAI,eAAe;IAClB;IACV,gBAAgB,CAAC,OAAQ;GAC1B;OAED,OAAM,IAAI,MAAM;GAElB,IAAI,QAAkB,IAAI,SAAS;IACjC,GAAG;IACH;IACA,QAAQ;GACT;AACD,UAAO;EACR,GAAE,CAAE,EAAiC;EAEtC,MAAM,YAAY,IAAI,kBAAkB;GACtC,GAAG;GACH;GACA,WAAW;EACZ;AAED,SAAO,IAAI,iBAAiB;GAC1B,GAAG;GACH;GACA;GACA,cAAc,gBAAgB;EAC/B;CACF;CAED,aAAqB;AACnB,SAAO;CACR;AACF"}
|
|
1
|
+
{"version":3,"file":"multi_prompt.js","names":["llm: BaseLanguageModelInterface","promptNames: string[]","promptDescriptions: string[]","promptTemplates: string[] | PromptTemplate[]","defaultChain?: BaseChain","options?: Omit<MultiRouteChainInput, \"defaultChain\">","myPrompt: string | PromptTemplate"],"sources":["../../../src/chains/router/multi_prompt.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { z } from \"zod/v3\";\nimport { interpolateFString, PromptTemplate } from \"@langchain/core/prompts\";\nimport { MultiRouteChain, MultiRouteChainInput } from \"./multi_route.js\";\nimport { STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE } from \"./multi_prompt_prompt.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMChain, LLMChainInput } from \"../../chains/llm_chain.js\";\nimport { LLMRouterChain } from \"./llm_router.js\";\nimport { ConversationChain } from \"../../chains/conversation.js\";\nimport { zipEntries } from \"./utils.js\";\nimport { RouterOutputParser } from \"../../output_parsers/router.js\";\n\n/**\n * A class that represents a multi-prompt chain in the LangChain\n * framework. It extends the MultiRouteChain class and provides additional\n * functionality specific to multi-prompt chains.\n * @example\n * ```typescript\n * const multiPromptChain = MultiPromptChain.fromLLMAndPrompts(\n * new ChatOpenAI({ model: \"gpt-4o-mini\" }),\n * {\n * promptNames: [\"physics\", \"math\", \"history\"],\n * promptDescriptions: [\n * \"Good for answering questions about physics\",\n * \"Good for answering math questions\",\n * \"Good for answering questions about history\",\n * ],\n * promptTemplates: [\n * `You are a very smart physics professor. Here is a question:\\n{input}\\n`,\n * `You are a very good mathematician. Here is a question:\\n{input}\\n`,\n * `You are a very smart history professor. Here is a question:\\n{input}\\n`,\n * ],\n * }\n * );\n * const result = await multiPromptChain.call({\n * input: \"What is the speed of light?\",\n * });\n * ```\n */\nexport class MultiPromptChain extends MultiRouteChain {\n /**\n * @deprecated Use `fromLLMAndPrompts` instead\n */\n static fromPrompts(\n llm: BaseLanguageModelInterface,\n promptNames: string[],\n promptDescriptions: string[],\n promptTemplates: string[] | PromptTemplate[],\n defaultChain?: BaseChain,\n options?: Omit<MultiRouteChainInput, \"defaultChain\">\n ) {\n return MultiPromptChain.fromLLMAndPrompts(llm, {\n promptNames,\n promptDescriptions,\n promptTemplates,\n defaultChain,\n multiRouteChainOpts: options,\n });\n }\n\n /**\n * A static method that creates an instance of MultiPromptChain from a\n * BaseLanguageModel and a set of prompts. It takes in optional parameters\n * for the default chain and additional options.\n * @param llm A BaseLanguageModel instance.\n * @param promptNames An array of prompt names.\n * @param promptDescriptions An array of prompt descriptions.\n * @param promptTemplates An array of prompt templates.\n * @param defaultChain An optional BaseChain instance to be used as the default chain.\n * @param llmChainOpts Optional parameters for the LLMChainInput, excluding 'llm' and 'prompt'.\n * @param conversationChainOpts Optional parameters for the LLMChainInput, excluding 'llm' and 'outputKey'.\n * @param multiRouteChainOpts Optional parameters for the MultiRouteChainInput, excluding 'defaultChain'.\n * @returns An instance of MultiPromptChain.\n */\n static fromLLMAndPrompts(\n llm: BaseLanguageModelInterface,\n {\n promptNames,\n promptDescriptions,\n promptTemplates,\n defaultChain,\n llmChainOpts,\n conversationChainOpts,\n multiRouteChainOpts,\n }: {\n promptNames: string[];\n promptDescriptions: string[];\n promptTemplates: string[] | PromptTemplate[];\n defaultChain?: BaseChain;\n llmChainOpts?: Omit<LLMChainInput, \"llm\" | \"prompt\">;\n conversationChainOpts?: Omit<LLMChainInput, \"llm\" | \"outputKey\">;\n multiRouteChainOpts?: Omit<MultiRouteChainInput, \"defaultChain\">;\n }\n ): MultiPromptChain {\n const destinations = zipEntries(promptNames, promptDescriptions).map(\n ([name, desc]) => `${name}: ${desc}`\n );\n\n const structuredOutputParserSchema = z.object({\n destination: z\n .string()\n .optional()\n .describe('name of the question answering system to use or \"DEFAULT\"'),\n next_inputs: z\n .object({\n input: z\n .string()\n .describe(\"a potentially modified version of the original input\"),\n })\n .describe(\"input to be fed to the next model\"),\n });\n\n const outputParser = new RouterOutputParser(structuredOutputParserSchema);\n\n const destinationsStr = destinations.join(\"\\n\");\n const routerTemplate = interpolateFString(\n STRUCTURED_MULTI_PROMPT_ROUTER_TEMPLATE(\n outputParser.getFormatInstructions({ interpolationDepth: 4 })\n ),\n {\n destinations: destinationsStr,\n }\n );\n\n const routerPrompt = new PromptTemplate({\n template: routerTemplate,\n inputVariables: [\"input\"],\n outputParser,\n });\n\n const routerChain = LLMRouterChain.fromLLM(llm, routerPrompt);\n const destinationChains = zipEntries<[string, string | PromptTemplate]>(\n promptNames,\n promptTemplates\n ).reduce((acc, [name, template]) => {\n let myPrompt: string | PromptTemplate;\n if (typeof template === \"object\") {\n myPrompt = template;\n } else if (typeof template === \"string\") {\n myPrompt = new PromptTemplate({\n template: template as string,\n inputVariables: [\"input\"],\n });\n } else {\n throw new Error(\"Invalid prompt template\");\n }\n acc[name as string] = new LLMChain({\n ...llmChainOpts,\n llm,\n prompt: myPrompt,\n });\n return acc;\n }, {} as { [name: string]: LLMChain });\n\n const convChain = new ConversationChain({\n ...conversationChainOpts,\n llm,\n outputKey: \"text\",\n });\n\n return new MultiPromptChain({\n ...multiRouteChainOpts,\n routerChain,\n destinationChains,\n defaultChain: defaultChain ?? convChain,\n });\n }\n\n _chainType(): string {\n return \"multi_prompt_chain\";\n }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAuCA,IAAa,mBAAb,MAAa,yBAAyB,gBAAgB;;;;CAIpD,OAAO,YACLA,KACAC,aACAC,oBACAC,iBACAC,cACAC,SACA;AACA,SAAO,iBAAiB,kBAAkB,KAAK;GAC7C;GACA;GACA;GACA;GACA,qBAAqB;EACtB,EAAC;CACH;;;;;;;;;;;;;;;CAgBD,OAAO,kBACLL,KACA,EACE,aACA,oBACA,iBACA,cACA,cACA,uBACA,qBASD,EACiB;EAClB,MAAM,eAAe,WAAW,aAAa,mBAAmB,CAAC,IAC/D,CAAC,CAAC,MAAM,KAAK,KAAK,GAAG,KAAK,EAAE,EAAE,MAAM,CACrC;EAED,MAAM,+BAA+B,EAAE,OAAO;GAC5C,aAAa,EACV,QAAQ,CACR,UAAU,CACV,SAAS,8DAA4D;GACxE,aAAa,EACV,OAAO,EACN,OAAO,EACJ,QAAQ,CACR,SAAS,uDAAuD,CACpE,EAAC,CACD,SAAS,oCAAoC;EACjD,EAAC;EAEF,MAAM,eAAe,IAAI,mBAAmB;EAE5C,MAAM,kBAAkB,aAAa,KAAK,KAAK;EAC/C,MAAM,iBAAiB,mBACrB,wCACE,aAAa,sBAAsB,EAAE,oBAAoB,EAAG,EAAC,CAC9D,EACD,EACE,cAAc,gBACf,EACF;EAED,MAAM,eAAe,IAAI,eAAe;GACtC,UAAU;GACV,gBAAgB,CAAC,OAAQ;GACzB;EACD;EAED,MAAM,cAAc,eAAe,QAAQ,KAAK,aAAa;EAC7D,MAAM,oBAAoB,WACxB,aACA,gBACD,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,SAAS,KAAK;GAClC,IAAIM;AACJ,OAAI,OAAO,aAAa,UACtB,WAAW;YACF,OAAO,aAAa,UAC7B,WAAW,IAAI,eAAe;IAClB;IACV,gBAAgB,CAAC,OAAQ;GAC1B;OAED,OAAM,IAAI,MAAM;GAElB,IAAI,QAAkB,IAAI,SAAS;IACjC,GAAG;IACH;IACA,QAAQ;GACT;AACD,UAAO;EACR,GAAE,CAAE,EAAiC;EAEtC,MAAM,YAAY,IAAI,kBAAkB;GACtC,GAAG;GACH;GACA,WAAW;EACZ;AAED,SAAO,IAAI,iBAAiB;GAC1B,GAAG;GACH;GACA;GACA,cAAc,gBAAgB;EAC/B;CACF;CAED,aAAqB;AACnB,SAAO;CACR;AACF"}
|
|
@@ -6,7 +6,7 @@ const require_llm_router = require('./llm_router.cjs');
|
|
|
6
6
|
const require_utils = require('./utils.cjs');
|
|
7
7
|
const require_router = require('../../output_parsers/router.cjs');
|
|
8
8
|
const require_multi_retrieval_prompt = require('./multi_retrieval_prompt.cjs');
|
|
9
|
-
const
|
|
9
|
+
const zod_v3 = require_rolldown_runtime.__toESM(require("zod/v3"));
|
|
10
10
|
const __langchain_core_prompts = require_rolldown_runtime.__toESM(require("@langchain/core/prompts"));
|
|
11
11
|
|
|
12
12
|
//#region src/chains/router/multi_retrieval_qa.ts
|
|
@@ -82,9 +82,9 @@ var MultiRetrievalQAChain = class MultiRetrievalQAChain extends require_multi_ro
|
|
|
82
82
|
const { defaultRetriever, defaultPrompt, defaultChain } = defaults ?? {};
|
|
83
83
|
if (defaultPrompt && !defaultRetriever) throw new Error("`default_retriever` must be specified if `default_prompt` is \nprovided. Received only `default_prompt`.");
|
|
84
84
|
const destinations = require_utils.zipEntries(retrieverNames, retrieverDescriptions).map(([name, desc]) => `${name}: ${desc}`);
|
|
85
|
-
const structuredOutputParserSchema =
|
|
86
|
-
destination:
|
|
87
|
-
next_inputs:
|
|
85
|
+
const structuredOutputParserSchema = zod_v3.z.object({
|
|
86
|
+
destination: zod_v3.z.string().optional().describe("name of the question answering system to use or \"DEFAULT\""),
|
|
87
|
+
next_inputs: zod_v3.z.object({ query: zod_v3.z.string().describe("a potentially modified version of the original input") }).describe("input to be fed to the next model")
|
|
88
88
|
});
|
|
89
89
|
const outputParser = new require_router.RouterOutputParser(structuredOutputParserSchema);
|
|
90
90
|
const destinationsStr = destinations.join("\n");
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"multi_retrieval_qa.cjs","names":["MultiRouteChain","llm: BaseLanguageModelInterface","retrieverNames: string[]","retrieverDescriptions: string[]","retrievers: BaseRetrieverInterface[]","retrieverPrompts?: PromptTemplate[]","defaults?: MultiRetrievalDefaults","options?: Omit<MultiRouteChainInput, \"defaultChain\">","zipEntries","z","RouterOutputParser","STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE","PromptTemplate","LLMRouterChain","opt: Partial<RetrievalQAChainInput> & {\n prompt?: PromptTemplate;\n }","RetrievalQAChain","DEFAULT_TEMPLATE","ConversationChain"],"sources":["../../../src/chains/router/multi_retrieval_qa.ts"],"sourcesContent":["import { z } from \"zod\";\nimport type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { interpolateFString, PromptTemplate } from \"@langchain/core/prompts\";\nimport { MultiRouteChain, MultiRouteChainInput } from \"./multi_route.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMRouterChain } from \"./llm_router.js\";\nimport {\n ConversationChain,\n DEFAULT_TEMPLATE,\n} from \"../../chains/conversation.js\";\nimport { STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE } from \"./multi_retrieval_prompt.js\";\nimport { zipEntries } from \"./utils.js\";\nimport {\n RetrievalQAChain,\n RetrievalQAChainInput,\n} from \"../../chains/retrieval_qa.js\";\nimport { RouterOutputParser } from \"../../output_parsers/router.js\";\n\n/**\n * A type that represents the default values for the MultiRetrievalQAChain\n * class. It includes optional properties for the default retriever,\n * default prompt, and default chain.\n */\nexport type MultiRetrievalDefaults = {\n defaultRetriever?: BaseRetrieverInterface;\n defaultPrompt?: PromptTemplate;\n defaultChain?: BaseChain;\n};\n\n/**\n * A class that represents a multi-retrieval question answering chain in\n * the LangChain framework. It extends the MultiRouteChain class and\n * provides additional functionality specific to multi-retrieval QA\n * chains.\n * @example\n * ```typescript\n * const multiRetrievalQAChain = MultiRetrievalQAChain.fromLLMAndRetrievers(\n * new ChatOpenAI({ model: \"gpt-4o-mini\" }),\n * {\n * retrieverNames: [\"aqua teen\", \"mst3k\", \"animaniacs\"],\n * retrieverDescriptions: [\n * \"Good for answering questions about Aqua Teen Hunger Force theme song\",\n * \"Good for answering questions about Mystery Science Theater 3000 theme song\",\n * \"Good for answering questions about Animaniacs theme song\",\n * ],\n * retrievers: [\n * new MemoryVectorStore().asRetriever(3),\n * new MemoryVectorStore().asRetriever(3),\n * new MemoryVectorStore().asRetriever(3),\n * ],\n * retrievalQAChainOpts: {\n * returnSourceDocuments: true,\n * },\n * },\n * );\n *\n * const result = await multiRetrievalQAChain.call({\n * input:\n * \"In the Aqua Teen Hunger Force theme song, who calls himself the mike rula?\",\n * });\n *\n * console.log(result.sourceDocuments, result.text);\n * ```\n */\nexport class MultiRetrievalQAChain extends MultiRouteChain {\n get outputKeys(): string[] {\n return [\"result\"];\n }\n\n /**\n * @deprecated Use `fromRetrieversAndPrompts` instead\n */\n static fromRetrievers(\n llm: BaseLanguageModelInterface,\n retrieverNames: string[],\n retrieverDescriptions: string[],\n retrievers: BaseRetrieverInterface[],\n retrieverPrompts?: PromptTemplate[],\n defaults?: MultiRetrievalDefaults,\n options?: Omit<MultiRouteChainInput, \"defaultChain\">\n ) {\n return MultiRetrievalQAChain.fromLLMAndRetrievers(llm, {\n retrieverNames,\n retrieverDescriptions,\n retrievers,\n retrieverPrompts,\n defaults,\n multiRetrievalChainOpts: options,\n });\n }\n\n /**\n * A static method that creates an instance of MultiRetrievalQAChain from\n * a BaseLanguageModel and a set of retrievers. It takes in optional\n * parameters for the retriever names, descriptions, prompts, defaults,\n * and additional options. It is an alternative method to fromRetrievers\n * and provides more flexibility in configuring the underlying chains.\n * @param llm A BaseLanguageModel instance.\n * @param retrieverNames An array of retriever names.\n * @param retrieverDescriptions An array of retriever descriptions.\n * @param retrievers An array of BaseRetrieverInterface instances.\n * @param retrieverPrompts An optional array of PromptTemplate instances for the retrievers.\n * @param defaults An optional MultiRetrievalDefaults instance.\n * @param multiRetrievalChainOpts Additional optional parameters for the multi-retrieval chain.\n * @param retrievalQAChainOpts Additional optional parameters for the retrieval QA chain.\n * @returns A new instance of MultiRetrievalQAChain.\n */\n static fromLLMAndRetrievers(\n llm: BaseLanguageModelInterface,\n {\n retrieverNames,\n retrieverDescriptions,\n retrievers,\n retrieverPrompts,\n defaults,\n multiRetrievalChainOpts,\n retrievalQAChainOpts,\n }: {\n retrieverNames: string[];\n retrieverDescriptions: string[];\n retrievers: BaseRetrieverInterface[];\n retrieverPrompts?: PromptTemplate[];\n defaults?: MultiRetrievalDefaults;\n multiRetrievalChainOpts?: Omit<MultiRouteChainInput, \"defaultChain\">;\n retrievalQAChainOpts?: Partial<\n Omit<RetrievalQAChainInput, \"retriever\" | \"combineDocumentsChain\">\n > & {\n prompt?: PromptTemplate;\n };\n }\n ): MultiRetrievalQAChain {\n const { defaultRetriever, defaultPrompt, defaultChain } = defaults ?? {};\n if (defaultPrompt && !defaultRetriever) {\n throw new Error(\n \"`default_retriever` must be specified if `default_prompt` is \\nprovided. Received only `default_prompt`.\"\n );\n }\n const destinations = zipEntries<[string, string]>(\n retrieverNames,\n retrieverDescriptions\n ).map(([name, desc]) => `${name}: ${desc}`);\n\n const structuredOutputParserSchema = z.object({\n destination: z\n .string()\n .optional()\n .describe('name of the question answering system to use or \"DEFAULT\"'),\n next_inputs: z\n .object({\n query: z\n .string()\n .describe(\"a potentially modified version of the original input\"),\n })\n .describe(\"input to be fed to the next model\"),\n });\n\n const outputParser = new RouterOutputParser<\n typeof structuredOutputParserSchema\n >(structuredOutputParserSchema);\n\n const destinationsStr = destinations.join(\"\\n\");\n const routerTemplate = interpolateFString(\n STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE(\n outputParser.getFormatInstructions({ interpolationDepth: 4 })\n ),\n {\n destinations: destinationsStr,\n }\n );\n const routerPrompt = new PromptTemplate({\n template: routerTemplate,\n inputVariables: [\"input\"],\n outputParser,\n });\n\n const routerChain = LLMRouterChain.fromLLM(llm, routerPrompt);\n const prompts = retrieverPrompts ?? retrievers.map(() => null);\n const destinationChains = zipEntries<\n [string, BaseRetrieverInterface, PromptTemplate | null]\n >(retrieverNames, retrievers, prompts).reduce(\n (acc, [name, retriever, prompt]) => {\n const opt: Partial<RetrievalQAChainInput> & {\n prompt?: PromptTemplate;\n } = retrievalQAChainOpts ?? {};\n if (prompt) {\n opt.prompt = prompt;\n }\n acc[name] = RetrievalQAChain.fromLLM(llm, retriever, opt);\n return acc;\n },\n {} as { [name: string]: RetrievalQAChain }\n );\n\n let _defaultChain;\n if (defaultChain) {\n _defaultChain = defaultChain;\n } else if (defaultRetriever) {\n _defaultChain = RetrievalQAChain.fromLLM(llm, defaultRetriever, {\n ...retrievalQAChainOpts,\n prompt: defaultPrompt,\n });\n } else {\n const promptTemplate = DEFAULT_TEMPLATE.replace(\"input\", \"query\");\n const prompt = new PromptTemplate({\n template: promptTemplate,\n inputVariables: [\"history\", \"query\"],\n });\n _defaultChain = new ConversationChain({\n llm,\n prompt,\n outputKey: \"result\",\n });\n }\n\n return new MultiRetrievalQAChain({\n ...multiRetrievalChainOpts,\n routerChain,\n destinationChains,\n defaultChain: _defaultChain,\n });\n }\n\n _chainType(): string {\n return \"multi_retrieval_qa_chain\";\n }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAiEA,IAAa,wBAAb,MAAa,8BAA8BA,oCAAgB;CACzD,IAAI,aAAuB;AACzB,SAAO,CAAC,QAAS;CAClB;;;;CAKD,OAAO,eACLC,KACAC,gBACAC,uBACAC,YACAC,kBACAC,UACAC,SACA;AACA,SAAO,sBAAsB,qBAAqB,KAAK;GACrD;GACA;GACA;GACA;GACA;GACA,yBAAyB;EAC1B,EAAC;CACH;;;;;;;;;;;;;;;;;CAkBD,OAAO,qBACLN,KACA,EACE,gBACA,uBACA,YACA,kBACA,UACA,yBACA,sBAaD,EACsB;EACvB,MAAM,EAAE,kBAAkB,eAAe,cAAc,GAAG,YAAY,CAAE;AACxE,MAAI,iBAAiB,CAAC,iBACpB,OAAM,IAAI,MACR;EAGJ,MAAM,eAAeO,yBACnB,gBACA,sBACD,CAAC,IAAI,CAAC,CAAC,MAAM,KAAK,KAAK,GAAG,KAAK,EAAE,EAAE,MAAM,CAAC;EAE3C,MAAM,+BAA+BC,MAAE,OAAO;GAC5C,aAAaA,MACV,QAAQ,CACR,UAAU,CACV,SAAS,8DAA4D;GACxE,aAAaA,MACV,OAAO,EACN,OAAOA,MACJ,QAAQ,CACR,SAAS,uDAAuD,CACpE,EAAC,CACD,SAAS,oCAAoC;EACjD,EAAC;EAEF,MAAM,eAAe,IAAIC,kCAEvB;EAEF,MAAM,kBAAkB,aAAa,KAAK,KAAK;EAC/C,MAAM,kEACJC,0EACE,aAAa,sBAAsB,EAAE,oBAAoB,EAAG,EAAC,CAC9D,EACD,EACE,cAAc,gBACf,EACF;EACD,MAAM,eAAe,IAAIC,wCAAe;GACtC,UAAU;GACV,gBAAgB,CAAC,OAAQ;GACzB;EACD;EAED,MAAM,cAAcC,kCAAe,QAAQ,KAAK,aAAa;EAC7D,MAAM,UAAU,oBAAoB,WAAW,IAAI,MAAM,KAAK;EAC9D,MAAM,oBAAoBL,yBAExB,gBAAgB,YAAY,QAAQ,CAAC,OACrC,CAAC,KAAK,CAAC,MAAM,WAAW,OAAO,KAAK;GAClC,MAAMM,MAEF,wBAAwB,CAAE;AAC9B,OAAI,QACF,IAAI,SAAS;GAEf,IAAI,QAAQC,sCAAiB,QAAQ,KAAK,WAAW,IAAI;AACzD,UAAO;EACR,GACD,CAAE,EACH;EAED,IAAI;AACJ,MAAI,cACF,gBAAgB;WACP,kBACT,gBAAgBA,sCAAiB,QAAQ,KAAK,kBAAkB;GAC9D,GAAG;GACH,QAAQ;EACT,EAAC;OACG;GACL,MAAM,iBAAiBC,sCAAiB,QAAQ,SAAS,QAAQ;GACjE,MAAM,SAAS,IAAIJ,wCAAe;IAChC,UAAU;IACV,gBAAgB,CAAC,WAAW,OAAQ;GACrC;GACD,gBAAgB,IAAIK,uCAAkB;IACpC;IACA;IACA,WAAW;GACZ;EACF;AAED,SAAO,IAAI,sBAAsB;GAC/B,GAAG;GACH;GACA;GACA,cAAc;EACf;CACF;CAED,aAAqB;AACnB,SAAO;CACR;AACF"}
|
|
1
|
+
{"version":3,"file":"multi_retrieval_qa.cjs","names":["MultiRouteChain","llm: BaseLanguageModelInterface","retrieverNames: string[]","retrieverDescriptions: string[]","retrievers: BaseRetrieverInterface[]","retrieverPrompts?: PromptTemplate[]","defaults?: MultiRetrievalDefaults","options?: Omit<MultiRouteChainInput, \"defaultChain\">","zipEntries","z","RouterOutputParser","STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE","PromptTemplate","LLMRouterChain","opt: Partial<RetrievalQAChainInput> & {\n prompt?: PromptTemplate;\n }","RetrievalQAChain","DEFAULT_TEMPLATE","ConversationChain"],"sources":["../../../src/chains/router/multi_retrieval_qa.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { interpolateFString, PromptTemplate } from \"@langchain/core/prompts\";\nimport { MultiRouteChain, MultiRouteChainInput } from \"./multi_route.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMRouterChain } from \"./llm_router.js\";\nimport {\n ConversationChain,\n DEFAULT_TEMPLATE,\n} from \"../../chains/conversation.js\";\nimport { STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE } from \"./multi_retrieval_prompt.js\";\nimport { zipEntries } from \"./utils.js\";\nimport {\n RetrievalQAChain,\n RetrievalQAChainInput,\n} from \"../../chains/retrieval_qa.js\";\nimport { RouterOutputParser } from \"../../output_parsers/router.js\";\n\n/**\n * A type that represents the default values for the MultiRetrievalQAChain\n * class. It includes optional properties for the default retriever,\n * default prompt, and default chain.\n */\nexport type MultiRetrievalDefaults = {\n defaultRetriever?: BaseRetrieverInterface;\n defaultPrompt?: PromptTemplate;\n defaultChain?: BaseChain;\n};\n\n/**\n * A class that represents a multi-retrieval question answering chain in\n * the LangChain framework. It extends the MultiRouteChain class and\n * provides additional functionality specific to multi-retrieval QA\n * chains.\n * @example\n * ```typescript\n * const multiRetrievalQAChain = MultiRetrievalQAChain.fromLLMAndRetrievers(\n * new ChatOpenAI({ model: \"gpt-4o-mini\" }),\n * {\n * retrieverNames: [\"aqua teen\", \"mst3k\", \"animaniacs\"],\n * retrieverDescriptions: [\n * \"Good for answering questions about Aqua Teen Hunger Force theme song\",\n * \"Good for answering questions about Mystery Science Theater 3000 theme song\",\n * \"Good for answering questions about Animaniacs theme song\",\n * ],\n * retrievers: [\n * new MemoryVectorStore().asRetriever(3),\n * new MemoryVectorStore().asRetriever(3),\n * new MemoryVectorStore().asRetriever(3),\n * ],\n * retrievalQAChainOpts: {\n * returnSourceDocuments: true,\n * },\n * },\n * );\n *\n * const result = await multiRetrievalQAChain.call({\n * input:\n * \"In the Aqua Teen Hunger Force theme song, who calls himself the mike rula?\",\n * });\n *\n * console.log(result.sourceDocuments, result.text);\n * ```\n */\nexport class MultiRetrievalQAChain extends MultiRouteChain {\n get outputKeys(): string[] {\n return [\"result\"];\n }\n\n /**\n * @deprecated Use `fromRetrieversAndPrompts` instead\n */\n static fromRetrievers(\n llm: BaseLanguageModelInterface,\n retrieverNames: string[],\n retrieverDescriptions: string[],\n retrievers: BaseRetrieverInterface[],\n retrieverPrompts?: PromptTemplate[],\n defaults?: MultiRetrievalDefaults,\n options?: Omit<MultiRouteChainInput, \"defaultChain\">\n ) {\n return MultiRetrievalQAChain.fromLLMAndRetrievers(llm, {\n retrieverNames,\n retrieverDescriptions,\n retrievers,\n retrieverPrompts,\n defaults,\n multiRetrievalChainOpts: options,\n });\n }\n\n /**\n * A static method that creates an instance of MultiRetrievalQAChain from\n * a BaseLanguageModel and a set of retrievers. It takes in optional\n * parameters for the retriever names, descriptions, prompts, defaults,\n * and additional options. It is an alternative method to fromRetrievers\n * and provides more flexibility in configuring the underlying chains.\n * @param llm A BaseLanguageModel instance.\n * @param retrieverNames An array of retriever names.\n * @param retrieverDescriptions An array of retriever descriptions.\n * @param retrievers An array of BaseRetrieverInterface instances.\n * @param retrieverPrompts An optional array of PromptTemplate instances for the retrievers.\n * @param defaults An optional MultiRetrievalDefaults instance.\n * @param multiRetrievalChainOpts Additional optional parameters for the multi-retrieval chain.\n * @param retrievalQAChainOpts Additional optional parameters for the retrieval QA chain.\n * @returns A new instance of MultiRetrievalQAChain.\n */\n static fromLLMAndRetrievers(\n llm: BaseLanguageModelInterface,\n {\n retrieverNames,\n retrieverDescriptions,\n retrievers,\n retrieverPrompts,\n defaults,\n multiRetrievalChainOpts,\n retrievalQAChainOpts,\n }: {\n retrieverNames: string[];\n retrieverDescriptions: string[];\n retrievers: BaseRetrieverInterface[];\n retrieverPrompts?: PromptTemplate[];\n defaults?: MultiRetrievalDefaults;\n multiRetrievalChainOpts?: Omit<MultiRouteChainInput, \"defaultChain\">;\n retrievalQAChainOpts?: Partial<\n Omit<RetrievalQAChainInput, \"retriever\" | \"combineDocumentsChain\">\n > & {\n prompt?: PromptTemplate;\n };\n }\n ): MultiRetrievalQAChain {\n const { defaultRetriever, defaultPrompt, defaultChain } = defaults ?? {};\n if (defaultPrompt && !defaultRetriever) {\n throw new Error(\n \"`default_retriever` must be specified if `default_prompt` is \\nprovided. Received only `default_prompt`.\"\n );\n }\n const destinations = zipEntries<[string, string]>(\n retrieverNames,\n retrieverDescriptions\n ).map(([name, desc]) => `${name}: ${desc}`);\n\n const structuredOutputParserSchema = z.object({\n destination: z\n .string()\n .optional()\n .describe('name of the question answering system to use or \"DEFAULT\"'),\n next_inputs: z\n .object({\n query: z\n .string()\n .describe(\"a potentially modified version of the original input\"),\n })\n .describe(\"input to be fed to the next model\"),\n });\n\n const outputParser = new RouterOutputParser<\n typeof structuredOutputParserSchema\n >(structuredOutputParserSchema);\n\n const destinationsStr = destinations.join(\"\\n\");\n const routerTemplate = interpolateFString(\n STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE(\n outputParser.getFormatInstructions({ interpolationDepth: 4 })\n ),\n {\n destinations: destinationsStr,\n }\n );\n const routerPrompt = new PromptTemplate({\n template: routerTemplate,\n inputVariables: [\"input\"],\n outputParser,\n });\n\n const routerChain = LLMRouterChain.fromLLM(llm, routerPrompt);\n const prompts = retrieverPrompts ?? retrievers.map(() => null);\n const destinationChains = zipEntries<\n [string, BaseRetrieverInterface, PromptTemplate | null]\n >(retrieverNames, retrievers, prompts).reduce(\n (acc, [name, retriever, prompt]) => {\n const opt: Partial<RetrievalQAChainInput> & {\n prompt?: PromptTemplate;\n } = retrievalQAChainOpts ?? {};\n if (prompt) {\n opt.prompt = prompt;\n }\n acc[name] = RetrievalQAChain.fromLLM(llm, retriever, opt);\n return acc;\n },\n {} as { [name: string]: RetrievalQAChain }\n );\n\n let _defaultChain;\n if (defaultChain) {\n _defaultChain = defaultChain;\n } else if (defaultRetriever) {\n _defaultChain = RetrievalQAChain.fromLLM(llm, defaultRetriever, {\n ...retrievalQAChainOpts,\n prompt: defaultPrompt,\n });\n } else {\n const promptTemplate = DEFAULT_TEMPLATE.replace(\"input\", \"query\");\n const prompt = new PromptTemplate({\n template: promptTemplate,\n inputVariables: [\"history\", \"query\"],\n });\n _defaultChain = new ConversationChain({\n llm,\n prompt,\n outputKey: \"result\",\n });\n }\n\n return new MultiRetrievalQAChain({\n ...multiRetrievalChainOpts,\n routerChain,\n destinationChains,\n defaultChain: _defaultChain,\n });\n }\n\n _chainType(): string {\n return \"multi_retrieval_qa_chain\";\n }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAiEA,IAAa,wBAAb,MAAa,8BAA8BA,oCAAgB;CACzD,IAAI,aAAuB;AACzB,SAAO,CAAC,QAAS;CAClB;;;;CAKD,OAAO,eACLC,KACAC,gBACAC,uBACAC,YACAC,kBACAC,UACAC,SACA;AACA,SAAO,sBAAsB,qBAAqB,KAAK;GACrD;GACA;GACA;GACA;GACA;GACA,yBAAyB;EAC1B,EAAC;CACH;;;;;;;;;;;;;;;;;CAkBD,OAAO,qBACLN,KACA,EACE,gBACA,uBACA,YACA,kBACA,UACA,yBACA,sBAaD,EACsB;EACvB,MAAM,EAAE,kBAAkB,eAAe,cAAc,GAAG,YAAY,CAAE;AACxE,MAAI,iBAAiB,CAAC,iBACpB,OAAM,IAAI,MACR;EAGJ,MAAM,eAAeO,yBACnB,gBACA,sBACD,CAAC,IAAI,CAAC,CAAC,MAAM,KAAK,KAAK,GAAG,KAAK,EAAE,EAAE,MAAM,CAAC;EAE3C,MAAM,+BAA+BC,SAAE,OAAO;GAC5C,aAAaA,SACV,QAAQ,CACR,UAAU,CACV,SAAS,8DAA4D;GACxE,aAAaA,SACV,OAAO,EACN,OAAOA,SACJ,QAAQ,CACR,SAAS,uDAAuD,CACpE,EAAC,CACD,SAAS,oCAAoC;EACjD,EAAC;EAEF,MAAM,eAAe,IAAIC,kCAEvB;EAEF,MAAM,kBAAkB,aAAa,KAAK,KAAK;EAC/C,MAAM,kEACJC,0EACE,aAAa,sBAAsB,EAAE,oBAAoB,EAAG,EAAC,CAC9D,EACD,EACE,cAAc,gBACf,EACF;EACD,MAAM,eAAe,IAAIC,wCAAe;GACtC,UAAU;GACV,gBAAgB,CAAC,OAAQ;GACzB;EACD;EAED,MAAM,cAAcC,kCAAe,QAAQ,KAAK,aAAa;EAC7D,MAAM,UAAU,oBAAoB,WAAW,IAAI,MAAM,KAAK;EAC9D,MAAM,oBAAoBL,yBAExB,gBAAgB,YAAY,QAAQ,CAAC,OACrC,CAAC,KAAK,CAAC,MAAM,WAAW,OAAO,KAAK;GAClC,MAAMM,MAEF,wBAAwB,CAAE;AAC9B,OAAI,QACF,IAAI,SAAS;GAEf,IAAI,QAAQC,sCAAiB,QAAQ,KAAK,WAAW,IAAI;AACzD,UAAO;EACR,GACD,CAAE,EACH;EAED,IAAI;AACJ,MAAI,cACF,gBAAgB;WACP,kBACT,gBAAgBA,sCAAiB,QAAQ,KAAK,kBAAkB;GAC9D,GAAG;GACH,QAAQ;EACT,EAAC;OACG;GACL,MAAM,iBAAiBC,sCAAiB,QAAQ,SAAS,QAAQ;GACjE,MAAM,SAAS,IAAIJ,wCAAe;IAChC,UAAU;IACV,gBAAgB,CAAC,WAAW,OAAQ;GACrC;GACD,gBAAgB,IAAIK,uCAAkB;IACpC;IACA;IACA,WAAW;GACZ;EACF;AAED,SAAO,IAAI,sBAAsB;GAC/B,GAAG;GACH;GACA;GACA,cAAc;EACf;CACF;CAED,aAAqB;AACnB,SAAO;CACR;AACF"}
|
|
@@ -5,7 +5,7 @@ import { LLMRouterChain } from "./llm_router.js";
|
|
|
5
5
|
import { zipEntries } from "./utils.js";
|
|
6
6
|
import { RouterOutputParser } from "../../output_parsers/router.js";
|
|
7
7
|
import { STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE } from "./multi_retrieval_prompt.js";
|
|
8
|
-
import { z } from "zod";
|
|
8
|
+
import { z } from "zod/v3";
|
|
9
9
|
import { PromptTemplate, interpolateFString } from "@langchain/core/prompts";
|
|
10
10
|
|
|
11
11
|
//#region src/chains/router/multi_retrieval_qa.ts
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"multi_retrieval_qa.js","names":["llm: BaseLanguageModelInterface","retrieverNames: string[]","retrieverDescriptions: string[]","retrievers: BaseRetrieverInterface[]","retrieverPrompts?: PromptTemplate[]","defaults?: MultiRetrievalDefaults","options?: Omit<MultiRouteChainInput, \"defaultChain\">","opt: Partial<RetrievalQAChainInput> & {\n prompt?: PromptTemplate;\n }"],"sources":["../../../src/chains/router/multi_retrieval_qa.ts"],"sourcesContent":["import { z } from \"zod\";\nimport type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { interpolateFString, PromptTemplate } from \"@langchain/core/prompts\";\nimport { MultiRouteChain, MultiRouteChainInput } from \"./multi_route.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMRouterChain } from \"./llm_router.js\";\nimport {\n ConversationChain,\n DEFAULT_TEMPLATE,\n} from \"../../chains/conversation.js\";\nimport { STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE } from \"./multi_retrieval_prompt.js\";\nimport { zipEntries } from \"./utils.js\";\nimport {\n RetrievalQAChain,\n RetrievalQAChainInput,\n} from \"../../chains/retrieval_qa.js\";\nimport { RouterOutputParser } from \"../../output_parsers/router.js\";\n\n/**\n * A type that represents the default values for the MultiRetrievalQAChain\n * class. It includes optional properties for the default retriever,\n * default prompt, and default chain.\n */\nexport type MultiRetrievalDefaults = {\n defaultRetriever?: BaseRetrieverInterface;\n defaultPrompt?: PromptTemplate;\n defaultChain?: BaseChain;\n};\n\n/**\n * A class that represents a multi-retrieval question answering chain in\n * the LangChain framework. It extends the MultiRouteChain class and\n * provides additional functionality specific to multi-retrieval QA\n * chains.\n * @example\n * ```typescript\n * const multiRetrievalQAChain = MultiRetrievalQAChain.fromLLMAndRetrievers(\n * new ChatOpenAI({ model: \"gpt-4o-mini\" }),\n * {\n * retrieverNames: [\"aqua teen\", \"mst3k\", \"animaniacs\"],\n * retrieverDescriptions: [\n * \"Good for answering questions about Aqua Teen Hunger Force theme song\",\n * \"Good for answering questions about Mystery Science Theater 3000 theme song\",\n * \"Good for answering questions about Animaniacs theme song\",\n * ],\n * retrievers: [\n * new MemoryVectorStore().asRetriever(3),\n * new MemoryVectorStore().asRetriever(3),\n * new MemoryVectorStore().asRetriever(3),\n * ],\n * retrievalQAChainOpts: {\n * returnSourceDocuments: true,\n * },\n * },\n * );\n *\n * const result = await multiRetrievalQAChain.call({\n * input:\n * \"In the Aqua Teen Hunger Force theme song, who calls himself the mike rula?\",\n * });\n *\n * console.log(result.sourceDocuments, result.text);\n * ```\n */\nexport class MultiRetrievalQAChain extends MultiRouteChain {\n get outputKeys(): string[] {\n return [\"result\"];\n }\n\n /**\n * @deprecated Use `fromRetrieversAndPrompts` instead\n */\n static fromRetrievers(\n llm: BaseLanguageModelInterface,\n retrieverNames: string[],\n retrieverDescriptions: string[],\n retrievers: BaseRetrieverInterface[],\n retrieverPrompts?: PromptTemplate[],\n defaults?: MultiRetrievalDefaults,\n options?: Omit<MultiRouteChainInput, \"defaultChain\">\n ) {\n return MultiRetrievalQAChain.fromLLMAndRetrievers(llm, {\n retrieverNames,\n retrieverDescriptions,\n retrievers,\n retrieverPrompts,\n defaults,\n multiRetrievalChainOpts: options,\n });\n }\n\n /**\n * A static method that creates an instance of MultiRetrievalQAChain from\n * a BaseLanguageModel and a set of retrievers. It takes in optional\n * parameters for the retriever names, descriptions, prompts, defaults,\n * and additional options. It is an alternative method to fromRetrievers\n * and provides more flexibility in configuring the underlying chains.\n * @param llm A BaseLanguageModel instance.\n * @param retrieverNames An array of retriever names.\n * @param retrieverDescriptions An array of retriever descriptions.\n * @param retrievers An array of BaseRetrieverInterface instances.\n * @param retrieverPrompts An optional array of PromptTemplate instances for the retrievers.\n * @param defaults An optional MultiRetrievalDefaults instance.\n * @param multiRetrievalChainOpts Additional optional parameters for the multi-retrieval chain.\n * @param retrievalQAChainOpts Additional optional parameters for the retrieval QA chain.\n * @returns A new instance of MultiRetrievalQAChain.\n */\n static fromLLMAndRetrievers(\n llm: BaseLanguageModelInterface,\n {\n retrieverNames,\n retrieverDescriptions,\n retrievers,\n retrieverPrompts,\n defaults,\n multiRetrievalChainOpts,\n retrievalQAChainOpts,\n }: {\n retrieverNames: string[];\n retrieverDescriptions: string[];\n retrievers: BaseRetrieverInterface[];\n retrieverPrompts?: PromptTemplate[];\n defaults?: MultiRetrievalDefaults;\n multiRetrievalChainOpts?: Omit<MultiRouteChainInput, \"defaultChain\">;\n retrievalQAChainOpts?: Partial<\n Omit<RetrievalQAChainInput, \"retriever\" | \"combineDocumentsChain\">\n > & {\n prompt?: PromptTemplate;\n };\n }\n ): MultiRetrievalQAChain {\n const { defaultRetriever, defaultPrompt, defaultChain } = defaults ?? {};\n if (defaultPrompt && !defaultRetriever) {\n throw new Error(\n \"`default_retriever` must be specified if `default_prompt` is \\nprovided. Received only `default_prompt`.\"\n );\n }\n const destinations = zipEntries<[string, string]>(\n retrieverNames,\n retrieverDescriptions\n ).map(([name, desc]) => `${name}: ${desc}`);\n\n const structuredOutputParserSchema = z.object({\n destination: z\n .string()\n .optional()\n .describe('name of the question answering system to use or \"DEFAULT\"'),\n next_inputs: z\n .object({\n query: z\n .string()\n .describe(\"a potentially modified version of the original input\"),\n })\n .describe(\"input to be fed to the next model\"),\n });\n\n const outputParser = new RouterOutputParser<\n typeof structuredOutputParserSchema\n >(structuredOutputParserSchema);\n\n const destinationsStr = destinations.join(\"\\n\");\n const routerTemplate = interpolateFString(\n STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE(\n outputParser.getFormatInstructions({ interpolationDepth: 4 })\n ),\n {\n destinations: destinationsStr,\n }\n );\n const routerPrompt = new PromptTemplate({\n template: routerTemplate,\n inputVariables: [\"input\"],\n outputParser,\n });\n\n const routerChain = LLMRouterChain.fromLLM(llm, routerPrompt);\n const prompts = retrieverPrompts ?? retrievers.map(() => null);\n const destinationChains = zipEntries<\n [string, BaseRetrieverInterface, PromptTemplate | null]\n >(retrieverNames, retrievers, prompts).reduce(\n (acc, [name, retriever, prompt]) => {\n const opt: Partial<RetrievalQAChainInput> & {\n prompt?: PromptTemplate;\n } = retrievalQAChainOpts ?? {};\n if (prompt) {\n opt.prompt = prompt;\n }\n acc[name] = RetrievalQAChain.fromLLM(llm, retriever, opt);\n return acc;\n },\n {} as { [name: string]: RetrievalQAChain }\n );\n\n let _defaultChain;\n if (defaultChain) {\n _defaultChain = defaultChain;\n } else if (defaultRetriever) {\n _defaultChain = RetrievalQAChain.fromLLM(llm, defaultRetriever, {\n ...retrievalQAChainOpts,\n prompt: defaultPrompt,\n });\n } else {\n const promptTemplate = DEFAULT_TEMPLATE.replace(\"input\", \"query\");\n const prompt = new PromptTemplate({\n template: promptTemplate,\n inputVariables: [\"history\", \"query\"],\n });\n _defaultChain = new ConversationChain({\n llm,\n prompt,\n outputKey: \"result\",\n });\n }\n\n return new MultiRetrievalQAChain({\n ...multiRetrievalChainOpts,\n routerChain,\n destinationChains,\n defaultChain: _defaultChain,\n });\n }\n\n _chainType(): string {\n return \"multi_retrieval_qa_chain\";\n }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAiEA,IAAa,wBAAb,MAAa,8BAA8B,gBAAgB;CACzD,IAAI,aAAuB;AACzB,SAAO,CAAC,QAAS;CAClB;;;;CAKD,OAAO,eACLA,KACAC,gBACAC,uBACAC,YACAC,kBACAC,UACAC,SACA;AACA,SAAO,sBAAsB,qBAAqB,KAAK;GACrD;GACA;GACA;GACA;GACA;GACA,yBAAyB;EAC1B,EAAC;CACH;;;;;;;;;;;;;;;;;CAkBD,OAAO,qBACLN,KACA,EACE,gBACA,uBACA,YACA,kBACA,UACA,yBACA,sBAaD,EACsB;EACvB,MAAM,EAAE,kBAAkB,eAAe,cAAc,GAAG,YAAY,CAAE;AACxE,MAAI,iBAAiB,CAAC,iBACpB,OAAM,IAAI,MACR;EAGJ,MAAM,eAAe,WACnB,gBACA,sBACD,CAAC,IAAI,CAAC,CAAC,MAAM,KAAK,KAAK,GAAG,KAAK,EAAE,EAAE,MAAM,CAAC;EAE3C,MAAM,+BAA+B,EAAE,OAAO;GAC5C,aAAa,EACV,QAAQ,CACR,UAAU,CACV,SAAS,8DAA4D;GACxE,aAAa,EACV,OAAO,EACN,OAAO,EACJ,QAAQ,CACR,SAAS,uDAAuD,CACpE,EAAC,CACD,SAAS,oCAAoC;EACjD,EAAC;EAEF,MAAM,eAAe,IAAI,mBAEvB;EAEF,MAAM,kBAAkB,aAAa,KAAK,KAAK;EAC/C,MAAM,iBAAiB,mBACrB,2CACE,aAAa,sBAAsB,EAAE,oBAAoB,EAAG,EAAC,CAC9D,EACD,EACE,cAAc,gBACf,EACF;EACD,MAAM,eAAe,IAAI,eAAe;GACtC,UAAU;GACV,gBAAgB,CAAC,OAAQ;GACzB;EACD;EAED,MAAM,cAAc,eAAe,QAAQ,KAAK,aAAa;EAC7D,MAAM,UAAU,oBAAoB,WAAW,IAAI,MAAM,KAAK;EAC9D,MAAM,oBAAoB,WAExB,gBAAgB,YAAY,QAAQ,CAAC,OACrC,CAAC,KAAK,CAAC,MAAM,WAAW,OAAO,KAAK;GAClC,MAAMO,MAEF,wBAAwB,CAAE;AAC9B,OAAI,QACF,IAAI,SAAS;GAEf,IAAI,QAAQ,iBAAiB,QAAQ,KAAK,WAAW,IAAI;AACzD,UAAO;EACR,GACD,CAAE,EACH;EAED,IAAI;AACJ,MAAI,cACF,gBAAgB;WACP,kBACT,gBAAgB,iBAAiB,QAAQ,KAAK,kBAAkB;GAC9D,GAAG;GACH,QAAQ;EACT,EAAC;OACG;GACL,MAAM,iBAAiB,iBAAiB,QAAQ,SAAS,QAAQ;GACjE,MAAM,SAAS,IAAI,eAAe;IAChC,UAAU;IACV,gBAAgB,CAAC,WAAW,OAAQ;GACrC;GACD,gBAAgB,IAAI,kBAAkB;IACpC;IACA;IACA,WAAW;GACZ;EACF;AAED,SAAO,IAAI,sBAAsB;GAC/B,GAAG;GACH;GACA;GACA,cAAc;EACf;CACF;CAED,aAAqB;AACnB,SAAO;CACR;AACF"}
|
|
1
|
+
{"version":3,"file":"multi_retrieval_qa.js","names":["llm: BaseLanguageModelInterface","retrieverNames: string[]","retrieverDescriptions: string[]","retrievers: BaseRetrieverInterface[]","retrieverPrompts?: PromptTemplate[]","defaults?: MultiRetrievalDefaults","options?: Omit<MultiRouteChainInput, \"defaultChain\">","opt: Partial<RetrievalQAChainInput> & {\n prompt?: PromptTemplate;\n }"],"sources":["../../../src/chains/router/multi_retrieval_qa.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { interpolateFString, PromptTemplate } from \"@langchain/core/prompts\";\nimport { MultiRouteChain, MultiRouteChainInput } from \"./multi_route.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMRouterChain } from \"./llm_router.js\";\nimport {\n ConversationChain,\n DEFAULT_TEMPLATE,\n} from \"../../chains/conversation.js\";\nimport { STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE } from \"./multi_retrieval_prompt.js\";\nimport { zipEntries } from \"./utils.js\";\nimport {\n RetrievalQAChain,\n RetrievalQAChainInput,\n} from \"../../chains/retrieval_qa.js\";\nimport { RouterOutputParser } from \"../../output_parsers/router.js\";\n\n/**\n * A type that represents the default values for the MultiRetrievalQAChain\n * class. It includes optional properties for the default retriever,\n * default prompt, and default chain.\n */\nexport type MultiRetrievalDefaults = {\n defaultRetriever?: BaseRetrieverInterface;\n defaultPrompt?: PromptTemplate;\n defaultChain?: BaseChain;\n};\n\n/**\n * A class that represents a multi-retrieval question answering chain in\n * the LangChain framework. It extends the MultiRouteChain class and\n * provides additional functionality specific to multi-retrieval QA\n * chains.\n * @example\n * ```typescript\n * const multiRetrievalQAChain = MultiRetrievalQAChain.fromLLMAndRetrievers(\n * new ChatOpenAI({ model: \"gpt-4o-mini\" }),\n * {\n * retrieverNames: [\"aqua teen\", \"mst3k\", \"animaniacs\"],\n * retrieverDescriptions: [\n * \"Good for answering questions about Aqua Teen Hunger Force theme song\",\n * \"Good for answering questions about Mystery Science Theater 3000 theme song\",\n * \"Good for answering questions about Animaniacs theme song\",\n * ],\n * retrievers: [\n * new MemoryVectorStore().asRetriever(3),\n * new MemoryVectorStore().asRetriever(3),\n * new MemoryVectorStore().asRetriever(3),\n * ],\n * retrievalQAChainOpts: {\n * returnSourceDocuments: true,\n * },\n * },\n * );\n *\n * const result = await multiRetrievalQAChain.call({\n * input:\n * \"In the Aqua Teen Hunger Force theme song, who calls himself the mike rula?\",\n * });\n *\n * console.log(result.sourceDocuments, result.text);\n * ```\n */\nexport class MultiRetrievalQAChain extends MultiRouteChain {\n get outputKeys(): string[] {\n return [\"result\"];\n }\n\n /**\n * @deprecated Use `fromRetrieversAndPrompts` instead\n */\n static fromRetrievers(\n llm: BaseLanguageModelInterface,\n retrieverNames: string[],\n retrieverDescriptions: string[],\n retrievers: BaseRetrieverInterface[],\n retrieverPrompts?: PromptTemplate[],\n defaults?: MultiRetrievalDefaults,\n options?: Omit<MultiRouteChainInput, \"defaultChain\">\n ) {\n return MultiRetrievalQAChain.fromLLMAndRetrievers(llm, {\n retrieverNames,\n retrieverDescriptions,\n retrievers,\n retrieverPrompts,\n defaults,\n multiRetrievalChainOpts: options,\n });\n }\n\n /**\n * A static method that creates an instance of MultiRetrievalQAChain from\n * a BaseLanguageModel and a set of retrievers. It takes in optional\n * parameters for the retriever names, descriptions, prompts, defaults,\n * and additional options. It is an alternative method to fromRetrievers\n * and provides more flexibility in configuring the underlying chains.\n * @param llm A BaseLanguageModel instance.\n * @param retrieverNames An array of retriever names.\n * @param retrieverDescriptions An array of retriever descriptions.\n * @param retrievers An array of BaseRetrieverInterface instances.\n * @param retrieverPrompts An optional array of PromptTemplate instances for the retrievers.\n * @param defaults An optional MultiRetrievalDefaults instance.\n * @param multiRetrievalChainOpts Additional optional parameters for the multi-retrieval chain.\n * @param retrievalQAChainOpts Additional optional parameters for the retrieval QA chain.\n * @returns A new instance of MultiRetrievalQAChain.\n */\n static fromLLMAndRetrievers(\n llm: BaseLanguageModelInterface,\n {\n retrieverNames,\n retrieverDescriptions,\n retrievers,\n retrieverPrompts,\n defaults,\n multiRetrievalChainOpts,\n retrievalQAChainOpts,\n }: {\n retrieverNames: string[];\n retrieverDescriptions: string[];\n retrievers: BaseRetrieverInterface[];\n retrieverPrompts?: PromptTemplate[];\n defaults?: MultiRetrievalDefaults;\n multiRetrievalChainOpts?: Omit<MultiRouteChainInput, \"defaultChain\">;\n retrievalQAChainOpts?: Partial<\n Omit<RetrievalQAChainInput, \"retriever\" | \"combineDocumentsChain\">\n > & {\n prompt?: PromptTemplate;\n };\n }\n ): MultiRetrievalQAChain {\n const { defaultRetriever, defaultPrompt, defaultChain } = defaults ?? {};\n if (defaultPrompt && !defaultRetriever) {\n throw new Error(\n \"`default_retriever` must be specified if `default_prompt` is \\nprovided. Received only `default_prompt`.\"\n );\n }\n const destinations = zipEntries<[string, string]>(\n retrieverNames,\n retrieverDescriptions\n ).map(([name, desc]) => `${name}: ${desc}`);\n\n const structuredOutputParserSchema = z.object({\n destination: z\n .string()\n .optional()\n .describe('name of the question answering system to use or \"DEFAULT\"'),\n next_inputs: z\n .object({\n query: z\n .string()\n .describe(\"a potentially modified version of the original input\"),\n })\n .describe(\"input to be fed to the next model\"),\n });\n\n const outputParser = new RouterOutputParser<\n typeof structuredOutputParserSchema\n >(structuredOutputParserSchema);\n\n const destinationsStr = destinations.join(\"\\n\");\n const routerTemplate = interpolateFString(\n STRUCTURED_MULTI_RETRIEVAL_ROUTER_TEMPLATE(\n outputParser.getFormatInstructions({ interpolationDepth: 4 })\n ),\n {\n destinations: destinationsStr,\n }\n );\n const routerPrompt = new PromptTemplate({\n template: routerTemplate,\n inputVariables: [\"input\"],\n outputParser,\n });\n\n const routerChain = LLMRouterChain.fromLLM(llm, routerPrompt);\n const prompts = retrieverPrompts ?? retrievers.map(() => null);\n const destinationChains = zipEntries<\n [string, BaseRetrieverInterface, PromptTemplate | null]\n >(retrieverNames, retrievers, prompts).reduce(\n (acc, [name, retriever, prompt]) => {\n const opt: Partial<RetrievalQAChainInput> & {\n prompt?: PromptTemplate;\n } = retrievalQAChainOpts ?? {};\n if (prompt) {\n opt.prompt = prompt;\n }\n acc[name] = RetrievalQAChain.fromLLM(llm, retriever, opt);\n return acc;\n },\n {} as { [name: string]: RetrievalQAChain }\n );\n\n let _defaultChain;\n if (defaultChain) {\n _defaultChain = defaultChain;\n } else if (defaultRetriever) {\n _defaultChain = RetrievalQAChain.fromLLM(llm, defaultRetriever, {\n ...retrievalQAChainOpts,\n prompt: defaultPrompt,\n });\n } else {\n const promptTemplate = DEFAULT_TEMPLATE.replace(\"input\", \"query\");\n const prompt = new PromptTemplate({\n template: promptTemplate,\n inputVariables: [\"history\", \"query\"],\n });\n _defaultChain = new ConversationChain({\n llm,\n prompt,\n outputKey: \"result\",\n });\n }\n\n return new MultiRetrievalQAChain({\n ...multiRetrievalChainOpts,\n routerChain,\n destinationChains,\n defaultChain: _defaultChain,\n });\n }\n\n _chainType(): string {\n return \"multi_retrieval_qa_chain\";\n }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAiEA,IAAa,wBAAb,MAAa,8BAA8B,gBAAgB;CACzD,IAAI,aAAuB;AACzB,SAAO,CAAC,QAAS;CAClB;;;;CAKD,OAAO,eACLA,KACAC,gBACAC,uBACAC,YACAC,kBACAC,UACAC,SACA;AACA,SAAO,sBAAsB,qBAAqB,KAAK;GACrD;GACA;GACA;GACA;GACA;GACA,yBAAyB;EAC1B,EAAC;CACH;;;;;;;;;;;;;;;;;CAkBD,OAAO,qBACLN,KACA,EACE,gBACA,uBACA,YACA,kBACA,UACA,yBACA,sBAaD,EACsB;EACvB,MAAM,EAAE,kBAAkB,eAAe,cAAc,GAAG,YAAY,CAAE;AACxE,MAAI,iBAAiB,CAAC,iBACpB,OAAM,IAAI,MACR;EAGJ,MAAM,eAAe,WACnB,gBACA,sBACD,CAAC,IAAI,CAAC,CAAC,MAAM,KAAK,KAAK,GAAG,KAAK,EAAE,EAAE,MAAM,CAAC;EAE3C,MAAM,+BAA+B,EAAE,OAAO;GAC5C,aAAa,EACV,QAAQ,CACR,UAAU,CACV,SAAS,8DAA4D;GACxE,aAAa,EACV,OAAO,EACN,OAAO,EACJ,QAAQ,CACR,SAAS,uDAAuD,CACpE,EAAC,CACD,SAAS,oCAAoC;EACjD,EAAC;EAEF,MAAM,eAAe,IAAI,mBAEvB;EAEF,MAAM,kBAAkB,aAAa,KAAK,KAAK;EAC/C,MAAM,iBAAiB,mBACrB,2CACE,aAAa,sBAAsB,EAAE,oBAAoB,EAAG,EAAC,CAC9D,EACD,EACE,cAAc,gBACf,EACF;EACD,MAAM,eAAe,IAAI,eAAe;GACtC,UAAU;GACV,gBAAgB,CAAC,OAAQ;GACzB;EACD;EAED,MAAM,cAAc,eAAe,QAAQ,KAAK,aAAa;EAC7D,MAAM,UAAU,oBAAoB,WAAW,IAAI,MAAM,KAAK;EAC9D,MAAM,oBAAoB,WAExB,gBAAgB,YAAY,QAAQ,CAAC,OACrC,CAAC,KAAK,CAAC,MAAM,WAAW,OAAO,KAAK;GAClC,MAAMO,MAEF,wBAAwB,CAAE;AAC9B,OAAI,QACF,IAAI,SAAS;GAEf,IAAI,QAAQ,iBAAiB,QAAQ,KAAK,WAAW,IAAI;AACzD,UAAO;EACR,GACD,CAAE,EACH;EAED,IAAI;AACJ,MAAI,cACF,gBAAgB;WACP,kBACT,gBAAgB,iBAAiB,QAAQ,KAAK,kBAAkB;GAC9D,GAAG;GACH,QAAQ;EACT,EAAC;OACG;GACL,MAAM,iBAAiB,iBAAiB,QAAQ,SAAS,QAAQ;GACjE,MAAM,SAAS,IAAI,eAAe;IAChC,UAAU;IACV,gBAAgB,CAAC,WAAW,OAAQ;GACrC;GACD,gBAAgB,IAAI,kBAAkB;IACpC;IACA;IACA,WAAW;GACZ;EACF;AAED,SAAO,IAAI,sBAAsB;GAC/B,GAAG;GACH;GACA;GACA,cAAc;EACf;CACF;CAED,aAAqB;AACnB,SAAO;CACR;AACF"}
|