@illuma-ai/agents 1.1.21 → 1.1.23
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/graphs/Graph.cjs +12 -1
- package/dist/cjs/graphs/Graph.cjs.map +1 -1
- package/dist/cjs/graphs/MultiAgentGraph.cjs +105 -1
- package/dist/cjs/graphs/MultiAgentGraph.cjs.map +1 -1
- package/dist/cjs/run.cjs +20 -9
- package/dist/cjs/run.cjs.map +1 -1
- package/dist/cjs/utils/llm.cjs.map +1 -1
- package/dist/esm/graphs/Graph.mjs +12 -1
- package/dist/esm/graphs/Graph.mjs.map +1 -1
- package/dist/esm/graphs/MultiAgentGraph.mjs +105 -1
- package/dist/esm/graphs/MultiAgentGraph.mjs.map +1 -1
- package/dist/esm/run.mjs +20 -9
- package/dist/esm/run.mjs.map +1 -1
- package/dist/esm/utils/llm.mjs.map +1 -1
- package/dist/types/graphs/MultiAgentGraph.d.ts +17 -0
- package/package.json +1 -1
- package/src/graphs/Graph.ts +13 -1
- package/src/graphs/MultiAgentGraph.ts +128 -1
- package/src/graphs/__tests__/multi-agent-delegate.test.ts +205 -0
- package/src/run.ts +20 -11
- package/src/scripts/test-bedrock-handoff-autonomous.ts +231 -0
- package/src/utils/llm.ts +1 -0
- package/src/agents/AgentContext.js +0 -782
- package/src/agents/AgentContext.test.js +0 -421
- package/src/agents/__tests__/AgentContext.test.js +0 -678
- package/src/agents/__tests__/resolveStructuredOutputMode.test.js +0 -117
- package/src/common/enum.js +0 -192
- package/src/common/index.js +0 -3
- package/src/events.js +0 -166
- package/src/graphs/Graph.js +0 -1857
- package/src/graphs/MultiAgentGraph.js +0 -1092
- package/src/graphs/__tests__/structured-output.integration.test.js +0 -624
- package/src/graphs/__tests__/structured-output.test.js +0 -144
- package/src/graphs/contextManagement.e2e.test.js +0 -718
- package/src/graphs/contextManagement.test.js +0 -485
- package/src/graphs/handoffValidation.test.js +0 -276
- package/src/graphs/index.js +0 -3
- package/src/index.js +0 -28
- package/src/instrumentation.js +0 -21
- package/src/llm/anthropic/index.js +0 -319
- package/src/llm/anthropic/types.js +0 -46
- package/src/llm/anthropic/utils/message_inputs.js +0 -627
- package/src/llm/anthropic/utils/message_outputs.js +0 -290
- package/src/llm/anthropic/utils/output_parsers.js +0 -89
- package/src/llm/anthropic/utils/tools.js +0 -25
- package/src/llm/bedrock/__tests__/bedrock-caching.test.js +0 -392
- package/src/llm/bedrock/index.js +0 -303
- package/src/llm/bedrock/types.js +0 -2
- package/src/llm/bedrock/utils/index.js +0 -6
- package/src/llm/bedrock/utils/message_inputs.js +0 -463
- package/src/llm/bedrock/utils/message_outputs.js +0 -269
- package/src/llm/fake.js +0 -92
- package/src/llm/google/index.js +0 -215
- package/src/llm/google/types.js +0 -12
- package/src/llm/google/utils/common.js +0 -670
- package/src/llm/google/utils/tools.js +0 -111
- package/src/llm/google/utils/zod_to_genai_parameters.js +0 -47
- package/src/llm/openai/index.js +0 -1033
- package/src/llm/openai/types.js +0 -2
- package/src/llm/openai/utils/index.js +0 -756
- package/src/llm/openai/utils/isReasoningModel.test.js +0 -79
- package/src/llm/openrouter/index.js +0 -261
- package/src/llm/openrouter/reasoning.test.js +0 -181
- package/src/llm/providers.js +0 -36
- package/src/llm/text.js +0 -65
- package/src/llm/vertexai/index.js +0 -402
- package/src/messages/__tests__/tools.test.js +0 -392
- package/src/messages/cache.js +0 -404
- package/src/messages/cache.test.js +0 -1167
- package/src/messages/content.js +0 -48
- package/src/messages/content.test.js +0 -314
- package/src/messages/core.js +0 -359
- package/src/messages/ensureThinkingBlock.test.js +0 -997
- package/src/messages/format.js +0 -973
- package/src/messages/formatAgentMessages.test.js +0 -2278
- package/src/messages/formatAgentMessages.tools.test.js +0 -362
- package/src/messages/formatMessage.test.js +0 -608
- package/src/messages/ids.js +0 -18
- package/src/messages/index.js +0 -9
- package/src/messages/labelContentByAgent.test.js +0 -725
- package/src/messages/prune.js +0 -438
- package/src/messages/reducer.js +0 -60
- package/src/messages/shiftIndexTokenCountMap.test.js +0 -63
- package/src/messages/summarize.js +0 -146
- package/src/messages/summarize.test.js +0 -332
- package/src/messages/tools.js +0 -90
- package/src/mockStream.js +0 -81
- package/src/prompts/collab.js +0 -7
- package/src/prompts/index.js +0 -3
- package/src/prompts/taskmanager.js +0 -58
- package/src/run.js +0 -427
- package/src/schemas/index.js +0 -3
- package/src/schemas/schema-preparation.test.js +0 -370
- package/src/schemas/validate.js +0 -314
- package/src/schemas/validate.test.js +0 -264
- package/src/scripts/abort.js +0 -127
- package/src/scripts/ant_web_search.js +0 -130
- package/src/scripts/ant_web_search_edge_case.js +0 -133
- package/src/scripts/ant_web_search_error_edge_case.js +0 -119
- package/src/scripts/args.js +0 -41
- package/src/scripts/bedrock-cache-debug.js +0 -186
- package/src/scripts/bedrock-content-aggregation-test.js +0 -195
- package/src/scripts/bedrock-merge-test.js +0 -80
- package/src/scripts/bedrock-parallel-tools-test.js +0 -150
- package/src/scripts/caching.js +0 -106
- package/src/scripts/cli.js +0 -152
- package/src/scripts/cli2.js +0 -119
- package/src/scripts/cli3.js +0 -163
- package/src/scripts/cli4.js +0 -165
- package/src/scripts/cli5.js +0 -165
- package/src/scripts/code_exec.js +0 -171
- package/src/scripts/code_exec_files.js +0 -180
- package/src/scripts/code_exec_multi_session.js +0 -185
- package/src/scripts/code_exec_ptc.js +0 -265
- package/src/scripts/code_exec_session.js +0 -217
- package/src/scripts/code_exec_simple.js +0 -120
- package/src/scripts/content.js +0 -111
- package/src/scripts/empty_input.js +0 -125
- package/src/scripts/handoff-test.js +0 -96
- package/src/scripts/image.js +0 -138
- package/src/scripts/memory.js +0 -83
- package/src/scripts/multi-agent-chain.js +0 -271
- package/src/scripts/multi-agent-conditional.js +0 -185
- package/src/scripts/multi-agent-document-review-chain.js +0 -171
- package/src/scripts/multi-agent-hybrid-flow.js +0 -264
- package/src/scripts/multi-agent-parallel-start.js +0 -214
- package/src/scripts/multi-agent-parallel.js +0 -346
- package/src/scripts/multi-agent-sequence.js +0 -184
- package/src/scripts/multi-agent-supervisor.js +0 -324
- package/src/scripts/multi-agent-test.js +0 -147
- package/src/scripts/parallel-asymmetric-tools-test.js +0 -202
- package/src/scripts/parallel-full-metadata-test.js +0 -176
- package/src/scripts/parallel-tools-test.js +0 -256
- package/src/scripts/programmatic_exec.js +0 -277
- package/src/scripts/programmatic_exec_agent.js +0 -168
- package/src/scripts/search.js +0 -118
- package/src/scripts/sequential-full-metadata-test.js +0 -143
- package/src/scripts/simple.js +0 -174
- package/src/scripts/single-agent-metadata-test.js +0 -152
- package/src/scripts/stream.js +0 -113
- package/src/scripts/test-custom-prompt-key.js +0 -132
- package/src/scripts/test-handoff-input.js +0 -143
- package/src/scripts/test-handoff-preamble.js +0 -227
- package/src/scripts/test-handoff-steering.js +0 -353
- package/src/scripts/test-multi-agent-list-handoff.js +0 -318
- package/src/scripts/test-parallel-agent-labeling.js +0 -253
- package/src/scripts/test-parallel-handoffs.js +0 -229
- package/src/scripts/test-thinking-handoff-bedrock.js +0 -132
- package/src/scripts/test-thinking-handoff.js +0 -132
- package/src/scripts/test-thinking-to-thinking-handoff-bedrock.js +0 -140
- package/src/scripts/test-tool-before-handoff-role-order.js +0 -223
- package/src/scripts/test-tools-before-handoff.js +0 -187
- package/src/scripts/test_code_api.js +0 -263
- package/src/scripts/thinking-bedrock.js +0 -128
- package/src/scripts/thinking-vertexai.js +0 -130
- package/src/scripts/thinking.js +0 -134
- package/src/scripts/tool_search.js +0 -114
- package/src/scripts/tools.js +0 -125
- package/src/specs/agent-handoffs-bedrock.integration.test.js +0 -280
- package/src/specs/agent-handoffs.test.js +0 -924
- package/src/specs/anthropic.simple.test.js +0 -287
- package/src/specs/azure.simple.test.js +0 -381
- package/src/specs/cache.simple.test.js +0 -282
- package/src/specs/custom-event-await.test.js +0 -148
- package/src/specs/deepseek.simple.test.js +0 -189
- package/src/specs/emergency-prune.test.js +0 -308
- package/src/specs/moonshot.simple.test.js +0 -237
- package/src/specs/observability.integration.test.js +0 -1337
- package/src/specs/openai.simple.test.js +0 -233
- package/src/specs/openrouter.simple.test.js +0 -202
- package/src/specs/prune.test.js +0 -733
- package/src/specs/reasoning.test.js +0 -144
- package/src/specs/spec.utils.js +0 -4
- package/src/specs/thinking-handoff.test.js +0 -486
- package/src/specs/thinking-prune.test.js +0 -600
- package/src/specs/token-distribution-edge-case.test.js +0 -246
- package/src/specs/token-memoization.test.js +0 -32
- package/src/specs/tokens.test.js +0 -49
- package/src/specs/tool-error.test.js +0 -139
- package/src/splitStream.js +0 -204
- package/src/splitStream.test.js +0 -504
- package/src/stream.js +0 -650
- package/src/stream.test.js +0 -225
- package/src/test/mockTools.js +0 -340
- package/src/tools/BrowserTools.js +0 -245
- package/src/tools/Calculator.js +0 -38
- package/src/tools/Calculator.test.js +0 -225
- package/src/tools/CodeExecutor.js +0 -233
- package/src/tools/ProgrammaticToolCalling.js +0 -602
- package/src/tools/StreamingToolCallBuffer.js +0 -179
- package/src/tools/ToolNode.js +0 -930
- package/src/tools/ToolSearch.js +0 -904
- package/src/tools/__tests__/BrowserTools.test.js +0 -306
- package/src/tools/__tests__/ProgrammaticToolCalling.integration.test.js +0 -276
- package/src/tools/__tests__/ProgrammaticToolCalling.test.js +0 -807
- package/src/tools/__tests__/StreamingToolCallBuffer.test.js +0 -175
- package/src/tools/__tests__/ToolApproval.test.js +0 -675
- package/src/tools/__tests__/ToolNode.recovery.test.js +0 -200
- package/src/tools/__tests__/ToolNode.session.test.js +0 -319
- package/src/tools/__tests__/ToolSearch.integration.test.js +0 -125
- package/src/tools/__tests__/ToolSearch.test.js +0 -812
- package/src/tools/__tests__/handlers.test.js +0 -799
- package/src/tools/__tests__/truncation-recovery.integration.test.js +0 -362
- package/src/tools/handlers.js +0 -306
- package/src/tools/schema.js +0 -25
- package/src/tools/search/anthropic.js +0 -34
- package/src/tools/search/content.js +0 -116
- package/src/tools/search/content.test.js +0 -133
- package/src/tools/search/firecrawl.js +0 -173
- package/src/tools/search/format.js +0 -198
- package/src/tools/search/highlights.js +0 -241
- package/src/tools/search/index.js +0 -3
- package/src/tools/search/jina-reranker.test.js +0 -106
- package/src/tools/search/rerankers.js +0 -165
- package/src/tools/search/schema.js +0 -102
- package/src/tools/search/search.js +0 -561
- package/src/tools/search/serper-scraper.js +0 -126
- package/src/tools/search/test.js +0 -129
- package/src/tools/search/tool.js +0 -453
- package/src/tools/search/types.js +0 -2
- package/src/tools/search/utils.js +0 -59
- package/src/types/graph.js +0 -24
- package/src/types/graph.test.js +0 -192
- package/src/types/index.js +0 -7
- package/src/types/llm.js +0 -2
- package/src/types/messages.js +0 -2
- package/src/types/run.js +0 -2
- package/src/types/stream.js +0 -2
- package/src/types/tools.js +0 -2
- package/src/utils/contextAnalytics.js +0 -79
- package/src/utils/contextAnalytics.test.js +0 -166
- package/src/utils/events.js +0 -26
- package/src/utils/graph.js +0 -11
- package/src/utils/handlers.js +0 -65
- package/src/utils/index.js +0 -10
- package/src/utils/llm.js +0 -21
- package/src/utils/llmConfig.js +0 -205
- package/src/utils/logging.js +0 -37
- package/src/utils/misc.js +0 -51
- package/src/utils/run.js +0 -69
- package/src/utils/schema.js +0 -21
- package/src/utils/title.js +0 -119
- package/src/utils/tokens.js +0 -92
- package/src/utils/toonFormat.js +0 -379
|
@@ -1,290 +0,0 @@
|
|
|
1
|
-
import { AIMessage, AIMessageChunk, } from '@langchain/core/messages';
|
|
2
|
-
import { extractToolCalls } from './output_parsers';
|
|
3
|
-
function _isAnthropicCompactionBlock(block) {
|
|
4
|
-
return (typeof block === 'object' &&
|
|
5
|
-
block !== null &&
|
|
6
|
-
'type' in block &&
|
|
7
|
-
block.type === 'compaction');
|
|
8
|
-
}
|
|
9
|
-
export function _makeMessageChunkFromAnthropicEvent(data, fields) {
|
|
10
|
-
if (data.type === 'message_start') {
|
|
11
|
-
// eslint-disable-next-line @typescript-eslint/no-unused-vars
|
|
12
|
-
const { content, usage, ...additionalKwargs } = data.message;
|
|
13
|
-
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
14
|
-
const filteredAdditionalKwargs = {};
|
|
15
|
-
for (const [key, value] of Object.entries(additionalKwargs)) {
|
|
16
|
-
if (value !== undefined && value !== null) {
|
|
17
|
-
filteredAdditionalKwargs[key] = value;
|
|
18
|
-
}
|
|
19
|
-
}
|
|
20
|
-
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
21
|
-
const { input_tokens, output_tokens, ...rest } = usage ?? {};
|
|
22
|
-
const usageMetadata = {
|
|
23
|
-
input_tokens,
|
|
24
|
-
output_tokens,
|
|
25
|
-
total_tokens: input_tokens + output_tokens,
|
|
26
|
-
input_token_details: {
|
|
27
|
-
cache_creation: rest.cache_creation_input_tokens,
|
|
28
|
-
cache_read: rest.cache_read_input_tokens,
|
|
29
|
-
},
|
|
30
|
-
};
|
|
31
|
-
return {
|
|
32
|
-
chunk: new AIMessageChunk({
|
|
33
|
-
content: fields.coerceContentToString ? '' : [],
|
|
34
|
-
additional_kwargs: filteredAdditionalKwargs,
|
|
35
|
-
usage_metadata: fields.streamUsage ? usageMetadata : undefined,
|
|
36
|
-
response_metadata: {
|
|
37
|
-
usage: {
|
|
38
|
-
...rest,
|
|
39
|
-
},
|
|
40
|
-
},
|
|
41
|
-
id: data.message.id,
|
|
42
|
-
}),
|
|
43
|
-
};
|
|
44
|
-
}
|
|
45
|
-
else if (data.type === 'message_delta') {
|
|
46
|
-
const usageMetadata = {
|
|
47
|
-
input_tokens: 0,
|
|
48
|
-
output_tokens: data.usage.output_tokens,
|
|
49
|
-
total_tokens: data.usage.output_tokens,
|
|
50
|
-
input_token_details: {
|
|
51
|
-
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
52
|
-
cache_creation: data.usage.cache_creation_input_tokens,
|
|
53
|
-
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
54
|
-
cache_read: data.usage.cache_read_input_tokens,
|
|
55
|
-
},
|
|
56
|
-
};
|
|
57
|
-
return {
|
|
58
|
-
chunk: new AIMessageChunk({
|
|
59
|
-
content: fields.coerceContentToString ? '' : [],
|
|
60
|
-
additional_kwargs: { ...data.delta },
|
|
61
|
-
usage_metadata: fields.streamUsage ? usageMetadata : undefined,
|
|
62
|
-
}),
|
|
63
|
-
};
|
|
64
|
-
}
|
|
65
|
-
else if (data.type === 'content_block_start' &&
|
|
66
|
-
[
|
|
67
|
-
'tool_use',
|
|
68
|
-
'document',
|
|
69
|
-
'server_tool_use',
|
|
70
|
-
'web_search_tool_result',
|
|
71
|
-
].includes(data.content_block.type)) {
|
|
72
|
-
const contentBlock = data.content_block;
|
|
73
|
-
let toolCallChunks;
|
|
74
|
-
if (contentBlock.type === 'tool_use') {
|
|
75
|
-
toolCallChunks = [
|
|
76
|
-
{
|
|
77
|
-
id: contentBlock.id,
|
|
78
|
-
index: data.index,
|
|
79
|
-
name: contentBlock.name,
|
|
80
|
-
args: '',
|
|
81
|
-
},
|
|
82
|
-
];
|
|
83
|
-
}
|
|
84
|
-
else if (contentBlock.type === 'server_tool_use') {
|
|
85
|
-
// Handle anthropic built-in server tool use (like web search)
|
|
86
|
-
toolCallChunks = [
|
|
87
|
-
{
|
|
88
|
-
id: contentBlock.id,
|
|
89
|
-
index: data.index,
|
|
90
|
-
name: contentBlock.name,
|
|
91
|
-
args: '',
|
|
92
|
-
},
|
|
93
|
-
];
|
|
94
|
-
}
|
|
95
|
-
else {
|
|
96
|
-
toolCallChunks = [];
|
|
97
|
-
}
|
|
98
|
-
return {
|
|
99
|
-
chunk: new AIMessageChunk({
|
|
100
|
-
content: fields.coerceContentToString
|
|
101
|
-
? ''
|
|
102
|
-
: [
|
|
103
|
-
{
|
|
104
|
-
index: data.index,
|
|
105
|
-
...data.content_block,
|
|
106
|
-
input: contentBlock.type === 'server_tool_use' ||
|
|
107
|
-
contentBlock.type === 'tool_use'
|
|
108
|
-
? ''
|
|
109
|
-
: undefined,
|
|
110
|
-
},
|
|
111
|
-
],
|
|
112
|
-
additional_kwargs: {},
|
|
113
|
-
tool_call_chunks: toolCallChunks,
|
|
114
|
-
}),
|
|
115
|
-
};
|
|
116
|
-
}
|
|
117
|
-
else if (data.type === 'content_block_delta' &&
|
|
118
|
-
[
|
|
119
|
-
'text_delta',
|
|
120
|
-
'citations_delta',
|
|
121
|
-
'thinking_delta',
|
|
122
|
-
'signature_delta',
|
|
123
|
-
].includes(data.delta.type)) {
|
|
124
|
-
if (fields.coerceContentToString && 'text' in data.delta) {
|
|
125
|
-
return {
|
|
126
|
-
chunk: new AIMessageChunk({
|
|
127
|
-
content: data.delta.text,
|
|
128
|
-
}),
|
|
129
|
-
};
|
|
130
|
-
}
|
|
131
|
-
else {
|
|
132
|
-
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
133
|
-
const contentBlock = data.delta;
|
|
134
|
-
if ('citation' in contentBlock) {
|
|
135
|
-
contentBlock.citations = [contentBlock.citation];
|
|
136
|
-
delete contentBlock.citation;
|
|
137
|
-
}
|
|
138
|
-
if (contentBlock.type === 'thinking_delta' ||
|
|
139
|
-
contentBlock.type === 'signature_delta') {
|
|
140
|
-
return {
|
|
141
|
-
chunk: new AIMessageChunk({
|
|
142
|
-
content: [{ index: data.index, ...contentBlock, type: 'thinking' }],
|
|
143
|
-
}),
|
|
144
|
-
};
|
|
145
|
-
}
|
|
146
|
-
return {
|
|
147
|
-
chunk: new AIMessageChunk({
|
|
148
|
-
content: [{ index: data.index, ...contentBlock, type: 'text' }],
|
|
149
|
-
}),
|
|
150
|
-
};
|
|
151
|
-
}
|
|
152
|
-
}
|
|
153
|
-
else if (data.type === 'content_block_delta' &&
|
|
154
|
-
data.delta.type === 'input_json_delta') {
|
|
155
|
-
return {
|
|
156
|
-
chunk: new AIMessageChunk({
|
|
157
|
-
content: fields.coerceContentToString
|
|
158
|
-
? ''
|
|
159
|
-
: [
|
|
160
|
-
{
|
|
161
|
-
index: data.index,
|
|
162
|
-
input: data.delta.partial_json,
|
|
163
|
-
type: data.delta.type,
|
|
164
|
-
},
|
|
165
|
-
],
|
|
166
|
-
additional_kwargs: {},
|
|
167
|
-
tool_call_chunks: [
|
|
168
|
-
{
|
|
169
|
-
index: data.index,
|
|
170
|
-
args: data.delta.partial_json,
|
|
171
|
-
},
|
|
172
|
-
],
|
|
173
|
-
}),
|
|
174
|
-
};
|
|
175
|
-
}
|
|
176
|
-
else if (data.type === 'content_block_start' &&
|
|
177
|
-
data.content_block.type === 'text') {
|
|
178
|
-
const content = data.content_block.text;
|
|
179
|
-
if (content !== undefined) {
|
|
180
|
-
return {
|
|
181
|
-
chunk: new AIMessageChunk({
|
|
182
|
-
content: fields.coerceContentToString
|
|
183
|
-
? content
|
|
184
|
-
: [
|
|
185
|
-
{
|
|
186
|
-
index: data.index,
|
|
187
|
-
...data.content_block,
|
|
188
|
-
},
|
|
189
|
-
],
|
|
190
|
-
additional_kwargs: {},
|
|
191
|
-
}),
|
|
192
|
-
};
|
|
193
|
-
}
|
|
194
|
-
}
|
|
195
|
-
else if (data.type === 'content_block_start' &&
|
|
196
|
-
data.content_block.type === 'redacted_thinking') {
|
|
197
|
-
return {
|
|
198
|
-
chunk: new AIMessageChunk({
|
|
199
|
-
content: fields.coerceContentToString
|
|
200
|
-
? ''
|
|
201
|
-
: [{ index: data.index, ...data.content_block }],
|
|
202
|
-
}),
|
|
203
|
-
};
|
|
204
|
-
}
|
|
205
|
-
else if (data.type === 'content_block_start' &&
|
|
206
|
-
data.content_block.type === 'thinking') {
|
|
207
|
-
const content = data.content_block.thinking;
|
|
208
|
-
return {
|
|
209
|
-
chunk: new AIMessageChunk({
|
|
210
|
-
content: fields.coerceContentToString
|
|
211
|
-
? content
|
|
212
|
-
: [{ index: data.index, ...data.content_block }],
|
|
213
|
-
}),
|
|
214
|
-
};
|
|
215
|
-
}
|
|
216
|
-
else if (data.type === 'content_block_start' &&
|
|
217
|
-
_isAnthropicCompactionBlock(data.content_block)) {
|
|
218
|
-
return {
|
|
219
|
-
chunk: new AIMessageChunk({
|
|
220
|
-
content: fields.coerceContentToString
|
|
221
|
-
? ''
|
|
222
|
-
: [{ index: data.index, ...data.content_block }],
|
|
223
|
-
}),
|
|
224
|
-
};
|
|
225
|
-
}
|
|
226
|
-
else if (data.type === 'content_block_delta' &&
|
|
227
|
-
data.delta.type === 'compaction_delta') {
|
|
228
|
-
return {
|
|
229
|
-
chunk: new AIMessageChunk({
|
|
230
|
-
content: fields.coerceContentToString
|
|
231
|
-
? ''
|
|
232
|
-
: [
|
|
233
|
-
{
|
|
234
|
-
index: data.index,
|
|
235
|
-
...data.delta,
|
|
236
|
-
type: 'compaction',
|
|
237
|
-
},
|
|
238
|
-
],
|
|
239
|
-
}),
|
|
240
|
-
};
|
|
241
|
-
}
|
|
242
|
-
return null;
|
|
243
|
-
}
|
|
244
|
-
export function anthropicResponseToChatMessages(messages, additionalKwargs) {
|
|
245
|
-
const usage = additionalKwargs.usage;
|
|
246
|
-
const usageMetadata = usage != null
|
|
247
|
-
? {
|
|
248
|
-
input_tokens: usage.input_tokens ?? 0,
|
|
249
|
-
output_tokens: usage.output_tokens ?? 0,
|
|
250
|
-
total_tokens: (usage.input_tokens ?? 0) + (usage.output_tokens ?? 0),
|
|
251
|
-
input_token_details: {
|
|
252
|
-
cache_creation: usage.cache_creation_input_tokens,
|
|
253
|
-
cache_read: usage.cache_read_input_tokens,
|
|
254
|
-
},
|
|
255
|
-
}
|
|
256
|
-
: undefined;
|
|
257
|
-
if (messages.length === 1 && messages[0].type === 'text') {
|
|
258
|
-
return [
|
|
259
|
-
{
|
|
260
|
-
text: messages[0].text,
|
|
261
|
-
message: new AIMessage({
|
|
262
|
-
content: messages[0].text,
|
|
263
|
-
additional_kwargs: additionalKwargs,
|
|
264
|
-
usage_metadata: usageMetadata,
|
|
265
|
-
response_metadata: additionalKwargs,
|
|
266
|
-
id: additionalKwargs.id,
|
|
267
|
-
}),
|
|
268
|
-
},
|
|
269
|
-
];
|
|
270
|
-
}
|
|
271
|
-
else {
|
|
272
|
-
const toolCalls = extractToolCalls(messages);
|
|
273
|
-
const generations = [
|
|
274
|
-
{
|
|
275
|
-
text: '',
|
|
276
|
-
message: new AIMessage({
|
|
277
|
-
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
278
|
-
content: messages,
|
|
279
|
-
additional_kwargs: additionalKwargs,
|
|
280
|
-
tool_calls: toolCalls,
|
|
281
|
-
usage_metadata: usageMetadata,
|
|
282
|
-
response_metadata: additionalKwargs,
|
|
283
|
-
id: additionalKwargs.id,
|
|
284
|
-
}),
|
|
285
|
-
},
|
|
286
|
-
];
|
|
287
|
-
return generations;
|
|
288
|
-
}
|
|
289
|
-
}
|
|
290
|
-
//# sourceMappingURL=message_outputs.js.map
|
|
@@ -1,89 +0,0 @@
|
|
|
1
|
-
/* eslint-disable @typescript-eslint/explicit-function-return-type */
|
|
2
|
-
/* eslint-disable @typescript-eslint/no-empty-object-type */
|
|
3
|
-
import { BaseLLMOutputParser, OutputParserException, } from '@langchain/core/output_parsers';
|
|
4
|
-
import { interopSafeParseAsync, } from '@langchain/core/utils/types';
|
|
5
|
-
export class AnthropicToolsOutputParser extends BaseLLMOutputParser {
|
|
6
|
-
static lc_name() {
|
|
7
|
-
return 'AnthropicToolsOutputParser';
|
|
8
|
-
}
|
|
9
|
-
lc_namespace = ['langchain', 'anthropic', 'output_parsers'];
|
|
10
|
-
returnId = false;
|
|
11
|
-
/** The type of tool calls to return. */
|
|
12
|
-
keyName;
|
|
13
|
-
/** Whether to return only the first tool call. */
|
|
14
|
-
returnSingle = false;
|
|
15
|
-
zodSchema;
|
|
16
|
-
constructor(params) {
|
|
17
|
-
super(params);
|
|
18
|
-
this.keyName = params.keyName;
|
|
19
|
-
this.returnSingle = params.returnSingle ?? this.returnSingle;
|
|
20
|
-
this.zodSchema = params.zodSchema;
|
|
21
|
-
}
|
|
22
|
-
async _validateResult(result) {
|
|
23
|
-
let parsedResult = result;
|
|
24
|
-
if (typeof result === 'string') {
|
|
25
|
-
try {
|
|
26
|
-
parsedResult = JSON.parse(result);
|
|
27
|
-
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
28
|
-
}
|
|
29
|
-
catch (e) {
|
|
30
|
-
throw new OutputParserException(`Failed to parse. Text: "${JSON.stringify(result, null, 2)}". Error: ${JSON.stringify(e.message)}`, result);
|
|
31
|
-
}
|
|
32
|
-
}
|
|
33
|
-
else {
|
|
34
|
-
parsedResult = result;
|
|
35
|
-
}
|
|
36
|
-
if (this.zodSchema === undefined) {
|
|
37
|
-
return parsedResult;
|
|
38
|
-
}
|
|
39
|
-
const zodParsedResult = await interopSafeParseAsync(this.zodSchema, parsedResult);
|
|
40
|
-
if (zodParsedResult.success) {
|
|
41
|
-
return zodParsedResult.data;
|
|
42
|
-
}
|
|
43
|
-
else {
|
|
44
|
-
throw new OutputParserException(`Failed to parse. Text: "${JSON.stringify(result, null, 2)}". Error: ${JSON.stringify(zodParsedResult.error.issues)}`, JSON.stringify(parsedResult, null, 2));
|
|
45
|
-
}
|
|
46
|
-
}
|
|
47
|
-
async parseResult(generations) {
|
|
48
|
-
const tools = generations.flatMap((generation) => {
|
|
49
|
-
const { message } = generation;
|
|
50
|
-
if (!Array.isArray(message.content)) {
|
|
51
|
-
return [];
|
|
52
|
-
}
|
|
53
|
-
const tool = extractToolCalls(message.content)[0];
|
|
54
|
-
return tool;
|
|
55
|
-
});
|
|
56
|
-
if (tools[0] === undefined) {
|
|
57
|
-
throw new Error('No parseable tool calls provided to AnthropicToolsOutputParser.');
|
|
58
|
-
}
|
|
59
|
-
const [tool] = tools;
|
|
60
|
-
const validatedResult = await this._validateResult(tool.args);
|
|
61
|
-
return validatedResult;
|
|
62
|
-
}
|
|
63
|
-
}
|
|
64
|
-
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
65
|
-
export function extractToolCalls(content) {
|
|
66
|
-
const toolCalls = [];
|
|
67
|
-
for (const block of content) {
|
|
68
|
-
if (block.type === 'tool_use') {
|
|
69
|
-
toolCalls.push({
|
|
70
|
-
name: block.name,
|
|
71
|
-
args: block.input,
|
|
72
|
-
id: block.id,
|
|
73
|
-
type: 'tool_call',
|
|
74
|
-
});
|
|
75
|
-
}
|
|
76
|
-
else if (block.type === 'server_tool_use' &&
|
|
77
|
-
block.name === 'web_search') {
|
|
78
|
-
// Handle Anthropic built-in web search tool
|
|
79
|
-
toolCalls.push({
|
|
80
|
-
name: block.name,
|
|
81
|
-
args: block.input,
|
|
82
|
-
id: block.id,
|
|
83
|
-
type: 'tool_call',
|
|
84
|
-
});
|
|
85
|
-
}
|
|
86
|
-
}
|
|
87
|
-
return toolCalls;
|
|
88
|
-
}
|
|
89
|
-
//# sourceMappingURL=output_parsers.js.map
|
|
@@ -1,25 +0,0 @@
|
|
|
1
|
-
export function handleToolChoice(toolChoice) {
|
|
2
|
-
if (toolChoice == null) {
|
|
3
|
-
return undefined;
|
|
4
|
-
}
|
|
5
|
-
else if (toolChoice === 'any') {
|
|
6
|
-
return {
|
|
7
|
-
type: 'any',
|
|
8
|
-
};
|
|
9
|
-
}
|
|
10
|
-
else if (toolChoice === 'auto') {
|
|
11
|
-
return {
|
|
12
|
-
type: 'auto',
|
|
13
|
-
};
|
|
14
|
-
}
|
|
15
|
-
else if (typeof toolChoice === 'string') {
|
|
16
|
-
return {
|
|
17
|
-
type: 'tool',
|
|
18
|
-
name: toolChoice,
|
|
19
|
-
};
|
|
20
|
-
}
|
|
21
|
-
else {
|
|
22
|
-
return toolChoice;
|
|
23
|
-
}
|
|
24
|
-
}
|
|
25
|
-
//# sourceMappingURL=tools.js.map
|