@n8n/agents 0.12.2 → 0.13.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/dist/build.tsbuildinfo +1 -1
- package/dist/index.d.ts +22 -21
- package/dist/index.js +13 -13
- package/dist/index.js.map +1 -1
- package/dist/runtime/{agent-runtime.d.ts → loop/agent-runtime.d.ts} +19 -51
- package/dist/runtime/loop/agent-runtime.js +468 -0
- package/dist/runtime/loop/agent-runtime.js.map +1 -0
- package/dist/runtime/loop/execution-counter.d.ts +12 -0
- package/dist/runtime/loop/execution-counter.js +36 -0
- package/dist/runtime/loop/execution-counter.js.map +1 -0
- package/dist/runtime/loop/generate-sink.d.ts +13 -0
- package/dist/runtime/loop/generate-sink.js +75 -0
- package/dist/runtime/loop/generate-sink.js.map +1 -0
- package/dist/runtime/loop/run-output-sink.d.ts +68 -0
- package/dist/runtime/loop/run-output-sink.js +3 -0
- package/dist/runtime/loop/run-output-sink.js.map +1 -0
- package/dist/runtime/loop/runtime-context.d.ts +38 -0
- package/dist/runtime/loop/runtime-context.js +188 -0
- package/dist/runtime/loop/runtime-context.js.map +1 -0
- package/dist/runtime/{runtime-helpers.d.ts → loop/runtime-helpers.d.ts} +3 -2
- package/dist/runtime/{runtime-helpers.js → loop/runtime-helpers.js} +4 -2
- package/dist/runtime/loop/runtime-helpers.js.map +1 -0
- package/dist/runtime/loop/stream-sink.d.ts +22 -0
- package/dist/runtime/loop/stream-sink.js +144 -0
- package/dist/runtime/loop/stream-sink.js.map +1 -0
- package/dist/runtime/{mcp-connection.d.ts → mcp/mcp-connection.d.ts} +2 -2
- package/dist/runtime/{mcp-connection.js → mcp/mcp-connection.js} +1 -1
- package/dist/runtime/mcp/mcp-connection.js.map +1 -0
- package/dist/runtime/{mcp-tool-resolver.d.ts → mcp/mcp-tool-resolver.d.ts} +1 -1
- package/dist/runtime/mcp/mcp-tool-resolver.js.map +1 -0
- package/dist/runtime/{episodic-memory-defaults.d.ts → memory/episodic-memory-defaults.d.ts} +1 -1
- package/dist/runtime/{episodic-memory-defaults.js → memory/episodic-memory-defaults.js} +2 -2
- package/dist/runtime/memory/episodic-memory-defaults.js.map +1 -0
- package/dist/runtime/{episodic-memory.d.ts → memory/episodic-memory.d.ts} +4 -4
- package/dist/runtime/{episodic-memory.js → memory/episodic-memory.js} +2 -2
- package/dist/runtime/memory/episodic-memory.js.map +1 -0
- package/dist/runtime/memory/memory-lifecycle.js.map +1 -0
- package/dist/runtime/memory/memory-orchestrator.d.ts +27 -0
- package/dist/runtime/memory/memory-orchestrator.js +232 -0
- package/dist/runtime/memory/memory-orchestrator.js.map +1 -0
- package/dist/runtime/{memory-store.d.ts → memory/memory-store.d.ts} +4 -4
- package/dist/runtime/{memory-store.js → memory/memory-store.js} +1 -1
- package/dist/runtime/memory/memory-store.js.map +1 -0
- package/dist/runtime/{observation-log-defaults.d.ts → memory/observation-log-defaults.d.ts} +2 -2
- package/dist/runtime/{observation-log-defaults.js → memory/observation-log-defaults.js} +6 -6
- package/dist/runtime/memory/observation-log-defaults.js.map +1 -0
- package/dist/runtime/{observation-log-observer.d.ts → memory/observation-log-observer.d.ts} +6 -5
- package/dist/runtime/{observation-log-observer.js → memory/observation-log-observer.js} +6 -2
- package/dist/runtime/memory/observation-log-observer.js.map +1 -0
- package/dist/runtime/{observation-log-reflector.d.ts → memory/observation-log-reflector.d.ts} +2 -2
- package/dist/runtime/{observation-log-reflector.js → memory/observation-log-reflector.js} +4 -6
- package/dist/runtime/memory/observation-log-reflector.js.map +1 -0
- package/dist/runtime/{observation-log-renderer.d.ts → memory/observation-log-renderer.d.ts} +1 -1
- package/dist/runtime/{observation-log-renderer.js → memory/observation-log-renderer.js} +1 -1
- package/dist/runtime/memory/observation-log-renderer.js.map +1 -0
- package/dist/runtime/{observation-log-store.d.ts → memory/observation-log-store.d.ts} +2 -1
- package/dist/runtime/{observation-log-store.js → memory/observation-log-store.js} +1 -0
- package/dist/runtime/memory/observation-log-store.js.map +1 -0
- package/dist/runtime/{scoped-memory-task-runner.d.ts → memory/scoped-memory-task-runner.d.ts} +5 -2
- package/dist/runtime/{scoped-memory-task-runner.js → memory/scoped-memory-task-runner.js} +2 -1
- package/dist/runtime/memory/scoped-memory-task-runner.js.map +1 -0
- package/dist/runtime/{strip-orphaned-tool-messages.d.ts → memory/strip-orphaned-tool-messages.d.ts} +1 -1
- package/dist/runtime/{strip-orphaned-tool-messages.js → memory/strip-orphaned-tool-messages.js} +1 -1
- package/dist/runtime/memory/strip-orphaned-tool-messages.js.map +1 -0
- package/dist/runtime/{title-generation.d.ts → memory/title-generation.d.ts} +5 -3
- package/dist/runtime/{title-generation.js → memory/title-generation.js} +5 -5
- package/dist/runtime/memory/title-generation.js.map +1 -0
- package/dist/runtime/model/lazy-ai.js.map +1 -0
- package/dist/runtime/{message-list.d.ts → model/message-list.d.ts} +3 -3
- package/dist/runtime/{message-list.js → model/message-list.js} +3 -3
- package/dist/runtime/model/message-list.js.map +1 -0
- package/dist/runtime/model/messages.d.ts +17 -0
- package/dist/runtime/{messages.js → model/messages.js} +64 -18
- package/dist/runtime/model/messages.js.map +1 -0
- package/dist/runtime/{model-factory.d.ts → model/model-factory.d.ts} +3 -2
- package/dist/runtime/{model-factory.js → model/model-factory.js} +3 -3
- package/dist/runtime/model/model-factory.js.map +1 -0
- package/dist/runtime/model/provider-credentials.js.map +1 -0
- package/dist/runtime/model/raw-usage.d.ts +11 -0
- package/dist/runtime/model/raw-usage.js +55 -0
- package/dist/runtime/model/raw-usage.js.map +1 -0
- package/dist/runtime/state/background-task-tracker.js.map +1 -0
- package/dist/runtime/{event-bus.d.ts → state/event-bus.d.ts} +2 -2
- package/dist/runtime/{event-bus.js → state/event-bus.js} +1 -1
- package/dist/runtime/state/event-bus.js.map +1 -0
- package/dist/runtime/{run-state.d.ts → state/run-state.d.ts} +1 -1
- package/dist/runtime/state/run-state.js.map +1 -0
- package/dist/runtime/streaming/stream-session.d.ts +21 -0
- package/dist/runtime/streaming/stream-session.js +67 -0
- package/dist/runtime/streaming/stream-session.js.map +1 -0
- package/dist/runtime/streaming/stream-writer-guard.d.ts +14 -0
- package/dist/runtime/streaming/stream-writer-guard.js +45 -0
- package/dist/runtime/streaming/stream-writer-guard.js.map +1 -0
- package/dist/runtime/{stream.d.ts → streaming/stream.d.ts} +2 -1
- package/dist/runtime/{stream.js → streaming/stream.js} +3 -1
- package/dist/runtime/streaming/stream.js.map +1 -0
- package/dist/runtime/telemetry/runtime-telemetry.d.ts +19 -0
- package/dist/runtime/telemetry/runtime-telemetry.js +174 -0
- package/dist/runtime/telemetry/runtime-telemetry.js.map +1 -0
- package/dist/runtime/{deferred-tool-manager.d.ts → tools/deferred-tool-manager.d.ts} +2 -2
- package/dist/runtime/{deferred-tool-manager.js → tools/deferred-tool-manager.js} +3 -5
- package/dist/runtime/tools/deferred-tool-manager.js.map +1 -0
- package/dist/runtime/{delegate-sub-agent-tool.d.ts → tools/delegate-sub-agent-tool.d.ts} +2 -2
- package/dist/runtime/{delegate-sub-agent-tool.js → tools/delegate-sub-agent-tool.js} +2 -2
- package/dist/runtime/tools/delegate-sub-agent-tool.js.map +1 -0
- package/dist/runtime/{fix-tool-call.d.ts → tools/fix-tool-call.d.ts} +1 -1
- package/dist/runtime/{fix-tool-call.js → tools/fix-tool-call.js} +1 -1
- package/dist/runtime/tools/fix-tool-call.js.map +1 -0
- package/dist/runtime/{sdk-owned-tool.d.ts → tools/sdk-owned-tool.d.ts} +1 -1
- package/dist/runtime/tools/sdk-owned-tool.js.map +1 -0
- package/dist/runtime/tools/sub-agent-task-path.js.map +1 -0
- package/dist/runtime/{tool-adapter.d.ts → tools/tool-adapter.d.ts} +1 -1
- package/dist/runtime/{tool-adapter.js → tools/tool-adapter.js} +4 -4
- package/dist/runtime/tools/tool-adapter.js.map +1 -0
- package/dist/runtime/tools/tool-call-executor.d.ts +86 -0
- package/dist/runtime/tools/tool-call-executor.js +498 -0
- package/dist/runtime/tools/tool-call-executor.js.map +1 -0
- package/dist/runtime/{write-todos-tool.d.ts → tools/write-todos-tool.d.ts} +1 -1
- package/dist/runtime/{write-todos-tool.js → tools/write-todos-tool.js} +1 -1
- package/dist/runtime/tools/write-todos-tool.js.map +1 -0
- package/dist/sdk/agent.d.ts +7 -1
- package/dist/sdk/agent.js +18 -8
- package/dist/sdk/agent.js.map +1 -1
- package/dist/sdk/eval.js +1 -1
- package/dist/sdk/eval.js.map +1 -1
- package/dist/sdk/guardrails/patterns.d.ts +3 -0
- package/dist/sdk/guardrails/patterns.js +95 -2
- package/dist/sdk/guardrails/patterns.js.map +1 -1
- package/dist/sdk/mcp-client.js +1 -1
- package/dist/sdk/mcp-client.js.map +1 -1
- package/dist/sdk/memory.d.ts +1 -1
- package/dist/sdk/memory.js +6 -6
- package/dist/sdk/memory.js.map +1 -1
- package/dist/skills/registry.js +2 -4
- package/dist/skills/registry.js.map +1 -1
- package/dist/skills/validator.js +4 -6
- package/dist/skills/validator.js.map +1 -1
- package/dist/types/sdk/agent-builder.d.ts +2 -0
- package/dist/types/sdk/agent.d.ts +6 -2
- package/dist/types/sdk/guardrail.d.ts +1 -1
- package/dist/types/sdk/provider.d.ts +1 -0
- package/dist/workspace/sandbox/daytona-auth-manager.d.ts +1 -1
- package/dist/workspace/sandbox/daytona-sandbox.d.ts +1 -1
- package/dist/workspace/sandbox/lazy-daytona.d.ts +1 -1
- package/dist/workspace/sandbox/lazy-daytona.js +1 -1
- package/dist/workspace/sandbox/lazy-daytona.js.map +1 -1
- package/dist/workspace/sandbox/types.d.ts +1 -1
- package/package.json +10 -10
- package/dist/runtime/agent-runtime.js +0 -1771
- package/dist/runtime/agent-runtime.js.map +0 -1
- package/dist/runtime/background-task-tracker.js.map +0 -1
- package/dist/runtime/deferred-tool-manager.js.map +0 -1
- package/dist/runtime/delegate-sub-agent-tool.js.map +0 -1
- package/dist/runtime/episodic-memory-defaults.js.map +0 -1
- package/dist/runtime/episodic-memory.js.map +0 -1
- package/dist/runtime/event-bus.js.map +0 -1
- package/dist/runtime/execution-counter.d.ts +0 -9
- package/dist/runtime/execution-counter.js +0 -16
- package/dist/runtime/execution-counter.js.map +0 -1
- package/dist/runtime/fix-tool-call.js.map +0 -1
- package/dist/runtime/lazy-ai.js.map +0 -1
- package/dist/runtime/mcp-connection.js.map +0 -1
- package/dist/runtime/mcp-tool-resolver.js.map +0 -1
- package/dist/runtime/memory-lifecycle.js.map +0 -1
- package/dist/runtime/memory-store.js.map +0 -1
- package/dist/runtime/message-list.js.map +0 -1
- package/dist/runtime/messages.d.ts +0 -6
- package/dist/runtime/messages.js.map +0 -1
- package/dist/runtime/model-factory.js.map +0 -1
- package/dist/runtime/observation-log-defaults.js.map +0 -1
- package/dist/runtime/observation-log-observer.js.map +0 -1
- package/dist/runtime/observation-log-reflector.js.map +0 -1
- package/dist/runtime/observation-log-renderer.js.map +0 -1
- package/dist/runtime/observation-log-store.js.map +0 -1
- package/dist/runtime/provider-credentials.js.map +0 -1
- package/dist/runtime/run-state.js.map +0 -1
- package/dist/runtime/runtime-helpers.js.map +0 -1
- package/dist/runtime/scoped-memory-task-runner.js.map +0 -1
- package/dist/runtime/sdk-owned-tool.js.map +0 -1
- package/dist/runtime/stream.js.map +0 -1
- package/dist/runtime/strip-orphaned-tool-messages.js.map +0 -1
- package/dist/runtime/sub-agent-task-path.js.map +0 -1
- package/dist/runtime/title-generation.js.map +0 -1
- package/dist/runtime/tool-adapter.js.map +0 -1
- package/dist/runtime/write-todos-tool.js.map +0 -1
- /package/dist/runtime/{mcp-tool-resolver.js → mcp/mcp-tool-resolver.js} +0 -0
- /package/dist/runtime/{memory-lifecycle.d.ts → memory/memory-lifecycle.d.ts} +0 -0
- /package/dist/runtime/{memory-lifecycle.js → memory/memory-lifecycle.js} +0 -0
- /package/dist/runtime/{lazy-ai.d.ts → model/lazy-ai.d.ts} +0 -0
- /package/dist/runtime/{lazy-ai.js → model/lazy-ai.js} +0 -0
- /package/dist/runtime/{provider-credentials.d.ts → model/provider-credentials.d.ts} +0 -0
- /package/dist/runtime/{provider-credentials.js → model/provider-credentials.js} +0 -0
- /package/dist/runtime/{background-task-tracker.d.ts → state/background-task-tracker.d.ts} +0 -0
- /package/dist/runtime/{background-task-tracker.js → state/background-task-tracker.js} +0 -0
- /package/dist/runtime/{run-state.js → state/run-state.js} +0 -0
- /package/dist/runtime/{sdk-owned-tool.js → tools/sdk-owned-tool.js} +0 -0
- /package/dist/runtime/{sub-agent-task-path.d.ts → tools/sub-agent-task-path.d.ts} +0 -0
- /package/dist/runtime/{sub-agent-task-path.js → tools/sub-agent-task-path.js} +0 -0
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { AgentEvent } from '
|
|
2
|
-
import type { AgentEventData, AgentEventHandler } from '
|
|
1
|
+
import { AgentEvent } from '../../types/runtime/event';
|
|
2
|
+
import type { AgentEventData, AgentEventHandler } from '../../types/runtime/event';
|
|
3
3
|
export interface AgentAbortScope {
|
|
4
4
|
readonly signal: AbortSignal;
|
|
5
5
|
readonly isAborted: boolean;
|
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.AgentEvent = exports.AgentEventBus = void 0;
|
|
4
|
-
const event_1 = require("
|
|
4
|
+
const event_1 = require("../../types/runtime/event");
|
|
5
5
|
Object.defineProperty(exports, "AgentEvent", { enumerable: true, get: function () { return event_1.AgentEvent; } });
|
|
6
6
|
class EventBusAbortScope {
|
|
7
7
|
constructor(externalSignal, onDispose) {
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"event-bus.js","sourceRoot":"","sources":["../../../src/runtime/state/event-bus.ts"],"names":[],"mappings":";;;AAAA,qDAAuD;AAkK9C,2FAlKA,kBAAU,OAkKA;AAxJnB,MAAM,kBAAkB;IAOvB,YACC,cAAuC,EACtB,SAA8C;QAA9C,cAAS,GAAT,SAAS,CAAqC;QAR/C,eAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QAI5C,aAAQ,GAAG,KAAK,CAAC;QAMxB,IAAI,CAAC,cAAc;YAAE,OAAO;QAE5B,IAAI,cAAc,CAAC,OAAO,EAAE,CAAC;YAC5B,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;YAC7C,OAAO;QACR,CAAC;QAED,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QACnE,cAAc,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAClE,IAAI,CAAC,eAAe,GAAG,GAAG,EAAE,CAAC,cAAc,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACnF,CAAC;IAED,IAAI,MAAM;QACT,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;IAC/B,CAAC;IAED,IAAI,SAAS;QACZ,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC;IACvC,CAAC;IAED,KAAK;QACJ,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;IAED,OAAO;QACN,IAAI,IAAI,CAAC,QAAQ;YAAE,OAAO;QAC1B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,IAAI,CAAC,eAAe,EAAE,EAAE,CAAC;QACzB,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;QACjC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IACtB,CAAC;CACD;AAiBD,MAAa,aAAa;IAA1B;QACS,aAAQ,GAAG,IAAI,GAAG,EAAsC,CAAC;QAEzD,eAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QAInC,gBAAW,GAAG,IAAI,GAAG,EAAsB,CAAC;IAoFrD,CAAC;IAlFA,EAAE,CAAC,KAAiB,EAAE,OAA0B;QAC/C,IAAI,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACnC,IAAI,CAAC,GAAG,EAAE,CAAC;YACV,GAAG,GAAG,IAAI,GAAG,EAAE,CAAC;YAChB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAC/B,CAAC;QACD,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAClB,CAAC;IAED,GAAG,CAAC,KAAiB,EAAE,OAA0B;QAChD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;IAC3C,CAAC;IAED,IAAI,CAAC,IAAoB;QACxB,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,IAAI,CAAC,GAAG;YAAE,OAAO;QACjB,KAAK,MAAM,OAAO,IAAI,GAAG,EAAE,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC,CAAC;QACf,CAAC;IACF,CAAC;IAED,KAAK;QACJ,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QACxB,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACtC,KAAK,CAAC,KAAK,EAAE,CAAC;QACf,CAAC;IACF,CAAC;IAED,gBAAgB,CAAC,cAA4B;QAC5C,MAAM,KAAK,GAAG,IAAI,kBAAkB,CAAC,cAAc,EAAE,CAAC,QAAQ,EAAE,EAAE;YACjE,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC5B,OAAO,KAAK,CAAC;IACd,CAAC;IAWD,UAAU,CAAC,cAA4B;QACtC,IAAI,CAAC,eAAe,EAAE,EAAE,CAAC;QACzB,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;QACjC,IAAI,CAAC,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QAExC,IAAI,cAAc,EAAE,CAAC;YACpB,IAAI,cAAc,CAAC,OAAO,EAAE,CAAC;gBAC5B,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;YAC9C,CAAC;iBAAM,CAAC;gBACP,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;gBACnE,cAAc,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;gBAClE,IAAI,CAAC,eAAe,GAAG,GAAG,EAAE,CAAC,cAAc,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACnF,CAAC;QACF,CAAC;IACF,CAAC;IAGD,IAAI,MAAM;QACT,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;IAC/B,CAAC;IAED,IAAI,SAAS;QACZ,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC;IACvC,CAAC;IAMD,OAAO;QACN,IAAI,CAAC,eAAe,EAAE,EAAE,CAAC;QACzB,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;QACjC,KAAK,MAAM,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC;YAC3C,KAAK,CAAC,OAAO,EAAE,CAAC;QACjB,CAAC;QACD,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC;CACD;AA3FD,sCA2FC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"run-state.js","sourceRoot":"","sources":["../../../src/runtime/state/run-state.ts"],"names":[],"mappings":";;;AAkEA,sCAEC;AA1DD,MAAM,qBAAqB;IAA3B;QACS,UAAK,GAAG,IAAI,GAAG,EAAkC,CAAC;IAa3D,CAAC;IAXA,KAAK,CAAC,IAAI,CAAC,GAAW,EAAE,KAA6B;QACpD,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,GAAW;QACrB,OAAO,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,GAAW;QACvB,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;IAC/C,CAAC;CACD;AAOD,MAAa,eAAe;IAG3B,YAAY,OAAoC;QAC/C,IAAI,CAAC,KAAK,GAAG,OAAO,IAAI,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,qBAAqB,EAAE,CAAC;IACtF,CAAC;IAGD,KAAK,CAAC,OAAO,CAAC,KAAa,EAAE,KAA6B;QACzD,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;IACjE,CAAC;IAID,KAAK,CAAC,MAAM,CAAC,KAAa;QACzB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,KAAK;YAAE,OAAO,SAAS,CAAC;QAC7B,IAAI,KAAK,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,OAAO,KAAK,mCAAmC,CAAC,CAAC;QAClE,CAAC;QACD,MAAM,QAAQ,GAA2B,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;QACzE,OAAO,QAAQ,CAAC;IACjB,CAAC;IAGD,KAAK,CAAC,QAAQ,CAAC,KAAa;QAC3B,IAAI,CAAC;YACJ,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,WAAoB,EAAE,CAAC;YAC/B,OAAO,CAAC,KAAK,CAAC,iDAAiD,KAAK,GAAG,EAAE,WAAW,CAAC,CAAC;QACvF,CAAC;IACF,CAAC;CACD;AAhCD,0CAgCC;AAGD,SAAgB,aAAa;IAC5B,OAAO,OAAO,MAAM,CAAC,UAAU,EAAE,EAAE,CAAC;AACrC,CAAC"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
import { StreamWriterGuard } from './stream-writer-guard';
|
|
2
|
+
import type { StreamChunk } from '../../types';
|
|
3
|
+
import type { ExecutionOptions, RunOptions, TokenUsage } from '../../types/sdk/agent';
|
|
4
|
+
import type { AgentAbortScope, AgentEventBus } from '../state/event-bus';
|
|
5
|
+
export interface StreamSessionDeps {
|
|
6
|
+
eventBus: AgentEventBus;
|
|
7
|
+
abortScope: AgentAbortScope;
|
|
8
|
+
runId: string;
|
|
9
|
+
options: (RunOptions & ExecutionOptions) | undefined;
|
|
10
|
+
withRootSpan: <T>(operation: 'generate' | 'stream', options: ExecutionOptions | undefined, runId: string, fn: () => Promise<T>) => Promise<T>;
|
|
11
|
+
runLoop: (guard: StreamWriterGuard) => Promise<void>;
|
|
12
|
+
flushTelemetry: (options: ExecutionOptions | undefined) => Promise<void>;
|
|
13
|
+
cleanupRun: () => Promise<void>;
|
|
14
|
+
updateState: (status: 'failed' | 'cancelled') => void;
|
|
15
|
+
emitError: (error: unknown) => void;
|
|
16
|
+
getAbortFinish?: () => {
|
|
17
|
+
usage?: TokenUsage;
|
|
18
|
+
model?: string;
|
|
19
|
+
};
|
|
20
|
+
}
|
|
21
|
+
export declare function startStreamSession(deps: StreamSessionDeps): ReadableStream<StreamChunk>;
|
|
@@ -0,0 +1,67 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.startStreamSession = startStreamSession;
|
|
4
|
+
const stream_writer_guard_1 = require("./stream-writer-guard");
|
|
5
|
+
function startStreamSession(deps) {
|
|
6
|
+
const { readable, writable } = new TransformStream();
|
|
7
|
+
const guard = new stream_writer_guard_1.StreamWriterGuard(writable.getWriter());
|
|
8
|
+
const onToolExecutionStart = (data) => {
|
|
9
|
+
if (data.type !== "tool_execution_start")
|
|
10
|
+
return;
|
|
11
|
+
void guard.write({
|
|
12
|
+
type: 'tool-execution-start',
|
|
13
|
+
toolCallId: data.toolCallId,
|
|
14
|
+
toolName: data.toolName,
|
|
15
|
+
startTime: Date.now(),
|
|
16
|
+
});
|
|
17
|
+
};
|
|
18
|
+
const onToolExecutionEnd = (data) => {
|
|
19
|
+
if (data.type !== "tool_execution_end")
|
|
20
|
+
return;
|
|
21
|
+
void guard.write({
|
|
22
|
+
type: 'tool-execution-end',
|
|
23
|
+
toolCallId: data.toolCallId,
|
|
24
|
+
toolName: data.toolName,
|
|
25
|
+
isError: data.isError,
|
|
26
|
+
endTime: Date.now(),
|
|
27
|
+
});
|
|
28
|
+
};
|
|
29
|
+
const onSubAgentStarted = (data) => {
|
|
30
|
+
if (data.type !== "subagent_started")
|
|
31
|
+
return;
|
|
32
|
+
const { type: _type, ...payload } = data;
|
|
33
|
+
void guard.write({ type: 'subagent-started', ...payload });
|
|
34
|
+
};
|
|
35
|
+
const onSubAgentCompleted = (data) => {
|
|
36
|
+
if (data.type !== "subagent_completed")
|
|
37
|
+
return;
|
|
38
|
+
const { type: _type, ...payload } = data;
|
|
39
|
+
void guard.write({ type: 'subagent-completed', ...payload });
|
|
40
|
+
};
|
|
41
|
+
deps.eventBus.on("tool_execution_start", onToolExecutionStart);
|
|
42
|
+
deps.eventBus.on("tool_execution_end", onToolExecutionEnd);
|
|
43
|
+
deps.eventBus.on("subagent_started", onSubAgentStarted);
|
|
44
|
+
deps.eventBus.on("subagent_completed", onSubAgentCompleted);
|
|
45
|
+
deps
|
|
46
|
+
.withRootSpan('stream', deps.options, deps.runId, async () => await deps.runLoop(guard))
|
|
47
|
+
.catch(async (error) => {
|
|
48
|
+
const isAbort = deps.abortScope.isAborted;
|
|
49
|
+
deps.updateState(isAbort ? 'cancelled' : 'failed');
|
|
50
|
+
if (!isAbort) {
|
|
51
|
+
deps.emitError(error);
|
|
52
|
+
}
|
|
53
|
+
await deps.cleanupRun();
|
|
54
|
+
await deps.flushTelemetry(deps.options);
|
|
55
|
+
await guard.fail(isAbort ? new Error('Agent run was aborted') : error, 'error', isAbort ? deps.getAbortFinish?.() : undefined);
|
|
56
|
+
})
|
|
57
|
+
.finally(() => {
|
|
58
|
+
deps.eventBus.off("tool_execution_start", onToolExecutionStart);
|
|
59
|
+
deps.eventBus.off("tool_execution_end", onToolExecutionEnd);
|
|
60
|
+
deps.eventBus.off("subagent_started", onSubAgentStarted);
|
|
61
|
+
deps.eventBus.off("subagent_completed", onSubAgentCompleted);
|
|
62
|
+
void guard.close();
|
|
63
|
+
deps.abortScope.dispose();
|
|
64
|
+
});
|
|
65
|
+
return readable;
|
|
66
|
+
}
|
|
67
|
+
//# sourceMappingURL=stream-session.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream-session.js","sourceRoot":"","sources":["../../../src/runtime/streaming/stream-session.ts"],"names":[],"mappings":";;AAuCA,gDAoEC;AA3GD,+DAA0D;AAuC1D,SAAgB,kBAAkB,CAAC,IAAuB;IACzD,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,IAAI,eAAe,EAA4B,CAAC;IAC/E,MAAM,KAAK,GAAG,IAAI,uCAAiB,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC,CAAC;IAK1D,MAAM,oBAAoB,GAAG,CAAC,IAAoB,EAAQ,EAAE;QAC3D,IAAI,IAAI,CAAC,IAAI,2BAAkC;YAAE,OAAO;QACxD,KAAK,KAAK,CAAC,KAAK,CAAC;YAChB,IAAI,EAAE,sBAAsB;YAC5B,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACrB,CAAC,CAAC;IACJ,CAAC,CAAC;IACF,MAAM,kBAAkB,GAAG,CAAC,IAAoB,EAAQ,EAAE;QACzD,IAAI,IAAI,CAAC,IAAI,yBAAgC;YAAE,OAAO;QACtD,KAAK,KAAK,CAAC,KAAK,CAAC;YAChB,IAAI,EAAE,oBAAoB;YAC1B,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE;SACnB,CAAC,CAAC;IACJ,CAAC,CAAC;IACF,MAAM,iBAAiB,GAAG,CAAC,IAAoB,EAAQ,EAAE;QACxD,IAAI,IAAI,CAAC,IAAI,uBAA+B;YAAE,OAAO;QACrD,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,GAAG,IAAI,CAAC;QACzC,KAAK,KAAK,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IAC5D,CAAC,CAAC;IACF,MAAM,mBAAmB,GAAG,CAAC,IAAoB,EAAQ,EAAE;QAC1D,IAAI,IAAI,CAAC,IAAI,yBAAiC;YAAE,OAAO;QACvD,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,GAAG,IAAI,CAAC;QACzC,KAAK,KAAK,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IAC9D,CAAC,CAAC;IAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,yBAAgC,oBAAoB,CAAC,CAAC;IACtE,IAAI,CAAC,QAAQ,CAAC,EAAE,uBAA8B,kBAAkB,CAAC,CAAC;IAClE,IAAI,CAAC,QAAQ,CAAC,EAAE,qBAA6B,iBAAiB,CAAC,CAAC;IAChE,IAAI,CAAC,QAAQ,CAAC,EAAE,uBAA+B,mBAAmB,CAAC,CAAC;IAEpE,IAAI;SACF,YAAY,CAAC,QAAQ,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;SACvF,KAAK,CAAC,KAAK,EAAE,KAAc,EAAE,EAAE;QAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC;QAC1C,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;QACnD,IAAI,CAAC,OAAO,EAAE,CAAC;YACd,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;QACD,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACxC,MAAM,KAAK,CAAC,IAAI,CACf,OAAO,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC,CAAC,CAAC,KAAK,EACpD,OAAO,EACP,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAC7C,CAAC;IACH,CAAC,CAAC;SACD,OAAO,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,QAAQ,CAAC,GAAG,yBAAgC,oBAAoB,CAAC,CAAC;QACvE,IAAI,CAAC,QAAQ,CAAC,GAAG,uBAA8B,kBAAkB,CAAC,CAAC;QACnE,IAAI,CAAC,QAAQ,CAAC,GAAG,qBAA6B,iBAAiB,CAAC,CAAC;QACjE,IAAI,CAAC,QAAQ,CAAC,GAAG,uBAA+B,mBAAmB,CAAC,CAAC;QACrE,KAAK,KAAK,CAAC,KAAK,EAAE,CAAC;QACnB,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEJ,OAAO,QAAQ,CAAC;AACjB,CAAC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import type { FinishReason, StreamChunk } from '../../types';
|
|
2
|
+
import type { TokenUsage } from '../../types/sdk/agent';
|
|
3
|
+
export declare class StreamWriterGuard {
|
|
4
|
+
private readonly writer;
|
|
5
|
+
private closed;
|
|
6
|
+
constructor(writer: WritableStreamDefaultWriter<StreamChunk>);
|
|
7
|
+
get isClosed(): boolean;
|
|
8
|
+
write(chunk: StreamChunk): Promise<void>;
|
|
9
|
+
close(): Promise<void>;
|
|
10
|
+
fail(error: unknown, finishReason?: FinishReason, finish?: {
|
|
11
|
+
usage?: TokenUsage;
|
|
12
|
+
model?: string;
|
|
13
|
+
}): Promise<void>;
|
|
14
|
+
}
|
|
@@ -0,0 +1,45 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.StreamWriterGuard = void 0;
|
|
4
|
+
class StreamWriterGuard {
|
|
5
|
+
constructor(writer) {
|
|
6
|
+
this.writer = writer;
|
|
7
|
+
this.closed = false;
|
|
8
|
+
writer.closed
|
|
9
|
+
.then(() => {
|
|
10
|
+
this.closed = true;
|
|
11
|
+
})
|
|
12
|
+
.catch(() => { });
|
|
13
|
+
}
|
|
14
|
+
get isClosed() {
|
|
15
|
+
return this.closed;
|
|
16
|
+
}
|
|
17
|
+
async write(chunk) {
|
|
18
|
+
if (this.closed)
|
|
19
|
+
return;
|
|
20
|
+
try {
|
|
21
|
+
await this.writer.write(chunk);
|
|
22
|
+
}
|
|
23
|
+
catch {
|
|
24
|
+
}
|
|
25
|
+
}
|
|
26
|
+
async close() {
|
|
27
|
+
if (this.closed)
|
|
28
|
+
return;
|
|
29
|
+
this.closed = true;
|
|
30
|
+
try {
|
|
31
|
+
await this.writer.close();
|
|
32
|
+
}
|
|
33
|
+
catch {
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
async fail(error, finishReason = 'error', finish) {
|
|
37
|
+
if (this.closed)
|
|
38
|
+
return;
|
|
39
|
+
await this.write({ type: 'error', error });
|
|
40
|
+
await this.write({ type: 'finish', finishReason, ...finish });
|
|
41
|
+
await this.close();
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
exports.StreamWriterGuard = StreamWriterGuard;
|
|
45
|
+
//# sourceMappingURL=stream-writer-guard.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream-writer-guard.js","sourceRoot":"","sources":["../../../src/runtime/streaming/stream-writer-guard.ts"],"names":[],"mappings":";;;AAWA,MAAa,iBAAiB;IAG7B,YAA6B,MAAgD;QAAhD,WAAM,GAAN,MAAM,CAA0C;QAFrE,WAAM,GAAG,KAAK,CAAC;QAGtB,MAAM,CAAC,MAAM;aACX,IAAI,CAAC,GAAG,EAAE;YACV,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACpB,CAAC,CAAC;aACD,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IACnB,CAAC;IAED,IAAI,QAAQ;QACX,OAAO,IAAI,CAAC,MAAM,CAAC;IACpB,CAAC;IAGD,KAAK,CAAC,KAAK,CAAC,KAAkB;QAC7B,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO;QACxB,IAAI,CAAC;YACJ,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;QAAC,MAAM,CAAC;QAET,CAAC;IACF,CAAC;IAGD,KAAK,CAAC,KAAK;QACV,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO;QACxB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC;YACJ,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAC3B,CAAC;QAAC,MAAM,CAAC;QAET,CAAC;IACF,CAAC;IASD,KAAK,CAAC,IAAI,CACT,KAAc,EACd,eAA6B,OAAO,EACpC,MAA+C;QAE/C,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO;QACxB,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;QAC3C,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,YAAY,EAAE,GAAG,MAAM,EAAE,CAAC,CAAC;QAC9D,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;IACpB,CAAC;CACD;AArDD,8CAqDC"}
|
|
@@ -1,10 +1,11 @@
|
|
|
1
1
|
import type { TextStreamPart, ToolSet } from 'ai';
|
|
2
|
-
import type { StreamChunk, TokenUsage } from '
|
|
2
|
+
import type { StreamChunk, TokenUsage } from '../../types';
|
|
3
3
|
export declare function toTokenUsage(usage: {
|
|
4
4
|
inputTokens?: number;
|
|
5
5
|
outputTokens?: number;
|
|
6
6
|
totalTokens?: number;
|
|
7
7
|
inputTokenDetails?: {
|
|
8
|
+
noCacheTokens?: number;
|
|
8
9
|
cacheReadTokens?: number;
|
|
9
10
|
cacheWriteTokens?: number;
|
|
10
11
|
};
|
|
@@ -10,10 +10,12 @@ function toTokenUsage(usage) {
|
|
|
10
10
|
completionTokens: usage.outputTokens ?? 0,
|
|
11
11
|
totalTokens: usage.totalTokens ?? (usage.inputTokens ?? 0) + (usage.outputTokens ?? 0),
|
|
12
12
|
};
|
|
13
|
+
const noCache = usage.inputTokenDetails?.noCacheTokens;
|
|
13
14
|
const cacheRead = usage.inputTokenDetails?.cacheReadTokens;
|
|
14
15
|
const cacheWrite = usage.inputTokenDetails?.cacheWriteTokens;
|
|
15
|
-
if (cacheRead || cacheWrite) {
|
|
16
|
+
if (noCache || cacheRead || cacheWrite) {
|
|
16
17
|
result.inputTokenDetails = {
|
|
18
|
+
...(noCache && { noCache }),
|
|
17
19
|
...(cacheRead && { cacheRead }),
|
|
18
20
|
...(cacheWrite && { cacheWrite }),
|
|
19
21
|
};
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream.js","sourceRoot":"","sources":["../../../src/runtime/streaming/stream.ts"],"names":[],"mappings":";;AAMA,oCAuCC;AAKD,oCA6FC;AAzID,SAAgB,YAAY,CAC3B,KAYY;IAEZ,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAE7B,MAAM,MAAM,GAAe;QAC1B,YAAY,EAAE,KAAK,CAAC,WAAW,IAAI,CAAC;QACpC,gBAAgB,EAAE,KAAK,CAAC,YAAY,IAAI,CAAC;QACzC,WAAW,EAAE,KAAK,CAAC,WAAW,IAAI,CAAC,KAAK,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC;KACtF,CAAC;IAEF,MAAM,OAAO,GAAG,KAAK,CAAC,iBAAiB,EAAE,aAAa,CAAC;IACvD,MAAM,SAAS,GAAG,KAAK,CAAC,iBAAiB,EAAE,eAAe,CAAC;IAC3D,MAAM,UAAU,GAAG,KAAK,CAAC,iBAAiB,EAAE,gBAAgB,CAAC;IAC7D,IAAI,OAAO,IAAI,SAAS,IAAI,UAAU,EAAE,CAAC;QACxC,MAAM,CAAC,iBAAiB,GAAG;YAC1B,GAAG,CAAC,OAAO,IAAI,EAAE,OAAO,EAAE,CAAC;YAC3B,GAAG,CAAC,SAAS,IAAI,EAAE,SAAS,EAAE,CAAC;YAC/B,GAAG,CAAC,UAAU,IAAI,EAAE,UAAU,EAAE,CAAC;SACjC,CAAC;IACH,CAAC;IAED,IAAI,KAAK,CAAC,kBAAkB,EAAE,eAAe,KAAK,SAAS,EAAE,CAAC;QAC7D,MAAM,CAAC,kBAAkB,GAAG,EAAE,SAAS,EAAE,KAAK,CAAC,kBAAkB,CAAC,eAAe,EAAE,CAAC;IACrF,CAAC;IAED,OAAO,MAAM,CAAC;AACf,CAAC;AAKD,SAAgB,YAAY,CAAC,CAA0B;IACtD,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;QAChB,KAAK,YAAY;YAChB,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;QAE/B,KAAK,aAAa;YACjB,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;QAEhC,KAAK,YAAY;YAChB,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC;QAEzC,KAAK,YAAY;YAChB,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC;QAE9D,KAAK,UAAU;YACd,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC;QAEvC,KAAK,iBAAiB;YACrB,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC;QAE9C,KAAK,iBAAiB;YACrB,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC;QAEnE,KAAK,eAAe;YACnB,OAAO,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC;QAE5C,KAAK,kBAAkB;YAEtB,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,UAAU,EAAE,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC;QAE7E,KAAK,kBAAkB;YACtB,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,UAAU,EAAE,CAAC,CAAC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC;QAEvE,KAAK,WAAW,CAAC,CAAC,CAAC;YAClB,IAAI,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,KAAK,EAAE,CAAC;gBAG1B,OAAO;oBACN,IAAI,EAAE,aAAa;oBACnB,UAAU,EAAE,CAAC,CAAC,UAAU;oBACxB,QAAQ,EAAE,CAAC,CAAC,QAAQ,IAAI,EAAE;oBAC1B,MAAM,EAAE,CAAC,CAAC,KAAK;oBACf,OAAO,EAAE,IAAI;iBACb,CAAC;YACH,CAAC;YACD,OAAO;gBACN,IAAI,EAAE,WAAW;gBACjB,UAAU,EAAE,CAAC,CAAC,UAAU;gBACxB,QAAQ,EAAE,CAAC,CAAC,QAAQ,IAAI,EAAE;gBAC1B,KAAK,EAAE,CAAC,CAAC,KAAkB;aAC3B,CAAC;QACH,CAAC;QAED,KAAK,aAAa;YAMjB,OAAO;gBACN,IAAI,EAAE,aAAa;gBACnB,UAAU,EAAE,CAAC,CAAC,UAAU,IAAI,EAAE;gBAC9B,QAAQ,EAAE,CAAC,CAAC,QAAQ,IAAI,EAAE;gBAC1B,MAAM,EAAE,CAAC,CAAC,MAAM;aAChB,CAAC;QAEH,KAAK,YAAY;YAIhB,OAAO;gBACN,IAAI,EAAE,aAAa;gBACnB,UAAU,EAAE,CAAC,CAAC,UAAU,IAAI,EAAE;gBAC9B,QAAQ,EAAE,CAAC,CAAC,QAAQ,IAAI,EAAE;gBAC1B,MAAM,EAAE,CAAC,CAAC,KAAK;gBACf,OAAO,EAAE,IAAI;aACb,CAAC;QAEH,KAAK,OAAO;YACX,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC;QAE1C,KAAK,QAAQ,CAAC,CAAC,CAAC;YACf,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;YACzC,OAAO;gBACN,IAAI,EAAE,QAAQ;gBACd,YAAY,EAAE,CAAC,CAAC,CAAC,YAAY,IAAI,MAAM,CAAiB;gBACxD,GAAG,CAAC,KAAK,IAAI,EAAE,KAAK,EAAE,CAAC;aACvB,CAAC;QACH,CAAC;QAED;YACC,OAAO,SAAS,CAAC;IACnB,CAAC;AACF,CAAC"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import type { TelemetrySettings } from 'ai';
|
|
2
|
+
import type { BuiltTelemetry } from '../../types';
|
|
3
|
+
import type { ExecutionOptions } from '../../types/sdk/agent';
|
|
4
|
+
import type { JSONValue } from '../../types/utils/json';
|
|
5
|
+
import type { AgentRuntimeConfig } from '../loop/agent-runtime';
|
|
6
|
+
export declare class RuntimeTelemetry {
|
|
7
|
+
private readonly config;
|
|
8
|
+
constructor(config: AgentRuntimeConfig);
|
|
9
|
+
resolve(options?: ExecutionOptions): BuiltTelemetry | undefined;
|
|
10
|
+
flush(options?: ExecutionOptions): Promise<void>;
|
|
11
|
+
buildTelemetryOptions(options?: ExecutionOptions): {
|
|
12
|
+
experimental_telemetry?: TelemetrySettings;
|
|
13
|
+
};
|
|
14
|
+
withRootSpan<T>(operation: 'generate' | 'stream', options: ExecutionOptions | undefined, runId: string, fn: () => Promise<T>): Promise<T>;
|
|
15
|
+
withToolSpan<T>(toolCallId: string, toolName: string, input: JSONValue, t: BuiltTelemetry | undefined, fn: () => Promise<T>): Promise<T>;
|
|
16
|
+
private buildTelemetryRootAttributes;
|
|
17
|
+
private buildTelemetryMetadataAttributes;
|
|
18
|
+
private buildAiSdkOperationAttributes;
|
|
19
|
+
}
|
|
@@ -0,0 +1,174 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.RuntimeTelemetry = void 0;
|
|
4
|
+
const zod_to_json_schema_1 = require("zod-to-json-schema");
|
|
5
|
+
const telemetry_1 = require("../../sdk/telemetry");
|
|
6
|
+
const zod_1 = require("../../utils/zod");
|
|
7
|
+
function isActiveSpanTracer(value) {
|
|
8
|
+
return (value !== null &&
|
|
9
|
+
typeof value === 'object' &&
|
|
10
|
+
typeof Reflect.get(value, 'startActiveSpan') === 'function');
|
|
11
|
+
}
|
|
12
|
+
function stringifyTelemetryValue(value) {
|
|
13
|
+
try {
|
|
14
|
+
return JSON.stringify(value);
|
|
15
|
+
}
|
|
16
|
+
catch {
|
|
17
|
+
return undefined;
|
|
18
|
+
}
|
|
19
|
+
}
|
|
20
|
+
function getToolInputSchema(tool) {
|
|
21
|
+
if (!tool.inputSchema) {
|
|
22
|
+
return undefined;
|
|
23
|
+
}
|
|
24
|
+
return (0, zod_1.isZodSchema)(tool.inputSchema) ? (0, zod_to_json_schema_1.zodToJsonSchema)(tool.inputSchema) : tool.inputSchema;
|
|
25
|
+
}
|
|
26
|
+
function summarizeToolForTelemetry(tool) {
|
|
27
|
+
return {
|
|
28
|
+
name: tool.name,
|
|
29
|
+
description: tool.description,
|
|
30
|
+
type: tool.mcpTool ? 'mcp' : 'local',
|
|
31
|
+
...(tool.mcpServerName ? { mcp_server: tool.mcpServerName } : {}),
|
|
32
|
+
...(tool.suspendSchema || tool.resumeSchema || tool.approval ? { approval: true } : {}),
|
|
33
|
+
...(tool.inputSchema ? { input_schema: getToolInputSchema(tool) } : {}),
|
|
34
|
+
};
|
|
35
|
+
}
|
|
36
|
+
function summarizeProviderToolForTelemetry(tool) {
|
|
37
|
+
const [provider] = tool.name.split('.');
|
|
38
|
+
return {
|
|
39
|
+
name: tool.name,
|
|
40
|
+
provider,
|
|
41
|
+
type: 'provider',
|
|
42
|
+
args: tool.args,
|
|
43
|
+
...(tool.inputSchema ? { input_schema: getToolInputSchema(tool) } : {}),
|
|
44
|
+
};
|
|
45
|
+
}
|
|
46
|
+
function buildAgentRootInputAttributes(config) {
|
|
47
|
+
const localTools = (config.tools ?? []).map(summarizeToolForTelemetry);
|
|
48
|
+
const providerTools = (config.providerTools ?? []).map(summarizeProviderToolForTelemetry);
|
|
49
|
+
const tools = [...localTools, ...providerTools];
|
|
50
|
+
const toolNames = tools
|
|
51
|
+
.map((tool) => (typeof tool.name === 'string' ? tool.name : undefined))
|
|
52
|
+
.filter((name) => name !== undefined);
|
|
53
|
+
const serialized = stringifyTelemetryValue({
|
|
54
|
+
agent: config.name,
|
|
55
|
+
tool_count: tools.length,
|
|
56
|
+
tools,
|
|
57
|
+
});
|
|
58
|
+
return {
|
|
59
|
+
...(toolNames.length > 0 ? { 'langsmith.metadata.available_tools': toolNames } : {}),
|
|
60
|
+
...(serialized ? { 'gen_ai.prompt': serialized } : {}),
|
|
61
|
+
};
|
|
62
|
+
}
|
|
63
|
+
class RuntimeTelemetry {
|
|
64
|
+
constructor(config) {
|
|
65
|
+
this.config = config;
|
|
66
|
+
}
|
|
67
|
+
resolve(options) {
|
|
68
|
+
if (this.config.telemetry)
|
|
69
|
+
return this.config.telemetry;
|
|
70
|
+
const inherited = options?.telemetry;
|
|
71
|
+
if (!inherited)
|
|
72
|
+
return undefined;
|
|
73
|
+
return { ...inherited, functionId: this.config.name };
|
|
74
|
+
}
|
|
75
|
+
async flush(options) {
|
|
76
|
+
await telemetry_1.Telemetry.forceFlush(this.resolve(options));
|
|
77
|
+
}
|
|
78
|
+
buildTelemetryOptions(options) {
|
|
79
|
+
const t = this.resolve(options);
|
|
80
|
+
if (!t?.enabled)
|
|
81
|
+
return {};
|
|
82
|
+
return {
|
|
83
|
+
experimental_telemetry: {
|
|
84
|
+
isEnabled: true,
|
|
85
|
+
functionId: t.functionId ?? this.config.name,
|
|
86
|
+
metadata: t.metadata,
|
|
87
|
+
recordInputs: t.recordInputs,
|
|
88
|
+
recordOutputs: t.recordOutputs,
|
|
89
|
+
tracer: t.tracer,
|
|
90
|
+
integrations: t.integrations.length > 0 ? t.integrations : undefined,
|
|
91
|
+
},
|
|
92
|
+
};
|
|
93
|
+
}
|
|
94
|
+
async withRootSpan(operation, options, runId, fn) {
|
|
95
|
+
const t = this.resolve(options);
|
|
96
|
+
if (!t?.enabled || t.runtimeRootSpanEnabled === false || !isActiveSpanTracer(t.tracer)) {
|
|
97
|
+
return await fn();
|
|
98
|
+
}
|
|
99
|
+
const spanName = `${t.functionId ?? this.config.name}.${operation}`;
|
|
100
|
+
return await t.tracer.startActiveSpan(spanName, { attributes: this.buildTelemetryRootAttributes(t, spanName, runId) }, async (span) => {
|
|
101
|
+
try {
|
|
102
|
+
return await fn();
|
|
103
|
+
}
|
|
104
|
+
catch (error) {
|
|
105
|
+
span.recordException?.(error);
|
|
106
|
+
span.setStatus?.({ code: 2, message: String(error) });
|
|
107
|
+
throw error;
|
|
108
|
+
}
|
|
109
|
+
finally {
|
|
110
|
+
span.end();
|
|
111
|
+
}
|
|
112
|
+
});
|
|
113
|
+
}
|
|
114
|
+
async withToolSpan(toolCallId, toolName, input, t, fn) {
|
|
115
|
+
if (!t?.enabled || !isActiveSpanTracer(t.tracer)) {
|
|
116
|
+
return await fn();
|
|
117
|
+
}
|
|
118
|
+
const shouldRecordInputs = t.recordInputs ?? true;
|
|
119
|
+
const inputValue = shouldRecordInputs ? stringifyTelemetryValue(input) : undefined;
|
|
120
|
+
return await t.tracer.startActiveSpan('ai.toolCall', {
|
|
121
|
+
attributes: {
|
|
122
|
+
...this.buildAiSdkOperationAttributes('ai.toolCall', t),
|
|
123
|
+
'ai.toolCall.name': toolName,
|
|
124
|
+
'ai.toolCall.id': toolCallId,
|
|
125
|
+
...(inputValue !== undefined ? { 'ai.toolCall.args': inputValue } : {}),
|
|
126
|
+
},
|
|
127
|
+
}, async (span) => {
|
|
128
|
+
try {
|
|
129
|
+
const result = await fn();
|
|
130
|
+
const shouldRecordOutputs = t.recordOutputs ?? true;
|
|
131
|
+
const outputValue = shouldRecordOutputs ? stringifyTelemetryValue(result) : undefined;
|
|
132
|
+
if (outputValue !== undefined) {
|
|
133
|
+
span.setAttributes?.({ 'ai.toolCall.result': outputValue });
|
|
134
|
+
}
|
|
135
|
+
return result;
|
|
136
|
+
}
|
|
137
|
+
catch (error) {
|
|
138
|
+
span.recordException?.(error);
|
|
139
|
+
span.setStatus?.({ code: 2, message: String(error) });
|
|
140
|
+
throw error;
|
|
141
|
+
}
|
|
142
|
+
finally {
|
|
143
|
+
span.end();
|
|
144
|
+
}
|
|
145
|
+
});
|
|
146
|
+
}
|
|
147
|
+
buildTelemetryRootAttributes(t, spanName, runId) {
|
|
148
|
+
const metadataAttributes = this.buildTelemetryMetadataAttributes(t, 'langsmith.metadata');
|
|
149
|
+
return {
|
|
150
|
+
'langsmith.traceable': 'true',
|
|
151
|
+
'langsmith.trace.name': spanName,
|
|
152
|
+
'langsmith.span.kind': 'chain',
|
|
153
|
+
'langsmith.metadata.agent_name': this.config.name,
|
|
154
|
+
'langsmith.metadata.agent_run_id': runId,
|
|
155
|
+
...metadataAttributes,
|
|
156
|
+
...buildAgentRootInputAttributes(this.config),
|
|
157
|
+
};
|
|
158
|
+
}
|
|
159
|
+
buildTelemetryMetadataAttributes(t, prefix) {
|
|
160
|
+
return Object.fromEntries(Object.entries(t.metadata ?? {}).map(([key, value]) => [`${prefix}.${key}`, value]));
|
|
161
|
+
}
|
|
162
|
+
buildAiSdkOperationAttributes(operationId, t) {
|
|
163
|
+
const functionId = t.functionId ?? this.config.name;
|
|
164
|
+
return {
|
|
165
|
+
'operation.name': `${operationId} ${functionId}`,
|
|
166
|
+
'resource.name': functionId,
|
|
167
|
+
'ai.operationId': operationId,
|
|
168
|
+
'ai.telemetry.functionId': functionId,
|
|
169
|
+
...this.buildTelemetryMetadataAttributes(t, 'ai.telemetry.metadata'),
|
|
170
|
+
};
|
|
171
|
+
}
|
|
172
|
+
}
|
|
173
|
+
exports.RuntimeTelemetry = RuntimeTelemetry;
|
|
174
|
+
//# sourceMappingURL=runtime-telemetry.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"runtime-telemetry.js","sourceRoot":"","sources":["../../../src/runtime/telemetry/runtime-telemetry.ts"],"names":[],"mappings":";;;AACA,2DAAqD;AAErD,mDAAgD;AAIhD,yCAA8C;AAkB9C,SAAS,kBAAkB,CAAC,KAAc;IACzC,OAAO,CACN,KAAK,KAAK,IAAI;QACd,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAO,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,iBAAiB,CAAC,KAAK,UAAU,CAC3D,CAAC;AACH,CAAC;AAED,SAAS,uBAAuB,CAAC,KAAc;IAC9C,IAAI,CAAC;QACJ,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IAC9B,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,SAAS,CAAC;IAClB,CAAC;AACF,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAmC;IAC9D,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;QACvB,OAAO,SAAS,CAAC;IAClB,CAAC;IAED,OAAO,IAAA,iBAAW,EAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAA,oCAAe,EAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC;AAC7F,CAAC;AAED,SAAS,yBAAyB,CAAC,IAAe;IACjD,OAAO;QACN,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,WAAW,EAAE,IAAI,CAAC,WAAW;QAC7B,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO;QACpC,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACjE,GAAG,CAAC,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACvF,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,kBAAkB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACvE,CAAC;AACH,CAAC;AAED,SAAS,iCAAiC,CAAC,IAAuB;IACjE,MAAM,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACxC,OAAO;QACN,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,QAAQ;QACR,IAAI,EAAE,UAAU;QAChB,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,kBAAkB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACvE,CAAC;AACH,CAAC;AAED,SAAS,6BAA6B,CAAC,MAA0B;IAChE,MAAM,UAAU,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;IACvE,MAAM,aAAa,GAAG,CAAC,MAAM,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;IAC1F,MAAM,KAAK,GAAG,CAAC,GAAG,UAAU,EAAE,GAAG,aAAa,CAAC,CAAC;IAChD,MAAM,SAAS,GAAG,KAAK;SACrB,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;SACtE,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;IAEvD,MAAM,UAAU,GAAG,uBAAuB,CAAC;QAC1C,KAAK,EAAE,MAAM,CAAC,IAAI;QAClB,UAAU,EAAE,KAAK,CAAC,MAAM;QACxB,KAAK;KACL,CAAC,CAAC;IAEH,OAAO;QACN,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,oCAAoC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACpF,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACtD,CAAC;AACH,CAAC;AAWD,MAAa,gBAAgB;IAC5B,YAA6B,MAA0B;QAA1B,WAAM,GAAN,MAAM,CAAoB;IAAG,CAAC;IAG3D,OAAO,CAAC,OAA0B;QACjC,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS;YAAE,OAAO,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;QACxD,MAAM,SAAS,GAAG,OAAO,EAAE,SAAS,CAAC;QACrC,IAAI,CAAC,SAAS;YAAE,OAAO,SAAS,CAAC;QACjC,OAAO,EAAE,GAAG,SAAS,EAAE,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;IACvD,CAAC;IAGD,KAAK,CAAC,KAAK,CAAC,OAA0B;QACrC,MAAM,qBAAS,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;IACnD,CAAC;IAGD,qBAAqB,CAAC,OAA0B;QAG/C,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAChC,IAAI,CAAC,CAAC,EAAE,OAAO;YAAE,OAAO,EAAE,CAAC;QAE3B,OAAO;YACN,sBAAsB,EAAE;gBACvB,SAAS,EAAE,IAAI;gBACf,UAAU,EAAE,CAAC,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI;gBAC5C,QAAQ,EAAE,CAAC,CAAC,QAAQ;gBACpB,YAAY,EAAE,CAAC,CAAC,YAAY;gBAC5B,aAAa,EAAE,CAAC,CAAC,aAAa;gBAE9B,MAAM,EAAE,CAAC,CAAC,MAAa;gBACvB,YAAY,EAAE,CAAC,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS;aACpE;SACD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CACjB,SAAgC,EAChC,OAAqC,EACrC,KAAa,EACb,EAAoB;QAEpB,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAChC,IAAI,CAAC,CAAC,EAAE,OAAO,IAAI,CAAC,CAAC,sBAAsB,KAAK,KAAK,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;YACxF,OAAO,MAAM,EAAE,EAAE,CAAC;QACnB,CAAC;QAED,MAAM,QAAQ,GAAG,GAAG,CAAC,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,SAAS,EAAE,CAAC;QACpE,OAAO,MAAM,CAAC,CAAC,MAAM,CAAC,eAAe,CACpC,QAAQ,EACR,EAAE,UAAU,EAAE,IAAI,CAAC,4BAA4B,CAAC,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,EAAE,EACrE,KAAK,EAAE,IAAI,EAAE,EAAE;YACd,IAAI,CAAC;gBACJ,OAAO,MAAM,EAAE,EAAE,CAAC;YACnB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAChB,IAAI,CAAC,eAAe,EAAE,CAAC,KAAK,CAAC,CAAC;gBAC9B,IAAI,CAAC,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBACtD,MAAM,KAAK,CAAC;YACb,CAAC;oBAAS,CAAC;gBACV,IAAI,CAAC,GAAG,EAAE,CAAC;YACZ,CAAC;QACF,CAAC,CACD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CACjB,UAAkB,EAClB,QAAgB,EAChB,KAAgB,EAChB,CAA6B,EAC7B,EAAoB;QAEpB,IAAI,CAAC,CAAC,EAAE,OAAO,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;YAClD,OAAO,MAAM,EAAE,EAAE,CAAC;QACnB,CAAC;QAED,MAAM,kBAAkB,GAAG,CAAC,CAAC,YAAY,IAAI,IAAI,CAAC;QAClD,MAAM,UAAU,GAAG,kBAAkB,CAAC,CAAC,CAAC,uBAAuB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAEnF,OAAO,MAAM,CAAC,CAAC,MAAM,CAAC,eAAe,CACpC,aAAa,EACb;YACC,UAAU,EAAE;gBACX,GAAG,IAAI,CAAC,6BAA6B,CAAC,aAAa,EAAE,CAAC,CAAC;gBACvD,kBAAkB,EAAE,QAAQ;gBAC5B,gBAAgB,EAAE,UAAU;gBAC5B,GAAG,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,kBAAkB,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACvE;SACD,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;YACd,IAAI,CAAC;gBACJ,MAAM,MAAM,GAAG,MAAM,EAAE,EAAE,CAAC;gBAC1B,MAAM,mBAAmB,GAAG,CAAC,CAAC,aAAa,IAAI,IAAI,CAAC;gBACpD,MAAM,WAAW,GAAG,mBAAmB,CAAC,CAAC,CAAC,uBAAuB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBACtF,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;oBAC/B,IAAI,CAAC,aAAa,EAAE,CAAC,EAAE,oBAAoB,EAAE,WAAW,EAAE,CAAC,CAAC;gBAC7D,CAAC;gBACD,OAAO,MAAM,CAAC;YACf,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAChB,IAAI,CAAC,eAAe,EAAE,CAAC,KAAK,CAAC,CAAC;gBAC9B,IAAI,CAAC,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBACtD,MAAM,KAAK,CAAC;YACb,CAAC;oBAAS,CAAC;gBACV,IAAI,CAAC,GAAG,EAAE,CAAC;YACZ,CAAC;QACF,CAAC,CACD,CAAC;IACH,CAAC;IAEO,4BAA4B,CACnC,CAAiB,EACjB,QAAgB,EAChB,KAAa;QAEb,MAAM,kBAAkB,GAAG,IAAI,CAAC,gCAAgC,CAAC,CAAC,EAAE,oBAAoB,CAAC,CAAC;QAE1F,OAAO;YACN,qBAAqB,EAAE,MAAM;YAC7B,sBAAsB,EAAE,QAAQ;YAChC,qBAAqB,EAAE,OAAO;YAC9B,+BAA+B,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;YACjD,iCAAiC,EAAE,KAAK;YACxC,GAAG,kBAAkB;YACrB,GAAG,6BAA6B,CAAC,IAAI,CAAC,MAAM,CAAC;SAC7C,CAAC;IACH,CAAC;IAEO,gCAAgC,CACvC,CAAiB,EACjB,MAAc;QAEd,OAAO,MAAM,CAAC,WAAW,CACxB,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,MAAM,IAAI,GAAG,EAAE,EAAE,KAAK,CAAC,CAAC,CACnF,CAAC;IACH,CAAC;IAEO,6BAA6B,CACpC,WAAmB,EACnB,CAAiB;QAEjB,MAAM,UAAU,GAAG,CAAC,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;QACpD,OAAO;YACN,gBAAgB,EAAE,GAAG,WAAW,IAAI,UAAU,EAAE;YAChD,eAAe,EAAE,UAAU;YAC3B,gBAAgB,EAAE,WAAW;YAC7B,yBAAyB,EAAE,UAAU;YACrC,GAAG,IAAI,CAAC,gCAAgC,CAAC,CAAC,EAAE,uBAAuB,CAAC;SACpE,CAAC;IACH,CAAC;CACD;AAtJD,4CAsJC"}
|
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import type { AgentDbMessage } from '
|
|
2
|
-
import type { BuiltTool } from '
|
|
1
|
+
import type { AgentDbMessage } from '../../types/sdk/message';
|
|
2
|
+
import type { BuiltTool } from '../../types/sdk/tool';
|
|
3
3
|
export declare const SEARCH_TOOLS_TOOL_NAME = "search_tools";
|
|
4
4
|
export declare const LOAD_TOOL_TOOL_NAME = "load_tool";
|
|
5
5
|
export interface DeferredToolManagerOptions {
|
|
@@ -1,6 +1,7 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.DeferredToolManager = exports.LOAD_TOOL_TOOL_NAME = exports.SEARCH_TOOLS_TOOL_NAME = void 0;
|
|
4
|
+
const utils_1 = require("@n8n/utils");
|
|
4
5
|
const zod_1 = require("zod");
|
|
5
6
|
exports.SEARCH_TOOLS_TOOL_NAME = 'search_tools';
|
|
6
7
|
exports.LOAD_TOOL_TOOL_NAME = 'load_tool';
|
|
@@ -26,9 +27,6 @@ const loadToolOutputSchema = zod_1.z.object({
|
|
|
26
27
|
candidates: zod_1.z.array(toolSummarySchema).optional(),
|
|
27
28
|
message: zod_1.z.string(),
|
|
28
29
|
});
|
|
29
|
-
function isRecord(value) {
|
|
30
|
-
return typeof value === 'object' && value !== null && !Array.isArray(value);
|
|
31
|
-
}
|
|
32
30
|
function tokenize(value) {
|
|
33
31
|
return new Set(value
|
|
34
32
|
.toLowerCase()
|
|
@@ -188,13 +186,13 @@ class DeferredToolManager {
|
|
|
188
186
|
};
|
|
189
187
|
}
|
|
190
188
|
isSuccessfulLoadToolCall(block) {
|
|
191
|
-
return (isRecord(block) &&
|
|
189
|
+
return ((0, utils_1.isRecord)(block) &&
|
|
192
190
|
block.type === 'tool-call' &&
|
|
193
191
|
block.toolName === exports.LOAD_TOOL_TOOL_NAME &&
|
|
194
192
|
block.state === 'resolved');
|
|
195
193
|
}
|
|
196
194
|
getLoadedToolNameFromOutput(output) {
|
|
197
|
-
if (!isRecord(output))
|
|
195
|
+
if (!(0, utils_1.isRecord)(output))
|
|
198
196
|
return undefined;
|
|
199
197
|
if (output.status !== 'loaded' && output.status !== 'already_loaded')
|
|
200
198
|
return undefined;
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"deferred-tool-manager.js","sourceRoot":"","sources":["../../../src/runtime/tools/deferred-tool-manager.ts"],"names":[],"mappings":";;;AAAA,sCAAsC;AACtC,6BAAwB;AAKX,QAAA,sBAAsB,GAAG,cAAc,CAAC;AACxC,QAAA,mBAAmB,GAAG,WAAW,CAAC;AAE/C,MAAM,aAAa,GAAG,CAAC,CAAC;AAExB,MAAM,sBAAsB,GAAG,OAAC,CAAC,MAAM,CAAC;IACvC,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,4DAA4D,CAAC;CAC/F,CAAC,CAAC;AAEH,MAAM,iBAAiB,GAAG,OAAC,CAAC,MAAM,CAAC;IAClC,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;IAChB,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE;IACvB,MAAM,EAAE,OAAC,CAAC,OAAO,EAAE;CACnB,CAAC,CAAC;AAEH,MAAM,uBAAuB,GAAG,OAAC,CAAC,MAAM,CAAC;IACxC,OAAO,EAAE,OAAC,CAAC,KAAK,CAAC,iBAAiB,CAAC;CACnC,CAAC,CAAC;AAEH,MAAM,mBAAmB,GAAG,OAAC,CAAC,MAAM,CAAC;IACpC,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,0CAA0C,CAAC;CAChF,CAAC,CAAC;AAEH,MAAM,oBAAoB,GAAG,OAAC,CAAC,MAAM,CAAC;IACrC,MAAM,EAAE,OAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,gBAAgB,EAAE,WAAW,CAAC,CAAC;IACzD,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE;IACpB,IAAI,EAAE,iBAAiB,CAAC,QAAQ,EAAE;IAClC,UAAU,EAAE,OAAC,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,QAAQ,EAAE;IACjD,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE;CACnB,CAAC,CAAC;AAMH,SAAS,QAAQ,CAAC,KAAa;IAC9B,OAAO,IAAI,GAAG,CACb,KAAK;SACH,WAAW,EAAE;SACb,KAAK,CAAC,kBAAkB,CAAC;SACzB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;SAC5B,MAAM,CAAC,OAAO,CAAC,CACjB,CAAC;AACH,CAAC;AAED,SAAS,SAAS,CAAC,IAAe,EAAE,KAAa;IAChD,MAAM,eAAe,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;IACnD,MAAM,WAAW,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IACpC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;IACrC,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC;IACnD,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;IAClC,MAAM,iBAAiB,GAAG,QAAQ,CAAC,WAAW,CAAC,CAAC;IAChD,IAAI,KAAK,GAAG,CAAC,CAAC;IAEd,IAAI,IAAI,KAAK,eAAe;QAAE,KAAK,IAAI,GAAG,CAAC;IAC3C,IAAI,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC;QAAE,KAAK,IAAI,EAAE,CAAC;IAChD,IAAI,WAAW,CAAC,QAAQ,CAAC,eAAe,CAAC;QAAE,KAAK,IAAI,EAAE,CAAC;IAEvD,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;QACjC,IAAI,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC;YAAE,KAAK,IAAI,EAAE,CAAC;QACvC,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,KAAK,IAAI,EAAE,CAAC;QACtC,IAAI,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC;YAAE,KAAK,IAAI,CAAC,CAAC;QAC7C,IAAI,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,KAAK,IAAI,CAAC,CAAC;IAC7C,CAAC;IAED,OAAO,KAAK,CAAC;AACd,CAAC;AAMD,MAAa,mBAAmB;IAW/B,YAAY,KAAkB,EAAE,UAAsC,EAAE;QAVvD,gBAAW,GAAG,IAAI,GAAG,EAAqB,CAAC;QAE3C,oBAAe,GAAG,IAAI,GAAG,EAAU,CAAC;QASpD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YAC1B,IAAI,IAAI,CAAC,IAAI,KAAK,8BAAsB,IAAI,IAAI,CAAC,IAAI,KAAK,2BAAmB,EAAE,CAAC;gBAC/E,MAAM,IAAI,KAAK,CAAC,uBAAuB,IAAI,CAAC,IAAI,eAAe,CAAC,CAAC;YAClE,CAAC;YACD,IAAI,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBACrC,MAAM,IAAI,KAAK,CAAC,iCAAiC,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC;YAChE,CAAC;YACD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QACvC,CAAC;QAED,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,aAAa,CAAC;QAC1C,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC1C,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;IACvC,CAAC;IAED,IAAI,QAAQ;QACX,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC,CAAC;IAClC,CAAC;IAED,IAAI,cAAc;QACjB,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;IAC9B,CAAC;IAED,IAAI,eAAe;QAClB,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC;IAClC,CAAC;IAED,kBAAkB;QACjB,IAAI,CAAC,IAAI,CAAC,QAAQ;YAAE,OAAO,EAAE,CAAC;QAC9B,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;IACzC,CAAC;IAED,cAAc;QACb,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC;aACrC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;aACzC,MAAM,CAAC,CAAC,IAAI,EAAqB,EAAE,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;IAC3D,CAAC;IAED,8BAA8B,CAAC,QAA0B;QACxD,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAE7B,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAChC,IAAI,CAAC,CAAC,SAAS,IAAI,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC;gBAAE,SAAS;YACzE,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBACrC,IAAI,CAAC,IAAI,CAAC,wBAAwB,CAAC,KAAK,CAAC;oBAAE,SAAS;gBACpD,MAAM,QAAQ,GAAG,IAAI,CAAC,2BAA2B,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAChE,IAAI,QAAQ,IAAI,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAChD,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;gBACpC,CAAC;YACF,CAAC;QACF,CAAC;IACF,CAAC;IAEO,gBAAgB;QACvB,OAAO;YACN,IAAI,EAAE,8BAAsB;YAC5B,WAAW,EACV,kGAAkG;YACnG,WAAW,EAAE,sBAAsB;YACnC,YAAY,EAAE,uBAAuB;YACrC,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;gBACxB,MAAM,EAAE,KAAK,EAAE,GAAG,sBAAsB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBACtD,OAAO,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YAClD,CAAC;SACD,CAAC;IACH,CAAC;IAEO,cAAc;QACrB,OAAO;YACN,IAAI,EAAE,2BAAmB;YACzB,WAAW,EACV,oIAAoI;YACrI,WAAW,EAAE,mBAAmB;YAChC,YAAY,EAAE,oBAAoB;YAClC,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;gBACxB,MAAM,EAAE,QAAQ,EAAE,GAAG,mBAAmB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBACtD,OAAO,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;YACnD,CAAC;SACD,CAAC;IACH,CAAC;IAEO,MAAM,CAAC,KAAa;QAC3B,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC;aAClD,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACf,IAAI;YACJ,KAAK,EAAE,SAAS,CAAC,IAAI,EAAE,KAAK,CAAC;SAC7B,CAAC,CAAC;aACF,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;YACrB,IAAI,IAAI,CAAC,KAAK,KAAK,KAAK,CAAC,KAAK;gBAAE,OAAO,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;YAChE,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEJ,MAAM,eAAe,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;QAChE,MAAM,OAAO,GAAG,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,MAAM,CAAC;QAEtE,OAAO;YACN,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;SAChF,CAAC;IACH,CAAC;IAEO,IAAI,CAAC,QAAgB;QAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,CAAC,IAAI,EAAE,CAAC;YACX,OAAO;gBACN,MAAM,EAAE,WAAW;gBACnB,QAAQ;gBACR,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO;gBACzC,OAAO,EAAE,SAAS,QAAQ,gEAAgE;aAC1F,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YACxC,OAAO;gBACN,MAAM,EAAE,gBAAgB;gBACxB,QAAQ;gBACR,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;gBAC9B,OAAO,EAAE,SAAS,QAAQ,sBAAsB;aAChD,CAAC;QACH,CAAC;QAED,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACnC,OAAO;YACN,MAAM,EAAE,QAAQ;YAChB,QAAQ;YACR,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;YAC9B,OAAO,EAAE,SAAS,QAAQ,2DAA2D;SACrF,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,IAAe;QACpC,OAAO;YACN,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;SAC3C,CAAC;IACH,CAAC;IAEO,wBAAwB,CAC/B,KAAc;QAEd,OAAO,CACN,IAAA,gBAAQ,EAAC,KAAK,CAAC;YACf,KAAK,CAAC,IAAI,KAAK,WAAW;YAC1B,KAAK,CAAC,QAAQ,KAAK,2BAAmB;YACtC,KAAK,CAAC,KAAK,KAAK,UAAU,CAC1B,CAAC;IACH,CAAC;IAEO,2BAA2B,CAAC,MAAe;QAClD,IAAI,CAAC,IAAA,gBAAQ,EAAC,MAAM,CAAC;YAAE,OAAO,SAAS,CAAC;QACxC,IAAI,MAAM,CAAC,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,MAAM,KAAK,gBAAgB;YAAE,OAAO,SAAS,CAAC;QACvF,OAAO,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;IAC1E,CAAC;CACD;AArKD,kDAqKC"}
|
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
import { z } from 'zod';
|
|
2
2
|
import { type SubAgentTaskPath, type SubAgentTaskPathPolicy } from './sub-agent-task-path';
|
|
3
|
-
import type { AgentExecutionCounter, FinishReason, GenerateResult, ModelConfig, TokenUsage } from '
|
|
4
|
-
import type { BuiltProviderTool, BuiltTool } from '
|
|
3
|
+
import type { AgentExecutionCounter, FinishReason, GenerateResult, ModelConfig, TokenUsage } from '../../types/sdk/agent';
|
|
4
|
+
import type { BuiltProviderTool, BuiltTool } from '../../types/sdk/tool';
|
|
5
5
|
export declare const DELEGATE_SUB_AGENT_TOOL_NAME = "delegate_subagent";
|
|
6
6
|
export declare const INLINE_SUB_AGENT_ID = "inline";
|
|
7
7
|
export declare const DELEGATED_CHILD_SUSPEND_UNSUPPORTED_MESSAGE = "agents.chat.delegate.childSuspendUnsupported";
|
|
@@ -9,8 +9,8 @@ exports.generateResultToDelegateSubAgentOutput = generateResultToDelegateSubAgen
|
|
|
9
9
|
const zod_1 = require("zod");
|
|
10
10
|
const sdk_owned_tool_1 = require("./sdk-owned-tool");
|
|
11
11
|
const sub_agent_task_path_1 = require("./sub-agent-task-path");
|
|
12
|
-
const message_1 = require("
|
|
13
|
-
const tool_1 = require("
|
|
12
|
+
const message_1 = require("../../sdk/message");
|
|
13
|
+
const tool_1 = require("../../sdk/tool");
|
|
14
14
|
exports.DELEGATE_SUB_AGENT_TOOL_NAME = 'delegate_subagent';
|
|
15
15
|
exports.INLINE_SUB_AGENT_ID = 'inline';
|
|
16
16
|
exports.DELEGATED_CHILD_SUSPEND_UNSUPPORTED_MESSAGE = 'agents.chat.delegate.childSuspendUnsupported';
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"delegate-sub-agent-tool.js","sourceRoot":"","sources":["../../../src/runtime/tools/delegate-sub-agent-tool.ts"],"names":[],"mappings":";;;AAoQA,gEAiEC;AAED,oFAMC;AAuLD,oEAiCC;AAeD,8EAWC;AAGD,wFA8BC;AAhmBD,6BAAwB;AAExB,qDAA+D;AAC/D,+DAK+B;AAC/B,+CAAsD;AACtD,yCAAsC;AAYzB,QAAA,4BAA4B,GAAG,mBAAmB,CAAC;AACnD,QAAA,mBAAmB,GAAG,QAAQ,CAAC;AAE/B,QAAA,2CAA2C,GACvD,8CAA8C,CAAC;AACnC,QAAA,2CAA2C,GAAG,wBAAwB,CAAC;AAEvE,QAAA,2BAA2B,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAU,CAAC;AAC9E,MAAM,4BAA4B,GAAG,OAAC,CAAC,IAAI,CAAC,mCAA2B,CAAC,CAAC;AAKzE,MAAM,2BAA2B,GAAG,OAAC,CAAC,MAAM,CAAC;IAC5C,UAAU,EAAE,OAAC;SACX,MAAM,EAAE;SACR,GAAG,CAAC,CAAC,CAAC;SACN,QAAQ,CACR,wIAAwI,CACxI;IACF,QAAQ,EAAE,OAAC;SACT,MAAM,EAAE;SACR,GAAG,CAAC,CAAC,CAAC;SACN,QAAQ,CAAC,yEAAyE,CAAC;IACrF,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,oDAAoD,CAAC;IACtF,OAAO,EAAE,OAAC;SACR,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CACR,wKAAwK,CACxK;IACF,cAAc,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,+CAA+C,CAAC;IAC/F,UAAU,EAAE,4BAA4B,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAC3D,8MAA8M,CAC9M;CACD,CAAC,CAAC;AAKH,MAAM,4BAA4B,GAAG,OAAC,CAAC,MAAM,CAAC;IAC7C,MAAM,EAAE,OAAC,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC;IACpD,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC/B,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC5B,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC/B,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC5B,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE;IAClB,gBAAgB,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;IACxC,KAAK,EAAE,OAAC;SACN,MAAM,CAAC;QACP,YAAY,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QACnC,gBAAgB,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QACvC,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QAClC,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;KAC3B,CAAC;SACD,QAAQ,EAAE;IACZ,YAAY,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACnC,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC5B,cAAc,EAAE,OAAC;SACf,KAAK,CACL,OAAC,CAAC,MAAM,CAAC;QACR,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE;QACjB,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE;QACtB,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE;QACpB,KAAK,EAAE,OAAC,CAAC,OAAO,EAAE;QAClB,cAAc,EAAE,OAAC,CAAC,OAAO,EAAE;QAC3B,YAAY,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;KACpC,CAAC,CACF;SACA,QAAQ,EAAE;CACZ,CAAC,CAAC;AAkJH,SAAS,6BAA6B,CACrC,MAA0C;IAE1C,MAAM,cAAc,GAAG;QACtB,GAAG,MAAM;QACT,WAAW,EAAE,MAAM,EAAE,WAAW,IAAI,oDAA8B;KAClE,CAAC;IAEF,IACC,CAAC,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,WAAW,CAAC;QAC5C,CAAC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,WAAW,CAAC,EAC5C,CAAC;QACF,MAAM,IAAI,KAAK,CAAC,wEAAwE,CAAC,CAAC;IAC3F,CAAC;IAED,IAAI,cAAc,CAAC,WAAW,GAAG,CAAC,EAAE,CAAC;QACpC,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;IAC5E,CAAC;IAED,OAAO,cAAc,CAAC;AACvB,CAAC;AAED,SAAgB,0BAA0B,CAAC,UAA6C,EAAE;IAEzF,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAkB,CAAC;IACnD,MAAM,eAAe,GAAsC;QAC1D,GAAG,OAAO;QACV,MAAM,EAAE,6BAA6B,CAAC,OAAO,CAAC,MAAM,CAAC;KACrD,CAAC;IACF,MAAM,6BAA6B,GAAG,eAAe,CAAC,kCAAkC;QACvF,CAAC,CAAC,mFAAmF;QACrF,CAAC,CAAC,wDAAwD,CAAC;IAE5D,MAAM,IAAI,GAAG,IAAI,WAAI,CAAC,oCAA4B,CAAC;SACjD,WAAW,CACX,gJAAgJ;QAC/I,gIAAgI;QAChI,0KAA0K,CAC3K;SACA,iBAAiB,CACjB;QACC,kQAAkQ,6BAA6B,qGAAqG;QACpY,oIAAoI;QACpI,GAAG,wBAAwB,CAAC,eAAe,CAAC,kBAAkB,CAAC;QAC/D,GAAG,kCAAkC,CAAC,eAAe,CAAC,MAAM,CAAC;QAC7D,iWAAiW;QACjW,yWAAyW;QACzW,6aAA6a,6BAA6B,8HAA8H;KACxkB,CAAC,IAAI,CAAC,IAAI,CAAC,CACZ;SACA,KAAK,CAAC,2BAA2B,CAAC;SAClC,MAAM,CAAC,4BAA4B,CAAC;SACpC,OAAO,CACP,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CACpB,MAAM,sBAAsB,CAAC,KAAK,EAAE,GAAG,EAAE,eAAe,EAAE,gBAAgB,CAAC,CAC5E;SACA,KAAK,EAAE,CAAC;IAEV,OAAO,IAAA,4CAA2B,EAAC;QAClC,GAAG,IAAI;QACP,QAAQ,EAAE;YACT,GAAG,IAAI,CAAC,QAAQ;YAChB,CAAC,mDAA2C,CAAC,EAAE;gBAC9C,GAAG,CAAC,eAAe,CAAC,kBAAkB,KAAK,SAAS;oBACnD,CAAC,CAAC,EAAE,kBAAkB,EAAE,eAAe,CAAC,kBAAkB,EAAE;oBAC5D,CAAC,CAAC,EAAE,CAAC;gBACN,MAAM,EAAE,eAAe,CAAC,MAAM;gBAC9B,GAAG,CAAC,eAAe,CAAC,0BAA0B,KAAK,SAAS;oBAC3D,CAAC,CAAC,EAAE,0BAA0B,EAAE,eAAe,CAAC,0BAA0B,EAAE;oBAC5E,CAAC,CAAC,EAAE,CAAC;gBACN,GAAG,CAAC,eAAe,CAAC,gCAAgC,KAAK,SAAS;oBACjE,CAAC,CAAC;wBACA,gCAAgC,EAAE,eAAe,CAAC,gCAAgC;qBAClF;oBACF,CAAC,CAAC,EAAE,CAAC;gBACN,GAAG,CAAC,eAAe,CAAC,kCAAkC,KAAK,SAAS;oBACnE,CAAC,CAAC;wBACA,kCAAkC,EACjC,eAAe,CAAC,kCAAkC;qBACnD;oBACF,CAAC,CAAC,EAAE,CAAC;gBACN,GAAG,CAAC,eAAe,CAAC,WAAW,KAAK,SAAS;oBAC5C,CAAC,CAAC,EAAE,WAAW,EAAE,eAAe,CAAC,WAAW,EAAE;oBAC9C,CAAC,CAAC,EAAE,CAAC;aACiC;SACxC;KACD,CAAC,CAAC;AACJ,CAAC;AAED,SAAgB,oCAAoC,CACnD,IAAe;IAEf,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC,mDAA2C,CAAC,CAAC;IAC3E,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI;QAAE,OAAO,SAAS,CAAC;IAClE,OAAO,KAAqC,CAAC;AAC9C,CAAC;AAED,SAAS,wBAAwB,CAChC,kBAA2E;IAE3E,IAAI,CAAC,kBAAkB,EAAE,MAAM;QAAE,OAAO,EAAE,CAAC;IAE3C,OAAO;QACN,yLAAyL;QACzL,GAAG,kBAAkB,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;YACtC,MAAM,WAAW,GAAG,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7E,OAAO,KAAK,QAAQ,CAAC,EAAE,KAAK,QAAQ,CAAC,IAAI,GAAG,WAAW,EAAE,CAAC;QAC3D,CAAC,CAAC;KACF,CAAC;AACH,CAAC;AAED,SAAS,kCAAkC,CAAC,MAA0C;IACrF,IAAI,MAAM,EAAE,WAAW,KAAK,SAAS;QAAE,OAAO,EAAE,CAAC;IAEjD,MAAM,QAAQ,GAAG,MAAM,CAAC,WAAW,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC;IAC3D,OAAO;QACN;YACC,yBAAyB;YACzB,WAAW,MAAM,CAAC,WAAW,oBAAoB,QAAQ,gCAAgC;YACzF,qEAAqE;YACrE,iIAAiI;SACjI,CAAC,IAAI,CAAC,IAAI,CAAC;KACZ,CAAC;AACH,CAAC;AAUD,KAAK,UAAU,sBAAsB,CACpC,KAA4B,EAC5B,GAAgB,EAChB,OAA0C,EAC1C,gBAAqC;IAErC,IAAI,QAAsC,CAAC;IAC3C,IAAI,OAA4C,CAAC;IACjD,IAAI,SAA6B,CAAC;IAClC,IAAI,CAAC;QACJ,MAAM,iBAAiB,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC;QACpD,MAAM,cAAc,GAAG,gBAAgB,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAEpE,QAAQ,GAAG,IAAA,iDAA2B,EAAC,KAAK,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;QACvE,gBAAgB,CAAC,GAAG,CAAC,iBAAiB,EAAE,cAAc,GAAG,CAAC,CAAC,CAAC;QAE5D,OAAO,GAAG;YACT,GAAG,KAAK;YACR,QAAQ;YACR,UAAU,EAAE,cAAc;YAC1B,GAAG,CAAC,GAAG,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC9D,GAAG,CAAC,GAAG,CAAC,WAAW,EAAE,QAAQ,KAAK,SAAS;gBAC1C,CAAC,CAAC,EAAE,cAAc,EAAE,GAAG,CAAC,WAAW,CAAC,QAAQ,EAAE;gBAC9C,CAAC,CAAC,EAAE,CAAC;YACN,GAAG,CAAC,GAAG,CAAC,WAAW,EAAE,UAAU,KAAK,SAAS;gBAC5C,CAAC,CAAC,EAAE,gBAAgB,EAAE,GAAG,CAAC,WAAW,CAAC,UAAU,EAAE;gBAClD,CAAC,CAAC,EAAE,CAAC;YACN,GAAG,CAAC,GAAG,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAChF,GAAG,CAAC,GAAG,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7E,GAAG,CAAC,GAAG,CAAC,gBAAgB,KAAK,SAAS;gBACrC,CAAC,CAAC,EAAE,sBAAsB,EAAE,GAAG,CAAC,gBAAgB,EAAE;gBAClD,CAAC,CAAC,EAAE,CAAC;YACN,GAAG,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACnE,CAAC;QAEF,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,mBAAmB,CAAC,GAAG,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CACd,wKAAwK,CACxK,CAAC;QACH,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,OAAO,EAAE;YACjD,iBAAiB,EAAE,GAAG,EAAE;gBACvB,MAAM,IAAI,KAAK,CACd,yHAAyH,CACzH,CAAC;YACH,CAAC;SACD,CAAC,CAAC;QACH,qBAAqB,CAAC,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;QACvD,OAAO,MAAM,CAAC;IACf,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,IAAI,OAAO,KAAK,SAAS,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YACtD,MAAM,YAAY,GAA+B;gBAChD,MAAM,EAAE,QAAQ;gBAChB,GAAG,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC/C,MAAM,EAAE,EAAE;gBACV,KAAK,EAAE,gBAAgB,CAAC,KAAK,CAAC;aAC9B,CAAC;YACF,qBAAqB,CAAC,GAAG,EAAE,OAAO,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;YAC7D,OAAO,YAAY,CAAC;QACrB,CAAC;QACD,OAAO;YACN,MAAM,EAAE,QAAQ;YAChB,GAAG,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC/C,MAAM,EAAE,EAAE;YACV,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAC7D,CAAC;IACH,CAAC;AACF,CAAC;AAED,SAAS,mBAAmB,CAC3B,GAAgB,EAChB,OAAgC,EAChC,SAAiB;IAEjB,GAAG,CAAC,SAAS,EAAE,CAAC;QACf,IAAI,oBAA4B;QAChC,GAAG,qBAAqB,CAAC,OAAO,CAAC;QACjC,SAAS;KACT,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAC7B,GAAgB,EAChB,OAAgC,EAChC,MAAkC,EAClC,SAAiB;IAEjB,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC9B,GAAG,CAAC,SAAS,EAAE,CAAC;QACf,IAAI,sBAA8B;QAClC,GAAG,qBAAqB,CAAC,OAAO,CAAC;QACjC,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,SAAS;QACT,UAAU;QACV,UAAU,EAAE,UAAU,GAAG,SAAS;QAClC,GAAG,CAAC,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC9D,GAAG,CAAC,MAAM,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACvE,GAAG,CAAC,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC9D,GAAG,CAAC,MAAM,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACnF,GAAG,CAAC,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC9D,GAAG,CAAC,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC9D,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,OAAgC;IAC9D,OAAO;QACN,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,GAAG,CAAC,OAAO,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAClF,GAAG,CAAC,OAAO,CAAC,gBAAgB,KAAK,SAAS;YACzC,CAAC,CAAC,EAAE,gBAAgB,EAAE,OAAO,CAAC,gBAAgB,EAAE;YAChD,CAAC,CAAC,EAAE,CAAC;QACN,GAAG,CAAC,OAAO,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC/E,CAAC;AACH,CAAC;AAED,SAAS,oBAAoB,CAAC,GAAgB;IAC7C,OAAO,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,WAAW,EAAE,QAAQ,IAAI,GAAG,CAAC,WAAW,EAAE,UAAU,IAAI,OAAO,CAAC;AACzF,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAc;IACvC,IAAI,KAAK,YAAY,KAAK;QAAE,OAAO,KAAK,CAAC,OAAO,CAAC;IACjD,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IAC5C,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QAC/E,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;IACtB,CAAC;IACD,IAAI,CAAC;QACJ,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IAC9B,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,eAAe,CAAC;IACxB,CAAC;AACF,CAAC;AAYD,SAAgB,4BAA4B,CAAC,OAI5C;IACA,MAAM,QAAQ,GAAG;QAChB,kEAAkE;QAClE,eAAe,OAAO,CAAC,IAAI,EAAE;KAC7B,CAAC;IAEF,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACrB,QAAQ,CAAC,IAAI,CAAC,aAAa,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IAC/C,CAAC;IAED,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;QAC5B,QAAQ,CAAC,IAAI,CAAC,qBAAqB,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;IAC9D,CAAC;IAED,QAAQ,CAAC,IAAI,CACZ;QACC,0GAA0G;QAC1G,gBAAgB;QAChB,kCAAkC;QAClC,6CAA6C;QAC7C,2CAA2C;QAC3C,EAAE;QACF,mJAAmJ;QACnJ,EAAE;QACF,wFAAwF;KACxF,CAAC,IAAI,CAAC,IAAI,CAAC,CACZ,CAAC;IAEF,OAAO,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAC9B,CAAC;AAED,SAAS,6BAA6B,CACrC,MAAsB;IAEtB,IAAI,MAAM,CAAC,YAAY,KAAK,OAAO,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QACnE,OAAO,QAAQ,CAAC;IACjB,CAAC;IACD,IAAI,MAAM,CAAC,cAAc,KAAK,SAAS,IAAI,MAAM,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7E,OAAO,WAAW,CAAC;IACpB,CAAC;IACD,OAAO,WAAW,CAAC;AACpB,CAAC;AAGD,SAAgB,iCAAiC,CAChD,QAA0B,EAC1B,KAAc;IAEd,OAAO;QACN,MAAM,EAAE,QAAQ;QAChB,QAAQ;QACR,MAAM,EAAE,EAAE;QACV,KAAK,EAAE,mDAA2C;QAClD,GAAG,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACzC,CAAC;AACH,CAAC;AAGD,SAAgB,sCAAsC,CACrD,QAA0B,EAC1B,MAAsB,EACtB,QAAiB;IAEjB,MAAM,MAAM,GAAG,6BAA6B,CAAC,MAAM,CAAC,CAAC;IACrD,OAAO;QACN,MAAM;QACN,QAAQ;QACR,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,GAAG,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/C,GAAG,CAAC,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC9D,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC;QACjC,GAAG,CAAC,MAAM,CAAC,gBAAgB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/F,GAAG,CAAC,MAAM,CAAC,KAAK,KAAK,SAAS;YAC7B,CAAC,CAAC;gBACA,KAAK,EAAE;oBACN,YAAY,EAAE,MAAM,CAAC,KAAK,CAAC,YAAY;oBACvC,gBAAgB,EAAE,MAAM,CAAC,KAAK,CAAC,gBAAgB;oBAC/C,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,WAAW;oBACrC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACvE;aACD;YACF,CAAC,CAAC,EAAE,CAAC;QACN,GAAG,CAAC,MAAM,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACnF,GAAG,CAAC,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,gBAAgB,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAChF,GAAG,CAAC,MAAM,KAAK,WAAW,IAAI,MAAM,CAAC,cAAc,KAAK,SAAS;YAChE,CAAC,CAAC,EAAE,cAAc,EAAE,MAAM,CAAC,cAAc,EAAE;YAC3C,CAAC,CAAC,EAAE,CAAC;KACN,CAAC;AACH,CAAC;AAGD,SAAS,QAAQ,CAAC,QAAwB;IACzC,MAAM,WAAW,GAAG,IAAA,2BAAiB,EAAC,QAAQ,CAAC,CAAC;IAChD,KAAK,IAAI,CAAC,GAAG,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAClD,MAAM,OAAO,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;QAC/B,IAAI,CAAC,OAAO;YAAE,SAAS;QAEvB,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO;aAC1B,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC;aAC5C,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;aAC9B,IAAI,CAAC,IAAI,CAAC;aACV,IAAI,EAAE,CAAC;QACT,IAAI,IAAI;YAAE,OAAO,IAAI,CAAC;IACvB,CAAC;IAED,OAAO,EAAE,CAAC;AACX,CAAC"}
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import type { ToolCall } from '@ai-sdk/provider-utils';
|
|
2
2
|
import type { InvalidToolInputError, NoSuchToolError, ToolCallRepairFunction, ToolSet } from 'ai';
|
|
3
|
-
import type { BuiltTool } from '
|
|
3
|
+
import type { BuiltTool } from '../../types';
|
|
4
4
|
export declare const fixToolCall: (options: {
|
|
5
5
|
toolCall: ToolCall<string, unknown>;
|
|
6
6
|
error: NoSuchToolError | InvalidToolInputError;
|
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.fixToolCall = void 0;
|
|
4
|
-
const zod_1 = require("
|
|
4
|
+
const zod_1 = require("../../utils/zod");
|
|
5
5
|
const isZodEmptyObject = (schema) => {
|
|
6
6
|
if (!(0, zod_1.isZodSchema)(schema))
|
|
7
7
|
return false;
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"fix-tool-call.js","sourceRoot":"","sources":["../../../src/runtime/tools/fix-tool-call.ts"],"names":[],"mappings":";;;AAIA,yCAA8C;AAE9C,MAAM,gBAAgB,GAAG,CAAC,MAAe,EAAE,EAAE;IAC5C,IAAI,CAAC,IAAA,iBAAW,EAAC,MAAM,CAAC;QAAE,OAAO,KAAK,CAAC;IACvC,MAAM,QAAQ,GAAI,MAAM,CAAC,IAA8B,CAAC,QAAQ,CAAC;IACjE,MAAM,KAAK,GAAG,OAAO,IAAI,MAAM,IAAI,QAAQ,KAAK,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;IACvF,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACpF,OAAO,IAAI,CAAC;IACb,CAAC;IACD,OAAO,KAAK,CAAC;AACd,CAAC,CAAC;AAEF,MAAM,iBAAiB,GAAG,CAAC,IAAe,EAAE,QAAmC,EAAE,EAAE;IAClF,MAAM,WAAW,GAAG,QAAQ,CAAC,KAAK,KAAK,MAAM,IAAI,QAAQ,CAAC,KAAK,KAAK,IAAI,CAAC;IACzE,MAAM,iBAAiB,GAAG,gBAAgB,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,WAAW,CAAC;IAC5E,OAAO,iBAAiB,IAAI,WAAW,CAAC;AACzC,CAAC,CAAC;AAEK,MAAM,WAAW,GAAG,KAAK,EAC/B,OAGC,EACD,OAA+B,EAEwB,EAAE;IACzD,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACpD,MAAM,uBAAuB,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,0BAA0B,CAAC;IAClF,IAAI,CAAC,IAAI,IAAI,CAAC,uBAAuB,IAAI,OAAO,CAAC,QAAQ,CAAC,gBAAgB;QAAE,OAAO,IAAI,CAAC;IAExF,IAAI,iBAAiB,CAAC,IAAI,EAAE,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC/C,OAAO;YACN,IAAI,EAAE,WAAoB;YAC1B,UAAU,EAAE,OAAO,CAAC,QAAQ,CAAC,UAAU;YACvC,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,QAAQ;YACnC,KAAK,EAAE,IAAI;YACX,gBAAgB,EAAE,KAAK;YACvB,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,OAAO;SACjC,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACb,CAAC,CAAC;AAvBW,QAAA,WAAW,eAuBtB"}
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import type { BuiltTool } from '
|
|
1
|
+
import type { BuiltTool } from '../../types/sdk/tool';
|
|
2
2
|
export declare const SDK_OWNED_BUILTIN_TOOL_METADATA_KEY = "sdkOwnedBuiltinTool";
|
|
3
3
|
export declare function isSdkOwnedBuiltInTool(tool: BuiltTool): boolean;
|
|
4
4
|
export declare function withSdkOwnedBuiltInMetadata(tool: BuiltTool): BuiltTool;
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sdk-owned-tool.js","sourceRoot":"","sources":["../../../src/runtime/tools/sdk-owned-tool.ts"],"names":[],"mappings":";;;AAIA,sDAEC;AAED,kEAQC;AAdY,QAAA,mCAAmC,GAAG,qBAAqB,CAAC;AAEzE,SAAgB,qBAAqB,CAAC,IAAe;IACpD,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC,2CAAmC,CAAC,KAAK,IAAI,CAAC;AACtE,CAAC;AAED,SAAgB,2BAA2B,CAAC,IAAe;IAC1D,OAAO;QACN,GAAG,IAAI;QACP,QAAQ,EAAE;YACT,GAAG,IAAI,CAAC,QAAQ;YAChB,CAAC,2CAAmC,CAAC,EAAE,IAAI;SAC3C;KACD,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sub-agent-task-path.js","sourceRoot":"","sources":["../../../src/runtime/tools/sub-agent-task-path.ts"],"names":[],"mappings":";;;AA6DA,4DAeC;AAGD,gDAEC;AAOD,wDAIC;AASD,kEAQC;AAtEY,QAAA,8BAA8B,GAAG,EAAE,CAAC;AAGpC,QAAA,wBAAwB,GAAG,OAAkC,CAAC;AAG3E,MAAM,oBAAoB,GAAG,EAAE,CAAC;AAEhC,MAAM,2BAA2B,GAAG,2BAA2B,CAAC;AAchE,SAAgB,wBAAwB,CAAC,QAAgB;IACxD,MAAM,SAAS,GAAG,QAAQ;SACxB,IAAI,EAAE;SACN,WAAW,EAAE;SACb,OAAO,CAAC,aAAa,EAAE,GAAG,CAAC;SAC3B,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC;SACnB,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC;SACvB,KAAK,CAAC,CAAC,EAAE,oBAAoB,CAAC;SAC9B,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IAEtB,IAAI,CAAC,SAAS,EAAE,CAAC;QAChB,MAAM,IAAI,KAAK,CAAC,sEAAsE,CAAC,CAAC;IACzF,CAAC;IAED,OAAO,SAAS,CAAC;AAClB,CAAC;AAGD,SAAgB,kBAAkB,CAAC,KAAa;IAC/C,OAAO,2BAA2B,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AAChD,CAAC;AAOD,SAAgB,sBAAsB,CAAC,KAAa;IACnD,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,EAAE,CAAC;QAChC,MAAM,IAAI,KAAK,CAAC,gCAAgC,KAAK,EAAE,CAAC,CAAC;IAC1D,CAAC;AACF,CAAC;AASD,SAAgB,2BAA2B,CAC1C,QAAgB,EAChB,UAAkB;IAElB,MAAM,SAAS,GAAG,GAAG,gCAAwB,IAAI,wBAAwB,CAAC,QAAQ,CAAC,IAAI,UAAU,EAAE,CAAC;IACpG,sBAAsB,CAAC,SAAS,CAAC,CAAC;IAElC,OAAO,SAAS,CAAC;AAClB,CAAC"}
|
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import type { Tool as AiSdkTool } from 'ai';
|
|
2
|
-
import { type BuiltProviderTool, type BuiltTool, type BuiltTelemetry, type ToolExecutionContext } from '
|
|
2
|
+
import { type BuiltProviderTool, type BuiltTool, type BuiltTelemetry, type ToolExecutionContext } from '../../types';
|
|
3
3
|
declare const SUSPEND_BRAND: unique symbol;
|
|
4
4
|
export interface SuspendedToolResult {
|
|
5
5
|
readonly [SUSPEND_BRAND]: true;
|