general-agent-runtime 0.0.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/README.md +311 -0
- package/dist/agent/agent.d.ts +14 -0
- package/dist/agent/agent.js +25 -0
- package/dist/agent/agent.js.map +1 -0
- package/dist/agent/builder.d.ts +28 -0
- package/dist/agent/builder.js +178 -0
- package/dist/agent/builder.js.map +1 -0
- package/dist/agent/create-agent.d.ts +2 -0
- package/dist/agent/create-agent.js +29 -0
- package/dist/agent/create-agent.js.map +1 -0
- package/dist/agent/index.d.ts +8 -0
- package/dist/agent/index.js +8 -0
- package/dist/agent/index.js.map +1 -0
- package/dist/agent/integrations/session-context-source.d.ts +12 -0
- package/dist/agent/integrations/session-context-source.js +84 -0
- package/dist/agent/integrations/session-context-source.js.map +1 -0
- package/dist/agent/integrations/session-run-recorder.d.ts +17 -0
- package/dist/agent/integrations/session-run-recorder.js +60 -0
- package/dist/agent/integrations/session-run-recorder.js.map +1 -0
- package/dist/agent/loop-factory.d.ts +11 -0
- package/dist/agent/loop-factory.js +16 -0
- package/dist/agent/loop-factory.js.map +1 -0
- package/dist/agent/run-manager.d.ts +38 -0
- package/dist/agent/run-manager.js +96 -0
- package/dist/agent/run-manager.js.map +1 -0
- package/dist/agent/types.d.ts +38 -0
- package/dist/agent/types.js +2 -0
- package/dist/agent/types.js.map +1 -0
- package/dist/cli/main.d.ts +2 -0
- package/dist/cli/main.js +88 -0
- package/dist/cli/main.js.map +1 -0
- package/dist/cli/renderer.d.ts +2 -0
- package/dist/cli/renderer.js +36 -0
- package/dist/cli/renderer.js.map +1 -0
- package/dist/compaction/budget.d.ts +14 -0
- package/dist/compaction/budget.js +28 -0
- package/dist/compaction/budget.js.map +1 -0
- package/dist/compaction/deterministic.d.ts +9 -0
- package/dist/compaction/deterministic.js +106 -0
- package/dist/compaction/deterministic.js.map +1 -0
- package/dist/compaction/index.d.ts +5 -0
- package/dist/compaction/index.js +5 -0
- package/dist/compaction/index.js.map +1 -0
- package/dist/compaction/manager.d.ts +22 -0
- package/dist/compaction/manager.js +111 -0
- package/dist/compaction/manager.js.map +1 -0
- package/dist/compaction/semantic.d.ts +13 -0
- package/dist/compaction/semantic.js +234 -0
- package/dist/compaction/semantic.js.map +1 -0
- package/dist/compaction/strategy.d.ts +1 -0
- package/dist/compaction/strategy.js +2 -0
- package/dist/compaction/strategy.js.map +1 -0
- package/dist/compaction/types.d.ts +40 -0
- package/dist/compaction/types.js +2 -0
- package/dist/compaction/types.js.map +1 -0
- package/dist/context/budget.d.ts +12 -0
- package/dist/context/budget.js +13 -0
- package/dist/context/budget.js.map +1 -0
- package/dist/context/controller.d.ts +17 -0
- package/dist/context/controller.js +78 -0
- package/dist/context/controller.js.map +1 -0
- package/dist/context/index.d.ts +14 -0
- package/dist/context/index.js +13 -0
- package/dist/context/index.js.map +1 -0
- package/dist/context/manager.d.ts +10 -0
- package/dist/context/manager.js +65 -0
- package/dist/context/manager.js.map +1 -0
- package/dist/context/pipeline.d.ts +16 -0
- package/dist/context/pipeline.js +45 -0
- package/dist/context/pipeline.js.map +1 -0
- package/dist/context/projection.d.ts +8 -0
- package/dist/context/projection.js +10 -0
- package/dist/context/projection.js.map +1 -0
- package/dist/context/source.d.ts +6 -0
- package/dist/context/source.js +2 -0
- package/dist/context/source.js.map +1 -0
- package/dist/context/sources/dynamic.d.ts +10 -0
- package/dist/context/sources/dynamic.js +13 -0
- package/dist/context/sources/dynamic.js.map +1 -0
- package/dist/context/sources/project.d.ts +19 -0
- package/dist/context/sources/project.js +75 -0
- package/dist/context/sources/project.js.map +1 -0
- package/dist/context/sources/system.d.ts +14 -0
- package/dist/context/sources/system.js +23 -0
- package/dist/context/sources/system.js.map +1 -0
- package/dist/context/sources/user.d.ts +14 -0
- package/dist/context/sources/user.js +23 -0
- package/dist/context/sources/user.js.map +1 -0
- package/dist/context/stages/deduplicate.d.ts +7 -0
- package/dist/context/stages/deduplicate.js +19 -0
- package/dist/context/stages/deduplicate.js.map +1 -0
- package/dist/context/stages/expiration.d.ts +5 -0
- package/dist/context/stages/expiration.js +14 -0
- package/dist/context/stages/expiration.js.map +1 -0
- package/dist/context/stages/visibility.d.ts +5 -0
- package/dist/context/stages/visibility.js +7 -0
- package/dist/context/stages/visibility.js.map +1 -0
- package/dist/context/types.d.ts +72 -0
- package/dist/context/types.js +2 -0
- package/dist/context/types.js.map +1 -0
- package/dist/errors/codes.d.ts +1 -0
- package/dist/errors/codes.js +2 -0
- package/dist/errors/codes.js.map +1 -0
- package/dist/errors/index.d.ts +4 -0
- package/dist/errors/index.js +4 -0
- package/dist/errors/index.js.map +1 -0
- package/dist/errors/normalize.d.ts +2 -0
- package/dist/errors/normalize.js +32 -0
- package/dist/errors/normalize.js.map +1 -0
- package/dist/errors/runtime-error.d.ts +18 -0
- package/dist/errors/runtime-error.js +21 -0
- package/dist/errors/runtime-error.js.map +1 -0
- package/dist/errors/serialize.d.ts +2 -0
- package/dist/errors/serialize.js +12 -0
- package/dist/errors/serialize.js.map +1 -0
- package/dist/events/bus.d.ts +17 -0
- package/dist/events/bus.js +52 -0
- package/dist/events/bus.js.map +1 -0
- package/dist/events/index.d.ts +6 -0
- package/dist/events/index.js +5 -0
- package/dist/events/index.js.map +1 -0
- package/dist/events/publisher.d.ts +12 -0
- package/dist/events/publisher.js +22 -0
- package/dist/events/publisher.js.map +1 -0
- package/dist/events/sequence.d.ts +6 -0
- package/dist/events/sequence.js +14 -0
- package/dist/events/sequence.js.map +1 -0
- package/dist/events/sink.d.ts +5 -0
- package/dist/events/sink.js +2 -0
- package/dist/events/sink.js.map +1 -0
- package/dist/events/stream.d.ts +12 -0
- package/dist/events/stream.js +46 -0
- package/dist/events/stream.js.map +1 -0
- package/dist/events/types.d.ts +42 -0
- package/dist/events/types.js +2 -0
- package/dist/events/types.js.map +1 -0
- package/dist/hooks/bus.d.ts +14 -0
- package/dist/hooks/bus.js +104 -0
- package/dist/hooks/bus.js.map +1 -0
- package/dist/hooks/index.d.ts +5 -0
- package/dist/hooks/index.js +5 -0
- package/dist/hooks/index.js.map +1 -0
- package/dist/hooks/policy.d.ts +3 -0
- package/dist/hooks/policy.js +33 -0
- package/dist/hooks/policy.js.map +1 -0
- package/dist/hooks/registry.d.ts +13 -0
- package/dist/hooks/registry.js +31 -0
- package/dist/hooks/registry.js.map +1 -0
- package/dist/hooks/timeout.d.ts +2 -0
- package/dist/hooks/timeout.js +63 -0
- package/dist/hooks/timeout.js.map +1 -0
- package/dist/hooks/types.d.ts +40 -0
- package/dist/hooks/types.js +2 -0
- package/dist/hooks/types.js.map +1 -0
- package/dist/index.d.ts +13 -0
- package/dist/index.js +14 -0
- package/dist/index.js.map +1 -0
- package/dist/loop/index.d.ts +7 -0
- package/dist/loop/index.js +6 -0
- package/dist/loop/index.js.map +1 -0
- package/dist/loop/loop.d.ts +22 -0
- package/dist/loop/loop.js +623 -0
- package/dist/loop/loop.js.map +1 -0
- package/dist/loop/policy.d.ts +21 -0
- package/dist/loop/policy.js +54 -0
- package/dist/loop/policy.js.map +1 -0
- package/dist/loop/recorder.d.ts +9 -0
- package/dist/loop/recorder.js +2 -0
- package/dist/loop/recorder.js.map +1 -0
- package/dist/loop/state.d.ts +18 -0
- package/dist/loop/state.js +81 -0
- package/dist/loop/state.js.map +1 -0
- package/dist/loop/stream-assembler.d.ts +12 -0
- package/dist/loop/stream-assembler.js +76 -0
- package/dist/loop/stream-assembler.js.map +1 -0
- package/dist/loop/transition.d.ts +2 -0
- package/dist/loop/transition.js +27 -0
- package/dist/loop/transition.js.map +1 -0
- package/dist/loop/types.d.ts +53 -0
- package/dist/loop/types.js +2 -0
- package/dist/loop/types.js.map +1 -0
- package/dist/model/index.d.ts +6 -0
- package/dist/model/index.js +5 -0
- package/dist/model/index.js.map +1 -0
- package/dist/model/provider.d.ts +6 -0
- package/dist/model/provider.js +2 -0
- package/dist/model/provider.js.map +1 -0
- package/dist/model/providers/openai-compatible/adapter.d.ts +3 -0
- package/dist/model/providers/openai-compatible/adapter.js +92 -0
- package/dist/model/providers/openai-compatible/adapter.js.map +1 -0
- package/dist/model/providers/openai-compatible/errors.d.ts +3 -0
- package/dist/model/providers/openai-compatible/errors.js +68 -0
- package/dist/model/providers/openai-compatible/errors.js.map +1 -0
- package/dist/model/providers/openai-compatible/provider.d.ts +18 -0
- package/dist/model/providers/openai-compatible/provider.js +182 -0
- package/dist/model/providers/openai-compatible/provider.js.map +1 -0
- package/dist/model/providers/openai-compatible/stream-parser.d.ts +1 -0
- package/dist/model/providers/openai-compatible/stream-parser.js +45 -0
- package/dist/model/providers/openai-compatible/stream-parser.js.map +1 -0
- package/dist/model/registry.d.ts +9 -0
- package/dist/model/registry.js +35 -0
- package/dist/model/registry.js.map +1 -0
- package/dist/model/types.d.ts +49 -0
- package/dist/model/types.js +2 -0
- package/dist/model/types.js.map +1 -0
- package/dist/session/factory.d.ts +13 -0
- package/dist/session/factory.js +14 -0
- package/dist/session/factory.js.map +1 -0
- package/dist/session/index.d.ts +8 -0
- package/dist/session/index.js +7 -0
- package/dist/session/index.js.map +1 -0
- package/dist/session/jsonl-store.d.ts +20 -0
- package/dist/session/jsonl-store.js +264 -0
- package/dist/session/jsonl-store.js.map +1 -0
- package/dist/session/projector.d.ts +4 -0
- package/dist/session/projector.js +104 -0
- package/dist/session/projector.js.map +1 -0
- package/dist/session/serializer.d.ts +5 -0
- package/dist/session/serializer.js +126 -0
- package/dist/session/serializer.js.map +1 -0
- package/dist/session/session.d.ts +32 -0
- package/dist/session/session.js +129 -0
- package/dist/session/session.js.map +1 -0
- package/dist/session/store.d.ts +9 -0
- package/dist/session/store.js +2 -0
- package/dist/session/store.js.map +1 -0
- package/dist/session/types.d.ts +29 -0
- package/dist/session/types.js +2 -0
- package/dist/session/types.js.map +1 -0
- package/dist/settings/defaults.d.ts +7 -0
- package/dist/settings/defaults.js +40 -0
- package/dist/settings/defaults.js.map +1 -0
- package/dist/settings/env-source.d.ts +8 -0
- package/dist/settings/env-source.js +26 -0
- package/dist/settings/env-source.js.map +1 -0
- package/dist/settings/file-source.d.ts +9 -0
- package/dist/settings/file-source.js +32 -0
- package/dist/settings/file-source.js.map +1 -0
- package/dist/settings/index.d.ts +8 -0
- package/dist/settings/index.js +7 -0
- package/dist/settings/index.js.map +1 -0
- package/dist/settings/resolver.d.ts +5 -0
- package/dist/settings/resolver.js +29 -0
- package/dist/settings/resolver.js.map +1 -0
- package/dist/settings/runtime-source.d.ts +8 -0
- package/dist/settings/runtime-source.js +11 -0
- package/dist/settings/runtime-source.js.map +1 -0
- package/dist/settings/schema.d.ts +37 -0
- package/dist/settings/schema.js +47 -0
- package/dist/settings/schema.js.map +1 -0
- package/dist/settings/source.d.ts +5 -0
- package/dist/settings/source.js +2 -0
- package/dist/settings/source.js.map +1 -0
- package/dist/settings/types.d.ts +46 -0
- package/dist/settings/types.js +2 -0
- package/dist/settings/types.js.map +1 -0
- package/dist/tools/builtin/list-directory.d.ts +17 -0
- package/dist/tools/builtin/list-directory.js +31 -0
- package/dist/tools/builtin/list-directory.js.map +1 -0
- package/dist/tools/builtin/read-file.d.ts +14 -0
- package/dist/tools/builtin/read-file.js +23 -0
- package/dist/tools/builtin/read-file.js.map +1 -0
- package/dist/tools/builtin/run-command.d.ts +19 -0
- package/dist/tools/builtin/run-command.js +46 -0
- package/dist/tools/builtin/run-command.js.map +1 -0
- package/dist/tools/builtin/write-file.d.ts +20 -0
- package/dist/tools/builtin/write-file.js +29 -0
- package/dist/tools/builtin/write-file.js.map +1 -0
- package/dist/tools/executor.d.ts +29 -0
- package/dist/tools/executor.js +212 -0
- package/dist/tools/executor.js.map +1 -0
- package/dist/tools/index.d.ts +10 -0
- package/dist/tools/index.js +9 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/registry.d.ts +19 -0
- package/dist/tools/registry.js +42 -0
- package/dist/tools/registry.js.map +1 -0
- package/dist/tools/result-processor.d.ts +4 -0
- package/dist/tools/result-processor.js +122 -0
- package/dist/tools/result-processor.js.map +1 -0
- package/dist/tools/tool.d.ts +8 -0
- package/dist/tools/tool.js +2 -0
- package/dist/tools/tool.js.map +1 -0
- package/dist/tools/types.d.ts +44 -0
- package/dist/tools/types.js +2 -0
- package/dist/tools/types.js.map +1 -0
- package/dist/tools/validator.d.ts +5 -0
- package/dist/tools/validator.js +112 -0
- package/dist/tools/validator.js.map +1 -0
- package/dist/types/compaction.d.ts +22 -0
- package/dist/types/compaction.js +2 -0
- package/dist/types/compaction.js.map +1 -0
- package/dist/types/error.d.ts +10 -0
- package/dist/types/error.js +2 -0
- package/dist/types/error.js.map +1 -0
- package/dist/types/ids.d.ts +12 -0
- package/dist/types/ids.js +2 -0
- package/dist/types/ids.js.map +1 -0
- package/dist/types/index.d.ts +9 -0
- package/dist/types/index.js +2 -0
- package/dist/types/index.js.map +1 -0
- package/dist/types/json.d.ts +7 -0
- package/dist/types/json.js +2 -0
- package/dist/types/json.js.map +1 -0
- package/dist/types/message.d.ts +39 -0
- package/dist/types/message.js +2 -0
- package/dist/types/message.js.map +1 -0
- package/dist/types/metadata.d.ts +2 -0
- package/dist/types/metadata.js +2 -0
- package/dist/types/metadata.js.map +1 -0
- package/dist/types/runtime.d.ts +20 -0
- package/dist/types/runtime.js +2 -0
- package/dist/types/runtime.js.map +1 -0
- package/dist/types/tool.d.ts +12 -0
- package/dist/types/tool.js +2 -0
- package/dist/types/tool.js.map +1 -0
- package/dist/types/usage.d.ts +17 -0
- package/dist/types/usage.js +2 -0
- package/dist/types/usage.js.map +1 -0
- package/dist/utils/deep.d.ts +2 -0
- package/dist/utils/deep.js +34 -0
- package/dist/utils/deep.js.map +1 -0
- package/dist/utils/id.d.ts +10 -0
- package/dist/utils/id.js +14 -0
- package/dist/utils/id.js.map +1 -0
- package/dist/utils/index.d.ts +3 -0
- package/dist/utils/index.js +4 -0
- package/dist/utils/index.js.map +1 -0
- package/dist/utils/message.d.ts +11 -0
- package/dist/utils/message.js +31 -0
- package/dist/utils/message.js.map +1 -0
- package/package.json +59 -0
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
import type { AgentRunState } from "../types/runtime.js";
|
|
2
|
+
import type { LoopPolicy, LoopPolicyDecision } from "./types.js";
|
|
3
|
+
export declare class MaxTurnsPolicy implements LoopPolicy {
|
|
4
|
+
private readonly maxTurns;
|
|
5
|
+
constructor(maxTurns: number);
|
|
6
|
+
evaluate(state: Readonly<AgentRunState>): LoopPolicyDecision;
|
|
7
|
+
}
|
|
8
|
+
export declare class MaxToolRoundsPolicy implements LoopPolicy {
|
|
9
|
+
private readonly maxToolRounds;
|
|
10
|
+
constructor(maxToolRounds: number);
|
|
11
|
+
evaluate(state: Readonly<AgentRunState>): LoopPolicyDecision;
|
|
12
|
+
}
|
|
13
|
+
export declare class CompositeLoopPolicy implements LoopPolicy {
|
|
14
|
+
private readonly policies;
|
|
15
|
+
constructor(policies: readonly LoopPolicy[]);
|
|
16
|
+
evaluate(state: Readonly<AgentRunState>): LoopPolicyDecision;
|
|
17
|
+
}
|
|
18
|
+
export declare function createDefaultLoopPolicy(options?: {
|
|
19
|
+
maxTurns?: number;
|
|
20
|
+
maxToolRounds?: number;
|
|
21
|
+
}): LoopPolicy;
|
|
@@ -0,0 +1,54 @@
|
|
|
1
|
+
export class MaxTurnsPolicy {
|
|
2
|
+
maxTurns;
|
|
3
|
+
constructor(maxTurns) {
|
|
4
|
+
this.maxTurns = maxTurns;
|
|
5
|
+
if (!Number.isSafeInteger(maxTurns) || maxTurns < 1) {
|
|
6
|
+
throw new RangeError("maxTurns must be a positive integer");
|
|
7
|
+
}
|
|
8
|
+
}
|
|
9
|
+
evaluate(state) {
|
|
10
|
+
if (state.status === "preparing" && state.turnCount >= this.maxTurns) {
|
|
11
|
+
return { continue: false, stopReason: "max_turns" };
|
|
12
|
+
}
|
|
13
|
+
return { continue: true };
|
|
14
|
+
}
|
|
15
|
+
}
|
|
16
|
+
export class MaxToolRoundsPolicy {
|
|
17
|
+
maxToolRounds;
|
|
18
|
+
constructor(maxToolRounds) {
|
|
19
|
+
this.maxToolRounds = maxToolRounds;
|
|
20
|
+
if (!Number.isSafeInteger(maxToolRounds) || maxToolRounds < 0) {
|
|
21
|
+
throw new RangeError("maxToolRounds must be a non-negative integer");
|
|
22
|
+
}
|
|
23
|
+
}
|
|
24
|
+
evaluate(state) {
|
|
25
|
+
if (state.status === "tool_executing" &&
|
|
26
|
+
state.pendingToolCalls.length > 0 &&
|
|
27
|
+
state.toolRoundCount >= this.maxToolRounds) {
|
|
28
|
+
return { continue: false, stopReason: "max_tool_rounds" };
|
|
29
|
+
}
|
|
30
|
+
return { continue: true };
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
export class CompositeLoopPolicy {
|
|
34
|
+
policies;
|
|
35
|
+
constructor(policies) {
|
|
36
|
+
this.policies = policies;
|
|
37
|
+
}
|
|
38
|
+
evaluate(state) {
|
|
39
|
+
for (const policy of this.policies) {
|
|
40
|
+
const decision = policy.evaluate(state);
|
|
41
|
+
if (!decision.continue) {
|
|
42
|
+
return decision;
|
|
43
|
+
}
|
|
44
|
+
}
|
|
45
|
+
return { continue: true };
|
|
46
|
+
}
|
|
47
|
+
}
|
|
48
|
+
export function createDefaultLoopPolicy(options = {}) {
|
|
49
|
+
return new CompositeLoopPolicy([
|
|
50
|
+
new MaxTurnsPolicy(options.maxTurns ?? 16),
|
|
51
|
+
new MaxToolRoundsPolicy(options.maxToolRounds ?? 8),
|
|
52
|
+
]);
|
|
53
|
+
}
|
|
54
|
+
//# sourceMappingURL=policy.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"policy.js","sourceRoot":"","sources":["../../src/loop/policy.ts"],"names":[],"mappings":"AAMA,MAAM,OAAO,cAAc;IACI;IAA7B,YAA6B,QAAgB;QAAhB,aAAQ,GAAR,QAAQ,CAAQ;QAC3C,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,IAAI,QAAQ,GAAG,CAAC,EAAE,CAAC;YACpD,MAAM,IAAI,UAAU,CAAC,qCAAqC,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,KAA8B;QACrC,IAAI,KAAK,CAAC,MAAM,KAAK,WAAW,IAAI,KAAK,CAAC,SAAS,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACrE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,CAAC;QACtD,CAAC;QACD,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;IAC5B,CAAC;CACF;AAED,MAAM,OAAO,mBAAmB;IACD;IAA7B,YAA6B,aAAqB;QAArB,kBAAa,GAAb,aAAa,CAAQ;QAChD,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,aAAa,CAAC,IAAI,aAAa,GAAG,CAAC,EAAE,CAAC;YAC9D,MAAM,IAAI,UAAU,CAAC,8CAA8C,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,KAA8B;QACrC,IACE,KAAK,CAAC,MAAM,KAAK,gBAAgB;YACjC,KAAK,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC;YACjC,KAAK,CAAC,cAAc,IAAI,IAAI,CAAC,aAAa,EAC1C,CAAC;YACD,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,iBAAiB,EAAE,CAAC;QAC5D,CAAC;QACD,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;IAC5B,CAAC;CACF;AAED,MAAM,OAAO,mBAAmB;IACD;IAA7B,YAA6B,QAA+B;QAA/B,aAAQ,GAAR,QAAQ,CAAuB;IAAG,CAAC;IAEhE,QAAQ,CAAC,KAA8B;QACrC,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnC,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;YACxC,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;gBACvB,OAAO,QAAQ,CAAC;YAClB,CAAC;QACH,CAAC;QACD,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;IAC5B,CAAC;CACF;AAED,MAAM,UAAU,uBAAuB,CAAC,UAGpC,EAAE;IACJ,OAAO,IAAI,mBAAmB,CAAC;QAC7B,IAAI,cAAc,CAAC,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC;QAC1C,IAAI,mBAAmB,CAAC,OAAO,CAAC,aAAa,IAAI,CAAC,CAAC;KACpD,CAAC,CAAC;AACL,CAAC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import type { AgentMessage, CompactionCheckpoint, SessionId, Usage } from "../types/index.js";
|
|
2
|
+
import type { ToolExecutionResult } from "../tools/types.js";
|
|
3
|
+
export interface RunRecorder {
|
|
4
|
+
readonly sessionId: SessionId;
|
|
5
|
+
appendUser(message: AgentMessage, signal?: AbortSignal): Promise<void>;
|
|
6
|
+
appendAssistant(message: AgentMessage, usage?: Usage, signal?: AbortSignal): Promise<void>;
|
|
7
|
+
appendToolResults(results: readonly ToolExecutionResult[], signal?: AbortSignal): Promise<void>;
|
|
8
|
+
appendCompaction(checkpoint: CompactionCheckpoint, signal?: AbortSignal): Promise<void>;
|
|
9
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"recorder.js","sourceRoot":"","sources":["../../src/loop/recorder.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import type { RuntimeError } from "../errors/runtime-error.js";
|
|
2
|
+
import type { AgentMessage, AgentRunState, LoopStatus, RunId, SessionId, StopReason, ToolCall, TurnId, Usage } from "../types/index.js";
|
|
3
|
+
export declare class RunStateController {
|
|
4
|
+
private readonly sessionId;
|
|
5
|
+
private state;
|
|
6
|
+
constructor(sessionId: SessionId, runId: RunId);
|
|
7
|
+
reset(runId: RunId): void;
|
|
8
|
+
transition(next: LoopStatus): void;
|
|
9
|
+
beginTurn(turnId: TurnId): void;
|
|
10
|
+
recordUsage(usage: Usage): void;
|
|
11
|
+
setAssistant(message: AgentMessage, pendingToolCalls: readonly ToolCall[]): void;
|
|
12
|
+
completeToolRound(): void;
|
|
13
|
+
setStop(reason: StopReason): void;
|
|
14
|
+
setFailure(error: RuntimeError): void;
|
|
15
|
+
setAbort(error: RuntimeError): void;
|
|
16
|
+
forceStatus(status: Extract<LoopStatus, "aborted" | "error">): void;
|
|
17
|
+
snapshot(): Readonly<AgentRunState>;
|
|
18
|
+
}
|
|
@@ -0,0 +1,81 @@
|
|
|
1
|
+
import { serializeRuntimeError } from "../errors/serialize.js";
|
|
2
|
+
import { deepFreeze } from "../utils/deep.js";
|
|
3
|
+
import { assertTransition } from "./transition.js";
|
|
4
|
+
function mergeUsage(current, next) {
|
|
5
|
+
if (current === undefined) {
|
|
6
|
+
return structuredClone(next);
|
|
7
|
+
}
|
|
8
|
+
const cachedInputTokens = (current.tokens.cachedInputTokens ?? 0) +
|
|
9
|
+
(next.tokens.cachedInputTokens ?? 0);
|
|
10
|
+
const reasoningTokens = (current.tokens.reasoningTokens ?? 0) +
|
|
11
|
+
(next.tokens.reasoningTokens ?? 0);
|
|
12
|
+
return {
|
|
13
|
+
tokens: {
|
|
14
|
+
inputTokens: current.tokens.inputTokens + next.tokens.inputTokens,
|
|
15
|
+
outputTokens: current.tokens.outputTokens + next.tokens.outputTokens,
|
|
16
|
+
totalTokens: current.tokens.totalTokens + next.tokens.totalTokens,
|
|
17
|
+
...(cachedInputTokens === 0 ? {} : { cachedInputTokens }),
|
|
18
|
+
...(reasoningTokens === 0 ? {} : { reasoningTokens }),
|
|
19
|
+
},
|
|
20
|
+
};
|
|
21
|
+
}
|
|
22
|
+
function initialState(sessionId, runId) {
|
|
23
|
+
return {
|
|
24
|
+
sessionId,
|
|
25
|
+
runId,
|
|
26
|
+
status: "idle",
|
|
27
|
+
turnCount: 0,
|
|
28
|
+
toolRoundCount: 0,
|
|
29
|
+
pendingToolCalls: [],
|
|
30
|
+
};
|
|
31
|
+
}
|
|
32
|
+
export class RunStateController {
|
|
33
|
+
sessionId;
|
|
34
|
+
state;
|
|
35
|
+
constructor(sessionId, runId) {
|
|
36
|
+
this.sessionId = sessionId;
|
|
37
|
+
this.state = initialState(sessionId, runId);
|
|
38
|
+
}
|
|
39
|
+
reset(runId) {
|
|
40
|
+
this.state = initialState(this.sessionId, runId);
|
|
41
|
+
}
|
|
42
|
+
transition(next) {
|
|
43
|
+
assertTransition(this.state.status, next);
|
|
44
|
+
this.state.status = next;
|
|
45
|
+
}
|
|
46
|
+
beginTurn(turnId) {
|
|
47
|
+
this.state.turnId = turnId;
|
|
48
|
+
this.state.turnCount += 1;
|
|
49
|
+
delete this.state.currentAssistantMessage;
|
|
50
|
+
this.state.pendingToolCalls = [];
|
|
51
|
+
}
|
|
52
|
+
recordUsage(usage) {
|
|
53
|
+
this.state.usage = mergeUsage(this.state.usage, usage);
|
|
54
|
+
}
|
|
55
|
+
setAssistant(message, pendingToolCalls) {
|
|
56
|
+
this.state.currentAssistantMessage = structuredClone(message);
|
|
57
|
+
this.state.pendingToolCalls = pendingToolCalls.map((call) => structuredClone(call));
|
|
58
|
+
}
|
|
59
|
+
completeToolRound() {
|
|
60
|
+
this.state.toolRoundCount += 1;
|
|
61
|
+
this.state.pendingToolCalls = [];
|
|
62
|
+
}
|
|
63
|
+
setStop(reason) {
|
|
64
|
+
this.state.stopReason = reason;
|
|
65
|
+
}
|
|
66
|
+
setFailure(error) {
|
|
67
|
+
this.state.lastError = serializeRuntimeError(error);
|
|
68
|
+
this.state.stopReason = "error";
|
|
69
|
+
}
|
|
70
|
+
setAbort(error) {
|
|
71
|
+
this.state.lastError = serializeRuntimeError(error);
|
|
72
|
+
this.state.stopReason = "user_abort";
|
|
73
|
+
}
|
|
74
|
+
forceStatus(status) {
|
|
75
|
+
this.state.status = status;
|
|
76
|
+
}
|
|
77
|
+
snapshot() {
|
|
78
|
+
return deepFreeze(structuredClone(this.state));
|
|
79
|
+
}
|
|
80
|
+
}
|
|
81
|
+
//# sourceMappingURL=state.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"state.js","sourceRoot":"","sources":["../../src/loop/state.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,qBAAqB,EAAE,MAAM,wBAAwB,CAAC;AAY/D,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAEnD,SAAS,UAAU,CAAC,OAA0B,EAAE,IAAW;IACzD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,OAAO,eAAe,CAAC,IAAI,CAAC,CAAC;IAC/B,CAAC;IAED,MAAM,iBAAiB,GACrB,CAAC,OAAO,CAAC,MAAM,CAAC,iBAAiB,IAAI,CAAC,CAAC;QACvC,CAAC,IAAI,CAAC,MAAM,CAAC,iBAAiB,IAAI,CAAC,CAAC,CAAC;IACvC,MAAM,eAAe,GACnB,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,IAAI,CAAC,CAAC;QACrC,CAAC,IAAI,CAAC,MAAM,CAAC,eAAe,IAAI,CAAC,CAAC,CAAC;IAErC,OAAO;QACL,MAAM,EAAE;YACN,WAAW,EAAE,OAAO,CAAC,MAAM,CAAC,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW;YACjE,YAAY,EAAE,OAAO,CAAC,MAAM,CAAC,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY;YACpE,WAAW,EAAE,OAAO,CAAC,MAAM,CAAC,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW;YACjE,GAAG,CAAC,iBAAiB,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,CAAC;YACzD,GAAG,CAAC,eAAe,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,CAAC;SACtD;KACF,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,SAAoB,EAAE,KAAY;IACtD,OAAO;QACL,SAAS;QACT,KAAK;QACL,MAAM,EAAE,MAAM;QACd,SAAS,EAAE,CAAC;QACZ,cAAc,EAAE,CAAC;QACjB,gBAAgB,EAAE,EAAE;KACrB,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,kBAAkB;IAIV;IAHX,KAAK,CAAgB;IAE7B,YACmB,SAAoB,EACrC,KAAY;QADK,cAAS,GAAT,SAAS,CAAW;QAGrC,IAAI,CAAC,KAAK,GAAG,YAAY,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,KAAY;QAChB,IAAI,CAAC,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IACnD,CAAC;IAED,UAAU,CAAC,IAAgB;QACzB,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC1C,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC;IAC3B,CAAC;IAED,SAAS,CAAC,MAAc;QACtB,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC;QAC3B,IAAI,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,CAAC;QAC1B,OAAO,IAAI,CAAC,KAAK,CAAC,uBAAuB,CAAC;QAC1C,IAAI,CAAC,KAAK,CAAC,gBAAgB,GAAG,EAAE,CAAC;IACnC,CAAC;IAED,WAAW,CAAC,KAAY;QACtB,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IACzD,CAAC;IAED,YAAY,CACV,OAAqB,EACrB,gBAAqC;QAErC,IAAI,CAAC,KAAK,CAAC,uBAAuB,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;QAC9D,IAAI,CAAC,KAAK,CAAC,gBAAgB,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAC1D,eAAe,CAAC,IAAI,CAAC,CACtB,CAAC;IACJ,CAAC;IAED,iBAAiB;QACf,IAAI,CAAC,KAAK,CAAC,cAAc,IAAI,CAAC,CAAC;QAC/B,IAAI,CAAC,KAAK,CAAC,gBAAgB,GAAG,EAAE,CAAC;IACnC,CAAC;IAED,OAAO,CAAC,MAAkB;QACxB,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,MAAM,CAAC;IACjC,CAAC;IAED,UAAU,CAAC,KAAmB;QAC5B,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;QACpD,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,OAAO,CAAC;IAClC,CAAC;IAED,QAAQ,CAAC,KAAmB;QAC1B,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;QACpD,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,YAAY,CAAC;IACvC,CAAC;IAED,WAAW,CAAC,MAAgD;QAC1D,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC;IAC7B,CAAC;IAED,QAAQ;QACN,OAAO,UAAU,CAAC,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,CAA4B,CAAC;IAC5E,CAAC;CACF"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { ModelEvent } from "../model/index.js";
|
|
2
|
+
import type { AssistantTurnResult } from "./types.js";
|
|
3
|
+
export declare class ModelStreamAssembler {
|
|
4
|
+
private text;
|
|
5
|
+
private reasoning;
|
|
6
|
+
private readonly toolCalls;
|
|
7
|
+
private usage;
|
|
8
|
+
private stopReason;
|
|
9
|
+
private ended;
|
|
10
|
+
accept(event: ModelEvent): void;
|
|
11
|
+
finish(): AssistantTurnResult;
|
|
12
|
+
}
|
|
@@ -0,0 +1,76 @@
|
|
|
1
|
+
import { RuntimeError } from "../errors/index.js";
|
|
2
|
+
import { createAssistantMessage } from "../utils/index.js";
|
|
3
|
+
export class ModelStreamAssembler {
|
|
4
|
+
text = "";
|
|
5
|
+
reasoning = "";
|
|
6
|
+
toolCalls = [];
|
|
7
|
+
usage;
|
|
8
|
+
stopReason;
|
|
9
|
+
ended = false;
|
|
10
|
+
accept(event) {
|
|
11
|
+
if (this.ended) {
|
|
12
|
+
throw new RuntimeError({
|
|
13
|
+
code: "INTERNAL",
|
|
14
|
+
category: "model",
|
|
15
|
+
message: "Received model event after message_end",
|
|
16
|
+
});
|
|
17
|
+
}
|
|
18
|
+
switch (event.type) {
|
|
19
|
+
case "message_start":
|
|
20
|
+
case "tool_call_start":
|
|
21
|
+
case "tool_call_delta":
|
|
22
|
+
break;
|
|
23
|
+
case "text_delta":
|
|
24
|
+
this.text += event.delta;
|
|
25
|
+
break;
|
|
26
|
+
case "reasoning_delta":
|
|
27
|
+
this.reasoning += event.delta;
|
|
28
|
+
break;
|
|
29
|
+
case "tool_call_end":
|
|
30
|
+
this.toolCalls.push({
|
|
31
|
+
callId: event.callId,
|
|
32
|
+
toolName: event.toolName,
|
|
33
|
+
input: structuredClone(event.input),
|
|
34
|
+
});
|
|
35
|
+
break;
|
|
36
|
+
case "usage":
|
|
37
|
+
this.usage = structuredClone(event.usage);
|
|
38
|
+
break;
|
|
39
|
+
case "message_end":
|
|
40
|
+
this.stopReason = event.stopReason;
|
|
41
|
+
this.ended = true;
|
|
42
|
+
break;
|
|
43
|
+
}
|
|
44
|
+
}
|
|
45
|
+
finish() {
|
|
46
|
+
if (!this.ended || this.stopReason === undefined) {
|
|
47
|
+
throw new RuntimeError({
|
|
48
|
+
code: "MODEL_REQUEST_FAILED",
|
|
49
|
+
category: "model",
|
|
50
|
+
message: "Model stream ended without message_end",
|
|
51
|
+
});
|
|
52
|
+
}
|
|
53
|
+
const parts = [];
|
|
54
|
+
if (this.reasoning.length > 0) {
|
|
55
|
+
parts.push({ type: "reasoning", text: this.reasoning });
|
|
56
|
+
}
|
|
57
|
+
if (this.text.length > 0) {
|
|
58
|
+
parts.push({ type: "text", text: this.text });
|
|
59
|
+
}
|
|
60
|
+
for (const call of this.toolCalls) {
|
|
61
|
+
parts.push({
|
|
62
|
+
type: "tool_call",
|
|
63
|
+
callId: call.callId,
|
|
64
|
+
toolName: call.toolName,
|
|
65
|
+
input: structuredClone(call.input),
|
|
66
|
+
});
|
|
67
|
+
}
|
|
68
|
+
return {
|
|
69
|
+
message: createAssistantMessage(parts),
|
|
70
|
+
toolCalls: this.toolCalls.map((call) => structuredClone(call)),
|
|
71
|
+
...(this.usage === undefined ? {} : { usage: structuredClone(this.usage) }),
|
|
72
|
+
stopReason: this.stopReason,
|
|
73
|
+
};
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
//# sourceMappingURL=stream-assembler.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream-assembler.js","sourceRoot":"","sources":["../../src/loop/stream-assembler.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAGlD,OAAO,EAAE,sBAAsB,EAAE,MAAM,mBAAmB,CAAC;AAI3D,MAAM,OAAO,oBAAoB;IACvB,IAAI,GAAG,EAAE,CAAC;IACV,SAAS,GAAG,EAAE,CAAC;IACN,SAAS,GAAe,EAAE,CAAC;IACpC,KAAK,CAAoB;IACzB,UAAU,CAA8B;IACxC,KAAK,GAAG,KAAK,CAAC;IAEtB,MAAM,CAAC,KAAiB;QACtB,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,YAAY,CAAC;gBACrB,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE,OAAO;gBACjB,OAAO,EAAE,wCAAwC;aAClD,CAAC,CAAC;QACL,CAAC;QAED,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,eAAe,CAAC;YACrB,KAAK,iBAAiB,CAAC;YACvB,KAAK,iBAAiB;gBACpB,MAAM;YACR,KAAK,YAAY;gBACf,IAAI,CAAC,IAAI,IAAI,KAAK,CAAC,KAAK,CAAC;gBACzB,MAAM;YACR,KAAK,iBAAiB;gBACpB,IAAI,CAAC,SAAS,IAAI,KAAK,CAAC,KAAK,CAAC;gBAC9B,MAAM;YACR,KAAK,eAAe;gBAClB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;oBAClB,MAAM,EAAE,KAAK,CAAC,MAAM;oBACpB,QAAQ,EAAE,KAAK,CAAC,QAAQ;oBACxB,KAAK,EAAE,eAAe,CAAC,KAAK,CAAC,KAAK,CAAC;iBACpC,CAAC,CAAC;gBACH,MAAM;YACR,KAAK,OAAO;gBACV,IAAI,CAAC,KAAK,GAAG,eAAe,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBAC1C,MAAM;YACR,KAAK,aAAa;gBAChB,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC,UAAU,CAAC;gBACnC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;gBAClB,MAAM;QACV,CAAC;IACH,CAAC;IAED,MAAM;QACJ,IAAI,CAAC,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;YACjD,MAAM,IAAI,YAAY,CAAC;gBACrB,IAAI,EAAE,sBAAsB;gBAC5B,QAAQ,EAAE,OAAO;gBACjB,OAAO,EAAE,wCAAwC;aAClD,CAAC,CAAC;QACL,CAAC;QAED,MAAM,KAAK,GAAkB,EAAE,CAAC;QAChC,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;QAC1D,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAChD,CAAC;QACD,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YAClC,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,WAAW;gBACjB,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,KAAK,EAAE,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC;aACnC,CAAC,CAAC;QACL,CAAC;QAED,OAAO;YACL,OAAO,EAAE,sBAAsB,CAAC,KAAK,CAAC;YACtC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC9D,GAAG,CAAC,IAAI,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;YAC3E,UAAU,EAAE,IAAI,CAAC,UAAU;SAC5B,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
import { RuntimeError } from "../errors/runtime-error.js";
|
|
2
|
+
const TRANSITIONS = {
|
|
3
|
+
idle: ["preparing", "completed", "aborted", "error"],
|
|
4
|
+
preparing: ["model_streaming", "completed", "aborted", "error"],
|
|
5
|
+
model_streaming: [
|
|
6
|
+
"tool_executing",
|
|
7
|
+
"completed",
|
|
8
|
+
"preparing",
|
|
9
|
+
"aborted",
|
|
10
|
+
"error",
|
|
11
|
+
],
|
|
12
|
+
tool_executing: ["integrating", "completed", "aborted", "error"],
|
|
13
|
+
integrating: ["preparing", "completed", "aborted", "error"],
|
|
14
|
+
completed: [],
|
|
15
|
+
aborted: [],
|
|
16
|
+
error: [],
|
|
17
|
+
};
|
|
18
|
+
export function assertTransition(current, next) {
|
|
19
|
+
if (!TRANSITIONS[current].includes(next)) {
|
|
20
|
+
throw new RuntimeError({
|
|
21
|
+
code: "INTERNAL",
|
|
22
|
+
category: "internal",
|
|
23
|
+
message: `Illegal AgentLoop transition: ${current} -> ${next}`,
|
|
24
|
+
});
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
//# sourceMappingURL=transition.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"transition.js","sourceRoot":"","sources":["../../src/loop/transition.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAG1D,MAAM,WAAW,GAAwD;IACvE,IAAI,EAAE,CAAC,WAAW,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,CAAC;IACpD,SAAS,EAAE,CAAC,iBAAiB,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,CAAC;IAC/D,eAAe,EAAE;QACf,gBAAgB;QAChB,WAAW;QACX,WAAW;QACX,SAAS;QACT,OAAO;KACR;IACD,cAAc,EAAE,CAAC,aAAa,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,CAAC;IAChE,WAAW,EAAE,CAAC,WAAW,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,CAAC;IAC3D,SAAS,EAAE,EAAE;IACb,OAAO,EAAE,EAAE;IACX,KAAK,EAAE,EAAE;CACV,CAAC;AAEF,MAAM,UAAU,gBAAgB,CAC9B,OAAmB,EACnB,IAAgB;IAEhB,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QACzC,MAAM,IAAI,YAAY,CAAC;YACrB,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE,UAAU;YACpB,OAAO,EAAE,iCAAiC,OAAO,OAAO,IAAI,EAAE;SAC/D,CAAC,CAAC;IACL,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,53 @@
|
|
|
1
|
+
import type { CompactionManager } from "../compaction/types.js";
|
|
2
|
+
import type { EventPublisher } from "../events/publisher.js";
|
|
3
|
+
import type { AgentEvent } from "../events/types.js";
|
|
4
|
+
import type { HookBus } from "../hooks/bus.js";
|
|
5
|
+
import type { ModelCapabilities, ModelStopReason } from "../model/types.js";
|
|
6
|
+
import type { ModelProvider } from "../model/provider.js";
|
|
7
|
+
import type { ContextManager } from "../context/types.js";
|
|
8
|
+
import type { ProjectionBudgetPolicy } from "../context/budget.js";
|
|
9
|
+
import type { ToolExecutor } from "../tools/executor.js";
|
|
10
|
+
import type { ToolRegistry } from "../tools/registry.js";
|
|
11
|
+
import type { AgentMessage, AgentRunState, JsonObject, StopReason, ToolCall, Usage } from "../types/index.js";
|
|
12
|
+
import type { RunRecorder } from "./recorder.js";
|
|
13
|
+
export interface LoopPolicyDecision {
|
|
14
|
+
continue: boolean;
|
|
15
|
+
stopReason?: StopReason;
|
|
16
|
+
}
|
|
17
|
+
export interface LoopPolicy {
|
|
18
|
+
evaluate(state: Readonly<AgentRunState>): LoopPolicyDecision;
|
|
19
|
+
}
|
|
20
|
+
export interface AgentLoopOptions {
|
|
21
|
+
readonly maxReactiveCompactions: number;
|
|
22
|
+
readonly maxCompactionReprojects: number;
|
|
23
|
+
}
|
|
24
|
+
export interface AgentLoopDependencies {
|
|
25
|
+
readonly modelProvider: ModelProvider;
|
|
26
|
+
readonly contextManager: ContextManager;
|
|
27
|
+
readonly compactionManager: CompactionManager;
|
|
28
|
+
readonly projectionBudgetPolicy: ProjectionBudgetPolicy;
|
|
29
|
+
readonly toolExecutor: ToolExecutor;
|
|
30
|
+
readonly toolRegistry: ToolRegistry;
|
|
31
|
+
readonly hookBus: HookBus;
|
|
32
|
+
readonly recorder: RunRecorder;
|
|
33
|
+
readonly eventPublisher: EventPublisher;
|
|
34
|
+
readonly policy: LoopPolicy;
|
|
35
|
+
}
|
|
36
|
+
export interface LoopInput {
|
|
37
|
+
input: AgentMessage;
|
|
38
|
+
model: string;
|
|
39
|
+
capabilities: ModelCapabilities;
|
|
40
|
+
temperature?: number;
|
|
41
|
+
maxOutputTokens?: number;
|
|
42
|
+
providerOptions?: JsonObject;
|
|
43
|
+
}
|
|
44
|
+
export interface AssistantTurnResult {
|
|
45
|
+
message: AgentMessage;
|
|
46
|
+
toolCalls: readonly ToolCall[];
|
|
47
|
+
usage?: Usage;
|
|
48
|
+
stopReason: ModelStopReason;
|
|
49
|
+
}
|
|
50
|
+
export interface AgentLoop {
|
|
51
|
+
run(input: LoopInput, signal?: AbortSignal): AsyncIterable<AgentEvent>;
|
|
52
|
+
getState(): Readonly<AgentRunState>;
|
|
53
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/loop/types.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
export type * from "./types.js";
|
|
2
|
+
export type * from "./provider.js";
|
|
3
|
+
export * from "./registry.js";
|
|
4
|
+
export * from "./providers/openai-compatible/provider.js";
|
|
5
|
+
export * from "./providers/openai-compatible/adapter.js";
|
|
6
|
+
export * from "./providers/openai-compatible/stream-parser.js";
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/model/index.ts"],"names":[],"mappings":"AAEA,cAAc,eAAe,CAAC;AAC9B,cAAc,2CAA2C,CAAC;AAC1D,cAAc,0CAA0C,CAAC;AACzD,cAAc,gDAAgD,CAAC"}
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
import type { ModelCapabilities, ModelEvent, ModelRequest } from "./types.js";
|
|
2
|
+
export interface ModelProvider {
|
|
3
|
+
readonly id: string;
|
|
4
|
+
getCapabilities(model: string): Promise<ModelCapabilities>;
|
|
5
|
+
stream(request: ModelRequest, signal?: AbortSignal): AsyncIterable<ModelEvent>;
|
|
6
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"provider.js","sourceRoot":"","sources":["../../src/model/provider.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,92 @@
|
|
|
1
|
+
import { RuntimeError } from "../../../errors/runtime-error.js";
|
|
2
|
+
export function toOpenAIRequest(request) {
|
|
3
|
+
const body = {
|
|
4
|
+
model: request.model,
|
|
5
|
+
messages: request.messages.flatMap(toOpenAIMessages),
|
|
6
|
+
stream: true,
|
|
7
|
+
stream_options: { include_usage: true }
|
|
8
|
+
};
|
|
9
|
+
if (request.temperature !== undefined) {
|
|
10
|
+
body.temperature = request.temperature;
|
|
11
|
+
}
|
|
12
|
+
if (request.maxOutputTokens !== undefined) {
|
|
13
|
+
body.max_tokens = request.maxOutputTokens;
|
|
14
|
+
}
|
|
15
|
+
if (request.tools && request.tools.length > 0) {
|
|
16
|
+
body.tools = request.tools.map((tool) => ({
|
|
17
|
+
type: "function",
|
|
18
|
+
function: {
|
|
19
|
+
name: tool.name,
|
|
20
|
+
description: tool.description,
|
|
21
|
+
parameters: tool.inputSchema
|
|
22
|
+
}
|
|
23
|
+
}));
|
|
24
|
+
}
|
|
25
|
+
if (request.providerOptions) {
|
|
26
|
+
Object.assign(body, request.providerOptions);
|
|
27
|
+
}
|
|
28
|
+
return body;
|
|
29
|
+
}
|
|
30
|
+
function toOpenAIMessages(message) {
|
|
31
|
+
if (message.role === "tool") {
|
|
32
|
+
const results = message.parts.filter((part) => part.type === "tool_result");
|
|
33
|
+
return results.map((result) => ({
|
|
34
|
+
role: "tool",
|
|
35
|
+
tool_call_id: result.callId,
|
|
36
|
+
content: stringifyContent(result.content)
|
|
37
|
+
}));
|
|
38
|
+
}
|
|
39
|
+
const text = message.parts
|
|
40
|
+
.filter((part) => part.type === "text")
|
|
41
|
+
.map((part) => part.text)
|
|
42
|
+
.join("");
|
|
43
|
+
const toolCalls = message.parts.filter((part) => part.type === "tool_call");
|
|
44
|
+
const images = message.parts.filter((part) => part.type === "image");
|
|
45
|
+
if (images.length > 0) {
|
|
46
|
+
if (message.role !== "user") {
|
|
47
|
+
throw new RuntimeError({
|
|
48
|
+
code: "MODEL_REQUEST_FAILED",
|
|
49
|
+
category: "model",
|
|
50
|
+
message: "OpenAI-compatible adapter only supports image parts on user messages"
|
|
51
|
+
});
|
|
52
|
+
}
|
|
53
|
+
const content = [];
|
|
54
|
+
if (text) {
|
|
55
|
+
content.push({ type: "text", text });
|
|
56
|
+
}
|
|
57
|
+
for (const image of images) {
|
|
58
|
+
const url = image.url ?? (image.data ? `data:${image.mimeType};base64,${image.data}` : undefined);
|
|
59
|
+
if (!url) {
|
|
60
|
+
throw new RuntimeError({
|
|
61
|
+
code: "MODEL_REQUEST_FAILED",
|
|
62
|
+
category: "model",
|
|
63
|
+
message: "Image part must contain url or data"
|
|
64
|
+
});
|
|
65
|
+
}
|
|
66
|
+
content.push({
|
|
67
|
+
type: "image_url",
|
|
68
|
+
image_url: { url }
|
|
69
|
+
});
|
|
70
|
+
}
|
|
71
|
+
return [{ role: "user", content }];
|
|
72
|
+
}
|
|
73
|
+
const result = {
|
|
74
|
+
role: message.role,
|
|
75
|
+
content: text || null
|
|
76
|
+
};
|
|
77
|
+
if (toolCalls.length > 0) {
|
|
78
|
+
result.tool_calls = toolCalls.map((call) => ({
|
|
79
|
+
id: call.callId,
|
|
80
|
+
type: "function",
|
|
81
|
+
function: {
|
|
82
|
+
name: call.toolName,
|
|
83
|
+
arguments: JSON.stringify(call.input)
|
|
84
|
+
}
|
|
85
|
+
}));
|
|
86
|
+
}
|
|
87
|
+
return [result];
|
|
88
|
+
}
|
|
89
|
+
function stringifyContent(value) {
|
|
90
|
+
return typeof value === "string" ? value : JSON.stringify(value);
|
|
91
|
+
}
|
|
92
|
+
//# sourceMappingURL=adapter.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"adapter.js","sourceRoot":"","sources":["../../../../src/model/providers/openai-compatible/adapter.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,kCAAkC,CAAC;AAahE,MAAM,UAAU,eAAe,CAAC,OAAqB;IACnD,MAAM,IAAI,GAA8B;QACtC,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,gBAAgB,CAAyB;QAC5E,MAAM,EAAE,IAAI;QACZ,cAAc,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;KACxC,CAAC;IAEF,IAAI,OAAO,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;QACtC,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;IACzC,CAAC;IAED,IAAI,OAAO,CAAC,eAAe,KAAK,SAAS,EAAE,CAAC;QAC1C,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,eAAe,CAAC;IAC5C,CAAC;IAED,IAAI,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9C,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACxC,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE;gBACR,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,UAAU,EAAE,IAAI,CAAC,WAAW;aAC7B;SACF,CAAC,CAAC,CAAC;IACN,CAAC;IAED,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;QAC5B,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,eAAe,CAAC,CAAC;IAC/C,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAqB;IAC7C,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QAC5B,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CAClC,CAAC,IAAI,EAA0B,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,aAAa,CAC9D,CAAC;QAEF,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YAC9B,IAAI,EAAE,MAAM;YACZ,YAAY,EAAE,MAAM,CAAC,MAAM;YAC3B,OAAO,EAAE,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC;SAC1C,CAAC,CAAC,CAAC;IACN,CAAC;IAED,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK;SACvB,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;SACtC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;SACxB,IAAI,CAAC,EAAE,CAAC,CAAC;IAEZ,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CACpC,CAAC,IAAI,EAAwB,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW,CAC1D,CAAC;IAEF,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CACjC,CAAC,IAAI,EAAqB,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,OAAO,CACnD,CAAC;IAEF,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtB,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC5B,MAAM,IAAI,YAAY,CAAC;gBACrB,IAAI,EAAE,sBAAsB;gBAC5B,QAAQ,EAAE,OAAO;gBACjB,OAAO,EAAE,sEAAsE;aAChF,CAAC,CAAC;QACL,CAAC;QAED,MAAM,OAAO,GAAmC,EAAE,CAAC;QACnD,IAAI,IAAI,EAAE,CAAC;YACT,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QACvC,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,IAAI,CACvB,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,KAAK,CAAC,QAAQ,WAAW,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CACvE,CAAC;YAEF,IAAI,CAAC,GAAG,EAAE,CAAC;gBACT,MAAM,IAAI,YAAY,CAAC;oBACrB,IAAI,EAAE,sBAAsB;oBAC5B,QAAQ,EAAE,OAAO;oBACjB,OAAO,EAAE,qCAAqC;iBAC/C,CAAC,CAAC;YACL,CAAC;YAED,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,WAAW;gBACjB,SAAS,EAAE,EAAE,GAAG,EAAE;aACnB,CAAC,CAAC;QACL,CAAC;QAED,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;IACrC,CAAC;IAED,MAAM,MAAM,GAAkB;QAC5B,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,OAAO,EAAE,IAAI,IAAI,IAAI;KACtB,CAAC;IAEF,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzB,MAAM,CAAC,UAAU,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC3C,EAAE,EAAE,IAAI,CAAC,MAAM;YACf,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE;gBACR,IAAI,EAAE,IAAI,CAAC,QAAQ;gBACnB,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC;aACtC;SACF,CAAC,CAAC,CAAC;IACN,CAAC;IAED,OAAO,CAAC,MAAM,CAAC,CAAC;AAClB,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAc;IACtC,OAAO,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;AACnE,CAAC"}
|
|
@@ -0,0 +1,68 @@
|
|
|
1
|
+
import { RuntimeError } from "../../../errors/runtime-error.js";
|
|
2
|
+
export async function responseToRuntimeError(response) {
|
|
3
|
+
const body = await safeText(response);
|
|
4
|
+
const lower = body.toLowerCase();
|
|
5
|
+
if (response.status === 401 || response.status === 403) {
|
|
6
|
+
return new RuntimeError({
|
|
7
|
+
code: "MODEL_AUTH",
|
|
8
|
+
category: "model",
|
|
9
|
+
message: `Model authentication failed (${response.status})`,
|
|
10
|
+
metadata: { status: response.status }
|
|
11
|
+
});
|
|
12
|
+
}
|
|
13
|
+
if (response.status === 429) {
|
|
14
|
+
return new RuntimeError({
|
|
15
|
+
code: "MODEL_RATE_LIMIT",
|
|
16
|
+
category: "model",
|
|
17
|
+
message: "Model rate limit exceeded",
|
|
18
|
+
retryable: true,
|
|
19
|
+
metadata: { status: response.status }
|
|
20
|
+
});
|
|
21
|
+
}
|
|
22
|
+
if (response.status === 400 &&
|
|
23
|
+
(lower.includes("context") || lower.includes("token"))) {
|
|
24
|
+
return new RuntimeError({
|
|
25
|
+
code: "MODEL_CONTEXT_LIMIT",
|
|
26
|
+
category: "model",
|
|
27
|
+
message: "Model context limit exceeded",
|
|
28
|
+
metadata: { status: response.status }
|
|
29
|
+
});
|
|
30
|
+
}
|
|
31
|
+
return new RuntimeError({
|
|
32
|
+
code: "MODEL_REQUEST_FAILED",
|
|
33
|
+
category: "model",
|
|
34
|
+
message: `Model request failed (${response.status})`,
|
|
35
|
+
retryable: response.status >= 500,
|
|
36
|
+
metadata: { status: response.status }
|
|
37
|
+
});
|
|
38
|
+
}
|
|
39
|
+
export function requestToRuntimeError(error) {
|
|
40
|
+
if ((error instanceof DOMException && error.name === "AbortError") ||
|
|
41
|
+
(error instanceof Error && error.name === "AbortError")) {
|
|
42
|
+
return new RuntimeError({
|
|
43
|
+
code: "ABORTED",
|
|
44
|
+
category: "abort",
|
|
45
|
+
message: "Model request aborted",
|
|
46
|
+
cause: error
|
|
47
|
+
});
|
|
48
|
+
}
|
|
49
|
+
if (error instanceof RuntimeError) {
|
|
50
|
+
return error;
|
|
51
|
+
}
|
|
52
|
+
return new RuntimeError({
|
|
53
|
+
code: "MODEL_REQUEST_FAILED",
|
|
54
|
+
category: "model",
|
|
55
|
+
message: error instanceof Error ? error.message : "Model request failed",
|
|
56
|
+
retryable: true,
|
|
57
|
+
cause: error
|
|
58
|
+
});
|
|
59
|
+
}
|
|
60
|
+
async function safeText(response) {
|
|
61
|
+
try {
|
|
62
|
+
return await response.text();
|
|
63
|
+
}
|
|
64
|
+
catch {
|
|
65
|
+
return "";
|
|
66
|
+
}
|
|
67
|
+
}
|
|
68
|
+
//# sourceMappingURL=errors.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"errors.js","sourceRoot":"","sources":["../../../../src/model/providers/openai-compatible/errors.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,kCAAkC,CAAC;AAEhE,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,QAAkB;IAElB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACtC,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IAEjC,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;QACvD,OAAO,IAAI,YAAY,CAAC;YACtB,IAAI,EAAE,YAAY;YAClB,QAAQ,EAAE,OAAO;YACjB,OAAO,EAAE,gCAAgC,QAAQ,CAAC,MAAM,GAAG;YAC3D,QAAQ,EAAE,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE;SACtC,CAAC,CAAC;IACL,CAAC;IAED,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;QAC5B,OAAO,IAAI,YAAY,CAAC;YACtB,IAAI,EAAE,kBAAkB;YACxB,QAAQ,EAAE,OAAO;YACjB,OAAO,EAAE,2BAA2B;YACpC,SAAS,EAAE,IAAI;YACf,QAAQ,EAAE,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE;SACtC,CAAC,CAAC;IACL,CAAC;IAED,IACE,QAAQ,CAAC,MAAM,KAAK,GAAG;QACvB,CAAC,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,EACtD,CAAC;QACD,OAAO,IAAI,YAAY,CAAC;YACtB,IAAI,EAAE,qBAAqB;YAC3B,QAAQ,EAAE,OAAO;YACjB,OAAO,EAAE,8BAA8B;YACvC,QAAQ,EAAE,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE;SACtC,CAAC,CAAC;IACL,CAAC;IAED,OAAO,IAAI,YAAY,CAAC;QACtB,IAAI,EAAE,sBAAsB;QAC5B,QAAQ,EAAE,OAAO;QACjB,OAAO,EAAE,yBAAyB,QAAQ,CAAC,MAAM,GAAG;QACpD,SAAS,EAAE,QAAQ,CAAC,MAAM,IAAI,GAAG;QACjC,QAAQ,EAAE,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE;KACtC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,KAAc;IAClD,IACE,CAAC,KAAK,YAAY,YAAY,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC;QAC9D,CAAC,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,EACvD,CAAC;QACD,OAAO,IAAI,YAAY,CAAC;YACtB,IAAI,EAAE,SAAS;YACf,QAAQ,EAAE,OAAO;YACjB,OAAO,EAAE,uBAAuB;YAChC,KAAK,EAAE,KAAK;SACb,CAAC,CAAC;IACL,CAAC;IAED,IAAI,KAAK,YAAY,YAAY,EAAE,CAAC;QAClC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,IAAI,YAAY,CAAC;QACtB,IAAI,EAAE,sBAAsB;QAC5B,QAAQ,EAAE,OAAO;QACjB,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,sBAAsB;QACxE,SAAS,EAAE,IAAI;QACf,KAAK,EAAE,KAAK;KACb,CAAC,CAAC;AACL,CAAC;AAED,KAAK,UAAU,QAAQ,CAAC,QAAkB;IACxC,IAAI,CAAC;QACH,OAAO,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC/B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import type { ModelProvider } from "../../provider.js";
|
|
2
|
+
import type { ModelCapabilities, ModelEvent, ModelRequest } from "../../types.js";
|
|
3
|
+
export interface OpenAICompatibleProviderOptions {
|
|
4
|
+
baseUrl: string;
|
|
5
|
+
apiKey?: string;
|
|
6
|
+
defaultCapabilities?: Partial<ModelCapabilities>;
|
|
7
|
+
fetch?: typeof globalThis.fetch;
|
|
8
|
+
}
|
|
9
|
+
export declare class OpenAICompatibleProvider implements ModelProvider {
|
|
10
|
+
readonly id = "openai-compatible";
|
|
11
|
+
private readonly baseUrl;
|
|
12
|
+
private readonly apiKey?;
|
|
13
|
+
private readonly fetchImpl;
|
|
14
|
+
private readonly capabilities;
|
|
15
|
+
constructor(options: OpenAICompatibleProviderOptions);
|
|
16
|
+
getCapabilities(_model: string): Promise<ModelCapabilities>;
|
|
17
|
+
stream(request: ModelRequest, signal?: AbortSignal): AsyncIterable<ModelEvent>;
|
|
18
|
+
}
|