neoctl 0.2.0 → 0.2.2
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +20 -9
- package/dist/context/compaction.d.ts +3 -0
- package/dist/context/compaction.js +231 -40
- package/dist/context/compaction.js.map +1 -1
- package/dist/context/prompts.js +4 -4
- package/dist/context/prompts.js.map +1 -1
- package/dist/core/commands.d.ts +66 -0
- package/dist/core/commands.js +151 -0
- package/dist/core/commands.js.map +1 -0
- package/dist/core/context-metrics.d.ts +5 -2
- package/dist/core/context-metrics.js +85 -9
- package/dist/core/context-metrics.js.map +1 -1
- package/dist/core/image-registry.d.ts +42 -0
- package/dist/core/image-registry.js +128 -0
- package/dist/core/image-registry.js.map +1 -0
- package/dist/core/message-pipeline.d.ts +6 -0
- package/dist/core/message-pipeline.js +12 -0
- package/dist/core/message-pipeline.js.map +1 -1
- package/dist/core/query-engine.d.ts +0 -4
- package/dist/core/query-engine.js +0 -22
- package/dist/core/query-engine.js.map +1 -1
- package/dist/core/query.d.ts +0 -4
- package/dist/core/query.js +21 -8
- package/dist/core/query.js.map +1 -1
- package/dist/core/runtime.d.ts +59 -0
- package/dist/core/runtime.js +161 -0
- package/dist/core/runtime.js.map +1 -0
- package/dist/core/state.d.ts +6 -7
- package/dist/core/state.js +2 -0
- package/dist/core/state.js.map +1 -1
- package/dist/index.d.ts +4 -1
- package/dist/index.js +4 -1
- package/dist/index.js.map +1 -1
- package/dist/model/anthropic-adapter.d.ts +30 -0
- package/dist/model/anthropic-adapter.js +106 -0
- package/dist/model/anthropic-adapter.js.map +1 -0
- package/dist/model/anthropic-mapper.d.ts +13 -0
- package/dist/model/anthropic-mapper.js +482 -0
- package/dist/model/anthropic-mapper.js.map +1 -0
- package/dist/model/config.d.ts +8 -2
- package/dist/model/config.js +24 -1
- package/dist/model/config.js.map +1 -1
- package/dist/model/openai-mappers.js +1 -1
- package/dist/model/openai-mappers.js.map +1 -1
- package/dist/model/provider-factory.js +17 -0
- package/dist/model/provider-factory.js.map +1 -1
- package/dist/model/smoke-anthropic-mapper.d.ts +1 -0
- package/dist/model/smoke-anthropic-mapper.js +85 -0
- package/dist/model/smoke-anthropic-mapper.js.map +1 -0
- package/dist/repl/index.js +56 -27
- package/dist/repl/index.js.map +1 -1
- package/dist/repl/transcript-flush.d.ts +6 -0
- package/dist/repl/transcript-flush.js +54 -0
- package/dist/repl/transcript-flush.js.map +1 -0
- package/dist/repl/transcript-format.d.ts +16 -0
- package/dist/repl/transcript-format.js +199 -0
- package/dist/repl/transcript-format.js.map +1 -0
- package/dist/server/agent-manager.d.ts +56 -0
- package/dist/server/agent-manager.js +171 -0
- package/dist/server/agent-manager.js.map +1 -0
- package/dist/server/connection-manager.d.ts +18 -0
- package/dist/server/connection-manager.js +52 -0
- package/dist/server/connection-manager.js.map +1 -0
- package/dist/server/event-mapper.d.ts +4 -0
- package/dist/server/event-mapper.js +31 -0
- package/dist/server/event-mapper.js.map +1 -0
- package/dist/server/query-runtime.d.ts +29 -0
- package/dist/server/query-runtime.js +61 -0
- package/dist/server/query-runtime.js.map +1 -0
- package/dist/server/rpc-router.d.ts +24 -0
- package/dist/server/rpc-router.js +165 -0
- package/dist/server/rpc-router.js.map +1 -0
- package/dist/server/rpc-types.d.ts +92 -0
- package/dist/server/rpc-types.js +2 -0
- package/dist/server/rpc-types.js.map +1 -0
- package/dist/server/ws-server.d.ts +22 -0
- package/dist/server/ws-server.js +84 -0
- package/dist/server/ws-server.js.map +1 -0
- package/dist/tasks/task-record.d.ts +50 -0
- package/dist/tasks/task-record.js +50 -0
- package/dist/tasks/task-record.js.map +1 -0
- package/dist/tips.js +1 -1
- package/dist/tips.js.map +1 -1
- package/dist/tools/builtins/image-loader-tool.d.ts +22 -0
- package/dist/tools/builtins/image-loader-tool.js +235 -0
- package/dist/tools/builtins/image-loader-tool.js.map +1 -0
- package/dist/web/html.js +3 -4
- package/dist/web/html.js.map +1 -1
- package/dist/web/index.d.ts +6 -0
- package/dist/web/index.js +60 -20
- package/dist/web/index.js.map +1 -1
- package/package.json +3 -1
|
@@ -0,0 +1,161 @@
|
|
|
1
|
+
import path from "node:path";
|
|
2
|
+
import { TaskStore } from "../tasks/task-store.js";
|
|
3
|
+
import { createTaskTools } from "../tasks/task-tools.js";
|
|
4
|
+
import { loadDefaultDotEnvFiles } from "../model/env.js";
|
|
5
|
+
import { readModelProviderConfig } from "../model/config.js";
|
|
6
|
+
import { CommunicationLogger, LoggingModelGateway } from "../model/communication-logger.js";
|
|
7
|
+
import { createModelGatewayFromProcessEnv } from "../model/provider-factory.js";
|
|
8
|
+
import { ToolRegistry } from "../tools/registry.js";
|
|
9
|
+
import { editTool, writeTool } from "../tools/builtins/edit-tool.js";
|
|
10
|
+
import { createExecTool } from "../tools/builtins/exec-tool.js";
|
|
11
|
+
import { listDirectoryTool, readFileTool } from "../tools/builtins/filesystem-tools.js";
|
|
12
|
+
import { grepTool } from "../tools/builtins/grep-tool.js";
|
|
13
|
+
import { searchTool } from "../tools/builtins/search-tool.js";
|
|
14
|
+
import { planTool } from "../tools/builtins/plan-tool.js";
|
|
15
|
+
import { createOpenAIImageGenerationTool } from "../tools/builtins/image-generation-tool.js";
|
|
16
|
+
import { createLoadImageTool } from "../tools/builtins/image-loader-tool.js";
|
|
17
|
+
import { QueryEngine } from "./query-engine.js";
|
|
18
|
+
import { commandUsages } from "./commands.js";
|
|
19
|
+
export class SessionUsageTracker {
|
|
20
|
+
totals = emptyUsageTotals();
|
|
21
|
+
lastUsage;
|
|
22
|
+
add(usage) {
|
|
23
|
+
if (usage === this.lastUsage)
|
|
24
|
+
return;
|
|
25
|
+
this.lastUsage = usage;
|
|
26
|
+
const inputTokens = usageTokenValue(usage.inputTokens);
|
|
27
|
+
const outputTokens = usageTokenValue(usage.outputTokens);
|
|
28
|
+
const reportedTotalTokens = usageTokenValue(usage.totalTokens);
|
|
29
|
+
const computedTotalTokens = reportedTotalTokens ?? sumUsageTokens(inputTokens, outputTokens);
|
|
30
|
+
const reasoningTokens = usageTokenValue(usage.reasoningTokens);
|
|
31
|
+
const cachedTokens = usageTokenValue(usage.cachedTokens);
|
|
32
|
+
if (inputTokens === undefined && outputTokens === undefined && computedTotalTokens === undefined && reasoningTokens === undefined && cachedTokens === undefined)
|
|
33
|
+
return;
|
|
34
|
+
this.totals = {
|
|
35
|
+
inputTokens: this.totals.inputTokens + (inputTokens ?? 0),
|
|
36
|
+
outputTokens: this.totals.outputTokens + (outputTokens ?? 0),
|
|
37
|
+
totalTokens: this.totals.totalTokens + (computedTotalTokens ?? 0),
|
|
38
|
+
reasoningTokens: this.totals.reasoningTokens + (reasoningTokens ?? 0),
|
|
39
|
+
cachedTokens: this.totals.cachedTokens + (cachedTokens ?? 0),
|
|
40
|
+
requests: this.totals.requests + 1,
|
|
41
|
+
computedTotalTokens: this.totals.computedTotalTokens || (reportedTotalTokens === undefined && computedTotalTokens !== undefined),
|
|
42
|
+
};
|
|
43
|
+
}
|
|
44
|
+
reset() {
|
|
45
|
+
this.totals = emptyUsageTotals();
|
|
46
|
+
this.lastUsage = undefined;
|
|
47
|
+
}
|
|
48
|
+
snapshot() {
|
|
49
|
+
return { ...this.totals };
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
function emptyUsageTotals() {
|
|
53
|
+
return { inputTokens: 0, outputTokens: 0, totalTokens: 0, reasoningTokens: 0, cachedTokens: 0, requests: 0, computedTotalTokens: false };
|
|
54
|
+
}
|
|
55
|
+
function usageTokenValue(value) {
|
|
56
|
+
return typeof value === "number" && Number.isFinite(value) && value >= 0 ? value : undefined;
|
|
57
|
+
}
|
|
58
|
+
function sumUsageTokens(left, right) {
|
|
59
|
+
if (left === undefined && right === undefined)
|
|
60
|
+
return undefined;
|
|
61
|
+
return (left ?? 0) + (right ?? 0);
|
|
62
|
+
}
|
|
63
|
+
export function createCoreServices() {
|
|
64
|
+
const envLoad = loadDefaultDotEnvFiles({ override: true });
|
|
65
|
+
const modelConfig = readModelProviderConfig(process.env);
|
|
66
|
+
const communicationLogger = new CommunicationLogger();
|
|
67
|
+
const modelGateway = new LoggingModelGateway(createModelGatewayFromProcessEnv(process.env), communicationLogger);
|
|
68
|
+
const taskStore = new TaskStore();
|
|
69
|
+
const tools = new ToolRegistry();
|
|
70
|
+
tools.register(editTool);
|
|
71
|
+
tools.register(writeTool);
|
|
72
|
+
tools.register(createExecTool({ taskStore }));
|
|
73
|
+
tools.register(listDirectoryTool);
|
|
74
|
+
tools.register(readFileTool);
|
|
75
|
+
tools.register(grepTool);
|
|
76
|
+
tools.register(searchTool);
|
|
77
|
+
tools.register(createLoadImageTool());
|
|
78
|
+
if (modelConfig?.provider === "openai")
|
|
79
|
+
tools.register(createOpenAIImageGenerationTool());
|
|
80
|
+
tools.register(planTool);
|
|
81
|
+
for (const tool of createTaskTools(taskStore))
|
|
82
|
+
tools.register(tool);
|
|
83
|
+
return {
|
|
84
|
+
communicationLogger,
|
|
85
|
+
modelGateway,
|
|
86
|
+
taskStore,
|
|
87
|
+
tools,
|
|
88
|
+
defaultReasoning: modelConfig?.defaultReasoning,
|
|
89
|
+
envPath: process.env.NEO_ENV_FILE?.trim() ? path.resolve(process.env.NEO_ENV_FILE.trim()) : envLoad.userDotEnvPath,
|
|
90
|
+
envNotice: envLoad.createdUserDotEnv ? formatCreatedEnvNotice(envLoad.userDotEnvPath) : undefined,
|
|
91
|
+
};
|
|
92
|
+
}
|
|
93
|
+
export async function createQueryEngineRuntime(services, options = {}) {
|
|
94
|
+
const modelConfig = readModelProviderConfig(process.env);
|
|
95
|
+
const engine = new QueryEngine({
|
|
96
|
+
agentId: options.agentId ?? "main",
|
|
97
|
+
model: modelConfig?.model,
|
|
98
|
+
fallbackModel: modelConfig?.fallbackModel,
|
|
99
|
+
reasoning: modelConfig?.defaultReasoning,
|
|
100
|
+
queryOrigin: options.queryOrigin,
|
|
101
|
+
modelGateway: services.modelGateway,
|
|
102
|
+
tools: services.tools,
|
|
103
|
+
taskNotificationSource: createTaskNotificationSource(services.taskStore),
|
|
104
|
+
commands: commandUsages,
|
|
105
|
+
session: {
|
|
106
|
+
enabled: process.env.AGENT_SESSION_TRANSCRIPT !== "0",
|
|
107
|
+
sessionId: options.sessionId ?? process.env.AGENT_SESSION_ID,
|
|
108
|
+
rootDir: process.env.AGENT_SESSION_DIR,
|
|
109
|
+
cwd: options.sessionCwd,
|
|
110
|
+
resume: options.resume ?? parseResumeFlag(process.env.AGENT_SESSION_RESUME),
|
|
111
|
+
toolResultThresholdChars: process.env.AGENT_TOOL_RESULT_THRESHOLD_CHARS ? Number(process.env.AGENT_TOOL_RESULT_THRESHOLD_CHARS) : undefined,
|
|
112
|
+
},
|
|
113
|
+
});
|
|
114
|
+
await engine.initialize();
|
|
115
|
+
const initialMetrics = await engine.contextMetrics();
|
|
116
|
+
return {
|
|
117
|
+
engine,
|
|
118
|
+
communicationLogger: services.communicationLogger,
|
|
119
|
+
modelGateway: services.modelGateway,
|
|
120
|
+
usage: new SessionUsageTracker(),
|
|
121
|
+
taskStore: services.taskStore,
|
|
122
|
+
tools: services.tools,
|
|
123
|
+
initialMetrics,
|
|
124
|
+
defaultReasoning: services.defaultReasoning,
|
|
125
|
+
envPath: services.envPath,
|
|
126
|
+
envNotice: services.envNotice,
|
|
127
|
+
};
|
|
128
|
+
}
|
|
129
|
+
export async function createCoreRuntime(options = {}) {
|
|
130
|
+
return createQueryEngineRuntime(createCoreServices(), options);
|
|
131
|
+
}
|
|
132
|
+
export function createTaskNotificationSource(taskStore) {
|
|
133
|
+
return {
|
|
134
|
+
collectUnnotifiedCompletions() {
|
|
135
|
+
return taskStore.collectUnnotifiedCompletions().map((task) => ({
|
|
136
|
+
taskId: task.taskId,
|
|
137
|
+
agentId: task.agentId,
|
|
138
|
+
status: task.status,
|
|
139
|
+
type: task.type,
|
|
140
|
+
content: task.result?.content ?? task.error ?? "",
|
|
141
|
+
}));
|
|
142
|
+
},
|
|
143
|
+
markNotified(taskId) {
|
|
144
|
+
taskStore.markNotified(taskId);
|
|
145
|
+
},
|
|
146
|
+
};
|
|
147
|
+
}
|
|
148
|
+
export function syncImageGenerationTool(runtime, provider) {
|
|
149
|
+
runtime.tools.unregister("image2");
|
|
150
|
+
if (provider === "openai")
|
|
151
|
+
runtime.tools.register(createOpenAIImageGenerationTool());
|
|
152
|
+
}
|
|
153
|
+
export function formatCreatedEnvNotice(dotEnvPath) {
|
|
154
|
+
return `Created default config file: ${dotEnvPath}\nSet MODEL_PROVIDER and the matching provider section (for example OPENAI_API_KEY, ANTHROPIC_API_KEY, or KIMI_API_KEY), then restart neo.`;
|
|
155
|
+
}
|
|
156
|
+
export function parseResumeFlag(value) {
|
|
157
|
+
if (!value)
|
|
158
|
+
return false;
|
|
159
|
+
return ["1", "true", "yes", "latest"].includes(value.toLowerCase());
|
|
160
|
+
}
|
|
161
|
+
//# sourceMappingURL=runtime.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"runtime.js","sourceRoot":"","sources":["../../src/core/runtime.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,sBAAsB,EAAE,MAAM,iBAAiB,CAAC;AACzD,OAAO,EAAE,uBAAuB,EAA0B,MAAM,oBAAoB,CAAC;AACrF,OAAO,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;AAC5F,OAAO,EAAE,gCAAgC,EAAE,MAAM,8BAA8B,CAAC;AAEhF,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,gCAAgC,CAAC;AACrE,OAAO,EAAE,cAAc,EAAE,MAAM,gCAAgC,CAAC;AAChE,OAAO,EAAE,iBAAiB,EAAE,YAAY,EAAE,MAAM,uCAAuC,CAAC;AACxF,OAAO,EAAE,QAAQ,EAAE,MAAM,gCAAgC,CAAC;AAC1D,OAAO,EAAE,UAAU,EAAE,MAAM,kCAAkC,CAAC;AAC9D,OAAO,EAAE,QAAQ,EAAE,MAAM,gCAAgC,CAAC;AAC1D,OAAO,EAAE,+BAA+B,EAAE,MAAM,4CAA4C,CAAC;AAC7F,OAAO,EAAE,mBAAmB,EAAE,MAAM,wCAAwC,CAAC;AAG7E,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AA2C9C,MAAM,OAAO,mBAAmB;IACtB,MAAM,GAAgB,gBAAgB,EAAE,CAAC;IACzC,SAAS,CAAc;IAE/B,GAAG,CAAC,KAAiB;QACnB,IAAI,KAAK,KAAK,IAAI,CAAC,SAAS;YAAE,OAAO;QACrC,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,MAAM,WAAW,GAAG,eAAe,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QACvD,MAAM,YAAY,GAAG,eAAe,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACzD,MAAM,mBAAmB,GAAG,eAAe,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QAC/D,MAAM,mBAAmB,GAAG,mBAAmB,IAAI,cAAc,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;QAC7F,MAAM,eAAe,GAAG,eAAe,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;QAC/D,MAAM,YAAY,GAAG,eAAe,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACzD,IAAI,WAAW,KAAK,SAAS,IAAI,YAAY,KAAK,SAAS,IAAI,mBAAmB,KAAK,SAAS,IAAI,eAAe,KAAK,SAAS,IAAI,YAAY,KAAK,SAAS;YAAE,OAAO;QACxK,IAAI,CAAC,MAAM,GAAG;YACZ,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC;YACzD,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,GAAG,CAAC,YAAY,IAAI,CAAC,CAAC;YAC5D,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,GAAG,CAAC,mBAAmB,IAAI,CAAC,CAAC;YACjE,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,GAAG,CAAC,eAAe,IAAI,CAAC,CAAC;YACrE,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,GAAG,CAAC,YAAY,IAAI,CAAC,CAAC;YAC5D,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,GAAG,CAAC;YAClC,mBAAmB,EAAE,IAAI,CAAC,MAAM,CAAC,mBAAmB,IAAI,CAAC,mBAAmB,KAAK,SAAS,IAAI,mBAAmB,KAAK,SAAS,CAAC;SACjI,CAAC;IACJ,CAAC;IAED,KAAK;QACH,IAAI,CAAC,MAAM,GAAG,gBAAgB,EAAE,CAAC;QACjC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,QAAQ;QACN,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;CACF;AAED,SAAS,gBAAgB;IACvB,OAAO,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,eAAe,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,mBAAmB,EAAE,KAAK,EAAE,CAAC;AAC3I,CAAC;AAED,SAAS,eAAe,CAAC,KAAyB;IAChD,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;AAC/F,CAAC;AAED,SAAS,cAAc,CAAC,IAAwB,EAAE,KAAyB;IACzE,IAAI,IAAI,KAAK,SAAS,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,SAAS,CAAC;IAChE,OAAO,CAAC,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC;AACpC,CAAC;AAED,MAAM,UAAU,kBAAkB;IAChC,MAAM,OAAO,GAAG,sBAAsB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3D,MAAM,WAAW,GAAG,uBAAuB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACzD,MAAM,mBAAmB,GAAG,IAAI,mBAAmB,EAAE,CAAC;IACtD,MAAM,YAAY,GAAG,IAAI,mBAAmB,CAAC,gCAAgC,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,mBAAmB,CAAC,CAAC;IACjH,MAAM,SAAS,GAAG,IAAI,SAAS,EAAE,CAAC;IAClC,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAC;IACjC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACzB,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IAC1B,KAAK,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;IAC9C,KAAK,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;IAClC,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;IAC7B,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACzB,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;IAC3B,KAAK,CAAC,QAAQ,CAAC,mBAAmB,EAAE,CAAC,CAAC;IACtC,IAAI,WAAW,EAAE,QAAQ,KAAK,QAAQ;QAAE,KAAK,CAAC,QAAQ,CAAC,+BAA+B,EAAE,CAAC,CAAC;IAC1F,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAEzB,KAAK,MAAM,IAAI,IAAI,eAAe,CAAC,SAAS,CAAC;QAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IAEpE,OAAO;QACL,mBAAmB;QACnB,YAAY;QACZ,SAAS;QACT,KAAK;QACL,gBAAgB,EAAE,WAAW,EAAE,gBAAgB;QAC/C,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc;QAClH,SAAS,EAAE,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,sBAAsB,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,SAAS;KAClG,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAAC,QAAsB,EAAE,UAAoC,EAAE;IAC3G,MAAM,WAAW,GAAG,uBAAuB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACzD,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC;QAC7B,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,MAAM;QAClC,KAAK,EAAE,WAAW,EAAE,KAAK;QACzB,aAAa,EAAE,WAAW,EAAE,aAAa;QACzC,SAAS,EAAE,WAAW,EAAE,gBAAgB;QACxC,WAAW,EAAE,OAAO,CAAC,WAAW;QAChC,YAAY,EAAE,QAAQ,CAAC,YAAY;QACnC,KAAK,EAAE,QAAQ,CAAC,KAAK;QACrB,sBAAsB,EAAE,4BAA4B,CAAC,QAAQ,CAAC,SAAS,CAAC;QACxE,QAAQ,EAAE,aAAa;QACvB,OAAO,EAAE;YACP,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,wBAAwB,KAAK,GAAG;YACrD,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB;YAC5D,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB;YACtC,GAAG,EAAE,OAAO,CAAC,UAAU;YACvB,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,eAAe,CAAC,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC;YAC3E,wBAAwB,EAAE,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC,CAAC,CAAC,SAAS;SAC5I;KACF,CAAC,CAAC;IACH,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;IAC1B,MAAM,cAAc,GAAG,MAAM,MAAM,CAAC,cAAc,EAAE,CAAC;IACrD,OAAO;QACL,MAAM;QACN,mBAAmB,EAAE,QAAQ,CAAC,mBAAmB;QACjD,YAAY,EAAE,QAAQ,CAAC,YAAY;QACnC,KAAK,EAAE,IAAI,mBAAmB,EAAE;QAChC,SAAS,EAAE,QAAQ,CAAC,SAAS;QAC7B,KAAK,EAAE,QAAQ,CAAC,KAAK;QACrB,cAAc;QACd,gBAAgB,EAAE,QAAQ,CAAC,gBAAgB;QAC3C,OAAO,EAAE,QAAQ,CAAC,OAAO;QACzB,SAAS,EAAE,QAAQ,CAAC,SAAS;KAC9B,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,UAAoC,EAAE;IAC5E,OAAO,wBAAwB,CAAC,kBAAkB,EAAE,EAAE,OAAO,CAAC,CAAC;AACjE,CAAC;AAED,MAAM,UAAU,4BAA4B,CAAC,SAAoB;IAC/D,OAAO;QACL,4BAA4B;YAC1B,OAAO,SAAS,CAAC,4BAA4B,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBAC7D,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,OAAO,IAAI,IAAI,CAAC,KAAK,IAAI,EAAE;aAClD,CAAC,CAAC,CAAC;QACN,CAAC;QACD,YAAY,CAAC,MAAc;YACzB,SAAS,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QACjC,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,OAAmC,EAAE,QAAuC;IAClH,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;IACnC,IAAI,QAAQ,KAAK,QAAQ;QAAE,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,+BAA+B,EAAE,CAAC,CAAC;AACvF,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,UAAkB;IACvD,OAAO,gCAAgC,UAAU,4IAA4I,CAAC;AAChM,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,KAAyB;IACvD,IAAI,CAAC,KAAK;QAAE,OAAO,KAAK,CAAC;IACzB,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC;AACtE,CAAC"}
|
package/dist/core/state.d.ts
CHANGED
|
@@ -1,8 +1,7 @@
|
|
|
1
1
|
import type { Message } from "../types/messages.js";
|
|
2
|
-
|
|
3
|
-
export type
|
|
4
|
-
export type
|
|
5
|
-
export type TerminalReason = "completed" | "blocking_limit" | "model_error" | "image_error" | "prompt_too_long" | "aborted_streaming" | "aborted_tools" | "hook_stopped" | "stop_hook_prevented" | "max_turns";
|
|
2
|
+
export type QueryPhase = "ready" | "preparing" | "compacting" | "calling_model" | "running_tools" | "injecting_context";
|
|
3
|
+
export type ContinueReason = "initial" | "next_turn" | "max_output_tokens_escalate" | "reactive_compact_retry";
|
|
4
|
+
export type TerminalReason = "completed" | "model_error" | "image_error" | "prompt_too_long" | "aborted_streaming" | "aborted_tools" | "max_turns";
|
|
6
5
|
export interface QueryTracking {
|
|
7
6
|
chainId: string;
|
|
8
7
|
depth: number;
|
|
@@ -22,14 +21,14 @@ export interface QueryState {
|
|
|
22
21
|
fallbackModel?: string;
|
|
23
22
|
queryTracking: QueryTracking;
|
|
24
23
|
maxOutputTokensRecoveryCount: number;
|
|
24
|
+
/** @deprecated Use reactiveCompactAttempts instead */
|
|
25
25
|
hasAttemptedReactiveCompact: boolean;
|
|
26
|
+
reactiveCompactAttempts: number;
|
|
26
27
|
maxOutputTokensOverride?: number;
|
|
27
|
-
pendingToolSummary?: string;
|
|
28
|
-
stopHookActive?: boolean;
|
|
29
28
|
turnCount: number;
|
|
30
29
|
transition: LoopTransition;
|
|
31
|
-
lastUsage?: ModelUsage;
|
|
32
30
|
}
|
|
31
|
+
export declare const MAX_REACTIVE_COMPACT_ATTEMPTS = 3;
|
|
33
32
|
export declare function createInitialState(messages: Message[], options?: {
|
|
34
33
|
model?: string;
|
|
35
34
|
fallbackModel?: string;
|
package/dist/core/state.js
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
1
|
+
export const MAX_REACTIVE_COMPACT_ATTEMPTS = 3;
|
|
1
2
|
export function createInitialState(messages, options) {
|
|
2
3
|
const chainId = cryptoId("chain");
|
|
3
4
|
return {
|
|
@@ -13,6 +14,7 @@ export function createInitialState(messages, options) {
|
|
|
13
14
|
},
|
|
14
15
|
maxOutputTokensRecoveryCount: 0,
|
|
15
16
|
hasAttemptedReactiveCompact: false,
|
|
17
|
+
reactiveCompactAttempts: 0,
|
|
16
18
|
turnCount: 0,
|
|
17
19
|
transition: { reason: "initial" },
|
|
18
20
|
};
|
package/dist/core/state.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"state.js","sourceRoot":"","sources":["../../src/core/state.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"state.js","sourceRoot":"","sources":["../../src/core/state.ts"],"names":[],"mappings":"AAsDA,MAAM,CAAC,MAAM,6BAA6B,GAAG,CAAC,CAAC;AAE/C,MAAM,UAAU,kBAAkB,CAAC,QAAmB,EAAE,OAAoD;IAC1G,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC;IAClC,OAAO;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,CAAC,GAAG,QAAQ,CAAC;QACvB,YAAY,EAAE,OAAO,EAAE,KAAK;QAC5B,aAAa,EAAE,OAAO,EAAE,aAAa;QACrC,aAAa,EAAE;YACb,OAAO;YACP,KAAK,EAAE,CAAC;YACR,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC;YACxB,WAAW,EAAE,CAAC;SACf;QACD,4BAA4B,EAAE,CAAC;QAC/B,2BAA2B,EAAE,KAAK;QAClC,uBAAuB,EAAE,CAAC;QAC1B,SAAS,EAAE,CAAC;QACZ,UAAU,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE;KAClC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,QAAuB;IAClD,OAAO;QACL,OAAO,EAAE,QAAQ,CAAC,OAAO;QACzB,KAAK,EAAE,QAAQ,CAAC,KAAK;QACrB,WAAW,EAAE,QAAQ,CAAC,WAAW,GAAG,CAAC;QACrC,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC;KACzB,CAAC;AACJ,CAAC;AAED,SAAS,QAAQ,CAAC,MAAc;IAC9B,OAAO,GAAG,MAAM,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;AAC1F,CAAC"}
|
package/dist/index.d.ts
CHANGED
|
@@ -24,6 +24,8 @@ export * from "./model/context-window.js";
|
|
|
24
24
|
export * from "./model/provider-factory.js";
|
|
25
25
|
export * from "./model/openai-adapter.js";
|
|
26
26
|
export * from "./model/openai-responses-adapter.js";
|
|
27
|
+
export * from "./model/anthropic-adapter.js";
|
|
28
|
+
export * from "./model/anthropic-mapper.js";
|
|
27
29
|
export * from "./model/deepseek-adapter.js";
|
|
28
30
|
export * from "./model/kimi-adapter.js";
|
|
29
31
|
export * from "./model/provider-adapter.js";
|
|
@@ -54,8 +56,9 @@ export * from "./tools/builtins/search-tool.js";
|
|
|
54
56
|
export * from "./tools/builtins/search-providers.js";
|
|
55
57
|
export * from "./tools/builtins/plan-tool.js";
|
|
56
58
|
export * from "./tools/builtins/image-generation-tool.js";
|
|
57
|
-
export * from "./tools/builtins/
|
|
59
|
+
export * from "./tools/builtins/image-loader-tool.js";
|
|
58
60
|
export * from "./tools/builtins/ripgrep-binary.js";
|
|
61
|
+
export * from "./core/image-registry.js";
|
|
59
62
|
export * from "./types/messages.js";
|
|
60
63
|
export * from "./types/events.js";
|
|
61
64
|
export * from "./ui/display-message.js";
|
package/dist/index.js
CHANGED
|
@@ -24,6 +24,8 @@ export * from "./model/context-window.js";
|
|
|
24
24
|
export * from "./model/provider-factory.js";
|
|
25
25
|
export * from "./model/openai-adapter.js";
|
|
26
26
|
export * from "./model/openai-responses-adapter.js";
|
|
27
|
+
export * from "./model/anthropic-adapter.js";
|
|
28
|
+
export * from "./model/anthropic-mapper.js";
|
|
27
29
|
export * from "./model/deepseek-adapter.js";
|
|
28
30
|
export * from "./model/kimi-adapter.js";
|
|
29
31
|
export * from "./model/provider-adapter.js";
|
|
@@ -54,8 +56,9 @@ export * from "./tools/builtins/search-tool.js";
|
|
|
54
56
|
export * from "./tools/builtins/search-providers.js";
|
|
55
57
|
export * from "./tools/builtins/plan-tool.js";
|
|
56
58
|
export * from "./tools/builtins/image-generation-tool.js";
|
|
57
|
-
export * from "./tools/builtins/
|
|
59
|
+
export * from "./tools/builtins/image-loader-tool.js";
|
|
58
60
|
export * from "./tools/builtins/ripgrep-binary.js";
|
|
61
|
+
export * from "./core/image-registry.js";
|
|
59
62
|
export * from "./types/messages.js";
|
|
60
63
|
export * from "./types/events.js";
|
|
61
64
|
export * from "./ui/display-message.js";
|
package/dist/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,wBAAwB,CAAC;AACvC,cAAc,iBAAiB,CAAC;AAChC,cAAc,iBAAiB,CAAC;AAChC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,qBAAqB,CAAC;AACpC,cAAc,mCAAmC,CAAC;AAClD,cAAc,2BAA2B,CAAC;AAC1C,cAAc,sBAAsB,CAAC;AACrC,cAAc,8BAA8B,CAAC;AAC7C,cAAc,yBAAyB,CAAC;AACxC,cAAc,8BAA8B,CAAC;AAC7C,cAAc,wBAAwB,CAAC;AACvC,cAAc,8BAA8B,CAAC;AAC7C,cAAc,kBAAkB,CAAC;AACjC,cAAc,uBAAuB,CAAC;AACtC,cAAc,uBAAuB,CAAC;AACtC,cAAc,wBAAwB,CAAC;AACvC,cAAc,8BAA8B,CAAC;AAC7C,cAAc,oCAAoC,CAAC;AACnD,cAAc,0BAA0B,CAAC;AACzC,cAAc,mBAAmB,CAAC;AAClC,cAAc,iCAAiC,CAAC;AAChD,cAAc,2BAA2B,CAAC;AAC1C,cAAc,6BAA6B,CAAC;AAC5C,cAAc,2BAA2B,CAAC;AAC1C,cAAc,qCAAqC,CAAC;AACpD,cAAc,6BAA6B,CAAC;AAC5C,cAAc,yBAAyB,CAAC;AACxC,cAAc,6BAA6B,CAAC;AAC5C,cAAc,wBAAwB,CAAC;AACvC,cAAc,2BAA2B,CAAC;AAC1C,cAAc,+BAA+B,CAAC;AAC9C,cAAc,2BAA2B,CAAC;AAC1C,cAAc,oCAAoC,CAAC;AACnD,cAAc,yBAAyB,CAAC;AACxC,cAAc,wBAAwB,CAAC;AACvC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,mBAAmB,CAAC;AAClC,cAAc,iBAAiB,CAAC;AAChC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,6BAA6B,CAAC;AAC5C,cAAc,iCAAiC,CAAC;AAChD,cAAc,qCAAqC,CAAC;AACpD,cAAc,qBAAqB,CAAC;AACpC,cAAc,yBAAyB,CAAC;AACxC,cAAc,+BAA+B,CAAC;AAC9C,cAAc,oCAAoC,CAAC;AACnD,cAAc,mBAAmB,CAAC;AAClC,cAAc,+BAA+B,CAAC;AAC9C,cAAc,+BAA+B,CAAC;AAC9C,cAAc,sCAAsC,CAAC;AACrD,cAAc,+BAA+B,CAAC;AAC9C,cAAc,iCAAiC,CAAC;AAChD,cAAc,sCAAsC,CAAC;AACrD,cAAc,+BAA+B,CAAC;AAC9C,cAAc,2CAA2C,CAAC;AAC1D,cAAc,
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,wBAAwB,CAAC;AACvC,cAAc,iBAAiB,CAAC;AAChC,cAAc,iBAAiB,CAAC;AAChC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,qBAAqB,CAAC;AACpC,cAAc,mCAAmC,CAAC;AAClD,cAAc,2BAA2B,CAAC;AAC1C,cAAc,sBAAsB,CAAC;AACrC,cAAc,8BAA8B,CAAC;AAC7C,cAAc,yBAAyB,CAAC;AACxC,cAAc,8BAA8B,CAAC;AAC7C,cAAc,wBAAwB,CAAC;AACvC,cAAc,8BAA8B,CAAC;AAC7C,cAAc,kBAAkB,CAAC;AACjC,cAAc,uBAAuB,CAAC;AACtC,cAAc,uBAAuB,CAAC;AACtC,cAAc,wBAAwB,CAAC;AACvC,cAAc,8BAA8B,CAAC;AAC7C,cAAc,oCAAoC,CAAC;AACnD,cAAc,0BAA0B,CAAC;AACzC,cAAc,mBAAmB,CAAC;AAClC,cAAc,iCAAiC,CAAC;AAChD,cAAc,2BAA2B,CAAC;AAC1C,cAAc,6BAA6B,CAAC;AAC5C,cAAc,2BAA2B,CAAC;AAC1C,cAAc,qCAAqC,CAAC;AACpD,cAAc,8BAA8B,CAAC;AAC7C,cAAc,6BAA6B,CAAC;AAC5C,cAAc,6BAA6B,CAAC;AAC5C,cAAc,yBAAyB,CAAC;AACxC,cAAc,6BAA6B,CAAC;AAC5C,cAAc,wBAAwB,CAAC;AACvC,cAAc,2BAA2B,CAAC;AAC1C,cAAc,+BAA+B,CAAC;AAC9C,cAAc,2BAA2B,CAAC;AAC1C,cAAc,oCAAoC,CAAC;AACnD,cAAc,yBAAyB,CAAC;AACxC,cAAc,wBAAwB,CAAC;AACvC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,mBAAmB,CAAC;AAClC,cAAc,iBAAiB,CAAC;AAChC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,6BAA6B,CAAC;AAC5C,cAAc,iCAAiC,CAAC;AAChD,cAAc,qCAAqC,CAAC;AACpD,cAAc,qBAAqB,CAAC;AACpC,cAAc,yBAAyB,CAAC;AACxC,cAAc,+BAA+B,CAAC;AAC9C,cAAc,oCAAoC,CAAC;AACnD,cAAc,mBAAmB,CAAC;AAClC,cAAc,+BAA+B,CAAC;AAC9C,cAAc,+BAA+B,CAAC;AAC9C,cAAc,sCAAsC,CAAC;AACrD,cAAc,+BAA+B,CAAC;AAC9C,cAAc,iCAAiC,CAAC;AAChD,cAAc,sCAAsC,CAAC;AACrD,cAAc,+BAA+B,CAAC;AAC9C,cAAc,2CAA2C,CAAC;AAC1D,cAAc,uCAAuC,CAAC;AACtD,cAAc,oCAAoC,CAAC;AACnD,cAAc,0BAA0B,CAAC;AACzC,cAAc,qBAAqB,CAAC;AACpC,cAAc,mBAAmB,CAAC;AAClC,cAAc,yBAAyB,CAAC;AACxC,cAAc,oBAAoB,CAAC;AACnC,cAAc,mBAAmB,CAAC;AAClC,cAAc,yBAAyB,CAAC;AACxC,cAAc,qBAAqB,CAAC;AACpC,cAAc,qBAAqB,CAAC;AACpC,cAAc,YAAY,CAAC;AAC3B,cAAc,WAAW,CAAC;AAC1B,cAAc,gBAAgB,CAAC"}
|
|
@@ -0,0 +1,30 @@
|
|
|
1
|
+
import type { CredentialProvider } from "./credentials.js";
|
|
2
|
+
import type { ModelGateway, ModelRequest, ModelStreamEvent, ReasoningConfig } from "./model-gateway.js";
|
|
3
|
+
import type { ProviderAdapter, ProviderCapabilities } from "./provider-adapter.js";
|
|
4
|
+
export interface AnthropicAdapterOptions {
|
|
5
|
+
apiKey?: string;
|
|
6
|
+
apiKeyEnvName?: string;
|
|
7
|
+
credentialProvider?: CredentialProvider;
|
|
8
|
+
baseUrl?: string;
|
|
9
|
+
model: string;
|
|
10
|
+
fallbackModel?: string;
|
|
11
|
+
anthropicVersion?: string;
|
|
12
|
+
timeoutMs?: number;
|
|
13
|
+
streamIdleTimeoutMs?: number;
|
|
14
|
+
maxRetries?: number;
|
|
15
|
+
defaultMaxOutputTokens?: number;
|
|
16
|
+
defaultReasoning?: ReasoningConfig | null;
|
|
17
|
+
}
|
|
18
|
+
export declare class AnthropicAdapter implements ProviderAdapter, ModelGateway {
|
|
19
|
+
private readonly options;
|
|
20
|
+
readonly name = "anthropic";
|
|
21
|
+
readonly capabilities: ProviderCapabilities;
|
|
22
|
+
private readonly credentialProvider;
|
|
23
|
+
private readonly transport;
|
|
24
|
+
private readonly baseUrl;
|
|
25
|
+
constructor(options: AnthropicAdapterOptions);
|
|
26
|
+
stream(request: ModelRequest): AsyncIterable<ModelStreamEvent>;
|
|
27
|
+
private streamMessages;
|
|
28
|
+
private assertRequestCapabilities;
|
|
29
|
+
private authHeaders;
|
|
30
|
+
}
|
|
@@ -0,0 +1,106 @@
|
|
|
1
|
+
import { EnvCredentialProvider, StaticCredentialProvider } from "./credentials.js";
|
|
2
|
+
import { ModelAPIError } from "./errors.js";
|
|
3
|
+
import { HttpTransport } from "./http-transport.js";
|
|
4
|
+
import { supportsImageInput } from "./context-window.js";
|
|
5
|
+
import { streamWithRetry } from "./retry-runner.js";
|
|
6
|
+
import { buildAnthropicRequest, normalizeAnthropicObject, normalizeAnthropicStream } from "./anthropic-mapper.js";
|
|
7
|
+
const DEFAULT_ANTHROPIC_VERSION = "2023-06-01";
|
|
8
|
+
export class AnthropicAdapter {
|
|
9
|
+
options;
|
|
10
|
+
name = "anthropic";
|
|
11
|
+
capabilities = {
|
|
12
|
+
streaming: true,
|
|
13
|
+
functionTools: true,
|
|
14
|
+
parallelToolCalls: false,
|
|
15
|
+
structuredOutput: false,
|
|
16
|
+
previousResponseId: false,
|
|
17
|
+
imageInput: true,
|
|
18
|
+
fileInput: false,
|
|
19
|
+
reasoningConfig: true,
|
|
20
|
+
builtInTools: false,
|
|
21
|
+
};
|
|
22
|
+
credentialProvider;
|
|
23
|
+
transport = new HttpTransport(this.name);
|
|
24
|
+
baseUrl;
|
|
25
|
+
constructor(options) {
|
|
26
|
+
this.options = options;
|
|
27
|
+
this.credentialProvider =
|
|
28
|
+
options.credentialProvider ??
|
|
29
|
+
(options.apiKey ? new StaticCredentialProvider(options.apiKey) : new EnvCredentialProvider(options.apiKeyEnvName ?? "ANTHROPIC_API_KEY"));
|
|
30
|
+
this.baseUrl = stripTrailingSlash(options.baseUrl ?? "https://api.anthropic.com");
|
|
31
|
+
}
|
|
32
|
+
async *stream(request) {
|
|
33
|
+
this.assertRequestCapabilities(request);
|
|
34
|
+
yield* streamWithRetry((attempt) => this.streamMessages(request, attempt), {
|
|
35
|
+
provider: this.name,
|
|
36
|
+
maxRetries: this.options.maxRetries ?? 2,
|
|
37
|
+
delayMs: 3000,
|
|
38
|
+
});
|
|
39
|
+
}
|
|
40
|
+
async *streamMessages(request, attempt) {
|
|
41
|
+
const body = buildAnthropicRequest(request, this.options);
|
|
42
|
+
const headers = await this.authHeaders();
|
|
43
|
+
const timeoutMs = request.timeoutMs ?? this.options.timeoutMs ?? 120000;
|
|
44
|
+
if (attempt > 0)
|
|
45
|
+
yield { type: "provider_event", event: { type: "anthropic_retry_attempt", attempt } };
|
|
46
|
+
if (request.stream !== false) {
|
|
47
|
+
const response = await this.transport.sendStream({
|
|
48
|
+
method: "POST",
|
|
49
|
+
url: `${this.baseUrl}/v1/messages`,
|
|
50
|
+
headers,
|
|
51
|
+
body: { ...body, stream: true },
|
|
52
|
+
timeoutMs,
|
|
53
|
+
signal: request.cancellation,
|
|
54
|
+
});
|
|
55
|
+
yield* normalizeAnthropicStream(response.body, this.options);
|
|
56
|
+
return;
|
|
57
|
+
}
|
|
58
|
+
const response = await this.transport.sendJson({
|
|
59
|
+
method: "POST",
|
|
60
|
+
url: `${this.baseUrl}/v1/messages`,
|
|
61
|
+
headers,
|
|
62
|
+
body: { ...body, stream: false },
|
|
63
|
+
timeoutMs,
|
|
64
|
+
signal: request.cancellation,
|
|
65
|
+
});
|
|
66
|
+
yield* normalizeAnthropicObject(response);
|
|
67
|
+
}
|
|
68
|
+
assertRequestCapabilities(request) {
|
|
69
|
+
const model = request.model ?? this.options.model;
|
|
70
|
+
const supported = supportsImageInput(model);
|
|
71
|
+
if (supported === false && hasImageInput(request)) {
|
|
72
|
+
throw new ModelAPIError({
|
|
73
|
+
category: "unsupported_image_input",
|
|
74
|
+
provider: this.name,
|
|
75
|
+
message: `Model ${model} does not support image input according to static model metadata`,
|
|
76
|
+
retryable: false,
|
|
77
|
+
});
|
|
78
|
+
}
|
|
79
|
+
}
|
|
80
|
+
async authHeaders() {
|
|
81
|
+
const credential = await this.credentialProvider.getCredential();
|
|
82
|
+
if (!credential) {
|
|
83
|
+
throw new ModelAPIError({
|
|
84
|
+
category: "auth_unavailable",
|
|
85
|
+
provider: this.name,
|
|
86
|
+
message: "Anthropic API key is not configured",
|
|
87
|
+
retryable: false,
|
|
88
|
+
});
|
|
89
|
+
}
|
|
90
|
+
return {
|
|
91
|
+
"x-api-key": credential,
|
|
92
|
+
"anthropic-version": this.options.anthropicVersion ?? DEFAULT_ANTHROPIC_VERSION,
|
|
93
|
+
Accept: "application/json, text/event-stream",
|
|
94
|
+
};
|
|
95
|
+
}
|
|
96
|
+
}
|
|
97
|
+
function hasImageInput(request) {
|
|
98
|
+
return request.messages.some((message) => message.blocks.some((block) => {
|
|
99
|
+
const type = block.type;
|
|
100
|
+
return type === "image" || type === "input_image" || (block.type === "text" && /!\[[^\]]*\]\([^\)]+\)/.test(block.text));
|
|
101
|
+
}));
|
|
102
|
+
}
|
|
103
|
+
function stripTrailingSlash(value) {
|
|
104
|
+
return value.replace(/\/+$/, "");
|
|
105
|
+
}
|
|
106
|
+
//# sourceMappingURL=anthropic-adapter.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic-adapter.js","sourceRoot":"","sources":["../../src/model/anthropic-adapter.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,qBAAqB,EAAE,wBAAwB,EAAE,MAAM,kBAAkB,CAAC;AACnF,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AAGzD,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,qBAAqB,EAAE,wBAAwB,EAAE,wBAAwB,EAAE,MAAM,uBAAuB,CAAC;AAElH,MAAM,yBAAyB,GAAG,YAAY,CAAC;AAiB/C,MAAM,OAAO,gBAAgB;IAkBE;IAjBpB,IAAI,GAAG,WAAW,CAAC;IACnB,YAAY,GAAyB;QAC5C,SAAS,EAAE,IAAI;QACf,aAAa,EAAE,IAAI;QACnB,iBAAiB,EAAE,KAAK;QACxB,gBAAgB,EAAE,KAAK;QACvB,kBAAkB,EAAE,KAAK;QACzB,UAAU,EAAE,IAAI;QAChB,SAAS,EAAE,KAAK;QAChB,eAAe,EAAE,IAAI;QACrB,YAAY,EAAE,KAAK;KACpB,CAAC;IAEe,kBAAkB,CAAqB;IACvC,SAAS,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACzC,OAAO,CAAS;IAEjC,YAA6B,OAAgC;QAAhC,YAAO,GAAP,OAAO,CAAyB;QAC3D,IAAI,CAAC,kBAAkB;YACrB,OAAO,CAAC,kBAAkB;gBAC1B,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,wBAAwB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,qBAAqB,CAAC,OAAO,CAAC,aAAa,IAAI,mBAAmB,CAAC,CAAC,CAAC;QAC5I,IAAI,CAAC,OAAO,GAAG,kBAAkB,CAAC,OAAO,CAAC,OAAO,IAAI,2BAA2B,CAAC,CAAC;IACpF,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,OAAqB;QACjC,IAAI,CAAC,yBAAyB,CAAC,OAAO,CAAC,CAAC;QACxC,KAAK,CAAC,CAAC,eAAe,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,OAAO,CAAC,EAAE;YACzE,QAAQ,EAAE,IAAI,CAAC,IAAI;YACnB,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,IAAI,CAAC;YACxC,OAAO,EAAE,IAAI;SACd,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,CAAC,cAAc,CAAC,OAAqB,EAAE,OAAe;QAClE,MAAM,IAAI,GAAG,qBAAqB,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAC1D,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QACzC,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,IAAI,MAAM,CAAC;QAExE,IAAI,OAAO,GAAG,CAAC;YAAE,MAAM,EAAE,IAAI,EAAE,gBAAgB,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,yBAAyB,EAAE,OAAO,EAAE,EAAE,CAAC;QAEvG,IAAI,OAAO,CAAC,MAAM,KAAK,KAAK,EAAE,CAAC;YAC7B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC;gBAC/C,MAAM,EAAE,MAAM;gBACd,GAAG,EAAE,GAAG,IAAI,CAAC,OAAO,cAAc;gBAClC,OAAO;gBACP,IAAI,EAAE,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;gBAC/B,SAAS;gBACT,MAAM,EAAE,OAAO,CAAC,YAAY;aAC7B,CAAC,CAAC;YACH,KAAK,CAAC,CAAC,wBAAwB,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;YAC7D,OAAO;QACT,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAA0B;YACtE,MAAM,EAAE,MAAM;YACd,GAAG,EAAE,GAAG,IAAI,CAAC,OAAO,cAAc;YAClC,OAAO;YACP,IAAI,EAAE,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE;YAChC,SAAS;YACT,MAAM,EAAE,OAAO,CAAC,YAAY;SAC7B,CAAC,CAAC;QACH,KAAK,CAAC,CAAC,wBAAwB,CAAC,QAAQ,CAAC,CAAC;IAC5C,CAAC;IAEO,yBAAyB,CAAC,OAAqB;QACrD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;QAClD,MAAM,SAAS,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;QAC5C,IAAI,SAAS,KAAK,KAAK,IAAI,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC;YAClD,MAAM,IAAI,aAAa,CAAC;gBACtB,QAAQ,EAAE,yBAAyB;gBACnC,QAAQ,EAAE,IAAI,CAAC,IAAI;gBACnB,OAAO,EAAE,SAAS,KAAK,kEAAkE;gBACzF,SAAS,EAAE,KAAK;aACjB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,WAAW;QACvB,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,aAAa,EAAE,CAAC;QACjE,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,MAAM,IAAI,aAAa,CAAC;gBACtB,QAAQ,EAAE,kBAAkB;gBAC5B,QAAQ,EAAE,IAAI,CAAC,IAAI;gBACnB,OAAO,EAAE,qCAAqC;gBAC9C,SAAS,EAAE,KAAK;aACjB,CAAC,CAAC;QACL,CAAC;QACD,OAAO;YACL,WAAW,EAAE,UAAU;YACvB,mBAAmB,EAAE,IAAI,CAAC,OAAO,CAAC,gBAAgB,IAAI,yBAAyB;YAC/E,MAAM,EAAE,qCAAqC;SAC9C,CAAC;IACJ,CAAC;CACF;AAED,SAAS,aAAa,CAAC,OAAqB;IAC1C,OAAO,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CACvC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE;QAC5B,MAAM,IAAI,GAAI,KAA0B,CAAC,IAAI,CAAC;QAC9C,OAAO,IAAI,KAAK,OAAO,IAAI,IAAI,KAAK,aAAa,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,uBAAuB,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;IAC3H,CAAC,CAAC,CACH,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAa;IACvC,OAAO,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;AACnC,CAAC"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import type { HttpJsonResponse } from "./http-transport.js";
|
|
2
|
+
import { ModelAPIError } from "./errors.js";
|
|
3
|
+
import type { ModelRequest, ModelStreamEvent, ReasoningConfig } from "./model-gateway.js";
|
|
4
|
+
export interface AnthropicMapperOptions {
|
|
5
|
+
model: string;
|
|
6
|
+
defaultMaxOutputTokens?: number;
|
|
7
|
+
defaultReasoning?: ReasoningConfig | null;
|
|
8
|
+
streamIdleTimeoutMs?: number;
|
|
9
|
+
}
|
|
10
|
+
export declare function buildAnthropicRequest(request: ModelRequest, options: AnthropicMapperOptions): Record<string, unknown>;
|
|
11
|
+
export declare function normalizeAnthropicStream(stream: ReadableStream<Uint8Array>, options: AnthropicMapperOptions): AsyncGenerator<ModelStreamEvent>;
|
|
12
|
+
export declare function normalizeAnthropicObject(response: HttpJsonResponse<Record<string, unknown>>): Generator<ModelStreamEvent>;
|
|
13
|
+
export declare function normalizeAnthropicStreamError(event: Record<string, unknown>): ModelAPIError;
|