@studio-foundation/runner 0.3.0-beta.1
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/ARCHITECTURE.md +53 -0
- package/LICENSE +663 -0
- package/README.md +125 -0
- package/configs/agents/analyst.agent.yaml +31 -0
- package/configs/agents/code-generator.agent.yaml +31 -0
- package/configs/agents/generic.agent.yaml +23 -0
- package/dist/__tests__/script-executor.test.d.ts +2 -0
- package/dist/__tests__/script-executor.test.d.ts.map +1 -0
- package/dist/__tests__/script-executor.test.js +173 -0
- package/dist/__tests__/script-executor.test.js.map +1 -0
- package/dist/index.d.ts +38 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +76 -0
- package/dist/index.js.map +1 -0
- package/dist/integrations/integration-loader.d.ts +20 -0
- package/dist/integrations/integration-loader.d.ts.map +1 -0
- package/dist/integrations/integration-loader.js +71 -0
- package/dist/integrations/integration-loader.js.map +1 -0
- package/dist/integrations/integration-loader.test.d.ts +2 -0
- package/dist/integrations/integration-loader.test.d.ts.map +1 -0
- package/dist/integrations/integration-loader.test.js +77 -0
- package/dist/integrations/integration-loader.test.js.map +1 -0
- package/dist/middleware/anonymization.d.ts +22 -0
- package/dist/middleware/anonymization.d.ts.map +1 -0
- package/dist/middleware/anonymization.js +37 -0
- package/dist/middleware/anonymization.js.map +1 -0
- package/dist/plugins/index.d.ts +5 -0
- package/dist/plugins/index.d.ts.map +1 -0
- package/dist/plugins/index.js +10 -0
- package/dist/plugins/index.js.map +1 -0
- package/dist/plugins/mcp-client.d.ts +19 -0
- package/dist/plugins/mcp-client.d.ts.map +1 -0
- package/dist/plugins/mcp-client.js +126 -0
- package/dist/plugins/mcp-client.js.map +1 -0
- package/dist/plugins/mcp-client.test.d.ts +2 -0
- package/dist/plugins/mcp-client.test.d.ts.map +1 -0
- package/dist/plugins/mcp-client.test.js +160 -0
- package/dist/plugins/mcp-client.test.js.map +1 -0
- package/dist/plugins/oauth-provider.d.ts +27 -0
- package/dist/plugins/oauth-provider.d.ts.map +1 -0
- package/dist/plugins/oauth-provider.js +140 -0
- package/dist/plugins/oauth-provider.js.map +1 -0
- package/dist/plugins/oauth-provider.test.d.ts +2 -0
- package/dist/plugins/oauth-provider.test.d.ts.map +1 -0
- package/dist/plugins/oauth-provider.test.js +144 -0
- package/dist/plugins/oauth-provider.test.js.map +1 -0
- package/dist/plugins/plugin-loader.d.ts +29 -0
- package/dist/plugins/plugin-loader.d.ts.map +1 -0
- package/dist/plugins/plugin-loader.js +71 -0
- package/dist/plugins/plugin-loader.js.map +1 -0
- package/dist/plugins/plugin-loader.test.d.ts +2 -0
- package/dist/plugins/plugin-loader.test.d.ts.map +1 -0
- package/dist/plugins/plugin-loader.test.js +87 -0
- package/dist/plugins/plugin-loader.test.js.map +1 -0
- package/dist/prompt-builder.d.ts +47 -0
- package/dist/prompt-builder.d.ts.map +1 -0
- package/dist/prompt-builder.js +263 -0
- package/dist/prompt-builder.js.map +1 -0
- package/dist/prompt-builder.test.d.ts +2 -0
- package/dist/prompt-builder.test.d.ts.map +1 -0
- package/dist/prompt-builder.test.js +154 -0
- package/dist/prompt-builder.test.js.map +1 -0
- package/dist/providers/anthropic.d.ts +14 -0
- package/dist/providers/anthropic.d.ts.map +1 -0
- package/dist/providers/anthropic.js +118 -0
- package/dist/providers/anthropic.js.map +1 -0
- package/dist/providers/anthropic.test.d.ts +2 -0
- package/dist/providers/anthropic.test.d.ts.map +1 -0
- package/dist/providers/anthropic.test.js +76 -0
- package/dist/providers/anthropic.test.js.map +1 -0
- package/dist/providers/mock.d.ts +17 -0
- package/dist/providers/mock.d.ts.map +1 -0
- package/dist/providers/mock.js +39 -0
- package/dist/providers/mock.js.map +1 -0
- package/dist/providers/ollama.d.ts +18 -0
- package/dist/providers/ollama.d.ts.map +1 -0
- package/dist/providers/ollama.js +155 -0
- package/dist/providers/ollama.js.map +1 -0
- package/dist/providers/ollama.test.d.ts +2 -0
- package/dist/providers/ollama.test.d.ts.map +1 -0
- package/dist/providers/ollama.test.js +123 -0
- package/dist/providers/ollama.test.js.map +1 -0
- package/dist/providers/openai-responses.d.ts +18 -0
- package/dist/providers/openai-responses.d.ts.map +1 -0
- package/dist/providers/openai-responses.js +175 -0
- package/dist/providers/openai-responses.js.map +1 -0
- package/dist/providers/openai.d.ts +16 -0
- package/dist/providers/openai.d.ts.map +1 -0
- package/dist/providers/openai.js +132 -0
- package/dist/providers/openai.js.map +1 -0
- package/dist/providers/openai.test.d.ts +2 -0
- package/dist/providers/openai.test.d.ts.map +1 -0
- package/dist/providers/openai.test.js +51 -0
- package/dist/providers/openai.test.js.map +1 -0
- package/dist/providers/provider.d.ts +42 -0
- package/dist/providers/provider.d.ts.map +1 -0
- package/dist/providers/provider.js +10 -0
- package/dist/providers/provider.js.map +1 -0
- package/dist/providers/registry.d.ts +42 -0
- package/dist/providers/registry.d.ts.map +1 -0
- package/dist/providers/registry.js +65 -0
- package/dist/providers/registry.js.map +1 -0
- package/dist/runner.d.ts +46 -0
- package/dist/runner.d.ts.map +1 -0
- package/dist/runner.js +324 -0
- package/dist/runner.js.map +1 -0
- package/dist/runner.test.d.ts +2 -0
- package/dist/runner.test.d.ts.map +1 -0
- package/dist/runner.test.js +280 -0
- package/dist/runner.test.js.map +1 -0
- package/dist/script-executor.d.ts +11 -0
- package/dist/script-executor.d.ts.map +1 -0
- package/dist/script-executor.js +85 -0
- package/dist/script-executor.js.map +1 -0
- package/dist/tools/builtin/git.d.ts +6 -0
- package/dist/tools/builtin/git.d.ts.map +1 -0
- package/dist/tools/builtin/git.js +294 -0
- package/dist/tools/builtin/git.js.map +1 -0
- package/dist/tools/builtin/patch.d.ts +6 -0
- package/dist/tools/builtin/patch.d.ts.map +1 -0
- package/dist/tools/builtin/patch.js +238 -0
- package/dist/tools/builtin/patch.js.map +1 -0
- package/dist/tools/builtin/repo-manager.d.ts +6 -0
- package/dist/tools/builtin/repo-manager.d.ts.map +1 -0
- package/dist/tools/builtin/repo-manager.js +171 -0
- package/dist/tools/builtin/repo-manager.js.map +1 -0
- package/dist/tools/builtin/search.d.ts +6 -0
- package/dist/tools/builtin/search.d.ts.map +1 -0
- package/dist/tools/builtin/search.js +104 -0
- package/dist/tools/builtin/search.js.map +1 -0
- package/dist/tools/builtin/shell.d.ts +6 -0
- package/dist/tools/builtin/shell.d.ts.map +1 -0
- package/dist/tools/builtin/shell.js +78 -0
- package/dist/tools/builtin/shell.js.map +1 -0
- package/dist/tools/builtin/studio-run.d.ts +12 -0
- package/dist/tools/builtin/studio-run.d.ts.map +1 -0
- package/dist/tools/builtin/studio-run.js +61 -0
- package/dist/tools/builtin/studio-run.js.map +1 -0
- package/dist/tools/builtin/web-search.d.ts +7 -0
- package/dist/tools/builtin/web-search.d.ts.map +1 -0
- package/dist/tools/builtin/web-search.js +94 -0
- package/dist/tools/builtin/web-search.js.map +1 -0
- package/dist/tools/builtin/web-search.test.d.ts +5 -0
- package/dist/tools/builtin/web-search.test.d.ts.map +1 -0
- package/dist/tools/builtin/web-search.test.js +90 -0
- package/dist/tools/builtin/web-search.test.js.map +1 -0
- package/dist/tools/errors.d.ts +4 -0
- package/dist/tools/errors.d.ts.map +1 -0
- package/dist/tools/errors.js +11 -0
- package/dist/tools/errors.js.map +1 -0
- package/dist/tools/errors.test.d.ts +2 -0
- package/dist/tools/errors.test.d.ts.map +1 -0
- package/dist/tools/errors.test.js +14 -0
- package/dist/tools/errors.test.js.map +1 -0
- package/dist/tools/plugin-loader.d.ts +31 -0
- package/dist/tools/plugin-loader.d.ts.map +1 -0
- package/dist/tools/plugin-loader.js +183 -0
- package/dist/tools/plugin-loader.js.map +1 -0
- package/dist/tools/plugin-loader.test.d.ts +2 -0
- package/dist/tools/plugin-loader.test.d.ts.map +1 -0
- package/dist/tools/plugin-loader.test.js +116 -0
- package/dist/tools/plugin-loader.test.js.map +1 -0
- package/dist/tools/skills/skill-loader.d.ts +23 -0
- package/dist/tools/skills/skill-loader.d.ts.map +1 -0
- package/dist/tools/skills/skill-loader.js +58 -0
- package/dist/tools/skills/skill-loader.js.map +1 -0
- package/dist/tools/skills/skill-loader.test.d.ts +2 -0
- package/dist/tools/skills/skill-loader.test.d.ts.map +1 -0
- package/dist/tools/skills/skill-loader.test.js +92 -0
- package/dist/tools/skills/skill-loader.test.js.map +1 -0
- package/dist/tools/tool-executor.d.ts +12 -0
- package/dist/tools/tool-executor.d.ts.map +1 -0
- package/dist/tools/tool-executor.js +76 -0
- package/dist/tools/tool-executor.js.map +1 -0
- package/dist/tools/tool-executor.test.d.ts +2 -0
- package/dist/tools/tool-executor.test.d.ts.map +1 -0
- package/dist/tools/tool-executor.test.js +83 -0
- package/dist/tools/tool-executor.test.js.map +1 -0
- package/dist/tools/tool-registry.d.ts +65 -0
- package/dist/tools/tool-registry.d.ts.map +1 -0
- package/dist/tools/tool-registry.js +112 -0
- package/dist/tools/tool-registry.js.map +1 -0
- package/dist/tools/yaml-executor.d.ts +22 -0
- package/dist/tools/yaml-executor.d.ts.map +1 -0
- package/dist/tools/yaml-executor.js +80 -0
- package/dist/tools/yaml-executor.js.map +1 -0
- package/dist/utils/race-signal.d.ts +6 -0
- package/dist/utils/race-signal.d.ts.map +1 -0
- package/dist/utils/race-signal.js +21 -0
- package/dist/utils/race-signal.js.map +1 -0
- package/dist/utils/race-signal.test.d.ts +2 -0
- package/dist/utils/race-signal.test.d.ts.map +1 -0
- package/dist/utils/race-signal.test.js +48 -0
- package/dist/utils/race-signal.test.js.map +1 -0
- package/package.json +36 -0
- package/src/__tests__/script-executor.test.ts +180 -0
- package/src/index.ts +68 -0
- package/src/integrations/integration-loader.test.ts +88 -0
- package/src/integrations/integration-loader.ts +68 -0
- package/src/middleware/anonymization.ts +38 -0
- package/src/plugins/index.ts +4 -0
- package/src/plugins/mcp-client.test.ts +148 -0
- package/src/plugins/mcp-client.ts +128 -0
- package/src/plugins/oauth-provider.test.ts +167 -0
- package/src/plugins/oauth-provider.ts +175 -0
- package/src/plugins/plugin-loader.test.ts +114 -0
- package/src/plugins/plugin-loader.ts +90 -0
- package/src/prompt-builder.test.ts +167 -0
- package/src/prompt-builder.ts +332 -0
- package/src/providers/anthropic.test.ts +101 -0
- package/src/providers/anthropic.ts +135 -0
- package/src/providers/mock.ts +57 -0
- package/src/providers/ollama.test.ts +166 -0
- package/src/providers/ollama.ts +152 -0
- package/src/providers/openai-responses.ts +212 -0
- package/src/providers/openai.test.ts +67 -0
- package/src/providers/openai.ts +139 -0
- package/src/providers/provider.ts +54 -0
- package/src/providers/registry.ts +77 -0
- package/src/runner.test.ts +343 -0
- package/src/runner.ts +396 -0
- package/src/script-executor.ts +107 -0
- package/src/tools/builtin/git.ts +311 -0
- package/src/tools/builtin/patch.ts +257 -0
- package/src/tools/builtin/repo-manager.ts +142 -0
- package/src/tools/builtin/search.ts +108 -0
- package/src/tools/builtin/shell.ts +82 -0
- package/src/tools/builtin/studio-run.ts +73 -0
- package/src/tools/builtin/web-search.test.ts +122 -0
- package/src/tools/builtin/web-search.ts +101 -0
- package/src/tools/errors.test.ts +12 -0
- package/src/tools/errors.ts +6 -0
- package/src/tools/plugin-loader.test.ts +130 -0
- package/src/tools/plugin-loader.ts +203 -0
- package/src/tools/skills/README.md +49 -0
- package/src/tools/skills/skill-loader.test.ts +106 -0
- package/src/tools/skills/skill-loader.ts +62 -0
- package/src/tools/tool-executor.test.ts +88 -0
- package/src/tools/tool-executor.ts +84 -0
- package/src/tools/tool-registry.ts +130 -0
- package/src/tools/yaml-executor.ts +120 -0
- package/src/utils/race-signal.test.ts +50 -0
- package/src/utils/race-signal.ts +17 -0
- package/templates/integrations/linear.integration.yaml +35 -0
- package/templates/integrations/slack.integration.yaml +22 -0
- package/templates/integrations/webhook.integration.yaml +17 -0
- package/templates/tools/git.tool.yaml +80 -0
- package/templates/tools/repo-manager.tool.yaml +64 -0
- package/templates/tools/search.tool.yaml +22 -0
- package/templates/tools/shell.tool.yaml +19 -0
- package/templates/tools/web-search.tool.yaml +24 -0
- package/tests/anonymization-middleware.test.ts +61 -0
- package/tests/anthropic.test.ts +87 -0
- package/tests/apply-patch.test.ts +355 -0
- package/tests/fixtures/tools/test-builtin.tool.yaml +14 -0
- package/tests/fixtures/tools/test-shell.tool.yaml +19 -0
- package/tests/mock-provider.test.ts +104 -0
- package/tests/openai.test.ts +72 -0
- package/tests/plugin-loader.test.ts +54 -0
- package/tests/prompt-builder.test.ts +468 -0
- package/tests/runner-anonymization.test.ts +89 -0
- package/tests/runner.test.ts +885 -0
- package/tests/studio-run.test.ts +94 -0
- package/tests/tool-executor.test.ts +115 -0
- package/tests/tool-registry.test.ts +84 -0
- package/tests/yaml-executor.test.ts +76 -0
- package/tsconfig.json +20 -0
- package/vitest.config.ts +7 -0
|
@@ -0,0 +1,76 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
const vitest_1 = require("vitest");
|
|
4
|
+
const anthropic_js_1 = require("./anthropic.js");
|
|
5
|
+
// We mock the entire SDK so no real HTTP calls are made.
|
|
6
|
+
// The fake stream hangs on finalMessage() to simulate the bug scenario.
|
|
7
|
+
vitest_1.vi.mock('@anthropic-ai/sdk', () => {
|
|
8
|
+
return {
|
|
9
|
+
default: class FakeAnthropic {
|
|
10
|
+
messages = {
|
|
11
|
+
stream: (_params, _opts) => {
|
|
12
|
+
const listeners = new Map();
|
|
13
|
+
return {
|
|
14
|
+
on(event, handler) {
|
|
15
|
+
if (!listeners.has(event))
|
|
16
|
+
listeners.set(event, []);
|
|
17
|
+
listeners.get(event).push(handler);
|
|
18
|
+
return this;
|
|
19
|
+
},
|
|
20
|
+
// finalMessage() hangs forever — this is the bug we're fixing
|
|
21
|
+
finalMessage: () => new Promise(() => { }),
|
|
22
|
+
};
|
|
23
|
+
},
|
|
24
|
+
create: (_params, _opts) => new Promise(() => { }),
|
|
25
|
+
};
|
|
26
|
+
},
|
|
27
|
+
};
|
|
28
|
+
});
|
|
29
|
+
(0, vitest_1.describe)('AnthropicProvider', () => {
|
|
30
|
+
let provider;
|
|
31
|
+
(0, vitest_1.beforeEach)(() => {
|
|
32
|
+
provider = new anthropic_js_1.AnthropicProvider('test-key');
|
|
33
|
+
});
|
|
34
|
+
(0, vitest_1.it)('aborts streaming call when signal fires', async () => {
|
|
35
|
+
const controller = new AbortController();
|
|
36
|
+
const onToken = vitest_1.vi.fn();
|
|
37
|
+
const callPromise = provider.call({
|
|
38
|
+
model: 'claude-haiku-4-20250514',
|
|
39
|
+
messages: [{ role: 'user', content: 'hello' }],
|
|
40
|
+
}, onToken, controller.signal);
|
|
41
|
+
// Fire signal after a tick to let the stream start
|
|
42
|
+
await Promise.resolve();
|
|
43
|
+
controller.abort();
|
|
44
|
+
await (0, vitest_1.expect)(callPromise).rejects.toSatisfy((e) => e instanceof DOMException && e.name === 'AbortError');
|
|
45
|
+
});
|
|
46
|
+
(0, vitest_1.it)('aborts non-streaming call when signal fires', async () => {
|
|
47
|
+
const controller = new AbortController();
|
|
48
|
+
const callPromise = provider.call({
|
|
49
|
+
model: 'claude-haiku-4-20250514',
|
|
50
|
+
messages: [{ role: 'user', content: 'hello' }],
|
|
51
|
+
}, undefined, controller.signal);
|
|
52
|
+
await Promise.resolve();
|
|
53
|
+
controller.abort();
|
|
54
|
+
await (0, vitest_1.expect)(callPromise).rejects.toSatisfy((e) => e instanceof DOMException && e.name === 'AbortError');
|
|
55
|
+
});
|
|
56
|
+
(0, vitest_1.it)('resolves normally when signal is not aborted', async () => {
|
|
57
|
+
// Directly override stream on the already-created client instance.
|
|
58
|
+
// (prototype override won't work because messages is an instance property.)
|
|
59
|
+
const client = provider.client;
|
|
60
|
+
client.messages.stream = () => ({
|
|
61
|
+
on: (_, __) => ({}),
|
|
62
|
+
finalMessage: () => Promise.resolve({
|
|
63
|
+
content: [{ type: 'text', text: '{"ok":true}' }],
|
|
64
|
+
stop_reason: 'end_turn',
|
|
65
|
+
usage: { input_tokens: 10, output_tokens: 5 },
|
|
66
|
+
}),
|
|
67
|
+
});
|
|
68
|
+
const controller = new AbortController();
|
|
69
|
+
const result = await provider.call({
|
|
70
|
+
model: 'claude-haiku-4-20250514',
|
|
71
|
+
messages: [{ role: 'user', content: 'hello' }],
|
|
72
|
+
}, vitest_1.vi.fn(), controller.signal);
|
|
73
|
+
(0, vitest_1.expect)(result).toBeDefined();
|
|
74
|
+
});
|
|
75
|
+
});
|
|
76
|
+
//# sourceMappingURL=anthropic.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic.test.js","sourceRoot":"","sources":["../../src/providers/anthropic.test.ts"],"names":[],"mappings":";;AAAA,mCAA8D;AAC9D,iDAAmD;AAEnD,yDAAyD;AACzD,wEAAwE;AACxE,WAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE;IAChC,OAAO;QACL,OAAO,EAAE,MAAM,aAAa;YAC1B,QAAQ,GAAG;gBACT,MAAM,EAAE,CAAC,OAAgB,EAAE,KAAc,EAAE,EAAE;oBAC3C,MAAM,SAAS,GAAG,IAAI,GAAG,EAA4C,CAAC;oBACtE,OAAO;wBACL,EAAE,CAAC,KAAa,EAAE,OAAqC;4BACrD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC;gCAAE,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;4BACpD,SAAS,CAAC,GAAG,CAAC,KAAK,CAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;4BACpC,OAAO,IAAI,CAAC;wBACd,CAAC;wBACD,8DAA8D;wBAC9D,YAAY,EAAE,GAAG,EAAE,CAAC,IAAI,OAAO,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;qBAC1C,CAAC;gBACJ,CAAC;gBACD,MAAM,EAAE,CAAC,OAAgB,EAAE,KAAc,EAAE,EAAE,CAAC,IAAI,OAAO,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;aACpE,CAAC;SACH;KACF,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,IAAA,iBAAQ,EAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,IAAI,QAA2B,CAAC;IAEhC,IAAA,mBAAU,EAAC,GAAG,EAAE;QACd,QAAQ,GAAG,IAAI,gCAAiB,CAAC,UAAU,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,OAAO,GAAG,WAAE,CAAC,EAAE,EAAE,CAAC;QAExB,MAAM,WAAW,GAAG,QAAQ,CAAC,IAAI,CAC/B;YACE,KAAK,EAAE,yBAAyB;YAChC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;SAC/C,EACD,OAAO,EACP,UAAU,CAAC,MAAM,CAClB,CAAC;QAEF,mDAAmD;QACnD,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC;QACxB,UAAU,CAAC,KAAK,EAAE,CAAC;QAEnB,MAAM,IAAA,eAAM,EAAC,WAAW,CAAC,CAAC,OAAO,CAAC,SAAS,CACzC,CAAC,CAAU,EAAE,EAAE,CAAC,CAAC,YAAY,YAAY,IAAK,CAAkB,CAAC,IAAI,KAAK,YAAY,CACvF,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QAEzC,MAAM,WAAW,GAAG,QAAQ,CAAC,IAAI,CAC/B;YACE,KAAK,EAAE,yBAAyB;YAChC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;SAC/C,EACD,SAAS,EACT,UAAU,CAAC,MAAM,CAClB,CAAC;QAEF,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC;QACxB,UAAU,CAAC,KAAK,EAAE,CAAC;QAEnB,MAAM,IAAA,eAAM,EAAC,WAAW,CAAC,CAAC,OAAO,CAAC,SAAS,CACzC,CAAC,CAAU,EAAE,EAAE,CAAC,CAAC,YAAY,YAAY,IAAK,CAAkB,CAAC,IAAI,KAAK,YAAY,CACvF,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,mEAAmE;QACnE,4EAA4E;QAC5E,MAAM,MAAM,GAAI,QAAyE,CAAC,MAAM,CAAC;QACjG,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC,CAAC;YAC9B,EAAE,EAAE,CAAC,CAAS,EAAE,EAAW,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC;YACpC,YAAY,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;gBAClC,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;gBAChD,WAAW,EAAE,UAAU;gBACvB,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE;aAC9C,CAAC;SACH,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,CAChC;YACE,KAAK,EAAE,yBAAyB;YAChC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;SAC/C,EACD,WAAE,CAAC,EAAE,EAAE,EACP,UAAU,CAAC,MAAM,CAClB,CAAC;QACF,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;IAC/B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
import type { LLMRequest, LLMResponse } from '@studio-foundation/contracts';
|
|
2
|
+
import type { AgentLoopProvider, AgentLoopResult, ToolCallOutcome } from './provider.js';
|
|
3
|
+
export interface MockStageConfig {
|
|
4
|
+
output: Record<string, unknown>;
|
|
5
|
+
tool_calls: Array<{
|
|
6
|
+
name: string;
|
|
7
|
+
arguments: Record<string, unknown>;
|
|
8
|
+
}>;
|
|
9
|
+
}
|
|
10
|
+
export declare class MockProvider implements AgentLoopProvider {
|
|
11
|
+
private readonly stages;
|
|
12
|
+
readonly name = "mock";
|
|
13
|
+
constructor(stages: Map<string, MockStageConfig>);
|
|
14
|
+
call(_request: LLMRequest, _onToken?: (token: string) => void, _signal?: AbortSignal): Promise<LLMResponse>;
|
|
15
|
+
runAgentLoop(request: LLMRequest, executeTool: (name: string, args: Record<string, unknown>, callId: string) => Promise<ToolCallOutcome>, onToken?: (token: string) => void, _signal?: AbortSignal): Promise<AgentLoopResult>;
|
|
16
|
+
}
|
|
17
|
+
//# sourceMappingURL=mock.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mock.d.ts","sourceRoot":"","sources":["../../src/providers/mock.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,8BAA8B,CAAC;AAC5E,OAAO,KAAK,EAAE,iBAAiB,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAEzF,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAChC,UAAU,EAAE,KAAK,CAAC;QAChB,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACpC,CAAC,CAAC;CACJ;AAED,qBAAa,YAAa,YAAW,iBAAiB;IAGxC,OAAO,CAAC,QAAQ,CAAC,MAAM;IAFnC,QAAQ,CAAC,IAAI,UAAU;gBAEM,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,eAAe,CAAC;IAE3D,IAAI,CAAC,QAAQ,EAAE,UAAU,EAAE,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,EAAE,OAAO,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;IAI3G,YAAY,CAChB,OAAO,EAAE,UAAU,EACnB,WAAW,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,eAAe,CAAC,EACtG,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,EACjC,OAAO,CAAC,EAAE,WAAW,GACpB,OAAO,CAAC,eAAe,CAAC;CA8B5B"}
|
|
@@ -0,0 +1,39 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.MockProvider = void 0;
|
|
4
|
+
const node_crypto_1 = require("node:crypto");
|
|
5
|
+
class MockProvider {
|
|
6
|
+
stages;
|
|
7
|
+
name = 'mock';
|
|
8
|
+
constructor(stages) {
|
|
9
|
+
this.stages = stages;
|
|
10
|
+
}
|
|
11
|
+
async call(_request, _onToken, _signal) {
|
|
12
|
+
throw new Error('MockProvider: use runAgentLoop, not call()');
|
|
13
|
+
}
|
|
14
|
+
async runAgentLoop(request, executeTool, onToken, _signal) {
|
|
15
|
+
if (!request.stage_name) {
|
|
16
|
+
throw new Error('MockProvider requires stage_name in LLMRequest');
|
|
17
|
+
}
|
|
18
|
+
const config = this.stages.get(request.stage_name);
|
|
19
|
+
if (!config) {
|
|
20
|
+
throw new Error(`Unknown mock stage: "${request.stage_name}". Add it to mock.yaml.`);
|
|
21
|
+
}
|
|
22
|
+
// Emit a fake token to exercise the streaming pipeline in tests
|
|
23
|
+
onToken?.('...');
|
|
24
|
+
const toolCallResults = [];
|
|
25
|
+
for (const tc of config.tool_calls) {
|
|
26
|
+
const callId = (0, node_crypto_1.randomUUID)();
|
|
27
|
+
const outcome = await executeTool(tc.name, tc.arguments, callId);
|
|
28
|
+
toolCallResults.push({ id: callId, name: tc.name, arguments: tc.arguments, ...outcome });
|
|
29
|
+
}
|
|
30
|
+
return {
|
|
31
|
+
content: JSON.stringify(config.output),
|
|
32
|
+
tool_calls: toolCallResults,
|
|
33
|
+
finish_reason: 'stop',
|
|
34
|
+
usage: { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 },
|
|
35
|
+
};
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
exports.MockProvider = MockProvider;
|
|
39
|
+
//# sourceMappingURL=mock.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mock.js","sourceRoot":"","sources":["../../src/providers/mock.ts"],"names":[],"mappings":";;;AAAA,6CAAyC;AAYzC,MAAa,YAAY;IAGM;IAFpB,IAAI,GAAG,MAAM,CAAC;IAEvB,YAA6B,MAAoC;QAApC,WAAM,GAAN,MAAM,CAA8B;IAAG,CAAC;IAErE,KAAK,CAAC,IAAI,CAAC,QAAoB,EAAE,QAAkC,EAAE,OAAqB;QACxF,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;IAChE,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,OAAmB,EACnB,WAAsG,EACtG,OAAiC,EACjC,OAAqB;QAErB,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;QACpE,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACnD,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CACb,wBAAwB,OAAO,CAAC,UAAU,yBAAyB,CACpE,CAAC;QACJ,CAAC;QAED,gEAAgE;QAChE,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC;QAEjB,MAAM,eAAe,GAAkC,EAAE,CAAC;QAE1D,KAAK,MAAM,EAAE,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;YACnC,MAAM,MAAM,GAAG,IAAA,wBAAU,GAAE,CAAC;YAC5B,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;YACjE,eAAe,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,SAAS,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;QAC3F,CAAC;QAED,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC;YACtC,UAAU,EAAE,eAAe;YAC3B,aAAa,EAAE,MAAM;YACrB,KAAK,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE;SACnE,CAAC;IACJ,CAAC;CACF;AA5CD,oCA4CC"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Ollama provider — OpenAI-compatible local LLM server
|
|
3
|
+
* Uses the openai npm SDK pointed at Ollama's /v1 endpoint.
|
|
4
|
+
*/
|
|
5
|
+
import type { LLMRequest, LLMResponse } from '@studio-foundation/contracts';
|
|
6
|
+
import type { Provider } from './provider.js';
|
|
7
|
+
export declare class OllamaProvider implements Provider {
|
|
8
|
+
readonly name = "ollama";
|
|
9
|
+
private client;
|
|
10
|
+
private baseUrl;
|
|
11
|
+
constructor(baseUrl?: string);
|
|
12
|
+
call(request: LLMRequest, onToken?: (token: string) => void, signal?: AbortSignal): Promise<LLMResponse>;
|
|
13
|
+
private callNonStreaming;
|
|
14
|
+
private callStreaming;
|
|
15
|
+
private buildMessages;
|
|
16
|
+
private buildTools;
|
|
17
|
+
}
|
|
18
|
+
//# sourceMappingURL=ollama.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ollama.d.ts","sourceRoot":"","sources":["../../src/providers/ollama.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,8BAA8B,CAAC;AAC5E,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAa9C,qBAAa,cAAe,YAAW,QAAQ;IAC7C,QAAQ,CAAC,IAAI,YAAY;IACzB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,OAAO,CAAS;gBAEZ,OAAO,SAA2B;IAQxC,IAAI,CAAC,OAAO,EAAE,UAAU,EAAE,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;YAgBhG,gBAAgB;YAiChB,aAAa;IAkD3B,OAAO,CAAC,aAAa;IASrB,OAAO,CAAC,UAAU;CAWnB"}
|
|
@@ -0,0 +1,155 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Ollama provider — OpenAI-compatible local LLM server
|
|
4
|
+
* Uses the openai npm SDK pointed at Ollama's /v1 endpoint.
|
|
5
|
+
*/
|
|
6
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
7
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
8
|
+
};
|
|
9
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
10
|
+
exports.OllamaProvider = void 0;
|
|
11
|
+
const openai_1 = __importDefault(require("openai"));
|
|
12
|
+
function isConnectionRefused(err) {
|
|
13
|
+
if (!(err instanceof Error))
|
|
14
|
+
return false;
|
|
15
|
+
if ('code' in err && err.code === 'ECONNREFUSED')
|
|
16
|
+
return true;
|
|
17
|
+
const cause = err.cause;
|
|
18
|
+
if (cause instanceof Error && 'code' in cause && cause.code === 'ECONNREFUSED')
|
|
19
|
+
return true;
|
|
20
|
+
if (err.message.includes('ECONNREFUSED'))
|
|
21
|
+
return true;
|
|
22
|
+
return false;
|
|
23
|
+
}
|
|
24
|
+
class OllamaProvider {
|
|
25
|
+
name = 'ollama';
|
|
26
|
+
client;
|
|
27
|
+
baseUrl;
|
|
28
|
+
constructor(baseUrl = 'http://localhost:11434') {
|
|
29
|
+
this.baseUrl = baseUrl;
|
|
30
|
+
this.client = new openai_1.default({
|
|
31
|
+
baseURL: `${baseUrl}/v1`,
|
|
32
|
+
apiKey: 'ollama', // required by SDK, ignored by Ollama
|
|
33
|
+
});
|
|
34
|
+
}
|
|
35
|
+
async call(request, onToken, signal) {
|
|
36
|
+
try {
|
|
37
|
+
if (onToken) {
|
|
38
|
+
return await this.callStreaming(request, onToken, signal);
|
|
39
|
+
}
|
|
40
|
+
return await this.callNonStreaming(request, signal);
|
|
41
|
+
}
|
|
42
|
+
catch (err) {
|
|
43
|
+
if (isConnectionRefused(err)) {
|
|
44
|
+
throw new Error(`Ollama is not running at ${this.baseUrl}. Start it with: ollama serve`);
|
|
45
|
+
}
|
|
46
|
+
throw err;
|
|
47
|
+
}
|
|
48
|
+
}
|
|
49
|
+
async callNonStreaming(request, signal) {
|
|
50
|
+
const completion = await this.client.chat.completions.create({
|
|
51
|
+
model: request.model,
|
|
52
|
+
messages: this.buildMessages(request),
|
|
53
|
+
tools: this.buildTools(request),
|
|
54
|
+
temperature: request.temperature,
|
|
55
|
+
max_tokens: request.max_tokens,
|
|
56
|
+
response_format: request.json_mode ? { type: 'json_object' } : undefined,
|
|
57
|
+
}, { signal });
|
|
58
|
+
const choice = completion.choices[0];
|
|
59
|
+
if (!choice)
|
|
60
|
+
throw new Error('Ollama returned an empty choices array');
|
|
61
|
+
const tool_calls = choice.message.tool_calls?.map(tc => ({
|
|
62
|
+
id: tc.id,
|
|
63
|
+
name: tc.function.name,
|
|
64
|
+
arguments: (() => {
|
|
65
|
+
try {
|
|
66
|
+
return JSON.parse(tc.function.arguments);
|
|
67
|
+
}
|
|
68
|
+
catch {
|
|
69
|
+
return {};
|
|
70
|
+
}
|
|
71
|
+
})(),
|
|
72
|
+
})) || [];
|
|
73
|
+
return {
|
|
74
|
+
content: choice.message.content || '',
|
|
75
|
+
tool_calls,
|
|
76
|
+
finish_reason: choice.finish_reason ?? 'stop',
|
|
77
|
+
usage: completion.usage ? {
|
|
78
|
+
prompt_tokens: completion.usage.prompt_tokens,
|
|
79
|
+
completion_tokens: completion.usage.completion_tokens,
|
|
80
|
+
total_tokens: completion.usage.total_tokens,
|
|
81
|
+
} : undefined,
|
|
82
|
+
};
|
|
83
|
+
}
|
|
84
|
+
async callStreaming(request, onToken, signal) {
|
|
85
|
+
// NOTE: No stream_options here — older Ollama versions reject it
|
|
86
|
+
const stream = this.client.chat.completions.create({
|
|
87
|
+
model: request.model,
|
|
88
|
+
messages: this.buildMessages(request),
|
|
89
|
+
tools: this.buildTools(request),
|
|
90
|
+
temperature: request.temperature,
|
|
91
|
+
max_tokens: request.max_tokens,
|
|
92
|
+
response_format: request.json_mode ? { type: 'json_object' } : undefined,
|
|
93
|
+
stream: true,
|
|
94
|
+
}, { signal });
|
|
95
|
+
let textContent = '';
|
|
96
|
+
const toolCallMap = new Map();
|
|
97
|
+
let finishReason = 'stop';
|
|
98
|
+
for await (const chunk of stream) {
|
|
99
|
+
if (signal?.aborted)
|
|
100
|
+
throw new DOMException('Aborted', 'AbortError');
|
|
101
|
+
const delta = chunk.choices[0]?.delta;
|
|
102
|
+
if (delta?.content) {
|
|
103
|
+
textContent += delta.content;
|
|
104
|
+
onToken(delta.content);
|
|
105
|
+
}
|
|
106
|
+
if (delta?.tool_calls) {
|
|
107
|
+
for (const tc of delta.tool_calls) {
|
|
108
|
+
if (!toolCallMap.has(tc.index)) {
|
|
109
|
+
toolCallMap.set(tc.index, { id: '', name: '', args: '' });
|
|
110
|
+
}
|
|
111
|
+
const acc = toolCallMap.get(tc.index);
|
|
112
|
+
if (tc.id)
|
|
113
|
+
acc.id = tc.id;
|
|
114
|
+
if (tc.function?.name)
|
|
115
|
+
acc.name += tc.function.name;
|
|
116
|
+
if (tc.function?.arguments)
|
|
117
|
+
acc.args += tc.function.arguments;
|
|
118
|
+
}
|
|
119
|
+
}
|
|
120
|
+
if (chunk.choices[0]?.finish_reason)
|
|
121
|
+
finishReason = chunk.choices[0].finish_reason;
|
|
122
|
+
}
|
|
123
|
+
const tool_calls = Array.from(toolCallMap.values()).map(tc => ({
|
|
124
|
+
id: tc.id,
|
|
125
|
+
name: tc.name,
|
|
126
|
+
arguments: JSON.parse(tc.args || '{}'),
|
|
127
|
+
}));
|
|
128
|
+
return { content: textContent, tool_calls, finish_reason: finishReason, usage: undefined };
|
|
129
|
+
}
|
|
130
|
+
buildMessages(request) {
|
|
131
|
+
return request.messages.map(msg => {
|
|
132
|
+
if (msg.role === 'system')
|
|
133
|
+
return { role: 'system', content: msg.content };
|
|
134
|
+
if (msg.role === 'user')
|
|
135
|
+
return { role: 'user', content: msg.content };
|
|
136
|
+
if (msg.role === 'assistant')
|
|
137
|
+
return { role: 'assistant', content: msg.content };
|
|
138
|
+
throw new Error(`Unsupported message role: ${msg.role}`);
|
|
139
|
+
});
|
|
140
|
+
}
|
|
141
|
+
buildTools(request) {
|
|
142
|
+
if (!request.tools || request.tools.length === 0)
|
|
143
|
+
return undefined;
|
|
144
|
+
return request.tools.map(tool => ({
|
|
145
|
+
type: 'function',
|
|
146
|
+
function: {
|
|
147
|
+
name: tool.name,
|
|
148
|
+
description: tool.description,
|
|
149
|
+
parameters: tool.parameters,
|
|
150
|
+
},
|
|
151
|
+
}));
|
|
152
|
+
}
|
|
153
|
+
}
|
|
154
|
+
exports.OllamaProvider = OllamaProvider;
|
|
155
|
+
//# sourceMappingURL=ollama.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ollama.js","sourceRoot":"","sources":["../../src/providers/ollama.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;;;;AAIH,oDAA4B;AAG5B,SAAS,mBAAmB,CAAC,GAAY;IACvC,IAAI,CAAC,CAAC,GAAG,YAAY,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IAC1C,IAAI,MAAM,IAAI,GAAG,IAAK,GAA6B,CAAC,IAAI,KAAK,cAAc;QAAE,OAAO,IAAI,CAAC;IACzF,MAAM,KAAK,GAAI,GAAmC,CAAC,KAAK,CAAC;IACzD,IAAI,KAAK,YAAY,KAAK,IAAI,MAAM,IAAI,KAAK,IAAK,KAA+B,CAAC,IAAI,KAAK,cAAc;QAAE,OAAO,IAAI,CAAC;IACvH,IAAI,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC;QAAE,OAAO,IAAI,CAAC;IACtD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAa,cAAc;IAChB,IAAI,GAAG,QAAQ,CAAC;IACjB,MAAM,CAAS;IACf,OAAO,CAAS;IAExB,YAAY,OAAO,GAAG,wBAAwB;QAC5C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,IAAI,gBAAM,CAAC;YACvB,OAAO,EAAE,GAAG,OAAO,KAAK;YACxB,MAAM,EAAE,QAAQ,EAAE,qCAAqC;SACxD,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAAmB,EAAE,OAAiC,EAAE,MAAoB;QACrF,IAAI,CAAC;YACH,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;YAC5D,CAAC;YACD,OAAO,MAAM,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACtD,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,mBAAmB,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CACb,4BAA4B,IAAI,CAAC,OAAO,+BAA+B,CACxE,CAAC;YACJ,CAAC;YACD,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,OAAmB,EAAE,MAAoB;QACtE,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;YAC3D,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,QAAQ,EAAE,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC;YACrC,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;YAC/B,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,eAAe,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,SAAS;SACzE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;QAEf,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QACrC,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QACvE,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YACvD,EAAE,EAAE,EAAE,CAAC,EAAE;YACT,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;YACtB,SAAS,EAAE,CAAC,GAAG,EAAE;gBACf,IAAI,CAAC;oBAAC,OAAO,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;gBAAC,CAAC;gBACjD,MAAM,CAAC;oBAAC,OAAO,EAAE,CAAC;gBAAC,CAAC;YACtB,CAAC,CAAC,EAAE;SACL,CAAC,CAAC,IAAI,EAAE,CAAC;QAEV,OAAO;YACL,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE;YACrC,UAAU;YACV,aAAa,EAAE,MAAM,CAAC,aAAa,IAAI,MAAM;YAC7C,KAAK,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC;gBACxB,aAAa,EAAE,UAAU,CAAC,KAAK,CAAC,aAAa;gBAC7C,iBAAiB,EAAE,UAAU,CAAC,KAAK,CAAC,iBAAiB;gBACrD,YAAY,EAAE,UAAU,CAAC,KAAK,CAAC,YAAY;aAC5C,CAAC,CAAC,CAAC,SAAS;SACd,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,aAAa,CACzB,OAAmB,EACnB,OAAgC,EAChC,MAAoB;QAEpB,iEAAiE;QACjE,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;YACjD,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,QAAQ,EAAE,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC;YACrC,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;YAC/B,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,eAAe,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,SAAS;YACxE,MAAM,EAAE,IAAa;SACtB,EAAE,EAAE,MAAM,EAAE,CAAkD,CAAC;QAEhE,IAAI,WAAW,GAAG,EAAE,CAAC;QACrB,MAAM,WAAW,GAAG,IAAI,GAAG,EAAsD,CAAC;QAClF,IAAI,YAAY,GAAG,MAAM,CAAC;QAE1B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,IAAI,MAAM,EAAE,OAAO;gBAAE,MAAM,IAAI,YAAY,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;YACrE,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC;YACtC,IAAI,KAAK,EAAE,OAAO,EAAE,CAAC;gBACnB,WAAW,IAAI,KAAK,CAAC,OAAO,CAAC;gBAC7B,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACzB,CAAC;YACD,IAAI,KAAK,EAAE,UAAU,EAAE,CAAC;gBACtB,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;oBAClC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;wBAC/B,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;oBAC5D,CAAC;oBACD,MAAM,GAAG,GAAG,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAE,CAAC;oBACvC,IAAI,EAAE,CAAC,EAAE;wBAAE,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;oBAC1B,IAAI,EAAE,CAAC,QAAQ,EAAE,IAAI;wBAAE,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC;oBACpD,IAAI,EAAE,CAAC,QAAQ,EAAE,SAAS;wBAAE,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;gBAChE,CAAC;YACH,CAAC;YACD,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa;gBAAE,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;QACrF,CAAC;QAED,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YAC7D,EAAE,EAAE,EAAE,CAAC,EAAE;YACT,IAAI,EAAE,EAAE,CAAC,IAAI;YACb,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,IAAI,IAAI,CAAC;SACvC,CAAC,CAAC,CAAC;QAEJ,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;IAC7F,CAAC;IAEO,aAAa,CAAC,OAAmB;QACvC,OAAO,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;YAChC,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ;gBAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;YAC3E,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM;gBAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;YACvE,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW;gBAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;YACjF,MAAM,IAAI,KAAK,CAAC,6BAA6B,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,UAAU,CAAC,OAAmB;QACpC,IAAI,CAAC,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,SAAS,CAAC;QACnE,OAAO,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAChC,IAAI,EAAE,UAAmB;YACzB,QAAQ,EAAE;gBACR,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;aAC5B;SACF,CAAC,CAAC,CAAC;IACN,CAAC;CACF;AApID,wCAoIC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ollama.test.d.ts","sourceRoot":"","sources":["../../src/providers/ollama.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,123 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
const vitest_1 = require("vitest");
|
|
4
|
+
const ollama_js_1 = require("./ollama.js");
|
|
5
|
+
// Capture constructor config so we can assert on baseURL
|
|
6
|
+
let capturedConfig = {};
|
|
7
|
+
const createMock = vitest_1.vi.fn();
|
|
8
|
+
vitest_1.vi.mock('openai', () => ({
|
|
9
|
+
default: class FakeOpenAI {
|
|
10
|
+
chat = { completions: { create: createMock } };
|
|
11
|
+
constructor(config) {
|
|
12
|
+
capturedConfig = config;
|
|
13
|
+
}
|
|
14
|
+
},
|
|
15
|
+
}));
|
|
16
|
+
(0, vitest_1.beforeEach)(() => {
|
|
17
|
+
vitest_1.vi.clearAllMocks();
|
|
18
|
+
capturedConfig = {};
|
|
19
|
+
});
|
|
20
|
+
(0, vitest_1.describe)('OllamaProvider', () => {
|
|
21
|
+
(0, vitest_1.it)('has name "ollama"', () => {
|
|
22
|
+
const provider = new ollama_js_1.OllamaProvider();
|
|
23
|
+
(0, vitest_1.expect)(provider.name).toBe('ollama');
|
|
24
|
+
});
|
|
25
|
+
(0, vitest_1.it)('passes correct baseURL to SDK (default)', () => {
|
|
26
|
+
new ollama_js_1.OllamaProvider();
|
|
27
|
+
(0, vitest_1.expect)(capturedConfig.baseURL).toBe('http://localhost:11434/v1');
|
|
28
|
+
(0, vitest_1.expect)(capturedConfig.apiKey).toBe('ollama');
|
|
29
|
+
});
|
|
30
|
+
(0, vitest_1.it)('passes custom baseURL to SDK', () => {
|
|
31
|
+
new ollama_js_1.OllamaProvider('http://my-server:11434');
|
|
32
|
+
(0, vitest_1.expect)(capturedConfig.baseURL).toBe('http://my-server:11434/v1');
|
|
33
|
+
});
|
|
34
|
+
(0, vitest_1.it)('returns content and tool calls on non-streaming call', async () => {
|
|
35
|
+
createMock.mockResolvedValueOnce({
|
|
36
|
+
choices: [{
|
|
37
|
+
message: {
|
|
38
|
+
content: '{"result":"ok"}',
|
|
39
|
+
tool_calls: [{
|
|
40
|
+
id: 'call_1',
|
|
41
|
+
function: { name: 'repo_manager-read_file', arguments: '{"path":"src/foo.ts"}' },
|
|
42
|
+
}],
|
|
43
|
+
},
|
|
44
|
+
finish_reason: 'tool_calls',
|
|
45
|
+
}],
|
|
46
|
+
usage: { prompt_tokens: 10, completion_tokens: 20, total_tokens: 30 },
|
|
47
|
+
});
|
|
48
|
+
const provider = new ollama_js_1.OllamaProvider();
|
|
49
|
+
const result = await provider.call({
|
|
50
|
+
model: 'llama3.2',
|
|
51
|
+
messages: [{ role: 'user', content: 'hello' }],
|
|
52
|
+
});
|
|
53
|
+
(0, vitest_1.expect)(result.content).toBe('{"result":"ok"}');
|
|
54
|
+
(0, vitest_1.expect)(result.tool_calls).toHaveLength(1);
|
|
55
|
+
(0, vitest_1.expect)(result.tool_calls[0].name).toBe('repo_manager-read_file');
|
|
56
|
+
(0, vitest_1.expect)(result.tool_calls[0].arguments).toEqual({ path: 'src/foo.ts' });
|
|
57
|
+
(0, vitest_1.expect)(result.usage?.total_tokens).toBe(30);
|
|
58
|
+
});
|
|
59
|
+
(0, vitest_1.it)('streaming: does NOT send stream_options', async () => {
|
|
60
|
+
async function* fakeStream() {
|
|
61
|
+
yield { choices: [{ delta: { content: 'hi' }, finish_reason: null }] };
|
|
62
|
+
yield { choices: [{ delta: {}, finish_reason: 'stop' }] };
|
|
63
|
+
}
|
|
64
|
+
createMock.mockReturnValueOnce(fakeStream());
|
|
65
|
+
const provider = new ollama_js_1.OllamaProvider();
|
|
66
|
+
await provider.call({ model: 'llama3.2', messages: [{ role: 'user', content: 'hi' }] }, () => { });
|
|
67
|
+
const callArgs = createMock.mock.calls[0][0];
|
|
68
|
+
(0, vitest_1.expect)(callArgs.stream_options).toBeUndefined();
|
|
69
|
+
(0, vitest_1.expect)(callArgs.stream).toBe(true);
|
|
70
|
+
});
|
|
71
|
+
(0, vitest_1.it)('streaming: accumulates content and calls onToken', async () => {
|
|
72
|
+
async function* fakeStream() {
|
|
73
|
+
yield { choices: [{ delta: { content: 'hello' }, finish_reason: null }] };
|
|
74
|
+
yield { choices: [{ delta: { content: ' world' }, finish_reason: 'stop' }] };
|
|
75
|
+
}
|
|
76
|
+
createMock.mockReturnValueOnce(fakeStream());
|
|
77
|
+
const provider = new ollama_js_1.OllamaProvider();
|
|
78
|
+
const tokens = [];
|
|
79
|
+
const result = await provider.call({ model: 'llama3.2', messages: [{ role: 'user', content: 'hi' }] }, (t) => tokens.push(t));
|
|
80
|
+
(0, vitest_1.expect)(result.content).toBe('hello world');
|
|
81
|
+
(0, vitest_1.expect)(tokens).toEqual(['hello', ' world']);
|
|
82
|
+
});
|
|
83
|
+
(0, vitest_1.it)('wraps ECONNREFUSED with a helpful message', async () => {
|
|
84
|
+
const connErr = Object.assign(new Error('connect ECONNREFUSED 127.0.0.1:11434'), {
|
|
85
|
+
code: 'ECONNREFUSED',
|
|
86
|
+
});
|
|
87
|
+
createMock.mockRejectedValueOnce(connErr);
|
|
88
|
+
const provider = new ollama_js_1.OllamaProvider();
|
|
89
|
+
await (0, vitest_1.expect)(provider.call({ model: 'llama3.2', messages: [{ role: 'user', content: 'hi' }] })).rejects.toThrow('Ollama is not running at http://localhost:11434');
|
|
90
|
+
});
|
|
91
|
+
(0, vitest_1.it)('wraps ECONNREFUSED from error.cause too', async () => {
|
|
92
|
+
const cause = Object.assign(new Error('connect ECONNREFUSED'), { code: 'ECONNREFUSED' });
|
|
93
|
+
const wrappedErr = new Error('fetch failed');
|
|
94
|
+
wrappedErr.cause = cause;
|
|
95
|
+
createMock.mockRejectedValueOnce(wrappedErr);
|
|
96
|
+
const provider = new ollama_js_1.OllamaProvider();
|
|
97
|
+
await (0, vitest_1.expect)(provider.call({ model: 'llama3.2', messages: [{ role: 'user', content: 'hi' }] })).rejects.toThrow('ollama serve');
|
|
98
|
+
});
|
|
99
|
+
(0, vitest_1.it)('streaming: usage is undefined (Ollama does not return usage in streaming)', async () => {
|
|
100
|
+
async function* fakeStream() {
|
|
101
|
+
yield { choices: [{ delta: { content: 'hi' }, finish_reason: 'stop' }] };
|
|
102
|
+
}
|
|
103
|
+
createMock.mockReturnValueOnce(fakeStream());
|
|
104
|
+
const provider = new ollama_js_1.OllamaProvider();
|
|
105
|
+
const result = await provider.call({ model: 'llama3.2', messages: [{ role: 'user', content: 'hi' }] }, () => { });
|
|
106
|
+
(0, vitest_1.expect)(result.usage).toBeUndefined();
|
|
107
|
+
});
|
|
108
|
+
(0, vitest_1.it)('aborts streaming when signal fires mid-iteration', async () => {
|
|
109
|
+
async function* slowStream() {
|
|
110
|
+
await new Promise((r) => setTimeout(r, 20));
|
|
111
|
+
yield { choices: [{ delta: { content: 'hello' }, finish_reason: null }] };
|
|
112
|
+
await new Promise((r) => setTimeout(r, 20));
|
|
113
|
+
yield { choices: [{ delta: { content: ' world' }, finish_reason: 'stop' }] };
|
|
114
|
+
}
|
|
115
|
+
createMock.mockReturnValueOnce(slowStream());
|
|
116
|
+
const provider = new ollama_js_1.OllamaProvider();
|
|
117
|
+
const controller = new AbortController();
|
|
118
|
+
const callPromise = provider.call({ model: 'llama3.2', messages: [{ role: 'user', content: 'hi' }] }, () => { }, controller.signal);
|
|
119
|
+
setTimeout(() => controller.abort(), 10);
|
|
120
|
+
await (0, vitest_1.expect)(callPromise).rejects.toSatisfy((e) => e instanceof DOMException && e.name === 'AbortError');
|
|
121
|
+
});
|
|
122
|
+
});
|
|
123
|
+
//# sourceMappingURL=ollama.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ollama.test.js","sourceRoot":"","sources":["../../src/providers/ollama.test.ts"],"names":[],"mappings":";;AAAA,mCAA8D;AAC9D,2CAA6C;AAE7C,yDAAyD;AACzD,IAAI,cAAc,GAA4B,EAAE,CAAC;AACjD,MAAM,UAAU,GAAG,WAAE,CAAC,EAAE,EAAE,CAAC;AAE3B,WAAE,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC;IACvB,OAAO,EAAE,MAAM,UAAU;QACvB,IAAI,GAAG,EAAE,WAAW,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,CAAC;QAC/C,YAAY,MAA+B;YACzC,cAAc,GAAG,MAAM,CAAC;QAC1B,CAAC;KACF;CACF,CAAC,CAAC,CAAC;AAEJ,IAAA,mBAAU,EAAC,GAAG,EAAE;IACd,WAAE,CAAC,aAAa,EAAE,CAAC;IACnB,cAAc,GAAG,EAAE,CAAC;AACtB,CAAC,CAAC,CAAC;AAEH,IAAA,iBAAQ,EAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAA,WAAE,EAAC,mBAAmB,EAAE,GAAG,EAAE;QAC3B,MAAM,QAAQ,GAAG,IAAI,0BAAc,EAAE,CAAC;QACtC,IAAA,eAAM,EAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,IAAI,0BAAc,EAAE,CAAC;QACrB,IAAA,eAAM,EAAC,cAAc,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;QACjE,IAAA,eAAM,EAAC,cAAc,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,IAAI,0BAAc,CAAC,wBAAwB,CAAC,CAAC;QAC7C,IAAA,eAAM,EAAC,cAAc,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,UAAU,CAAC,qBAAqB,CAAC;YAC/B,OAAO,EAAE,CAAC;oBACR,OAAO,EAAE;wBACP,OAAO,EAAE,iBAAiB;wBAC1B,UAAU,EAAE,CAAC;gCACX,EAAE,EAAE,QAAQ;gCACZ,QAAQ,EAAE,EAAE,IAAI,EAAE,wBAAwB,EAAE,SAAS,EAAE,uBAAuB,EAAE;6BACjF,CAAC;qBACH;oBACD,aAAa,EAAE,YAAY;iBAC5B,CAAC;YACF,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,iBAAiB,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;SACtE,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,0BAAc,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC;YACjC,KAAK,EAAE,UAAU;YACjB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;SAC/C,CAAC,CAAC;QAEH,IAAA,eAAM,EAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC/C,IAAA,eAAM,EAAC,MAAM,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC1C,IAAA,eAAM,EAAC,MAAM,CAAC,UAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QAClE,IAAA,eAAM,EAAC,MAAM,CAAC,UAAW,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;QACxE,IAAA,eAAM,EAAC,MAAM,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,KAAK,SAAS,CAAC,CAAC,UAAU;YACxB,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;YACvE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;QAC5D,CAAC;QACD,UAAU,CAAC,mBAAmB,CAAC,UAAU,EAAE,CAAC,CAAC;QAE7C,MAAM,QAAQ,GAAG,IAAI,0BAAc,EAAE,CAAC;QACtC,MAAM,QAAQ,CAAC,IAAI,CACjB,EAAE,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,EAClE,GAAG,EAAE,GAAE,CAAC,CACT,CAAC;QAEF,MAAM,QAAQ,GAAG,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAA4B,CAAC;QACxE,IAAA,eAAM,EAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,aAAa,EAAE,CAAC;QAChD,IAAA,eAAM,EAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,KAAK,SAAS,CAAC,CAAC,UAAU;YACxB,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;YAC1E,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;QAC/E,CAAC;QACD,UAAU,CAAC,mBAAmB,CAAC,UAAU,EAAE,CAAC,CAAC;QAE7C,MAAM,QAAQ,GAAG,IAAI,0BAAc,EAAE,CAAC;QACtC,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,CAChC,EAAE,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,EAClE,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CACtB,CAAC;QAEF,IAAA,eAAM,EAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC3C,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,sCAAsC,CAAC,EAAE;YAC/E,IAAI,EAAE,cAAc;SACrB,CAAC,CAAC;QACH,UAAU,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC;QAE1C,MAAM,QAAQ,GAAG,IAAI,0BAAc,EAAE,CAAC;QACtC,MAAM,IAAA,eAAM,EACV,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAClF,CAAC,OAAO,CAAC,OAAO,CAAC,iDAAiD,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,sBAAsB,CAAC,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC;QACzF,MAAM,UAAU,GAAG,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC;QAC5C,UAAyC,CAAC,KAAK,GAAG,KAAK,CAAC;QACzD,UAAU,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;QAE7C,MAAM,QAAQ,GAAG,IAAI,0BAAc,EAAE,CAAC;QACtC,MAAM,IAAA,eAAM,EACV,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAClF,CAAC,OAAO,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QACzF,KAAK,SAAS,CAAC,CAAC,UAAU;YACxB,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;QAC3E,CAAC;QACD,UAAU,CAAC,mBAAmB,CAAC,UAAU,EAAE,CAAC,CAAC;QAE7C,MAAM,QAAQ,GAAG,IAAI,0BAAc,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,CAChC,EAAE,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,EAClE,GAAG,EAAE,GAAE,CAAC,CACT,CAAC;QAEF,IAAA,eAAM,EAAC,MAAM,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,IAAA,WAAE,EAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,KAAK,SAAS,CAAC,CAAC,UAAU;YACxB,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;YAC5C,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;YAC1E,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;YAC5C,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;QAC/E,CAAC;QACD,UAAU,CAAC,mBAAmB,CAAC,UAAU,EAAE,CAAC,CAAC;QAE7C,MAAM,QAAQ,GAAG,IAAI,0BAAc,EAAE,CAAC;QACtC,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QAEzC,MAAM,WAAW,GAAG,QAAQ,CAAC,IAAI,CAC/B,EAAE,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,EAClE,GAAG,EAAE,GAAE,CAAC,EACR,UAAU,CAAC,MAAM,CAClB,CAAC;QAEF,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QAEzC,MAAM,IAAA,eAAM,EAAC,WAAW,CAAC,CAAC,OAAO,CAAC,SAAS,CACzC,CAAC,CAAU,EAAE,EAAE,CAAC,CAAC,YAAY,YAAY,IAAK,CAAkB,CAAC,IAAI,KAAK,YAAY,CACvF,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* OpenAI Responses API provider — supports models only available on /v1/responses
|
|
3
|
+
* (e.g. gpt-5.1-codex-mini).
|
|
4
|
+
*
|
|
5
|
+
* Implements AgentLoopProvider to own the multi-turn tool-calling loop,
|
|
6
|
+
* because the Responses API uses typed function_call / function_call_output
|
|
7
|
+
* items that cannot be expressed as plain text messages.
|
|
8
|
+
*/
|
|
9
|
+
import type { LLMRequest, LLMResponse } from '@studio-foundation/contracts';
|
|
10
|
+
import type { AgentLoopProvider, AgentLoopResult, ToolCallOutcome } from './provider.js';
|
|
11
|
+
export declare class OpenAIResponsesProvider implements AgentLoopProvider {
|
|
12
|
+
readonly name = "openai-responses";
|
|
13
|
+
private client;
|
|
14
|
+
constructor(apiKey?: string);
|
|
15
|
+
call(request: LLMRequest, _onToken?: (token: string) => void, signal?: AbortSignal): Promise<LLMResponse>;
|
|
16
|
+
runAgentLoop(request: LLMRequest, executeTool: (name: string, args: Record<string, unknown>, callId: string) => Promise<ToolCallOutcome>, onToken?: (token: string) => void, signal?: AbortSignal): Promise<AgentLoopResult>;
|
|
17
|
+
}
|
|
18
|
+
//# sourceMappingURL=openai-responses.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai-responses.d.ts","sourceRoot":"","sources":["../../src/providers/openai-responses.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,UAAU,EAAE,WAAW,EAAW,MAAM,8BAA8B,CAAC;AACrF,OAAO,KAAK,EAAE,iBAAiB,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AASzF,qBAAa,uBAAwB,YAAW,iBAAiB;IAC/D,QAAQ,CAAC,IAAI,sBAAsB;IACnC,OAAO,CAAC,MAAM,CAAS;gBAEX,MAAM,CAAC,EAAE,MAAM;IAKrB,IAAI,CAAC,OAAO,EAAE,UAAU,EAAE,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;IA0BzG,YAAY,CAChB,OAAO,EAAE,UAAU,EACnB,WAAW,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,eAAe,CAAC,EACtG,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,EACjC,MAAM,CAAC,EAAE,WAAW,GACnB,OAAO,CAAC,eAAe,CAAC;CAgJ5B"}
|