@librechat/agents 3.2.31 → 3.2.33
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/cjs/_virtual/_rolldown/runtime.cjs +23 -0
- package/dist/cjs/agents/AgentContext.cjs +844 -1046
- package/dist/cjs/agents/AgentContext.cjs.map +1 -1
- package/dist/cjs/common/constants.cjs +13 -13
- package/dist/cjs/common/constants.cjs.map +1 -1
- package/dist/cjs/common/enum.cjs +233 -240
- package/dist/cjs/common/enum.cjs.map +1 -1
- package/dist/cjs/common/index.cjs +2 -0
- package/dist/cjs/events.cjs +121 -169
- package/dist/cjs/events.cjs.map +1 -1
- package/dist/cjs/graphs/Graph.cjs +1389 -1807
- package/dist/cjs/graphs/Graph.cjs.map +1 -1
- package/dist/cjs/graphs/MultiAgentGraph.cjs +713 -945
- package/dist/cjs/graphs/MultiAgentGraph.cjs.map +1 -1
- package/dist/cjs/graphs/index.cjs +2 -0
- package/dist/cjs/hitl/askUserQuestion.cjs +60 -62
- package/dist/cjs/hitl/askUserQuestion.cjs.map +1 -1
- package/dist/cjs/hitl/index.cjs +1 -0
- package/dist/cjs/hooks/HookRegistry.cjs +176 -202
- package/dist/cjs/hooks/HookRegistry.cjs.map +1 -1
- package/dist/cjs/hooks/createToolPolicyHook.cjs +71 -101
- package/dist/cjs/hooks/createToolPolicyHook.cjs.map +1 -1
- package/dist/cjs/hooks/createWorkspacePolicyHook.cjs +170 -273
- package/dist/cjs/hooks/createWorkspacePolicyHook.cjs.map +1 -1
- package/dist/cjs/hooks/executeHooks.cjs +227 -282
- package/dist/cjs/hooks/executeHooks.cjs.map +1 -1
- package/dist/cjs/hooks/index.cjs +6 -0
- package/dist/cjs/hooks/matchers.cjs +196 -230
- package/dist/cjs/hooks/matchers.cjs.map +1 -1
- package/dist/cjs/hooks/types.cjs +24 -24
- package/dist/cjs/hooks/types.cjs.map +1 -1
- package/dist/cjs/instrumentation.cjs +110 -137
- package/dist/cjs/instrumentation.cjs.map +1 -1
- package/dist/cjs/langchain/google-common.cjs +0 -3
- package/dist/cjs/langchain/index.cjs +80 -43
- package/dist/cjs/langchain/language_models/chat_models.cjs +0 -3
- package/dist/cjs/langchain/messages/tool.cjs +0 -3
- package/dist/cjs/langchain/messages.cjs +35 -18
- package/dist/cjs/langchain/openai.cjs +0 -3
- package/dist/cjs/langchain/prompts.cjs +5 -8
- package/dist/cjs/langchain/runnables.cjs +11 -10
- package/dist/cjs/langchain/tools.cjs +14 -11
- package/dist/cjs/langchain/utils/env.cjs +5 -8
- package/dist/cjs/langfuse.cjs +60 -79
- package/dist/cjs/langfuse.cjs.map +1 -1
- package/dist/cjs/langfuseToolOutputTracing.cjs +267 -399
- package/dist/cjs/langfuseToolOutputTracing.cjs.map +1 -1
- package/dist/cjs/llm/anthropic/index.cjs +432 -562
- package/dist/cjs/llm/anthropic/index.cjs.map +1 -1
- package/dist/cjs/llm/anthropic/types.cjs +23 -47
- package/dist/cjs/llm/anthropic/types.cjs.map +1 -1
- package/dist/cjs/llm/anthropic/utils/message_inputs.cjs +441 -696
- package/dist/cjs/llm/anthropic/utils/message_inputs.cjs.map +1 -1
- package/dist/cjs/llm/anthropic/utils/message_outputs.cjs +171 -252
- package/dist/cjs/llm/anthropic/utils/message_outputs.cjs.map +1 -1
- package/dist/cjs/llm/anthropic/utils/output_parsers.cjs +2 -0
- package/dist/cjs/llm/anthropic/utils/tools.cjs +12 -26
- package/dist/cjs/llm/anthropic/utils/tools.cjs.map +1 -1
- package/dist/cjs/llm/bedrock/index.cjs +195 -240
- package/dist/cjs/llm/bedrock/index.cjs.map +1 -1
- package/dist/cjs/llm/bedrock/toolCache.cjs +84 -106
- package/dist/cjs/llm/bedrock/toolCache.cjs.map +1 -1
- package/dist/cjs/llm/bedrock/utils/index.cjs +2 -0
- package/dist/cjs/llm/bedrock/utils/message_inputs.cjs +357 -620
- package/dist/cjs/llm/bedrock/utils/message_inputs.cjs.map +1 -1
- package/dist/cjs/llm/bedrock/utils/message_outputs.cjs +105 -149
- package/dist/cjs/llm/bedrock/utils/message_outputs.cjs.map +1 -1
- package/dist/cjs/llm/fake.cjs +86 -96
- package/dist/cjs/llm/fake.cjs.map +1 -1
- package/dist/cjs/llm/google/index.cjs +183 -237
- package/dist/cjs/llm/google/index.cjs.map +1 -1
- package/dist/cjs/llm/google/utils/common.cjs +398 -674
- package/dist/cjs/llm/google/utils/common.cjs.map +1 -1
- package/dist/cjs/llm/google/utils/zod_to_genai_parameters.cjs +2 -0
- package/dist/cjs/llm/init.cjs +44 -53
- package/dist/cjs/llm/init.cjs.map +1 -1
- package/dist/cjs/llm/invoke.cjs +142 -182
- package/dist/cjs/llm/invoke.cjs.map +1 -1
- package/dist/cjs/llm/openai/index.cjs +991 -1276
- package/dist/cjs/llm/openai/index.cjs.map +1 -1
- package/dist/cjs/llm/openai/utils/index.cjs +189 -316
- package/dist/cjs/llm/openai/utils/index.cjs.map +1 -1
- package/dist/cjs/llm/openrouter/index.cjs +102 -153
- package/dist/cjs/llm/openrouter/index.cjs.map +1 -1
- package/dist/cjs/llm/openrouter/toolCache.cjs +35 -44
- package/dist/cjs/llm/openrouter/toolCache.cjs.map +1 -1
- package/dist/cjs/llm/providers.cjs +29 -37
- package/dist/cjs/llm/providers.cjs.map +1 -1
- package/dist/cjs/llm/request.cjs +20 -33
- package/dist/cjs/llm/request.cjs.map +1 -1
- package/dist/cjs/llm/vertexai/index.cjs +427 -453
- package/dist/cjs/llm/vertexai/index.cjs.map +1 -1
- package/dist/cjs/main.cjs +547 -528
- package/dist/cjs/messages/anthropicToolCache.cjs +68 -119
- package/dist/cjs/messages/anthropicToolCache.cjs.map +1 -1
- package/dist/cjs/messages/cache.cjs +308 -401
- package/dist/cjs/messages/cache.cjs.map +1 -1
- package/dist/cjs/messages/content.cjs +36 -49
- package/dist/cjs/messages/content.cjs.map +1 -1
- package/dist/cjs/messages/contextPruning.cjs +112 -145
- package/dist/cjs/messages/contextPruning.cjs.map +1 -1
- package/dist/cjs/messages/contextPruningSettings.cjs +36 -46
- package/dist/cjs/messages/contextPruningSettings.cjs.map +1 -1
- package/dist/cjs/messages/core.cjs +256 -397
- package/dist/cjs/messages/core.cjs.map +1 -1
- package/dist/cjs/messages/format.cjs +904 -1382
- package/dist/cjs/messages/format.cjs.map +1 -1
- package/dist/cjs/messages/ids.cjs +16 -20
- package/dist/cjs/messages/ids.cjs.map +1 -1
- package/dist/cjs/messages/index.cjs +12 -0
- package/dist/cjs/messages/langchain.cjs +18 -18
- package/dist/cjs/messages/langchain.cjs.map +1 -1
- package/dist/cjs/messages/prune.cjs +1054 -1517
- package/dist/cjs/messages/prune.cjs.map +1 -1
- package/dist/cjs/messages/recency.cjs +77 -95
- package/dist/cjs/messages/recency.cjs.map +1 -1
- package/dist/cjs/messages/reducer.cjs +63 -78
- package/dist/cjs/messages/reducer.cjs.map +1 -1
- package/dist/cjs/messages/tools.cjs +51 -79
- package/dist/cjs/messages/tools.cjs.map +1 -1
- package/dist/cjs/openai/index.cjs +171 -217
- package/dist/cjs/openai/index.cjs.map +1 -1
- package/dist/cjs/responses/index.cjs +302 -391
- package/dist/cjs/responses/index.cjs.map +1 -1
- package/dist/cjs/run.cjs +903 -1113
- package/dist/cjs/run.cjs.map +1 -1
- package/dist/cjs/session/AgentSession.cjs +805 -986
- package/dist/cjs/session/AgentSession.cjs.map +1 -1
- package/dist/cjs/session/JsonlSessionStore.cjs +327 -410
- package/dist/cjs/session/JsonlSessionStore.cjs.map +1 -1
- package/dist/cjs/session/handlers.cjs +192 -208
- package/dist/cjs/session/handlers.cjs.map +1 -1
- package/dist/cjs/session/ids.cjs +9 -10
- package/dist/cjs/session/ids.cjs.map +1 -1
- package/dist/cjs/session/index.cjs +4 -0
- package/dist/cjs/session/messageSerialization.cjs +94 -156
- package/dist/cjs/session/messageSerialization.cjs.map +1 -1
- package/dist/cjs/splitStream.cjs +147 -206
- package/dist/cjs/splitStream.cjs.map +1 -1
- package/dist/cjs/stream.cjs +856 -1344
- package/dist/cjs/stream.cjs.map +1 -1
- package/dist/cjs/summarization/index.cjs +57 -101
- package/dist/cjs/summarization/index.cjs.map +1 -1
- package/dist/cjs/summarization/node.cjs +643 -796
- package/dist/cjs/summarization/node.cjs.map +1 -1
- package/dist/cjs/tools/BashExecutor.cjs +110 -136
- package/dist/cjs/tools/BashExecutor.cjs.map +1 -1
- package/dist/cjs/tools/BashProgrammaticToolCalling.cjs +165 -245
- package/dist/cjs/tools/BashProgrammaticToolCalling.cjs.map +1 -1
- package/dist/cjs/tools/Calculator.cjs +36 -57
- package/dist/cjs/tools/Calculator.cjs.map +1 -1
- package/dist/cjs/tools/CodeExecutor.cjs +126 -168
- package/dist/cjs/tools/CodeExecutor.cjs.map +1 -1
- package/dist/cjs/tools/CodeSessionFileSummary.cjs +36 -46
- package/dist/cjs/tools/CodeSessionFileSummary.cjs.map +1 -1
- package/dist/cjs/tools/ProgrammaticToolCalling.cjs +459 -649
- package/dist/cjs/tools/ProgrammaticToolCalling.cjs.map +1 -1
- package/dist/cjs/tools/ReadFile.cjs +17 -20
- package/dist/cjs/tools/ReadFile.cjs.map +1 -1
- package/dist/cjs/tools/SkillTool.cjs +26 -27
- package/dist/cjs/tools/SkillTool.cjs.map +1 -1
- package/dist/cjs/tools/SubagentTool.cjs +59 -61
- package/dist/cjs/tools/SubagentTool.cjs.map +1 -1
- package/dist/cjs/tools/ToolNode.cjs +2109 -2686
- package/dist/cjs/tools/ToolNode.cjs.map +1 -1
- package/dist/cjs/tools/ToolSearch.cjs +663 -825
- package/dist/cjs/tools/ToolSearch.cjs.map +1 -1
- package/dist/cjs/tools/cloudflare/CloudflareBridgeRuntime.cjs +248 -340
- package/dist/cjs/tools/cloudflare/CloudflareBridgeRuntime.cjs.map +1 -1
- package/dist/cjs/tools/cloudflare/CloudflareProgrammaticToolCalling.cjs +170 -197
- package/dist/cjs/tools/cloudflare/CloudflareProgrammaticToolCalling.cjs.map +1 -1
- package/dist/cjs/tools/cloudflare/CloudflareSandboxExecutionEngine.cjs +425 -520
- package/dist/cjs/tools/cloudflare/CloudflareSandboxExecutionEngine.cjs.map +1 -1
- package/dist/cjs/tools/cloudflare/CloudflareSandboxTools.cjs +91 -124
- package/dist/cjs/tools/cloudflare/CloudflareSandboxTools.cjs.map +1 -1
- package/dist/cjs/tools/cloudflare/index.cjs +4 -0
- package/dist/cjs/tools/eagerEventExecution.cjs +75 -99
- package/dist/cjs/tools/eagerEventExecution.cjs.map +1 -1
- package/dist/cjs/tools/handlers.cjs +200 -262
- package/dist/cjs/tools/handlers.cjs.map +1 -1
- package/dist/cjs/tools/local/CompileCheckTool.cjs +150 -212
- package/dist/cjs/tools/local/CompileCheckTool.cjs.map +1 -1
- package/dist/cjs/tools/local/FileCheckpointer.cjs +77 -85
- package/dist/cjs/tools/local/FileCheckpointer.cjs.map +1 -1
- package/dist/cjs/tools/local/LocalCodingTools.cjs +763 -1022
- package/dist/cjs/tools/local/LocalCodingTools.cjs.map +1 -1
- package/dist/cjs/tools/local/LocalExecutionEngine.cjs +666 -941
- package/dist/cjs/tools/local/LocalExecutionEngine.cjs.map +1 -1
- package/dist/cjs/tools/local/LocalExecutionTools.cjs +49 -92
- package/dist/cjs/tools/local/LocalExecutionTools.cjs.map +1 -1
- package/dist/cjs/tools/local/LocalProgrammaticToolCalling.cjs +286 -354
- package/dist/cjs/tools/local/LocalProgrammaticToolCalling.cjs.map +1 -1
- package/dist/cjs/tools/local/attachments.cjs +108 -165
- package/dist/cjs/tools/local/attachments.cjs.map +1 -1
- package/dist/cjs/tools/local/bashAst.cjs +99 -113
- package/dist/cjs/tools/local/bashAst.cjs.map +1 -1
- package/dist/cjs/tools/local/editStrategies.cjs +126 -169
- package/dist/cjs/tools/local/editStrategies.cjs.map +1 -1
- package/dist/cjs/tools/local/index.cjs +12 -0
- package/dist/cjs/tools/local/resolveLocalExecutionTools.cjs +136 -218
- package/dist/cjs/tools/local/resolveLocalExecutionTools.cjs.map +1 -1
- package/dist/cjs/tools/local/syntaxCheck.cjs +142 -161
- package/dist/cjs/tools/local/syntaxCheck.cjs.map +1 -1
- package/dist/cjs/tools/local/textEncoding.cjs +25 -23
- package/dist/cjs/tools/local/textEncoding.cjs.map +1 -1
- package/dist/cjs/tools/local/workspaceFS.cjs +38 -46
- package/dist/cjs/tools/local/workspaceFS.cjs.map +1 -1
- package/dist/cjs/tools/ptcTimeout.cjs +27 -47
- package/dist/cjs/tools/ptcTimeout.cjs.map +1 -1
- package/dist/cjs/tools/schema.cjs +24 -23
- package/dist/cjs/tools/schema.cjs.map +1 -1
- package/dist/cjs/tools/search/anthropic.cjs +24 -33
- package/dist/cjs/tools/search/anthropic.cjs.map +1 -1
- package/dist/cjs/tools/search/content.cjs +95 -137
- package/dist/cjs/tools/search/content.cjs.map +1 -1
- package/dist/cjs/tools/search/firecrawl.cjs +141 -172
- package/dist/cjs/tools/search/firecrawl.cjs.map +1 -1
- package/dist/cjs/tools/search/format.cjs +128 -196
- package/dist/cjs/tools/search/format.cjs.map +1 -1
- package/dist/cjs/tools/search/highlights.cjs +165 -232
- package/dist/cjs/tools/search/highlights.cjs.map +1 -1
- package/dist/cjs/tools/search/index.cjs +2 -0
- package/dist/cjs/tools/search/rerankers.cjs +151 -174
- package/dist/cjs/tools/search/rerankers.cjs.map +1 -1
- package/dist/cjs/tools/search/schema.cjs +40 -39
- package/dist/cjs/tools/search/schema.cjs.map +1 -1
- package/dist/cjs/tools/search/search.cjs +428 -530
- package/dist/cjs/tools/search/search.cjs.map +1 -1
- package/dist/cjs/tools/search/serper-scraper.cjs +106 -127
- package/dist/cjs/tools/search/serper-scraper.cjs.map +1 -1
- package/dist/cjs/tools/search/tavily-scraper.cjs +129 -181
- package/dist/cjs/tools/search/tavily-scraper.cjs.map +1 -1
- package/dist/cjs/tools/search/tavily-search.cjs +295 -359
- package/dist/cjs/tools/search/tavily-search.cjs.map +1 -1
- package/dist/cjs/tools/search/tool.cjs +260 -299
- package/dist/cjs/tools/search/tool.cjs.map +1 -1
- package/dist/cjs/tools/search/utils.cjs +74 -117
- package/dist/cjs/tools/search/utils.cjs.map +1 -1
- package/dist/cjs/tools/skillCatalog.cjs +54 -72
- package/dist/cjs/tools/skillCatalog.cjs.map +1 -1
- package/dist/cjs/tools/streamedToolCallSeals.cjs +19 -36
- package/dist/cjs/tools/streamedToolCallSeals.cjs.map +1 -1
- package/dist/cjs/tools/subagent/SubagentExecutor.cjs +612 -771
- package/dist/cjs/tools/subagent/SubagentExecutor.cjs.map +1 -1
- package/dist/cjs/tools/subagent/index.cjs +1 -0
- package/dist/cjs/tools/toolOutputReferences.cjs +523 -630
- package/dist/cjs/tools/toolOutputReferences.cjs.map +1 -1
- package/dist/cjs/utils/callbacks.cjs +11 -21
- package/dist/cjs/utils/callbacks.cjs.map +1 -1
- package/dist/cjs/utils/errors.cjs +70 -95
- package/dist/cjs/utils/errors.cjs.map +1 -1
- package/dist/cjs/utils/events.cjs +32 -42
- package/dist/cjs/utils/events.cjs.map +1 -1
- package/dist/cjs/utils/graph.cjs +8 -12
- package/dist/cjs/utils/graph.cjs.map +1 -1
- package/dist/cjs/utils/handlers.cjs +60 -82
- package/dist/cjs/utils/handlers.cjs.map +1 -1
- package/dist/cjs/utils/index.cjs +9 -0
- package/dist/cjs/utils/llm.cjs +19 -27
- package/dist/cjs/utils/llm.cjs.map +1 -1
- package/dist/cjs/utils/misc.cjs +30 -46
- package/dist/cjs/utils/misc.cjs.map +1 -1
- package/dist/cjs/utils/run.cjs +50 -66
- package/dist/cjs/utils/run.cjs.map +1 -1
- package/dist/cjs/utils/schema.cjs +11 -19
- package/dist/cjs/utils/schema.cjs.map +1 -1
- package/dist/cjs/utils/title.cjs +71 -106
- package/dist/cjs/utils/title.cjs.map +1 -1
- package/dist/cjs/utils/tokens.cjs +186 -283
- package/dist/cjs/utils/tokens.cjs.map +1 -1
- package/dist/cjs/utils/truncation.cjs +95 -114
- package/dist/cjs/utils/truncation.cjs.map +1 -1
- package/dist/esm/agents/AgentContext.mjs +844 -1044
- package/dist/esm/agents/AgentContext.mjs.map +1 -1
- package/dist/esm/common/constants.mjs +13 -11
- package/dist/esm/common/constants.mjs.map +1 -1
- package/dist/esm/common/enum.mjs +221 -238
- package/dist/esm/common/enum.mjs.map +1 -1
- package/dist/esm/common/index.mjs +3 -0
- package/dist/esm/events.mjs +121 -167
- package/dist/esm/events.mjs.map +1 -1
- package/dist/esm/graphs/Graph.mjs +1388 -1804
- package/dist/esm/graphs/Graph.mjs.map +1 -1
- package/dist/esm/graphs/MultiAgentGraph.mjs +713 -943
- package/dist/esm/graphs/MultiAgentGraph.mjs.map +1 -1
- package/dist/esm/graphs/index.mjs +3 -0
- package/dist/esm/hitl/askUserQuestion.mjs +60 -60
- package/dist/esm/hitl/askUserQuestion.mjs.map +1 -1
- package/dist/esm/hitl/index.mjs +2 -0
- package/dist/esm/hooks/HookRegistry.mjs +176 -200
- package/dist/esm/hooks/HookRegistry.mjs.map +1 -1
- package/dist/esm/hooks/createToolPolicyHook.mjs +71 -99
- package/dist/esm/hooks/createToolPolicyHook.mjs.map +1 -1
- package/dist/esm/hooks/createWorkspacePolicyHook.mjs +170 -271
- package/dist/esm/hooks/createWorkspacePolicyHook.mjs.map +1 -1
- package/dist/esm/hooks/executeHooks.mjs +227 -280
- package/dist/esm/hooks/executeHooks.mjs.map +1 -1
- package/dist/esm/hooks/index.mjs +7 -0
- package/dist/esm/hooks/matchers.mjs +196 -228
- package/dist/esm/hooks/matchers.mjs.map +1 -1
- package/dist/esm/hooks/types.mjs +24 -22
- package/dist/esm/hooks/types.mjs.map +1 -1
- package/dist/esm/instrumentation.mjs +109 -132
- package/dist/esm/instrumentation.mjs.map +1 -1
- package/dist/esm/langchain/google-common.mjs +1 -2
- package/dist/esm/langchain/index.mjs +5 -5
- package/dist/esm/langchain/language_models/chat_models.mjs +1 -2
- package/dist/esm/langchain/messages/tool.mjs +1 -2
- package/dist/esm/langchain/messages.mjs +2 -2
- package/dist/esm/langchain/openai.mjs +1 -2
- package/dist/esm/langchain/prompts.mjs +2 -2
- package/dist/esm/langchain/runnables.mjs +2 -2
- package/dist/esm/langchain/tools.mjs +2 -2
- package/dist/esm/langchain/utils/env.mjs +2 -2
- package/dist/esm/langfuse.mjs +60 -76
- package/dist/esm/langfuse.mjs.map +1 -1
- package/dist/esm/langfuseToolOutputTracing.mjs +267 -395
- package/dist/esm/langfuseToolOutputTracing.mjs.map +1 -1
- package/dist/esm/llm/anthropic/index.mjs +432 -559
- package/dist/esm/llm/anthropic/index.mjs.map +1 -1
- package/dist/esm/llm/anthropic/types.mjs +23 -45
- package/dist/esm/llm/anthropic/types.mjs.map +1 -1
- package/dist/esm/llm/anthropic/utils/message_inputs.mjs +439 -690
- package/dist/esm/llm/anthropic/utils/message_inputs.mjs.map +1 -1
- package/dist/esm/llm/anthropic/utils/message_outputs.mjs +171 -249
- package/dist/esm/llm/anthropic/utils/message_outputs.mjs.map +1 -1
- package/dist/esm/llm/anthropic/utils/output_parsers.mjs +3 -0
- package/dist/esm/llm/anthropic/utils/tools.mjs +12 -24
- package/dist/esm/llm/anthropic/utils/tools.mjs.map +1 -1
- package/dist/esm/llm/bedrock/index.mjs +195 -238
- package/dist/esm/llm/bedrock/index.mjs.map +1 -1
- package/dist/esm/llm/bedrock/toolCache.mjs +84 -104
- package/dist/esm/llm/bedrock/toolCache.mjs.map +1 -1
- package/dist/esm/llm/bedrock/utils/index.mjs +3 -0
- package/dist/esm/llm/bedrock/utils/message_inputs.mjs +357 -618
- package/dist/esm/llm/bedrock/utils/message_inputs.mjs.map +1 -1
- package/dist/esm/llm/bedrock/utils/message_outputs.mjs +105 -147
- package/dist/esm/llm/bedrock/utils/message_outputs.mjs.map +1 -1
- package/dist/esm/llm/fake.mjs +86 -94
- package/dist/esm/llm/fake.mjs.map +1 -1
- package/dist/esm/llm/google/index.mjs +183 -235
- package/dist/esm/llm/google/index.mjs.map +1 -1
- package/dist/esm/llm/google/utils/common.mjs +397 -666
- package/dist/esm/llm/google/utils/common.mjs.map +1 -1
- package/dist/esm/llm/google/utils/zod_to_genai_parameters.mjs +3 -0
- package/dist/esm/llm/init.mjs +44 -51
- package/dist/esm/llm/init.mjs.map +1 -1
- package/dist/esm/llm/invoke.mjs +142 -180
- package/dist/esm/llm/invoke.mjs.map +1 -1
- package/dist/esm/llm/openai/index.mjs +991 -1271
- package/dist/esm/llm/openai/index.mjs.map +1 -1
- package/dist/esm/llm/openai/utils/index.mjs +188 -312
- package/dist/esm/llm/openai/utils/index.mjs.map +1 -1
- package/dist/esm/llm/openrouter/index.mjs +102 -151
- package/dist/esm/llm/openrouter/index.mjs.map +1 -1
- package/dist/esm/llm/openrouter/toolCache.mjs +35 -42
- package/dist/esm/llm/openrouter/toolCache.mjs.map +1 -1
- package/dist/esm/llm/providers.mjs +29 -34
- package/dist/esm/llm/providers.mjs.map +1 -1
- package/dist/esm/llm/request.mjs +20 -31
- package/dist/esm/llm/request.mjs.map +1 -1
- package/dist/esm/llm/vertexai/index.mjs +427 -449
- package/dist/esm/llm/vertexai/index.mjs.map +1 -1
- package/dist/esm/main.mjs +99 -87
- package/dist/esm/messages/anthropicToolCache.mjs +68 -117
- package/dist/esm/messages/anthropicToolCache.mjs.map +1 -1
- package/dist/esm/messages/cache.mjs +308 -399
- package/dist/esm/messages/cache.mjs.map +1 -1
- package/dist/esm/messages/content.mjs +36 -47
- package/dist/esm/messages/content.mjs.map +1 -1
- package/dist/esm/messages/contextPruning.mjs +112 -143
- package/dist/esm/messages/contextPruning.mjs.map +1 -1
- package/dist/esm/messages/contextPruningSettings.mjs +36 -44
- package/dist/esm/messages/contextPruningSettings.mjs.map +1 -1
- package/dist/esm/messages/core.mjs +254 -393
- package/dist/esm/messages/core.mjs.map +1 -1
- package/dist/esm/messages/format.mjs +902 -1378
- package/dist/esm/messages/format.mjs.map +1 -1
- package/dist/esm/messages/ids.mjs +16 -18
- package/dist/esm/messages/ids.mjs.map +1 -1
- package/dist/esm/messages/index.mjs +13 -0
- package/dist/esm/messages/langchain.mjs +18 -16
- package/dist/esm/messages/langchain.mjs.map +1 -1
- package/dist/esm/messages/prune.mjs +1053 -1514
- package/dist/esm/messages/prune.mjs.map +1 -1
- package/dist/esm/messages/recency.mjs +77 -93
- package/dist/esm/messages/recency.mjs.map +1 -1
- package/dist/esm/messages/reducer.mjs +63 -76
- package/dist/esm/messages/reducer.mjs.map +1 -1
- package/dist/esm/messages/tools.mjs +49 -75
- package/dist/esm/messages/tools.mjs.map +1 -1
- package/dist/esm/openai/index.mjs +170 -215
- package/dist/esm/openai/index.mjs.map +1 -1
- package/dist/esm/responses/index.mjs +301 -389
- package/dist/esm/responses/index.mjs.map +1 -1
- package/dist/esm/run.mjs +903 -1111
- package/dist/esm/run.mjs.map +1 -1
- package/dist/esm/session/AgentSession.mjs +806 -985
- package/dist/esm/session/AgentSession.mjs.map +1 -1
- package/dist/esm/session/JsonlSessionStore.mjs +326 -407
- package/dist/esm/session/JsonlSessionStore.mjs.map +1 -1
- package/dist/esm/session/handlers.mjs +192 -206
- package/dist/esm/session/handlers.mjs.map +1 -1
- package/dist/esm/session/ids.mjs +9 -8
- package/dist/esm/session/ids.mjs.map +1 -1
- package/dist/esm/session/index.mjs +5 -0
- package/dist/esm/session/messageSerialization.mjs +94 -154
- package/dist/esm/session/messageSerialization.mjs.map +1 -1
- package/dist/esm/splitStream.mjs +147 -204
- package/dist/esm/splitStream.mjs.map +1 -1
- package/dist/esm/stream.mjs +854 -1341
- package/dist/esm/stream.mjs.map +1 -1
- package/dist/esm/summarization/index.mjs +57 -99
- package/dist/esm/summarization/index.mjs.map +1 -1
- package/dist/esm/summarization/node.mjs +640 -790
- package/dist/esm/summarization/node.mjs.map +1 -1
- package/dist/esm/tools/BashExecutor.mjs +103 -129
- package/dist/esm/tools/BashExecutor.mjs.map +1 -1
- package/dist/esm/tools/BashProgrammaticToolCalling.mjs +162 -239
- package/dist/esm/tools/BashProgrammaticToolCalling.mjs.map +1 -1
- package/dist/esm/tools/Calculator.mjs +34 -36
- package/dist/esm/tools/Calculator.mjs.map +1 -1
- package/dist/esm/tools/CodeExecutor.mjs +123 -164
- package/dist/esm/tools/CodeExecutor.mjs.map +1 -1
- package/dist/esm/tools/CodeSessionFileSummary.mjs +36 -44
- package/dist/esm/tools/CodeSessionFileSummary.mjs.map +1 -1
- package/dist/esm/tools/ProgrammaticToolCalling.mjs +454 -644
- package/dist/esm/tools/ProgrammaticToolCalling.mjs.map +1 -1
- package/dist/esm/tools/ReadFile.mjs +17 -18
- package/dist/esm/tools/ReadFile.mjs.map +1 -1
- package/dist/esm/tools/SkillTool.mjs +26 -25
- package/dist/esm/tools/SkillTool.mjs.map +1 -1
- package/dist/esm/tools/SubagentTool.mjs +59 -59
- package/dist/esm/tools/SubagentTool.mjs.map +1 -1
- package/dist/esm/tools/ToolNode.mjs +2107 -2684
- package/dist/esm/tools/ToolNode.mjs.map +1 -1
- package/dist/esm/tools/ToolSearch.mjs +659 -804
- package/dist/esm/tools/ToolSearch.mjs.map +1 -1
- package/dist/esm/tools/cloudflare/CloudflareBridgeRuntime.mjs +248 -338
- package/dist/esm/tools/cloudflare/CloudflareBridgeRuntime.mjs.map +1 -1
- package/dist/esm/tools/cloudflare/CloudflareProgrammaticToolCalling.mjs +170 -195
- package/dist/esm/tools/cloudflare/CloudflareProgrammaticToolCalling.mjs.map +1 -1
- package/dist/esm/tools/cloudflare/CloudflareSandboxExecutionEngine.mjs +424 -517
- package/dist/esm/tools/cloudflare/CloudflareSandboxExecutionEngine.mjs.map +1 -1
- package/dist/esm/tools/cloudflare/CloudflareSandboxTools.mjs +91 -122
- package/dist/esm/tools/cloudflare/CloudflareSandboxTools.mjs.map +1 -1
- package/dist/esm/tools/cloudflare/index.mjs +5 -0
- package/dist/esm/tools/eagerEventExecution.mjs +75 -96
- package/dist/esm/tools/eagerEventExecution.mjs.map +1 -1
- package/dist/esm/tools/handlers.mjs +200 -260
- package/dist/esm/tools/handlers.mjs.map +1 -1
- package/dist/esm/tools/local/CompileCheckTool.mjs +150 -210
- package/dist/esm/tools/local/CompileCheckTool.mjs.map +1 -1
- package/dist/esm/tools/local/FileCheckpointer.mjs +77 -83
- package/dist/esm/tools/local/FileCheckpointer.mjs.map +1 -1
- package/dist/esm/tools/local/LocalCodingTools.mjs +760 -1017
- package/dist/esm/tools/local/LocalCodingTools.mjs.map +1 -1
- package/dist/esm/tools/local/LocalExecutionEngine.mjs +663 -936
- package/dist/esm/tools/local/LocalExecutionEngine.mjs.map +1 -1
- package/dist/esm/tools/local/LocalExecutionTools.mjs +49 -90
- package/dist/esm/tools/local/LocalExecutionTools.mjs.map +1 -1
- package/dist/esm/tools/local/LocalProgrammaticToolCalling.mjs +283 -349
- package/dist/esm/tools/local/LocalProgrammaticToolCalling.mjs.map +1 -1
- package/dist/esm/tools/local/attachments.mjs +108 -163
- package/dist/esm/tools/local/attachments.mjs.map +1 -1
- package/dist/esm/tools/local/bashAst.mjs +99 -111
- package/dist/esm/tools/local/bashAst.mjs.map +1 -1
- package/dist/esm/tools/local/editStrategies.mjs +126 -167
- package/dist/esm/tools/local/editStrategies.mjs.map +1 -1
- package/dist/esm/tools/local/index.mjs +13 -0
- package/dist/esm/tools/local/resolveLocalExecutionTools.mjs +136 -216
- package/dist/esm/tools/local/resolveLocalExecutionTools.mjs.map +1 -1
- package/dist/esm/tools/local/syntaxCheck.mjs +138 -155
- package/dist/esm/tools/local/syntaxCheck.mjs.map +1 -1
- package/dist/esm/tools/local/textEncoding.mjs +25 -21
- package/dist/esm/tools/local/textEncoding.mjs.map +1 -1
- package/dist/esm/tools/local/workspaceFS.mjs +38 -44
- package/dist/esm/tools/local/workspaceFS.mjs.map +1 -1
- package/dist/esm/tools/ptcTimeout.mjs +27 -42
- package/dist/esm/tools/ptcTimeout.mjs.map +1 -1
- package/dist/esm/tools/schema.mjs +24 -21
- package/dist/esm/tools/schema.mjs.map +1 -1
- package/dist/esm/tools/search/anthropic.mjs +24 -31
- package/dist/esm/tools/search/anthropic.mjs.map +1 -1
- package/dist/esm/tools/search/content.mjs +93 -116
- package/dist/esm/tools/search/content.mjs.map +1 -1
- package/dist/esm/tools/search/firecrawl.mjs +139 -169
- package/dist/esm/tools/search/firecrawl.mjs.map +1 -1
- package/dist/esm/tools/search/format.mjs +128 -194
- package/dist/esm/tools/search/format.mjs.map +1 -1
- package/dist/esm/tools/search/highlights.mjs +165 -230
- package/dist/esm/tools/search/highlights.mjs.map +1 -1
- package/dist/esm/tools/search/index.mjs +3 -0
- package/dist/esm/tools/search/rerankers.mjs +149 -168
- package/dist/esm/tools/search/rerankers.mjs.map +1 -1
- package/dist/esm/tools/search/schema.mjs +39 -37
- package/dist/esm/tools/search/schema.mjs.map +1 -1
- package/dist/esm/tools/search/search.mjs +426 -528
- package/dist/esm/tools/search/search.mjs.map +1 -1
- package/dist/esm/tools/search/serper-scraper.mjs +104 -124
- package/dist/esm/tools/search/serper-scraper.mjs.map +1 -1
- package/dist/esm/tools/search/tavily-scraper.mjs +127 -178
- package/dist/esm/tools/search/tavily-scraper.mjs.map +1 -1
- package/dist/esm/tools/search/tavily-search.mjs +293 -357
- package/dist/esm/tools/search/tavily-search.mjs.map +1 -1
- package/dist/esm/tools/search/tool.mjs +259 -297
- package/dist/esm/tools/search/tool.mjs.map +1 -1
- package/dist/esm/tools/search/utils.mjs +74 -115
- package/dist/esm/tools/search/utils.mjs.map +1 -1
- package/dist/esm/tools/skillCatalog.mjs +54 -70
- package/dist/esm/tools/skillCatalog.mjs.map +1 -1
- package/dist/esm/tools/streamedToolCallSeals.mjs +19 -31
- package/dist/esm/tools/streamedToolCallSeals.mjs.map +1 -1
- package/dist/esm/tools/subagent/SubagentExecutor.mjs +612 -768
- package/dist/esm/tools/subagent/SubagentExecutor.mjs.map +1 -1
- package/dist/esm/tools/subagent/index.mjs +2 -0
- package/dist/esm/tools/toolOutputReferences.mjs +523 -624
- package/dist/esm/tools/toolOutputReferences.mjs.map +1 -1
- package/dist/esm/utils/callbacks.mjs +11 -19
- package/dist/esm/utils/callbacks.mjs.map +1 -1
- package/dist/esm/utils/errors.mjs +70 -93
- package/dist/esm/utils/errors.mjs.map +1 -1
- package/dist/esm/utils/events.mjs +32 -40
- package/dist/esm/utils/events.mjs.map +1 -1
- package/dist/esm/utils/graph.mjs +8 -10
- package/dist/esm/utils/graph.mjs.map +1 -1
- package/dist/esm/utils/handlers.mjs +60 -80
- package/dist/esm/utils/handlers.mjs.map +1 -1
- package/dist/esm/utils/index.mjs +10 -0
- package/dist/esm/utils/llm.mjs +19 -25
- package/dist/esm/utils/llm.mjs.map +1 -1
- package/dist/esm/utils/misc.mjs +30 -44
- package/dist/esm/utils/misc.mjs.map +1 -1
- package/dist/esm/utils/run.mjs +50 -64
- package/dist/esm/utils/run.mjs.map +1 -1
- package/dist/esm/utils/schema.mjs +11 -17
- package/dist/esm/utils/schema.mjs.map +1 -1
- package/dist/esm/utils/title.mjs +71 -104
- package/dist/esm/utils/title.mjs.map +1 -1
- package/dist/esm/utils/tokens.mjs +186 -281
- package/dist/esm/utils/tokens.mjs.map +1 -1
- package/dist/esm/utils/truncation.mjs +95 -112
- package/dist/esm/utils/truncation.mjs.map +1 -1
- package/dist/types/messages/format.d.ts +5 -0
- package/dist/types/tools/search/tool.d.ts +17 -0
- package/dist/types/tools/search/types.d.ts +4 -0
- package/package.json +11 -17
- package/src/llm/anthropic/llm.spec.ts +36 -0
- package/src/llm/anthropic/utils/message_inputs.ts +45 -3
- package/src/llm/anthropic/utils/message_outputs.ts +6 -2
- package/src/llm/anthropic/utils/streaming-tool-input.test.ts +186 -0
- package/src/messages/cache.test.ts +122 -0
- package/src/messages/cache.ts +25 -1
- package/src/messages/format.ts +9 -0
- package/src/messages/formatAgentMessages.skills.test.ts +100 -0
- package/src/tools/search/highlights.ts +9 -1
- package/src/tools/search/search.ts +41 -3
- package/src/tools/search/source-processing.test.ts +373 -0
- package/src/tools/search/tool.ts +22 -2
- package/src/tools/search/types.ts +4 -0
- package/dist/cjs/langchain/google-common.cjs.map +0 -1
- package/dist/cjs/langchain/index.cjs.map +0 -1
- package/dist/cjs/langchain/language_models/chat_models.cjs.map +0 -1
- package/dist/cjs/langchain/messages/tool.cjs.map +0 -1
- package/dist/cjs/langchain/messages.cjs.map +0 -1
- package/dist/cjs/langchain/openai.cjs.map +0 -1
- package/dist/cjs/langchain/prompts.cjs.map +0 -1
- package/dist/cjs/langchain/runnables.cjs.map +0 -1
- package/dist/cjs/langchain/tools.cjs.map +0 -1
- package/dist/cjs/langchain/utils/env.cjs.map +0 -1
- package/dist/cjs/main.cjs.map +0 -1
- package/dist/esm/langchain/google-common.mjs.map +0 -1
- package/dist/esm/langchain/index.mjs.map +0 -1
- package/dist/esm/langchain/language_models/chat_models.mjs.map +0 -1
- package/dist/esm/langchain/messages/tool.mjs.map +0 -1
- package/dist/esm/langchain/messages.mjs.map +0 -1
- package/dist/esm/langchain/openai.mjs.map +0 -1
- package/dist/esm/langchain/prompts.mjs.map +0 -1
- package/dist/esm/langchain/runnables.mjs.map +0 -1
- package/dist/esm/langchain/tools.mjs.map +0 -1
- package/dist/esm/langchain/utils/env.mjs.map +0 -1
- package/dist/esm/main.mjs.map +0 -1
|
@@ -1,457 +1,435 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import {
|
|
3
|
-
import {
|
|
4
|
-
|
|
1
|
+
import { AIMessageChunk, isAIMessage } from "@langchain/core/messages";
|
|
2
|
+
import { ChatGoogle } from "@langchain/google-gauth";
|
|
3
|
+
import { ChatConnection } from "@langchain/google-common";
|
|
4
|
+
//#region src/llm/vertexai/index.ts
|
|
5
5
|
/**
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
|
|
6
|
+
* `@langchain/google-common`'s `_streamResponseChunks` emits usage on TWO
|
|
7
|
+
* different paths within the same stream:
|
|
8
|
+
*
|
|
9
|
+
* - Streaming chunks set `chunk.generationInfo.usage_metadata` via
|
|
10
|
+
* `responseToUsageMetadata`, which correctly sums
|
|
11
|
+
* `candidatesTokenCount + thoughtsTokenCount` and includes
|
|
12
|
+
* `output_token_details.reasoning`.
|
|
13
|
+
* - The trailing fallback chunk (emitted after the API stream exhausts)
|
|
14
|
+
* attaches its own `chunk.message.usage_metadata` built inline as
|
|
15
|
+
* `output_tokens = candidatesTokenCount` only — dropping
|
|
16
|
+
* `thoughtsTokenCount` and `output_token_details` entirely.
|
|
17
|
+
*
|
|
18
|
+
* After `AIMessageChunk.concat`, only `message.usage_metadata` survives —
|
|
19
|
+
* which is the buggy fallback value. This breaks the documented
|
|
20
|
+
* `total_tokens === input_tokens + output_tokens` invariant and silently
|
|
21
|
+
* undercharges thinking models for reasoning tokens.
|
|
22
|
+
*
|
|
23
|
+
* The repair: track the last `generationInfo.usage_metadata` we see, and
|
|
24
|
+
* when the fallback chunk arrives with its buggy `message.usage_metadata`,
|
|
25
|
+
* replace it with the tracked good value. `CustomChatGoogleGenerativeAI`
|
|
26
|
+
* solves the same problem for the Google API path differently — by
|
|
27
|
+
* overriding `_convertToUsageMetadata`.
|
|
28
|
+
*/
|
|
29
29
|
function repairStreamUsageMetadata(current, generationInfoUsage) {
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
return current;
|
|
36
|
-
if (generationInfoUsage.output_tokens <= current.output_tokens)
|
|
37
|
-
return current;
|
|
38
|
-
return generationInfoUsage;
|
|
30
|
+
if (!current) return current;
|
|
31
|
+
if (!generationInfoUsage) return current;
|
|
32
|
+
if (generationInfoUsage.total_tokens !== current.total_tokens) return current;
|
|
33
|
+
if (generationInfoUsage.output_tokens <= current.output_tokens) return current;
|
|
34
|
+
return generationInfoUsage;
|
|
39
35
|
}
|
|
40
36
|
/**
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
37
|
+
* Fixes thought signatures on functionCall parts in the formatted Gemini request.
|
|
38
|
+
*
|
|
39
|
+
* `@langchain/google-common` stores signatures as a flat array in
|
|
40
|
+
* `additional_kwargs.signatures` (one per response part) and re-attaches them
|
|
41
|
+
* by index only when `signatures.length === parts.length`. This fails when:
|
|
42
|
+
* - The API omits a signature (length mismatch)
|
|
43
|
+
* - Streaming chunks merge with different part counts
|
|
44
|
+
* - The signature for a functionCall part is an empty string
|
|
45
|
+
*
|
|
46
|
+
* This function correlates each "model" content block in the formatted request
|
|
47
|
+
* back to its originating AI message by *position*, then re-attaches non-empty
|
|
48
|
+
* signatures that the library failed to apply. AI messages without signatures
|
|
49
|
+
* still consume their slot — filtering them out shifted later messages onto
|
|
50
|
+
* the wrong content block and dropped real signatures on the floor.
|
|
51
|
+
*/
|
|
56
52
|
function fixThoughtSignatures(contents, input) {
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
|
|
72
|
-
let sigIdx = 0;
|
|
73
|
-
for (const part of content.parts) {
|
|
74
|
-
if ('functionCall' in part &&
|
|
75
|
-
(part.thoughtSignature == null || part.thoughtSignature === '') &&
|
|
76
|
-
sigIdx < availableSignatures.length) {
|
|
77
|
-
part.thoughtSignature = availableSignatures[sigIdx];
|
|
78
|
-
sigIdx++;
|
|
79
|
-
}
|
|
80
|
-
}
|
|
81
|
-
}
|
|
82
|
-
}
|
|
83
|
-
class CustomChatConnection extends ChatConnection {
|
|
84
|
-
thinkingConfig;
|
|
85
|
-
async formatData(input, parameters) {
|
|
86
|
-
const formattedData = (await super.formatData(input, parameters));
|
|
87
|
-
if (formattedData.generationConfig?.thinkingConfig?.thinkingBudget === -1) {
|
|
88
|
-
// -1 means "let the model decide" - delete the property so the API doesn't receive an invalid value
|
|
89
|
-
if (formattedData.generationConfig.thinkingConfig.includeThoughts === false) {
|
|
90
|
-
formattedData.generationConfig.thinkingConfig.includeThoughts = true;
|
|
91
|
-
}
|
|
92
|
-
delete formattedData.generationConfig.thinkingConfig.thinkingBudget;
|
|
93
|
-
}
|
|
94
|
-
if (this.thinkingConfig?.thinkingLevel != null) {
|
|
95
|
-
formattedData.generationConfig ??= {};
|
|
96
|
-
// thinkingLevel and thinkingBudget cannot coexist — the API rejects the request.
|
|
97
|
-
// Remove thinkingBudget when thinkingLevel is set.
|
|
98
|
-
const { thinkingBudget: _, ...existingThinkingConfig } = formattedData.generationConfig.thinkingConfig ?? {};
|
|
99
|
-
formattedData.generationConfig.thinkingConfig = {
|
|
100
|
-
...existingThinkingConfig,
|
|
101
|
-
thinkingLevel: this.thinkingConfig.thinkingLevel,
|
|
102
|
-
...(this.thinkingConfig.includeThoughts != null && {
|
|
103
|
-
includeThoughts: this.thinkingConfig.includeThoughts,
|
|
104
|
-
}),
|
|
105
|
-
};
|
|
106
|
-
}
|
|
107
|
-
if (formattedData.contents) {
|
|
108
|
-
fixThoughtSignatures(formattedData.contents, input);
|
|
109
|
-
// gemini-3.1+ models reject role="function"; convert to role="user"
|
|
110
|
-
for (const content of formattedData.contents) {
|
|
111
|
-
if (content.role === 'function') {
|
|
112
|
-
content.role = 'user';
|
|
113
|
-
}
|
|
114
|
-
}
|
|
115
|
-
}
|
|
116
|
-
return formattedData;
|
|
117
|
-
}
|
|
53
|
+
const aiMessages = input.filter(isAIMessage);
|
|
54
|
+
const modelContents = contents.filter((c) => c.role === "model");
|
|
55
|
+
const count = Math.min(aiMessages.length, modelContents.length);
|
|
56
|
+
for (let i = 0; i < count; i++) {
|
|
57
|
+
const signatures = aiMessages[i].additional_kwargs?.signatures;
|
|
58
|
+
if (!Array.isArray(signatures) || signatures.length === 0) continue;
|
|
59
|
+
const content = modelContents[i];
|
|
60
|
+
const attachedSignatures = new Set(content.parts.map((p) => p.thoughtSignature).filter((s) => s != null && s !== ""));
|
|
61
|
+
const availableSignatures = signatures.filter((s) => s != null && s !== "" && !attachedSignatures.has(s));
|
|
62
|
+
let sigIdx = 0;
|
|
63
|
+
for (const part of content.parts) if ("functionCall" in part && (part.thoughtSignature == null || part.thoughtSignature === "") && sigIdx < availableSignatures.length) {
|
|
64
|
+
part.thoughtSignature = availableSignatures[sigIdx];
|
|
65
|
+
sigIdx++;
|
|
66
|
+
}
|
|
67
|
+
}
|
|
118
68
|
}
|
|
69
|
+
var CustomChatConnection = class extends ChatConnection {
|
|
70
|
+
thinkingConfig;
|
|
71
|
+
async formatData(input, parameters) {
|
|
72
|
+
const formattedData = await super.formatData(input, parameters);
|
|
73
|
+
if (formattedData.generationConfig?.thinkingConfig?.thinkingBudget === -1) {
|
|
74
|
+
if (formattedData.generationConfig.thinkingConfig.includeThoughts === false) formattedData.generationConfig.thinkingConfig.includeThoughts = true;
|
|
75
|
+
delete formattedData.generationConfig.thinkingConfig.thinkingBudget;
|
|
76
|
+
}
|
|
77
|
+
if (this.thinkingConfig?.thinkingLevel != null) {
|
|
78
|
+
formattedData.generationConfig ??= {};
|
|
79
|
+
const { thinkingBudget: _, ...existingThinkingConfig } = formattedData.generationConfig.thinkingConfig ?? {};
|
|
80
|
+
formattedData.generationConfig.thinkingConfig = {
|
|
81
|
+
...existingThinkingConfig,
|
|
82
|
+
thinkingLevel: this.thinkingConfig.thinkingLevel,
|
|
83
|
+
...this.thinkingConfig.includeThoughts != null && { includeThoughts: this.thinkingConfig.includeThoughts }
|
|
84
|
+
};
|
|
85
|
+
}
|
|
86
|
+
if (formattedData.contents) {
|
|
87
|
+
fixThoughtSignatures(formattedData.contents, input);
|
|
88
|
+
for (const content of formattedData.contents) if (content.role === "function") content.role = "user";
|
|
89
|
+
}
|
|
90
|
+
return formattedData;
|
|
91
|
+
}
|
|
92
|
+
};
|
|
119
93
|
/**
|
|
120
|
-
|
|
121
|
-
|
|
122
|
-
|
|
123
|
-
|
|
124
|
-
|
|
125
|
-
|
|
126
|
-
|
|
127
|
-
|
|
128
|
-
|
|
129
|
-
|
|
130
|
-
|
|
131
|
-
|
|
132
|
-
|
|
133
|
-
|
|
134
|
-
|
|
135
|
-
|
|
136
|
-
|
|
137
|
-
|
|
138
|
-
|
|
139
|
-
|
|
140
|
-
|
|
141
|
-
|
|
142
|
-
|
|
143
|
-
|
|
144
|
-
|
|
145
|
-
|
|
146
|
-
|
|
147
|
-
|
|
148
|
-
|
|
149
|
-
|
|
150
|
-
|
|
151
|
-
|
|
152
|
-
|
|
153
|
-
|
|
154
|
-
|
|
155
|
-
|
|
156
|
-
|
|
157
|
-
|
|
158
|
-
|
|
159
|
-
|
|
160
|
-
|
|
161
|
-
|
|
162
|
-
|
|
163
|
-
|
|
164
|
-
|
|
165
|
-
|
|
166
|
-
|
|
167
|
-
|
|
168
|
-
|
|
169
|
-
|
|
170
|
-
|
|
171
|
-
|
|
172
|
-
|
|
173
|
-
|
|
174
|
-
|
|
175
|
-
|
|
176
|
-
|
|
177
|
-
|
|
178
|
-
|
|
179
|
-
|
|
180
|
-
|
|
181
|
-
|
|
182
|
-
|
|
183
|
-
|
|
184
|
-
|
|
185
|
-
|
|
186
|
-
|
|
187
|
-
|
|
188
|
-
|
|
189
|
-
|
|
190
|
-
|
|
191
|
-
|
|
192
|
-
|
|
193
|
-
|
|
194
|
-
|
|
195
|
-
|
|
196
|
-
|
|
197
|
-
|
|
198
|
-
|
|
199
|
-
|
|
200
|
-
|
|
201
|
-
|
|
202
|
-
|
|
203
|
-
|
|
204
|
-
|
|
205
|
-
|
|
206
|
-
|
|
207
|
-
|
|
208
|
-
|
|
209
|
-
|
|
210
|
-
|
|
211
|
-
|
|
212
|
-
|
|
213
|
-
|
|
214
|
-
|
|
215
|
-
|
|
216
|
-
|
|
217
|
-
|
|
218
|
-
|
|
219
|
-
|
|
220
|
-
|
|
221
|
-
|
|
222
|
-
|
|
223
|
-
|
|
224
|
-
|
|
225
|
-
|
|
226
|
-
|
|
227
|
-
|
|
228
|
-
|
|
229
|
-
|
|
230
|
-
|
|
231
|
-
|
|
232
|
-
|
|
233
|
-
|
|
234
|
-
|
|
235
|
-
|
|
236
|
-
|
|
237
|
-
|
|
238
|
-
|
|
239
|
-
|
|
240
|
-
|
|
241
|
-
|
|
242
|
-
|
|
243
|
-
|
|
244
|
-
|
|
245
|
-
|
|
246
|
-
|
|
247
|
-
|
|
248
|
-
|
|
249
|
-
|
|
250
|
-
|
|
251
|
-
|
|
252
|
-
|
|
253
|
-
|
|
254
|
-
|
|
255
|
-
|
|
256
|
-
|
|
257
|
-
|
|
258
|
-
|
|
259
|
-
|
|
260
|
-
|
|
261
|
-
|
|
262
|
-
|
|
263
|
-
|
|
264
|
-
|
|
265
|
-
|
|
266
|
-
|
|
267
|
-
|
|
268
|
-
|
|
269
|
-
|
|
270
|
-
|
|
271
|
-
|
|
272
|
-
|
|
273
|
-
|
|
274
|
-
|
|
275
|
-
|
|
276
|
-
|
|
277
|
-
|
|
278
|
-
|
|
279
|
-
|
|
280
|
-
|
|
281
|
-
|
|
282
|
-
|
|
283
|
-
|
|
284
|
-
|
|
285
|
-
|
|
286
|
-
|
|
287
|
-
|
|
288
|
-
|
|
289
|
-
|
|
290
|
-
|
|
291
|
-
|
|
292
|
-
|
|
293
|
-
|
|
294
|
-
|
|
295
|
-
|
|
296
|
-
|
|
297
|
-
|
|
298
|
-
|
|
299
|
-
|
|
300
|
-
|
|
301
|
-
|
|
302
|
-
|
|
303
|
-
|
|
304
|
-
|
|
305
|
-
|
|
306
|
-
|
|
307
|
-
|
|
308
|
-
|
|
309
|
-
|
|
310
|
-
|
|
311
|
-
|
|
312
|
-
|
|
313
|
-
|
|
314
|
-
|
|
315
|
-
|
|
316
|
-
|
|
317
|
-
|
|
318
|
-
|
|
319
|
-
|
|
320
|
-
|
|
321
|
-
|
|
322
|
-
|
|
323
|
-
|
|
324
|
-
|
|
325
|
-
|
|
326
|
-
|
|
327
|
-
|
|
328
|
-
|
|
329
|
-
|
|
330
|
-
|
|
331
|
-
|
|
332
|
-
|
|
333
|
-
|
|
334
|
-
|
|
335
|
-
|
|
336
|
-
|
|
337
|
-
|
|
338
|
-
|
|
339
|
-
|
|
340
|
-
|
|
341
|
-
|
|
342
|
-
|
|
343
|
-
|
|
344
|
-
|
|
345
|
-
|
|
346
|
-
|
|
347
|
-
|
|
348
|
-
|
|
349
|
-
|
|
350
|
-
|
|
351
|
-
|
|
352
|
-
|
|
353
|
-
|
|
354
|
-
|
|
355
|
-
|
|
356
|
-
|
|
357
|
-
|
|
358
|
-
|
|
359
|
-
|
|
360
|
-
|
|
361
|
-
|
|
362
|
-
|
|
363
|
-
|
|
364
|
-
|
|
365
|
-
|
|
366
|
-
|
|
367
|
-
|
|
368
|
-
|
|
369
|
-
|
|
370
|
-
|
|
371
|
-
|
|
372
|
-
|
|
373
|
-
|
|
374
|
-
|
|
375
|
-
|
|
376
|
-
|
|
377
|
-
|
|
378
|
-
|
|
379
|
-
|
|
380
|
-
|
|
381
|
-
|
|
382
|
-
|
|
383
|
-
|
|
384
|
-
|
|
385
|
-
|
|
386
|
-
|
|
387
|
-
|
|
388
|
-
|
|
389
|
-
|
|
390
|
-
|
|
391
|
-
|
|
392
|
-
|
|
393
|
-
|
|
394
|
-
|
|
395
|
-
|
|
396
|
-
|
|
397
|
-
|
|
398
|
-
|
|
399
|
-
|
|
400
|
-
|
|
401
|
-
|
|
402
|
-
|
|
403
|
-
|
|
404
|
-
|
|
405
|
-
|
|
406
|
-
|
|
407
|
-
|
|
408
|
-
|
|
409
|
-
|
|
410
|
-
|
|
411
|
-
|
|
412
|
-
|
|
413
|
-
|
|
414
|
-
|
|
415
|
-
|
|
416
|
-
|
|
417
|
-
|
|
418
|
-
|
|
419
|
-
|
|
420
|
-
|
|
421
|
-
|
|
422
|
-
|
|
423
|
-
|
|
424
|
-
|
|
425
|
-
|
|
426
|
-
|
|
427
|
-
|
|
428
|
-
|
|
429
|
-
|
|
430
|
-
|
|
431
|
-
|
|
432
|
-
|
|
433
|
-
|
|
434
|
-
|
|
435
|
-
|
|
436
|
-
|
|
437
|
-
|
|
438
|
-
|
|
439
|
-
|
|
440
|
-
|
|
441
|
-
|
|
442
|
-
|
|
443
|
-
|
|
444
|
-
|
|
445
|
-
|
|
446
|
-
|
|
447
|
-
|
|
448
|
-
|
|
449
|
-
|
|
450
|
-
|
|
451
|
-
|
|
452
|
-
|
|
453
|
-
|
|
454
|
-
|
|
94
|
+
* Integration with Google Vertex AI chat models.
|
|
95
|
+
*
|
|
96
|
+
* Setup:
|
|
97
|
+
* Install `@langchain/google-vertexai` and set your stringified
|
|
98
|
+
* Vertex AI credentials as an environment variable named `GOOGLE_APPLICATION_CREDENTIALS`.
|
|
99
|
+
*
|
|
100
|
+
* ```bash
|
|
101
|
+
* npm install @langchain/google-vertexai
|
|
102
|
+
* export GOOGLE_APPLICATION_CREDENTIALS="path/to/credentials"
|
|
103
|
+
* ```
|
|
104
|
+
*
|
|
105
|
+
* ## [Constructor args](https://api.js.langchain.com/classes/_langchain_google_vertexai.index.ChatVertexAI.html#constructor.new_ChatVertexAI)
|
|
106
|
+
*
|
|
107
|
+
* ## [Runtime args](https://api.js.langchain.com/interfaces/langchain_google_common_types.GoogleAIBaseLanguageModelCallOptions.html)
|
|
108
|
+
*
|
|
109
|
+
* Runtime args can be passed as the second argument to any of the base runnable methods `.invoke`. `.stream`, `.batch`, etc.
|
|
110
|
+
* They can also be passed via `.withConfig`, or the second arg in `.bindTools`, like shown in the examples below:
|
|
111
|
+
*
|
|
112
|
+
* ```typescript
|
|
113
|
+
* // When calling `.withConfig`, call options should be passed via the first argument
|
|
114
|
+
* const llmWithArgsBound = llm.withConfig({
|
|
115
|
+
* stop: ["\n"],
|
|
116
|
+
* tools: [...],
|
|
117
|
+
* });
|
|
118
|
+
*
|
|
119
|
+
* // When calling `.bindTools`, call options should be passed via the second argument
|
|
120
|
+
* const llmWithTools = llm.bindTools(
|
|
121
|
+
* [...],
|
|
122
|
+
* {
|
|
123
|
+
* tool_choice: "auto",
|
|
124
|
+
* }
|
|
125
|
+
* );
|
|
126
|
+
* ```
|
|
127
|
+
*
|
|
128
|
+
* ## Examples
|
|
129
|
+
*
|
|
130
|
+
* <details open>
|
|
131
|
+
* <summary><strong>Instantiate</strong></summary>
|
|
132
|
+
*
|
|
133
|
+
* ```typescript
|
|
134
|
+
* import { ChatVertexAI } from '@langchain/google-vertexai';
|
|
135
|
+
*
|
|
136
|
+
* const llm = new ChatVertexAI({
|
|
137
|
+
* model: "gemini-1.5-pro",
|
|
138
|
+
* temperature: 0,
|
|
139
|
+
* // other params...
|
|
140
|
+
* });
|
|
141
|
+
* ```
|
|
142
|
+
* </details>
|
|
143
|
+
*
|
|
144
|
+
* <br />
|
|
145
|
+
*
|
|
146
|
+
* <details>
|
|
147
|
+
* <summary><strong>Invoking</strong></summary>
|
|
148
|
+
*
|
|
149
|
+
* ```typescript
|
|
150
|
+
* const input = `Translate "I love programming" into French.`;
|
|
151
|
+
*
|
|
152
|
+
* // Models also accept a list of chat messages or a formatted prompt
|
|
153
|
+
* const result = await llm.invoke(input);
|
|
154
|
+
* console.log(result);
|
|
155
|
+
* ```
|
|
156
|
+
*
|
|
157
|
+
* ```txt
|
|
158
|
+
* AIMessageChunk {
|
|
159
|
+
* "content": "\"J'adore programmer\" \n\nHere's why this is the best translation:\n\n* **J'adore** means \"I love\" and conveys a strong passion.\n* **Programmer** is the French verb for \"to program.\"\n\nThis translation is natural and idiomatic in French. \n",
|
|
160
|
+
* "additional_kwargs": {},
|
|
161
|
+
* "response_metadata": {},
|
|
162
|
+
* "tool_calls": [],
|
|
163
|
+
* "tool_call_chunks": [],
|
|
164
|
+
* "invalid_tool_calls": [],
|
|
165
|
+
* "usage_metadata": {
|
|
166
|
+
* "input_tokens": 9,
|
|
167
|
+
* "output_tokens": 63,
|
|
168
|
+
* "total_tokens": 72
|
|
169
|
+
* }
|
|
170
|
+
* }
|
|
171
|
+
* ```
|
|
172
|
+
* </details>
|
|
173
|
+
*
|
|
174
|
+
* <br />
|
|
175
|
+
*
|
|
176
|
+
* <details>
|
|
177
|
+
* <summary><strong>Streaming Chunks</strong></summary>
|
|
178
|
+
*
|
|
179
|
+
* ```typescript
|
|
180
|
+
* for await (const chunk of await llm.stream(input)) {
|
|
181
|
+
* console.log(chunk);
|
|
182
|
+
* }
|
|
183
|
+
* ```
|
|
184
|
+
*
|
|
185
|
+
* ```txt
|
|
186
|
+
* AIMessageChunk {
|
|
187
|
+
* "content": "\"",
|
|
188
|
+
* "additional_kwargs": {},
|
|
189
|
+
* "response_metadata": {},
|
|
190
|
+
* "tool_calls": [],
|
|
191
|
+
* "tool_call_chunks": [],
|
|
192
|
+
* "invalid_tool_calls": []
|
|
193
|
+
* }
|
|
194
|
+
* AIMessageChunk {
|
|
195
|
+
* "content": "J'adore programmer\" \n",
|
|
196
|
+
* "additional_kwargs": {},
|
|
197
|
+
* "response_metadata": {},
|
|
198
|
+
* "tool_calls": [],
|
|
199
|
+
* "tool_call_chunks": [],
|
|
200
|
+
* "invalid_tool_calls": []
|
|
201
|
+
* }
|
|
202
|
+
* AIMessageChunk {
|
|
203
|
+
* "content": "",
|
|
204
|
+
* "additional_kwargs": {},
|
|
205
|
+
* "response_metadata": {},
|
|
206
|
+
* "tool_calls": [],
|
|
207
|
+
* "tool_call_chunks": [],
|
|
208
|
+
* "invalid_tool_calls": []
|
|
209
|
+
* }
|
|
210
|
+
* AIMessageChunk {
|
|
211
|
+
* "content": "",
|
|
212
|
+
* "additional_kwargs": {},
|
|
213
|
+
* "response_metadata": {
|
|
214
|
+
* "finishReason": "stop"
|
|
215
|
+
* },
|
|
216
|
+
* "tool_calls": [],
|
|
217
|
+
* "tool_call_chunks": [],
|
|
218
|
+
* "invalid_tool_calls": [],
|
|
219
|
+
* "usage_metadata": {
|
|
220
|
+
* "input_tokens": 9,
|
|
221
|
+
* "output_tokens": 8,
|
|
222
|
+
* "total_tokens": 17
|
|
223
|
+
* }
|
|
224
|
+
* }
|
|
225
|
+
* ```
|
|
226
|
+
* </details>
|
|
227
|
+
*
|
|
228
|
+
* <br />
|
|
229
|
+
*
|
|
230
|
+
* <details>
|
|
231
|
+
* <summary><strong>Aggregate Streamed Chunks</strong></summary>
|
|
232
|
+
*
|
|
233
|
+
* ```typescript
|
|
234
|
+
* import { AIMessageChunk } from '@langchain/core/messages';
|
|
235
|
+
* import { concat } from '@langchain/core/utils/stream';
|
|
236
|
+
*
|
|
237
|
+
* const stream = await llm.stream(input);
|
|
238
|
+
* let full: AIMessageChunk | undefined;
|
|
239
|
+
* for await (const chunk of stream) {
|
|
240
|
+
* full = !full ? chunk : concat(full, chunk);
|
|
241
|
+
* }
|
|
242
|
+
* console.log(full);
|
|
243
|
+
* ```
|
|
244
|
+
*
|
|
245
|
+
* ```txt
|
|
246
|
+
* AIMessageChunk {
|
|
247
|
+
* "content": "\"J'adore programmer\" \n",
|
|
248
|
+
* "additional_kwargs": {},
|
|
249
|
+
* "response_metadata": {
|
|
250
|
+
* "finishReason": "stop"
|
|
251
|
+
* },
|
|
252
|
+
* "tool_calls": [],
|
|
253
|
+
* "tool_call_chunks": [],
|
|
254
|
+
* "invalid_tool_calls": [],
|
|
255
|
+
* "usage_metadata": {
|
|
256
|
+
* "input_tokens": 9,
|
|
257
|
+
* "output_tokens": 8,
|
|
258
|
+
* "total_tokens": 17
|
|
259
|
+
* }
|
|
260
|
+
* }
|
|
261
|
+
* ```
|
|
262
|
+
* </details>
|
|
263
|
+
*
|
|
264
|
+
* <br />
|
|
265
|
+
*
|
|
266
|
+
* <details>
|
|
267
|
+
* <summary><strong>Bind tools</strong></summary>
|
|
268
|
+
*
|
|
269
|
+
* ```typescript
|
|
270
|
+
* import { z } from 'zod';
|
|
271
|
+
*
|
|
272
|
+
* const GetWeather = {
|
|
273
|
+
* name: "GetWeather",
|
|
274
|
+
* description: "Get the current weather in a given location",
|
|
275
|
+
* schema: z.object({
|
|
276
|
+
* location: z.string().describe("The city and state, e.g. San Francisco, CA")
|
|
277
|
+
* }),
|
|
278
|
+
* }
|
|
279
|
+
*
|
|
280
|
+
* const GetPopulation = {
|
|
281
|
+
* name: "GetPopulation",
|
|
282
|
+
* description: "Get the current population in a given location",
|
|
283
|
+
* schema: z.object({
|
|
284
|
+
* location: z.string().describe("The city and state, e.g. San Francisco, CA")
|
|
285
|
+
* }),
|
|
286
|
+
* }
|
|
287
|
+
*
|
|
288
|
+
* const llmWithTools = llm.bindTools([GetWeather, GetPopulation]);
|
|
289
|
+
* const aiMsg = await llmWithTools.invoke(
|
|
290
|
+
* "Which city is hotter today and which is bigger: LA or NY?"
|
|
291
|
+
* );
|
|
292
|
+
* console.log(aiMsg.tool_calls);
|
|
293
|
+
* ```
|
|
294
|
+
*
|
|
295
|
+
* ```txt
|
|
296
|
+
* [
|
|
297
|
+
* {
|
|
298
|
+
* name: 'GetPopulation',
|
|
299
|
+
* args: { location: 'New York City, NY' },
|
|
300
|
+
* id: '33c1c1f47e2f492799c77d2800a43912',
|
|
301
|
+
* type: 'tool_call'
|
|
302
|
+
* }
|
|
303
|
+
* ]
|
|
304
|
+
* ```
|
|
305
|
+
* </details>
|
|
306
|
+
*
|
|
307
|
+
* <br />
|
|
308
|
+
*
|
|
309
|
+
* <details>
|
|
310
|
+
* <summary><strong>Structured Output</strong></summary>
|
|
311
|
+
*
|
|
312
|
+
* ```typescript
|
|
313
|
+
* import { z } from 'zod';
|
|
314
|
+
*
|
|
315
|
+
* const Joke = z.object({
|
|
316
|
+
* setup: z.string().describe("The setup of the joke"),
|
|
317
|
+
* punchline: z.string().describe("The punchline to the joke"),
|
|
318
|
+
* rating: z.number().optional().describe("How funny the joke is, from 1 to 10")
|
|
319
|
+
* }).describe('Joke to tell user.');
|
|
320
|
+
*
|
|
321
|
+
* const structuredLlm = llm.withStructuredOutput(Joke, { name: "Joke" });
|
|
322
|
+
* const jokeResult = await structuredLlm.invoke("Tell me a joke about cats");
|
|
323
|
+
* console.log(jokeResult);
|
|
324
|
+
* ```
|
|
325
|
+
*
|
|
326
|
+
* ```txt
|
|
327
|
+
* {
|
|
328
|
+
* setup: 'What do you call a cat that loves to bowl?',
|
|
329
|
+
* punchline: 'An alley cat!'
|
|
330
|
+
* }
|
|
331
|
+
* ```
|
|
332
|
+
* </details>
|
|
333
|
+
*
|
|
334
|
+
* <br />
|
|
335
|
+
*
|
|
336
|
+
* <details>
|
|
337
|
+
* <summary><strong>Usage Metadata</strong></summary>
|
|
338
|
+
*
|
|
339
|
+
* ```typescript
|
|
340
|
+
* const aiMsgForMetadata = await llm.invoke(input);
|
|
341
|
+
* console.log(aiMsgForMetadata.usage_metadata);
|
|
342
|
+
* ```
|
|
343
|
+
*
|
|
344
|
+
* ```txt
|
|
345
|
+
* { input_tokens: 9, output_tokens: 8, total_tokens: 17 }
|
|
346
|
+
* ```
|
|
347
|
+
* </details>
|
|
348
|
+
*
|
|
349
|
+
* <br />
|
|
350
|
+
*
|
|
351
|
+
* <details>
|
|
352
|
+
* <summary><strong>Stream Usage Metadata</strong></summary>
|
|
353
|
+
*
|
|
354
|
+
* ```typescript
|
|
355
|
+
* const streamForMetadata = await llm.stream(
|
|
356
|
+
* input,
|
|
357
|
+
* {
|
|
358
|
+
* streamUsage: true
|
|
359
|
+
* }
|
|
360
|
+
* );
|
|
361
|
+
* let fullForMetadata: AIMessageChunk | undefined;
|
|
362
|
+
* for await (const chunk of streamForMetadata) {
|
|
363
|
+
* fullForMetadata = !fullForMetadata ? chunk : concat(fullForMetadata, chunk);
|
|
364
|
+
* }
|
|
365
|
+
* console.log(fullForMetadata?.usage_metadata);
|
|
366
|
+
* ```
|
|
367
|
+
*
|
|
368
|
+
* ```txt
|
|
369
|
+
* { input_tokens: 9, output_tokens: 8, total_tokens: 17 }
|
|
370
|
+
* ```
|
|
371
|
+
* </details>
|
|
372
|
+
*
|
|
373
|
+
* <br />
|
|
374
|
+
*/
|
|
375
|
+
var ChatVertexAI = class extends ChatGoogle {
|
|
376
|
+
lc_namespace = [
|
|
377
|
+
"langchain",
|
|
378
|
+
"chat_models",
|
|
379
|
+
"vertexai"
|
|
380
|
+
];
|
|
381
|
+
dynamicThinkingBudget = false;
|
|
382
|
+
thinkingConfig;
|
|
383
|
+
static lc_name() {
|
|
384
|
+
return "LibreChatVertexAI";
|
|
385
|
+
}
|
|
386
|
+
constructor(modelOrFields, params) {
|
|
387
|
+
const fields = typeof modelOrFields === "string" ? {
|
|
388
|
+
...params ?? {},
|
|
389
|
+
model: modelOrFields
|
|
390
|
+
} : modelOrFields;
|
|
391
|
+
const dynamicThinkingBudget = fields?.thinkingBudget === -1;
|
|
392
|
+
super({
|
|
393
|
+
...fields,
|
|
394
|
+
platformType: "gcp"
|
|
395
|
+
});
|
|
396
|
+
this.dynamicThinkingBudget = dynamicThinkingBudget;
|
|
397
|
+
this.thinkingConfig = fields?.thinkingConfig;
|
|
398
|
+
}
|
|
399
|
+
invocationParams(options) {
|
|
400
|
+
const params = super.invocationParams(options);
|
|
401
|
+
if (this.dynamicThinkingBudget) params.maxReasoningTokens = -1;
|
|
402
|
+
return params;
|
|
403
|
+
}
|
|
404
|
+
async *_streamResponseChunks(messages, options, runManager) {
|
|
405
|
+
let lastGoodUsage;
|
|
406
|
+
for await (const chunk of super._streamResponseChunks(messages, options, runManager)) {
|
|
407
|
+
const genUsage = chunk.generationInfo?.usage_metadata;
|
|
408
|
+
if (genUsage) lastGoodUsage = genUsage;
|
|
409
|
+
if (chunk.message instanceof AIMessageChunk) {
|
|
410
|
+
const repaired = repairStreamUsageMetadata(chunk.message.usage_metadata, lastGoodUsage);
|
|
411
|
+
if (repaired !== chunk.message.usage_metadata) chunk.message.usage_metadata = repaired;
|
|
412
|
+
}
|
|
413
|
+
yield chunk;
|
|
414
|
+
}
|
|
415
|
+
}
|
|
416
|
+
buildConnection(fields, client) {
|
|
417
|
+
const thinkingConfig = fields?.thinkingConfig ?? this.thinkingConfig;
|
|
418
|
+
const connection = new CustomChatConnection({
|
|
419
|
+
...fields,
|
|
420
|
+
...this
|
|
421
|
+
}, this.caller, client, false);
|
|
422
|
+
connection.thinkingConfig = thinkingConfig;
|
|
423
|
+
this.connection = connection;
|
|
424
|
+
const streamedConnection = new CustomChatConnection({
|
|
425
|
+
...fields,
|
|
426
|
+
...this
|
|
427
|
+
}, this.caller, client, true);
|
|
428
|
+
streamedConnection.thinkingConfig = thinkingConfig;
|
|
429
|
+
this.streamedConnection = streamedConnection;
|
|
430
|
+
}
|
|
431
|
+
};
|
|
432
|
+
//#endregion
|
|
433
|
+
export { ChatVertexAI };
|
|
455
434
|
|
|
456
|
-
|
|
457
|
-
//# sourceMappingURL=index.mjs.map
|
|
435
|
+
//# sourceMappingURL=index.mjs.map
|