@keel_flow/runtime 0.2.0

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.
Files changed (99) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +15 -0
  3. package/dist/agent-runtime.d.ts +8 -0
  4. package/dist/agent-runtime.d.ts.map +1 -0
  5. package/dist/agent-runtime.js +9 -0
  6. package/dist/agent-runtime.js.map +1 -0
  7. package/dist/client.d.ts +15 -0
  8. package/dist/client.d.ts.map +1 -0
  9. package/dist/client.js +16 -0
  10. package/dist/client.js.map +1 -0
  11. package/dist/cost.d.ts +3 -0
  12. package/dist/cost.d.ts.map +1 -0
  13. package/dist/cost.js +52 -0
  14. package/dist/cost.js.map +1 -0
  15. package/dist/dispatch-subagent.d.ts +24 -0
  16. package/dist/dispatch-subagent.d.ts.map +1 -0
  17. package/dist/dispatch-subagent.js +90 -0
  18. package/dist/dispatch-subagent.js.map +1 -0
  19. package/dist/index.d.ts +16 -0
  20. package/dist/index.d.ts.map +1 -0
  21. package/dist/index.js +9 -0
  22. package/dist/index.js.map +1 -0
  23. package/dist/provider.d.ts +55 -0
  24. package/dist/provider.d.ts.map +1 -0
  25. package/dist/provider.js +18 -0
  26. package/dist/provider.js.map +1 -0
  27. package/dist/providers/anthropic.d.ts +9 -0
  28. package/dist/providers/anthropic.d.ts.map +1 -0
  29. package/dist/providers/anthropic.js +116 -0
  30. package/dist/providers/anthropic.js.map +1 -0
  31. package/dist/providers/claude-bridge.d.ts +20 -0
  32. package/dist/providers/claude-bridge.d.ts.map +1 -0
  33. package/dist/providers/claude-bridge.js +109 -0
  34. package/dist/providers/claude-bridge.js.map +1 -0
  35. package/dist/providers/openai-compatible.d.ts +9 -0
  36. package/dist/providers/openai-compatible.d.ts.map +1 -0
  37. package/dist/providers/openai-compatible.js +171 -0
  38. package/dist/providers/openai-compatible.js.map +1 -0
  39. package/dist/run-agent.d.ts +43 -0
  40. package/dist/run-agent.d.ts.map +1 -0
  41. package/dist/run-agent.js +318 -0
  42. package/dist/run-agent.js.map +1 -0
  43. package/dist/spec-checker/__fixtures__/mock-provider.d.ts +20 -0
  44. package/dist/spec-checker/__fixtures__/mock-provider.d.ts.map +1 -0
  45. package/dist/spec-checker/__fixtures__/mock-provider.js +34 -0
  46. package/dist/spec-checker/__fixtures__/mock-provider.js.map +1 -0
  47. package/dist/spec-checker/adversarial.d.ts +15 -0
  48. package/dist/spec-checker/adversarial.d.ts.map +1 -0
  49. package/dist/spec-checker/adversarial.js +77 -0
  50. package/dist/spec-checker/adversarial.js.map +1 -0
  51. package/dist/spec-checker/aggregate.d.ts +17 -0
  52. package/dist/spec-checker/aggregate.d.ts.map +1 -0
  53. package/dist/spec-checker/aggregate.js +25 -0
  54. package/dist/spec-checker/aggregate.js.map +1 -0
  55. package/dist/spec-checker/bias.d.ts +16 -0
  56. package/dist/spec-checker/bias.d.ts.map +1 -0
  57. package/dist/spec-checker/bias.js +26 -0
  58. package/dist/spec-checker/bias.js.map +1 -0
  59. package/dist/spec-checker/bidirectional.d.ts +21 -0
  60. package/dist/spec-checker/bidirectional.d.ts.map +1 -0
  61. package/dist/spec-checker/bidirectional.js +97 -0
  62. package/dist/spec-checker/bidirectional.js.map +1 -0
  63. package/dist/spec-checker/calibration.d.ts +15 -0
  64. package/dist/spec-checker/calibration.d.ts.map +1 -0
  65. package/dist/spec-checker/calibration.js +58 -0
  66. package/dist/spec-checker/calibration.js.map +1 -0
  67. package/dist/spec-checker/claims.d.ts +26 -0
  68. package/dist/spec-checker/claims.d.ts.map +1 -0
  69. package/dist/spec-checker/claims.js +104 -0
  70. package/dist/spec-checker/claims.js.map +1 -0
  71. package/dist/spec-checker/index.d.ts +40 -0
  72. package/dist/spec-checker/index.d.ts.map +1 -0
  73. package/dist/spec-checker/index.js +308 -0
  74. package/dist/spec-checker/index.js.map +1 -0
  75. package/dist/spec-checker/prompts.d.ts +11 -0
  76. package/dist/spec-checker/prompts.d.ts.map +1 -0
  77. package/dist/spec-checker/prompts.js +11 -0
  78. package/dist/spec-checker/prompts.js.map +1 -0
  79. package/dist/spec-checker/rubric.d.ts +14 -0
  80. package/dist/spec-checker/rubric.d.ts.map +1 -0
  81. package/dist/spec-checker/rubric.js +68 -0
  82. package/dist/spec-checker/rubric.js.map +1 -0
  83. package/dist/spec-checker/score.d.ts +16 -0
  84. package/dist/spec-checker/score.d.ts.map +1 -0
  85. package/dist/spec-checker/score.js +45 -0
  86. package/dist/spec-checker/score.js.map +1 -0
  87. package/dist/spec-checker/shim.d.ts +4 -0
  88. package/dist/spec-checker/shim.d.ts.map +1 -0
  89. package/dist/spec-checker/shim.js +69 -0
  90. package/dist/spec-checker/shim.js.map +1 -0
  91. package/dist/spec-checker.d.ts +4 -0
  92. package/dist/spec-checker.d.ts.map +1 -0
  93. package/dist/spec-checker.js +2 -0
  94. package/dist/spec-checker.js.map +1 -0
  95. package/dist/tools.d.ts +6 -0
  96. package/dist/tools.d.ts.map +1 -0
  97. package/dist/tools.js +18 -0
  98. package/dist/tools.js.map +1 -0
  99. package/package.json +45 -0
@@ -0,0 +1,109 @@
1
+ import { randomUUID } from "node:crypto";
2
+ export class ClaudeBridgeUnsupportedError extends Error {
3
+ constructor(detail) {
4
+ super(`claude-bridge supports single-shot generation only (${detail}). ` +
5
+ "Agentic tool loops route through Pi (@keel_flow/pi-gateway) or a key-based provider.");
6
+ }
7
+ }
8
+ async function loadSdkQuery() {
9
+ const sdk = await import("@anthropic-ai/claude-agent-sdk");
10
+ return sdk.query;
11
+ }
12
+ function userPromptFrom(req) {
13
+ const parts = [];
14
+ for (const message of req.messages) {
15
+ if (message.role !== "user") {
16
+ throw new ClaudeBridgeUnsupportedError("conversation history with assistant turns");
17
+ }
18
+ for (const block of message.content) {
19
+ if (block.type !== "text") {
20
+ throw new ClaudeBridgeUnsupportedError(`${block.type} blocks in the request`);
21
+ }
22
+ parts.push(block.text);
23
+ }
24
+ }
25
+ if (parts.length === 0)
26
+ throw new ClaudeBridgeUnsupportedError("empty prompt");
27
+ return parts.join("\n\n");
28
+ }
29
+ function forcedToolFrom(req) {
30
+ if (!req.tools || req.tools.length === 0)
31
+ return null;
32
+ const choice = req.toolChoice;
33
+ if (typeof choice === "object" && choice !== null) {
34
+ const tool = req.tools.find((t) => t.name === choice.name);
35
+ if (!tool)
36
+ throw new ClaudeBridgeUnsupportedError(`forced tool ${choice.name} not in tools`);
37
+ return { name: tool.name, inputSchema: tool.inputSchema };
38
+ }
39
+ if (choice === "any" && req.tools.length === 1) {
40
+ const tool = req.tools[0];
41
+ return { name: tool.name, inputSchema: tool.inputSchema };
42
+ }
43
+ throw new ClaudeBridgeUnsupportedError(`toolChoice ${JSON.stringify(choice ?? "auto")} with ${req.tools.length} tools`);
44
+ }
45
+ function scrubbedEnv() {
46
+ const env = { ...process.env };
47
+ delete env["ANTHROPIC_API_KEY"];
48
+ return env;
49
+ }
50
+ export class ClaudeBridgeProvider {
51
+ kind = "claude-bridge";
52
+ defaultModel;
53
+ queryFn;
54
+ constructor(config, queryFn) {
55
+ this.defaultModel = config.defaultModel ?? "claude-sonnet-4-6";
56
+ this.queryFn = queryFn ? () => Promise.resolve(queryFn) : loadSdkQuery;
57
+ }
58
+ async generate(req) {
59
+ const prompt = userPromptFrom(req);
60
+ const forcedTool = forcedToolFrom(req);
61
+ const options = {
62
+ systemPrompt: req.system,
63
+ model: req.model,
64
+ maxTurns: 8,
65
+ allowedTools: [],
66
+ settingSources: [],
67
+ permissionMode: "dontAsk",
68
+ env: scrubbedEnv(),
69
+ };
70
+ if (forcedTool) {
71
+ options["outputFormat"] = { type: "json_schema", schema: forcedTool.inputSchema };
72
+ }
73
+ const query = await this.queryFn();
74
+ let result = null;
75
+ for await (const message of query({ prompt, options })) {
76
+ if (message.type === "result") {
77
+ result = message;
78
+ }
79
+ }
80
+ if (!result)
81
+ throw new Error("claude-bridge: query ended without a result message");
82
+ if (result.is_error || result.subtype !== "success") {
83
+ const detail = result.errors?.join("; ") ?? result.subtype;
84
+ throw new Error(`claude-bridge: ${detail}`);
85
+ }
86
+ const usage = {
87
+ inputTokens: result.usage?.input_tokens ?? 0,
88
+ outputTokens: result.usage?.output_tokens ?? 0,
89
+ };
90
+ if (forcedTool) {
91
+ if (result.structured_output === undefined) {
92
+ throw new Error("claude-bridge: success result missing structured_output");
93
+ }
94
+ const block = {
95
+ type: "tool_use",
96
+ id: `bridge_${randomUUID()}`,
97
+ name: forcedTool.name,
98
+ input: result.structured_output,
99
+ };
100
+ return { content: [block], stopReason: "tool_use", usage };
101
+ }
102
+ return {
103
+ content: [{ type: "text", text: result.result ?? "" }],
104
+ stopReason: "end_turn",
105
+ usage,
106
+ };
107
+ }
108
+ }
109
+ //# sourceMappingURL=claude-bridge.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"claude-bridge.js","sourceRoot":"","sources":["../../src/providers/claude-bridge.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AASzC,MAAM,OAAO,4BAA6B,SAAQ,KAAK;IACrD,YAAY,MAAc;QACxB,KAAK,CACH,uDAAuD,MAAM,KAAK;YAChE,sFAAsF,CACzF,CAAC;IACJ,CAAC;CACF;AAmBD,KAAK,UAAU,YAAY;IACzB,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,gCAAgC,CAAC,CAAC;IAC3D,OAAO,GAAG,CAAC,KAAiC,CAAC;AAC/C,CAAC;AAED,SAAS,cAAc,CAAC,GAAsB;IAC5C,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,MAAM,OAAO,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;QACnC,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC5B,MAAM,IAAI,4BAA4B,CAAC,2CAA2C,CAAC,CAAC;QACtF,CAAC;QACD,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACpC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC1B,MAAM,IAAI,4BAA4B,CAAC,GAAG,KAAK,CAAC,IAAI,wBAAwB,CAAC,CAAC;YAChF,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IACD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,MAAM,IAAI,4BAA4B,CAAC,cAAc,CAAC,CAAC;IAC/E,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAC5B,CAAC;AAED,SAAS,cAAc,CAAC,GAAsB;IAC5C,IAAI,CAAC,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IACtD,MAAM,MAAM,GAAG,GAAG,CAAC,UAAU,CAAC;IAC9B,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;QAClD,MAAM,IAAI,GAAG,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC;QAC3D,IAAI,CAAC,IAAI;YAAE,MAAM,IAAI,4BAA4B,CAAC,eAAe,MAAM,CAAC,IAAI,eAAe,CAAC,CAAC;QAC7F,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC;IAC5D,CAAC;IACD,IAAI,MAAM,KAAK,KAAK,IAAI,GAAG,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/C,MAAM,IAAI,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC;QAC3B,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC;IAC5D,CAAC;IACD,MAAM,IAAI,4BAA4B,CACpC,cAAc,IAAI,CAAC,SAAS,CAAC,MAAM,IAAI,MAAM,CAAC,SAAS,GAAG,CAAC,KAAK,CAAC,MAAM,QAAQ,CAChF,CAAC;AACJ,CAAC;AAED,SAAS,WAAW;IAClB,MAAM,GAAG,GAAuC,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IACnE,OAAO,GAAG,CAAC,mBAAmB,CAAC,CAAC;IAChC,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,OAAO,oBAAoB;IACtB,IAAI,GAAG,eAAwB,CAAC;IAChC,YAAY,CAAS;IACb,OAAO,CAA+B;IAEvD,YAAY,MAAsB,EAAE,OAAuB;QACzD,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,mBAAmB,CAAC;QAC/D,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC;IACzE,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,GAAsB;QACnC,MAAM,MAAM,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;QACnC,MAAM,UAAU,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;QAEvC,MAAM,OAAO,GAA4B;YACvC,YAAY,EAAE,GAAG,CAAC,MAAM;YACxB,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,QAAQ,EAAE,CAAC;YACX,YAAY,EAAE,EAAE;YAChB,cAAc,EAAE,EAAE;YAClB,cAAc,EAAE,SAAS;YACzB,GAAG,EAAE,WAAW,EAAE;SACnB,CAAC;QACF,IAAI,UAAU,EAAE,CAAC;YACf,OAAO,CAAC,cAAc,CAAC,GAAG,EAAE,IAAI,EAAE,aAAa,EAAE,MAAM,EAAE,UAAU,CAAC,WAAW,EAAE,CAAC;QACpF,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACnC,IAAI,MAAM,GAA+B,IAAI,CAAC;QAC9C,IAAI,KAAK,EAAE,MAAM,OAAO,IAAI,KAAK,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;YACvD,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC9B,MAAM,GAAG,OAAyC,CAAC;YACrD,CAAC;QACH,CAAC;QACD,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QACpF,IAAI,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YACpD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC;YAC3D,MAAM,IAAI,KAAK,CAAC,kBAAkB,MAAM,EAAE,CAAC,CAAC;QAC9C,CAAC;QAED,MAAM,KAAK,GAAG;YACZ,WAAW,EAAE,MAAM,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC;YAC5C,YAAY,EAAE,MAAM,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;SAC/C,CAAC;QAEF,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,MAAM,CAAC,iBAAiB,KAAK,SAAS,EAAE,CAAC;gBAC3C,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;YAC7E,CAAC;YACD,MAAM,KAAK,GAAoB;gBAC7B,IAAI,EAAE,UAAU;gBAChB,EAAE,EAAE,UAAU,UAAU,EAAE,EAAE;gBAC5B,IAAI,EAAE,UAAU,CAAC,IAAI;gBACrB,KAAK,EAAE,MAAM,CAAC,iBAAiB;aAChC,CAAC;YACF,OAAO,EAAE,OAAO,EAAE,CAAC,KAAK,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;QAC7D,CAAC;QAED,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,IAAI,EAAE,EAAE,CAAC;YACtD,UAAU,EAAE,UAAU;YACtB,KAAK;SACN,CAAC;IACJ,CAAC;CACF"}
@@ -0,0 +1,9 @@
1
+ import type { ModelProvider, NormalizedRequest, NormalizedResponse, ProviderConfig } from "../provider.js";
2
+ export declare class OpenAICompatibleProvider implements ModelProvider {
3
+ readonly kind: "openai-compatible";
4
+ readonly defaultModel: string;
5
+ private readonly client;
6
+ constructor(config: ProviderConfig);
7
+ generate(req: NormalizedRequest): Promise<NormalizedResponse>;
8
+ }
9
+ //# sourceMappingURL=openai-compatible.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"openai-compatible.d.ts","sourceRoot":"","sources":["../../src/providers/openai-compatible.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,aAAa,EACb,iBAAiB,EACjB,kBAAkB,EAElB,cAAc,EACf,MAAM,gBAAgB,CAAC;AAexB,qBAAa,wBAAyB,YAAW,aAAa;IAC5D,QAAQ,CAAC,IAAI,EAAG,mBAAmB,CAAU;IAC7C,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;gBAEpB,MAAM,EAAE,cAAc;IAW5B,QAAQ,CAAC,GAAG,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC;CA2EpE"}
@@ -0,0 +1,171 @@
1
+ import OpenAI from "openai";
2
+ function resolveDefaultModel(baseURL) {
3
+ if (!baseURL)
4
+ return "gpt-4o";
5
+ try {
6
+ const host = new URL(baseURL).hostname;
7
+ if (host.includes("deepseek"))
8
+ return "deepseek-chat";
9
+ if (host === "localhost" || host === "127.0.0.1")
10
+ return "llama3.1:70b";
11
+ if (host.includes("moonshot") || host.includes("kimi"))
12
+ return "kimi-k2";
13
+ }
14
+ catch {
15
+ // unparseable baseURL — fall through to default
16
+ }
17
+ return "gpt-4o";
18
+ }
19
+ export class OpenAICompatibleProvider {
20
+ kind = "openai-compatible";
21
+ defaultModel;
22
+ client;
23
+ constructor(config) {
24
+ // Ollama and similar local servers don't require a real API key,
25
+ // but the OpenAI SDK requires a non-empty string.
26
+ const apiKey = config.apiKey || process.env["OPENAI_API_KEY"] || "ollama";
27
+ this.client = new OpenAI({
28
+ apiKey,
29
+ ...(config.baseURL ? { baseURL: config.baseURL } : {}),
30
+ });
31
+ this.defaultModel = config.defaultModel ?? resolveDefaultModel(config.baseURL);
32
+ }
33
+ async generate(req) {
34
+ const openaiMessages = toOpenAIMessages(req);
35
+ const tools = req.tools && req.tools.length > 0
36
+ ? req.tools.map((t) => ({
37
+ type: "function",
38
+ function: {
39
+ name: t.name,
40
+ description: t.description,
41
+ // eslint-disable-next-line @typescript-eslint/no-explicit-any -- FunctionParameters requires index signature; inputSchema is a valid JSON Schema object
42
+ parameters: t.inputSchema,
43
+ },
44
+ }))
45
+ : undefined;
46
+ let tool_choice;
47
+ if (req.toolChoice === "any") {
48
+ tool_choice = "required";
49
+ }
50
+ else if (req.toolChoice === "auto") {
51
+ tool_choice = "auto";
52
+ }
53
+ else if (req.toolChoice && typeof req.toolChoice === "object") {
54
+ tool_choice = { type: "function", function: { name: req.toolChoice.name } };
55
+ }
56
+ const response = await this.client.chat.completions.create({
57
+ model: req.model,
58
+ messages: openaiMessages,
59
+ max_tokens: req.maxTokens ?? 4096,
60
+ ...(tools ? { tools } : {}),
61
+ ...(tool_choice ? { tool_choice } : {}),
62
+ });
63
+ const choice = response.choices[0];
64
+ if (!choice) {
65
+ return { content: [], stopReason: "end_turn", usage: { inputTokens: 0, outputTokens: 0 } };
66
+ }
67
+ const content = [];
68
+ const msg = choice.message;
69
+ if (msg.content) {
70
+ content.push({ type: "text", text: msg.content });
71
+ }
72
+ if (msg.tool_calls) {
73
+ for (const tc of msg.tool_calls) {
74
+ let parsed;
75
+ try {
76
+ parsed = JSON.parse(tc.function.arguments);
77
+ }
78
+ catch {
79
+ parsed = tc.function.arguments;
80
+ }
81
+ content.push({
82
+ type: "tool_use",
83
+ id: tc.id,
84
+ name: tc.function.name,
85
+ input: parsed,
86
+ });
87
+ }
88
+ }
89
+ const stopReason = msg.tool_calls?.length
90
+ ? "tool_use"
91
+ : mapFinishReason(choice.finish_reason);
92
+ return {
93
+ content,
94
+ stopReason,
95
+ usage: {
96
+ inputTokens: response.usage?.prompt_tokens ?? 0,
97
+ outputTokens: response.usage?.completion_tokens ?? 0,
98
+ },
99
+ };
100
+ }
101
+ }
102
+ function toOpenAIMessages(req) {
103
+ const result = [];
104
+ if (req.system) {
105
+ result.push({ role: "system", content: req.system });
106
+ }
107
+ for (const msg of req.messages) {
108
+ if (msg.role === "user") {
109
+ const toolResults = msg.content.filter((b) => b.type === "tool_result");
110
+ const textBlocks = msg.content.filter((b) => b.type === "text");
111
+ if (toolResults.length > 0) {
112
+ for (const tr of toolResults) {
113
+ if (tr.type !== "tool_result")
114
+ continue;
115
+ result.push({
116
+ role: "tool",
117
+ tool_call_id: tr.toolUseId,
118
+ content: tr.content,
119
+ });
120
+ }
121
+ }
122
+ else {
123
+ const text = textBlocks
124
+ .filter((b) => b.type === "text")
125
+ .map((b) => (b.type === "text" ? b.text : ""))
126
+ .join("\n");
127
+ result.push({ role: "user", content: text });
128
+ }
129
+ }
130
+ else {
131
+ const textBlocks = msg.content.filter((b) => b.type === "text");
132
+ const toolUseBlocks = msg.content.filter((b) => b.type === "tool_use");
133
+ const text = textBlocks.length > 0
134
+ ? textBlocks
135
+ .filter((b) => b.type === "text")
136
+ .map((b) => (b.type === "text" ? b.text : ""))
137
+ .join("\n")
138
+ : null;
139
+ const tool_calls = toolUseBlocks
140
+ .filter((b) => b.type === "tool_use")
141
+ .map((b) => {
142
+ if (b.type !== "tool_use")
143
+ throw new Error("unreachable");
144
+ return {
145
+ id: b.id,
146
+ type: "function",
147
+ function: {
148
+ name: b.name,
149
+ arguments: JSON.stringify(b.input),
150
+ },
151
+ };
152
+ });
153
+ result.push({
154
+ role: "assistant",
155
+ content: text,
156
+ ...(tool_calls.length > 0 ? { tool_calls } : {}),
157
+ });
158
+ }
159
+ }
160
+ return result;
161
+ }
162
+ function mapFinishReason(reason) {
163
+ if (reason === "tool_calls")
164
+ return "tool_use";
165
+ if (reason === "length")
166
+ return "max_tokens";
167
+ if (reason === "stop")
168
+ return "end_turn";
169
+ return "end_turn";
170
+ }
171
+ //# sourceMappingURL=openai-compatible.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"openai-compatible.js","sourceRoot":"","sources":["../../src/providers/openai-compatible.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAS5B,SAAS,mBAAmB,CAAC,OAA2B;IACtD,IAAI,CAAC,OAAO;QAAE,OAAO,QAAQ,CAAC;IAC9B,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC;QACvC,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC;YAAE,OAAO,eAAe,CAAC;QACtD,IAAI,IAAI,KAAK,WAAW,IAAI,IAAI,KAAK,WAAW;YAAE,OAAO,cAAc,CAAC;QACxE,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,SAAS,CAAC;IAC3E,CAAC;IAAC,MAAM,CAAC;QACP,gDAAgD;IAClD,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,OAAO,wBAAwB;IAC1B,IAAI,GAAG,mBAA4B,CAAC;IACpC,YAAY,CAAS;IACb,MAAM,CAAS;IAEhC,YAAY,MAAsB;QAChC,iEAAiE;QACjE,kDAAkD;QAClD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,IAAI,QAAQ,CAAC;QAC1E,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC;YACvB,MAAM;YACN,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACvD,CAAC,CAAC;QACH,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,mBAAmB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACjF,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,GAAsB;QACnC,MAAM,cAAc,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC;QAE7C,MAAM,KAAK,GACT,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC;YAC/B,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACpB,IAAI,EAAE,UAAmB;gBACzB,QAAQ,EAAE;oBACR,IAAI,EAAE,CAAC,CAAC,IAAI;oBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;oBAC1B,wJAAwJ;oBACxJ,UAAU,EAAE,CAAC,CAAC,WAAkB;iBACjC;aACF,CAAC,CAAC;YACL,CAAC,CAAC,SAAS,CAAC;QAEhB,IAAI,WAAmE,CAAC;QACxE,IAAI,GAAG,CAAC,UAAU,KAAK,KAAK,EAAE,CAAC;YAC7B,WAAW,GAAG,UAAU,CAAC;QAC3B,CAAC;aAAM,IAAI,GAAG,CAAC,UAAU,KAAK,MAAM,EAAE,CAAC;YACrC,WAAW,GAAG,MAAM,CAAC;QACvB,CAAC;aAAM,IAAI,GAAG,CAAC,UAAU,IAAI,OAAO,GAAG,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;YAChE,WAAW,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,EAAE,CAAC;QAC9E,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;YACzD,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,QAAQ,EAAE,cAAc;YACxB,UAAU,EAAE,GAAG,CAAC,SAAS,IAAI,IAAI;YACjC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3B,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACxC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QACnC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,UAAU,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;QAC7F,CAAC;QAED,MAAM,OAAO,GAAsB,EAAE,CAAC;QACtC,MAAM,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC;QAE3B,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;YAChB,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;QACpD,CAAC;QAED,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;YACnB,KAAK,MAAM,EAAE,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;gBAChC,IAAI,MAAe,CAAC;gBACpB,IAAI,CAAC;oBACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;gBAC7C,CAAC;gBAAC,MAAM,CAAC;oBACP,MAAM,GAAG,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;gBACjC,CAAC;gBACD,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,UAAU;oBAChB,EAAE,EAAE,EAAE,CAAC,EAAE;oBACT,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;oBACtB,KAAK,EAAE,MAAM;iBACd,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,MAAM,UAAU,GAAG,GAAG,CAAC,UAAU,EAAE,MAAM;YACvC,CAAC,CAAC,UAAU;YACZ,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAE1C,OAAO;YACL,OAAO;YACP,UAAU;YACV,KAAK,EAAE;gBACL,WAAW,EAAE,QAAQ,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;gBAC/C,YAAY,EAAE,QAAQ,CAAC,KAAK,EAAE,iBAAiB,IAAI,CAAC;aACrD;SACF,CAAC;IACJ,CAAC;CACF;AAED,SAAS,gBAAgB,CAAC,GAAsB;IAC9C,MAAM,MAAM,GAA6C,EAAE,CAAC;IAE5D,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;QACf,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;IACvD,CAAC;IAED,KAAK,MAAM,GAAG,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC/B,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACxB,MAAM,WAAW,GAAG,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAAC,CAAC;YACxE,MAAM,UAAU,GAAG,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;YAEhE,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3B,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;oBAC7B,IAAI,EAAE,CAAC,IAAI,KAAK,aAAa;wBAAE,SAAS;oBACxC,MAAM,CAAC,IAAI,CAAC;wBACV,IAAI,EAAE,MAAM;wBACZ,YAAY,EAAE,EAAE,CAAC,SAAS;wBAC1B,OAAO,EAAE,EAAE,CAAC,OAAO;qBACpB,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,GAAG,UAAU;qBACpB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;qBAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;qBAC7C,IAAI,CAAC,IAAI,CAAC,CAAC;gBACd,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;YAC/C,CAAC;QACH,CAAC;aAAM,CAAC;YACN,MAAM,UAAU,GAAG,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;YAChE,MAAM,aAAa,GAAG,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;YAEvE,MAAM,IAAI,GACR,UAAU,CAAC,MAAM,GAAG,CAAC;gBACnB,CAAC,CAAC,UAAU;qBACP,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;qBAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;qBAC7C,IAAI,CAAC,IAAI,CAAC;gBACf,CAAC,CAAC,IAAI,CAAC;YAEX,MAAM,UAAU,GACd,aAAa;iBACV,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC;iBACpC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBACT,IAAI,CAAC,CAAC,IAAI,KAAK,UAAU;oBAAE,MAAM,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC;gBAC1D,OAAO;oBACL,EAAE,EAAE,CAAC,CAAC,EAAE;oBACR,IAAI,EAAE,UAAmB;oBACzB,QAAQ,EAAE;wBACR,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC;qBACnC;iBACF,CAAC;YACJ,CAAC,CAAC,CAAC;YAEP,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,IAAI;gBACb,GAAG,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACjD,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,eAAe,CACtB,MAAiC;IAEjC,IAAI,MAAM,KAAK,YAAY;QAAE,OAAO,UAAU,CAAC;IAC/C,IAAI,MAAM,KAAK,QAAQ;QAAE,OAAO,YAAY,CAAC;IAC7C,IAAI,MAAM,KAAK,MAAM;QAAE,OAAO,UAAU,CAAC;IACzC,OAAO,UAAU,CAAC;AACpB,CAAC"}
@@ -0,0 +1,43 @@
1
+ import type Anthropic from "@anthropic-ai/sdk";
2
+ import type { AgentConfig, KnowledgeBase } from "@keel_flow/core";
3
+ import type { TelemetryEmitter } from "@keel_flow/telemetry";
4
+ import type { ToolDefinition } from "./tools.js";
5
+ import type { ModelProvider } from "./provider.js";
6
+ export interface UsageEvent {
7
+ agentId: string;
8
+ toolIds: string[];
9
+ model: string;
10
+ inputTokens: number;
11
+ outputTokens: number;
12
+ costUsd: number;
13
+ }
14
+ export interface RunAgentArgs {
15
+ agent: AgentConfig;
16
+ tools: ToolDefinition[];
17
+ prompt: string;
18
+ kb?: KnowledgeBase | undefined;
19
+ provider?: ModelProvider | undefined;
20
+ client?: Anthropic | undefined;
21
+ model?: string | undefined;
22
+ maxIterations?: number | undefined;
23
+ onUsage?: ((event: UsageEvent) => void | Promise<void>) | undefined;
24
+ telemetry?: TelemetryEmitter | undefined;
25
+ repoMap?: string | undefined;
26
+ }
27
+ export interface ToolCallRecord {
28
+ toolId: string;
29
+ input: unknown;
30
+ output: unknown;
31
+ }
32
+ export interface RunResult {
33
+ finalText: string;
34
+ toolCalls: ToolCallRecord[];
35
+ iterations: number;
36
+ stopReason: "end_turn" | "max_iterations" | "stop_sequence" | "max_tokens";
37
+ usage: {
38
+ inputTokens: number;
39
+ outputTokens: number;
40
+ };
41
+ }
42
+ export declare function runAgent({ agent, tools, prompt, kb, provider: providerOverride, client: clientOverride, model: modelOverride, maxIterations, onUsage, telemetry, repoMap, }: RunAgentArgs): Promise<RunResult>;
43
+ //# sourceMappingURL=run-agent.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"run-agent.d.ts","sourceRoot":"","sources":["../src/run-agent.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,SAAS,MAAM,mBAAmB,CAAC;AAC/C,OAAO,KAAK,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAClE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAI7D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AACjD,OAAO,KAAK,EAAE,aAAa,EAA6E,MAAM,eAAe,CAAC;AA6F9H,MAAM,WAAW,UAAU;IACzB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,YAAY;IAC3B,KAAK,EAAE,WAAW,CAAC;IACnB,KAAK,EAAE,cAAc,EAAE,CAAC;IACxB,MAAM,EAAE,MAAM,CAAC;IACf,EAAE,CAAC,EAAE,aAAa,GAAG,SAAS,CAAC;IAC/B,QAAQ,CAAC,EAAE,aAAa,GAAG,SAAS,CAAC;IACrC,MAAM,CAAC,EAAE,SAAS,GAAG,SAAS,CAAC;IAC/B,KAAK,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,aAAa,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACnC,OAAO,CAAC,EAAE,CAAC,CAAC,KAAK,EAAE,UAAU,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,SAAS,CAAC;IACpE,SAAS,CAAC,EAAE,gBAAgB,GAAG,SAAS,CAAC;IACzC,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC9B;AAED,MAAM,WAAW,cAAc;IAC7B,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,OAAO,CAAC;IACf,MAAM,EAAE,OAAO,CAAC;CACjB;AAED,MAAM,WAAW,SAAS;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,cAAc,EAAE,CAAC;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,UAAU,GAAG,gBAAgB,GAAG,eAAe,GAAG,YAAY,CAAC;IAC3E,KAAK,EAAE;QAAE,WAAW,EAAE,MAAM,CAAC;QAAC,YAAY,EAAE,MAAM,CAAA;KAAE,CAAC;CACtD;AAED,wBAAsB,QAAQ,CAAC,EAC7B,KAAK,EACL,KAAK,EACL,MAAM,EACN,EAAE,EACF,QAAQ,EAAE,gBAAgB,EAC1B,MAAM,EAAE,cAAc,EACtB,KAAK,EAAE,aAAa,EACpB,aAAiB,EACjB,OAAO,EACP,SAAS,EACT,OAAO,GACR,EAAE,YAAY,GAAG,OAAO,CAAC,SAAS,CAAC,CAgQnC"}