@zhin.js/agent 0.1.31 → 0.1.32
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 +42 -2
- package/lib/builtin/memory-search-tool.d.ts +2 -0
- package/lib/builtin/memory-search-tool.d.ts.map +1 -0
- package/lib/builtin/memory-search-tool.js +61 -0
- package/lib/builtin/memory-search-tool.js.map +1 -0
- package/lib/builtin/memory-upsert-tool.d.ts +2 -0
- package/lib/builtin/memory-upsert-tool.d.ts.map +1 -0
- package/lib/builtin/memory-upsert-tool.js +75 -0
- package/lib/builtin/memory-upsert-tool.js.map +1 -0
- package/lib/builtin/orchestration-tools.d.ts +7 -0
- package/lib/builtin/orchestration-tools.d.ts.map +1 -0
- package/lib/builtin/orchestration-tools.js +283 -0
- package/lib/builtin/orchestration-tools.js.map +1 -0
- package/lib/builtin/run-deferred-task-tool.d.ts +1 -1
- package/lib/builtin/run-deferred-task-tool.d.ts.map +1 -1
- package/lib/builtin/run-deferred-task-tool.js +1 -1
- package/lib/builtin/run-deferred-task-tool.js.map +1 -1
- package/lib/builtin/run-spec-dry-run-tool.d.ts +6 -0
- package/lib/builtin/run-spec-dry-run-tool.d.ts.map +1 -0
- package/lib/builtin/run-spec-dry-run-tool.js +51 -0
- package/lib/builtin/run-spec-dry-run-tool.js.map +1 -0
- package/lib/builtin/run-validation-spec-tool.d.ts +8 -0
- package/lib/builtin/run-validation-spec-tool.d.ts.map +1 -0
- package/lib/builtin/run-validation-spec-tool.js +44 -0
- package/lib/builtin/run-validation-spec-tool.js.map +1 -0
- package/lib/builtin/spawn-task-tool.d.ts +1 -1
- package/lib/builtin/spawn-task-tool.d.ts.map +1 -1
- package/lib/builtin/spawn-task-tool.js +53 -1
- package/lib/builtin/spawn-task-tool.js.map +1 -1
- package/lib/builtin-tools.d.ts +2 -0
- package/lib/builtin-tools.d.ts.map +1 -1
- package/lib/builtin-tools.js +6 -0
- package/lib/builtin-tools.js.map +1 -1
- package/lib/create-agent-session.d.ts +16 -0
- package/lib/create-agent-session.d.ts.map +1 -0
- package/lib/create-agent-session.js +15 -0
- package/lib/create-agent-session.js.map +1 -0
- package/lib/discovery/agents.d.ts +5 -0
- package/lib/discovery/agents.d.ts.map +1 -1
- package/lib/discovery/agents.js +7 -0
- package/lib/discovery/agents.js.map +1 -1
- package/lib/discovery/skills.d.ts +1 -1
- package/lib/discovery/skills.d.ts.map +1 -1
- package/lib/discovery/skills.js +2 -9
- package/lib/discovery/skills.js.map +1 -1
- package/lib/discovery/utils.d.ts +7 -1
- package/lib/discovery/utils.d.ts.map +1 -1
- package/lib/discovery/utils.js +66 -5
- package/lib/discovery/utils.js.map +1 -1
- package/lib/index.d.ts +17 -1
- package/lib/index.d.ts.map +1 -1
- package/lib/index.js +11 -1
- package/lib/index.js.map +1 -1
- package/lib/init/activate-ai-database-storage.d.ts.map +1 -1
- package/lib/init/activate-ai-database-storage.js +23 -1
- package/lib/init/activate-ai-database-storage.js.map +1 -1
- package/lib/init/create-zhin-agent.d.ts.map +1 -1
- package/lib/init/create-zhin-agent.js +69 -7
- package/lib/init/create-zhin-agent.js.map +1 -1
- package/lib/init/introspection-args.d.ts +34 -0
- package/lib/init/introspection-args.d.ts.map +1 -0
- package/lib/init/introspection-args.js +71 -0
- package/lib/init/introspection-args.js.map +1 -0
- package/lib/init/introspection-collectors.d.ts +21 -0
- package/lib/init/introspection-collectors.d.ts.map +1 -0
- package/lib/init/introspection-collectors.js +74 -0
- package/lib/init/introspection-collectors.js.map +1 -0
- package/lib/init/introspection-commands-format.d.ts +1 -1
- package/lib/init/introspection-commands-format.js +4 -4
- package/lib/init/introspection-commands-format.js.map +1 -1
- package/lib/init/introspection-output.d.ts +6 -0
- package/lib/init/introspection-output.d.ts.map +1 -0
- package/lib/init/introspection-output.js +52 -0
- package/lib/init/introspection-output.js.map +1 -0
- package/lib/init/introspection-pagination.d.ts +14 -0
- package/lib/init/introspection-pagination.d.ts.map +1 -0
- package/lib/init/introspection-pagination.js +33 -0
- package/lib/init/introspection-pagination.js.map +1 -0
- package/lib/init/introspection-rest.d.ts +24 -0
- package/lib/init/introspection-rest.d.ts.map +1 -0
- package/lib/init/introspection-rest.js +65 -0
- package/lib/init/introspection-rest.js.map +1 -0
- package/lib/init/register-agent-mesh-mcp.d.ts +2 -0
- package/lib/init/register-agent-mesh-mcp.d.ts.map +1 -0
- package/lib/init/register-agent-mesh-mcp.js +13 -0
- package/lib/init/register-agent-mesh-mcp.js.map +1 -0
- package/lib/init/register-ai-trigger.d.ts.map +1 -1
- package/lib/init/register-ai-trigger.js +29 -4
- package/lib/init/register-ai-trigger.js.map +1 -1
- package/lib/init/register-builtin-tools.d.ts.map +1 -1
- package/lib/init/register-builtin-tools.js +4 -0
- package/lib/init/register-builtin-tools.js.map +1 -1
- package/lib/init/register-db-models.d.ts.map +1 -1
- package/lib/init/register-db-models.js +5 -2
- package/lib/init/register-db-models.js.map +1 -1
- package/lib/init/register-db-upgrade.d.ts.map +1 -1
- package/lib/init/register-db-upgrade.js +20 -1
- package/lib/init/register-db-upgrade.js.map +1 -1
- package/lib/init/register-group-session-passive.d.ts +3 -0
- package/lib/init/register-group-session-passive.d.ts.map +1 -0
- package/lib/init/register-group-session-passive.js +55 -0
- package/lib/init/register-group-session-passive.js.map +1 -0
- package/lib/init/register-introspection-commands.d.ts.map +1 -1
- package/lib/init/register-introspection-commands.js +5 -6
- package/lib/init/register-introspection-commands.js.map +1 -1
- package/lib/init/register-management-tools.d.ts +1 -1
- package/lib/init/register-management-tools.d.ts.map +1 -1
- package/lib/init/register-management-tools.js +57 -36
- package/lib/init/register-management-tools.js.map +1 -1
- package/lib/init/upgrade-agent-db-schema.d.ts +41 -0
- package/lib/init/upgrade-agent-db-schema.d.ts.map +1 -0
- package/lib/init/upgrade-agent-db-schema.js +170 -0
- package/lib/init/upgrade-agent-db-schema.js.map +1 -0
- package/lib/init.d.ts.map +1 -1
- package/lib/init.js +4 -0
- package/lib/init.js.map +1 -1
- package/lib/mcp-server/agent-mesh-tools.d.ts +8 -0
- package/lib/mcp-server/agent-mesh-tools.d.ts.map +1 -0
- package/lib/mcp-server/agent-mesh-tools.js +138 -0
- package/lib/mcp-server/agent-mesh-tools.js.map +1 -0
- package/lib/media/deliver-subagent-result.d.ts +7 -0
- package/lib/media/deliver-subagent-result.d.ts.map +1 -1
- package/lib/media/deliver-subagent-result.js +3 -1
- package/lib/media/deliver-subagent-result.js.map +1 -1
- package/lib/media/media-normalize.d.ts +2 -0
- package/lib/media/media-normalize.d.ts.map +1 -1
- package/lib/media/media-normalize.js +89 -0
- package/lib/media/media-normalize.js.map +1 -1
- package/lib/memory-entry-registry.d.ts +7 -0
- package/lib/memory-entry-registry.d.ts.map +1 -0
- package/lib/memory-entry-registry.js +8 -0
- package/lib/memory-entry-registry.js.map +1 -0
- package/lib/orchestration-memory-hook.d.ts +3 -0
- package/lib/orchestration-memory-hook.d.ts.map +1 -0
- package/lib/orchestration-memory-hook.js +22 -0
- package/lib/orchestration-memory-hook.js.map +1 -0
- package/lib/orchestration-runtime-registry.d.ts +14 -0
- package/lib/orchestration-runtime-registry.d.ts.map +1 -0
- package/lib/orchestration-runtime-registry.js +15 -0
- package/lib/orchestration-runtime-registry.js.map +1 -0
- package/lib/orchestrator/agent-dispatcher.d.ts +36 -3
- package/lib/orchestrator/agent-dispatcher.d.ts.map +1 -1
- package/lib/orchestrator/agent-dispatcher.js +203 -13
- package/lib/orchestrator/agent-dispatcher.js.map +1 -1
- package/lib/orchestrator/delegation-processor.d.ts +33 -0
- package/lib/orchestrator/delegation-processor.d.ts.map +1 -0
- package/lib/orchestrator/delegation-processor.js +140 -0
- package/lib/orchestrator/delegation-processor.js.map +1 -0
- package/lib/orchestrator/mission-milestone-notify.d.ts +17 -0
- package/lib/orchestrator/mission-milestone-notify.d.ts.map +1 -0
- package/lib/orchestrator/mission-milestone-notify.js +49 -0
- package/lib/orchestrator/mission-milestone-notify.js.map +1 -0
- package/lib/orchestrator/mission-negotiation.d.ts +17 -0
- package/lib/orchestrator/mission-negotiation.d.ts.map +1 -0
- package/lib/orchestrator/mission-negotiation.js +89 -0
- package/lib/orchestrator/mission-negotiation.js.map +1 -0
- package/lib/orchestrator/mission-patch-acl.d.ts +9 -0
- package/lib/orchestrator/mission-patch-acl.d.ts.map +1 -0
- package/lib/orchestrator/mission-patch-acl.js +29 -0
- package/lib/orchestrator/mission-patch-acl.js.map +1 -0
- package/lib/orchestrator/mission-runner.d.ts +24 -0
- package/lib/orchestrator/mission-runner.d.ts.map +1 -0
- package/lib/orchestrator/mission-runner.js +179 -0
- package/lib/orchestrator/mission-runner.js.map +1 -0
- package/lib/orchestrator/mission-spec.d.ts +44 -0
- package/lib/orchestrator/mission-spec.d.ts.map +1 -0
- package/lib/orchestrator/mission-spec.js +121 -0
- package/lib/orchestrator/mission-spec.js.map +1 -0
- package/lib/orchestrator/mission-state.d.ts +48 -0
- package/lib/orchestrator/mission-state.d.ts.map +1 -0
- package/lib/orchestrator/mission-state.js +66 -0
- package/lib/orchestrator/mission-state.js.map +1 -0
- package/lib/orchestrator/orchestration-repository.d.ts +85 -0
- package/lib/orchestrator/orchestration-repository.d.ts.map +1 -0
- package/lib/orchestrator/orchestration-repository.js +325 -0
- package/lib/orchestrator/orchestration-repository.js.map +1 -0
- package/lib/orchestrator/orchestration-service.d.ts +58 -0
- package/lib/orchestrator/orchestration-service.d.ts.map +1 -0
- package/lib/orchestrator/orchestration-service.js +221 -0
- package/lib/orchestrator/orchestration-service.js.map +1 -0
- package/lib/orchestrator/remote-agent-registry.d.ts +32 -0
- package/lib/orchestrator/remote-agent-registry.d.ts.map +1 -0
- package/lib/orchestrator/remote-agent-registry.js +120 -0
- package/lib/orchestrator/remote-agent-registry.js.map +1 -0
- package/lib/orchestrator/remote-task-executor.d.ts +10 -0
- package/lib/orchestrator/remote-task-executor.d.ts.map +1 -0
- package/lib/orchestrator/remote-task-executor.js +142 -0
- package/lib/orchestrator/remote-task-executor.js.map +1 -0
- package/lib/orchestrator/remote-task-poller.d.ts +16 -0
- package/lib/orchestrator/remote-task-poller.d.ts.map +1 -0
- package/lib/orchestrator/remote-task-poller.js +75 -0
- package/lib/orchestrator/remote-task-poller.js.map +1 -0
- package/lib/plugin-loop-hooks.d.ts +32 -0
- package/lib/plugin-loop-hooks.d.ts.map +1 -0
- package/lib/plugin-loop-hooks.js +69 -0
- package/lib/plugin-loop-hooks.js.map +1 -0
- package/lib/service.d.ts +7 -0
- package/lib/service.d.ts.map +1 -1
- package/lib/service.js +13 -0
- package/lib/service.js.map +1 -1
- package/lib/session-tree-runtime-registry.d.ts +20 -0
- package/lib/session-tree-runtime-registry.d.ts.map +1 -0
- package/lib/session-tree-runtime-registry.js +25 -0
- package/lib/session-tree-runtime-registry.js.map +1 -0
- package/lib/subagent-artifact.d.ts +7 -0
- package/lib/subagent-artifact.d.ts.map +1 -0
- package/lib/subagent-artifact.js +26 -0
- package/lib/subagent-artifact.js.map +1 -0
- package/lib/subagent-parent-context.d.ts +9 -0
- package/lib/subagent-parent-context.d.ts.map +1 -0
- package/lib/subagent-parent-context.js +50 -0
- package/lib/subagent-parent-context.js.map +1 -0
- package/lib/subagent-preset.d.ts +10 -0
- package/lib/subagent-preset.d.ts.map +1 -0
- package/lib/subagent-preset.js +44 -0
- package/lib/subagent-preset.js.map +1 -0
- package/lib/subagent-prompt.d.ts +7 -0
- package/lib/subagent-prompt.d.ts.map +1 -0
- package/lib/subagent-prompt.js +14 -0
- package/lib/subagent-prompt.js.map +1 -0
- package/lib/subagent.d.ts +19 -0
- package/lib/subagent.d.ts.map +1 -1
- package/lib/subagent.js +134 -36
- package/lib/subagent.js.map +1 -1
- package/lib/typing-indicator/adapter-integration.d.ts +7 -3
- package/lib/typing-indicator/adapter-integration.d.ts.map +1 -1
- package/lib/typing-indicator/adapter-integration.js +30 -6
- package/lib/typing-indicator/adapter-integration.js.map +1 -1
- package/lib/typing-indicator/index.d.ts +26 -3
- package/lib/typing-indicator/index.d.ts.map +1 -1
- package/lib/typing-indicator/index.js +94 -6
- package/lib/typing-indicator/index.js.map +1 -1
- package/lib/typing-indicator/integration.d.ts.map +1 -1
- package/lib/typing-indicator/integration.js +7 -2
- package/lib/typing-indicator/integration.js.map +1 -1
- package/lib/zhin-agent/agent-loop-turn.d.ts +2 -1
- package/lib/zhin-agent/agent-loop-turn.d.ts.map +1 -1
- package/lib/zhin-agent/agent-loop-turn.js +45 -4
- package/lib/zhin-agent/agent-loop-turn.js.map +1 -1
- package/lib/zhin-agent/branch-summarization-runtime.d.ts +16 -0
- package/lib/zhin-agent/branch-summarization-runtime.d.ts.map +1 -0
- package/lib/zhin-agent/branch-summarization-runtime.js +58 -0
- package/lib/zhin-agent/branch-summarization-runtime.js.map +1 -0
- package/lib/zhin-agent/compaction-runtime.d.ts +25 -0
- package/lib/zhin-agent/compaction-runtime.d.ts.map +1 -0
- package/lib/zhin-agent/compaction-runtime.js +78 -0
- package/lib/zhin-agent/compaction-runtime.js.map +1 -0
- package/lib/zhin-agent/config.d.ts +17 -0
- package/lib/zhin-agent/config.d.ts.map +1 -1
- package/lib/zhin-agent/config.js +22 -0
- package/lib/zhin-agent/config.js.map +1 -1
- package/lib/zhin-agent/deferred-auto-continue.d.ts +11 -0
- package/lib/zhin-agent/deferred-auto-continue.d.ts.map +1 -0
- package/lib/zhin-agent/deferred-auto-continue.js +41 -0
- package/lib/zhin-agent/deferred-auto-continue.js.map +1 -0
- package/lib/zhin-agent/deferred-delivery.d.ts +4 -0
- package/lib/zhin-agent/deferred-delivery.d.ts.map +1 -1
- package/lib/zhin-agent/deferred-delivery.js +7 -1
- package/lib/zhin-agent/deferred-delivery.js.map +1 -1
- package/lib/zhin-agent/index.d.ts +20 -3
- package/lib/zhin-agent/index.d.ts.map +1 -1
- package/lib/zhin-agent/index.js +103 -6
- package/lib/zhin-agent/index.js.map +1 -1
- package/lib/zhin-agent/persist-deferred-context.d.ts +5 -0
- package/lib/zhin-agent/persist-deferred-context.d.ts.map +1 -0
- package/lib/zhin-agent/persist-deferred-context.js +40 -0
- package/lib/zhin-agent/persist-deferred-context.js.map +1 -0
- package/lib/zhin-agent/prompt-builder.d.ts.map +1 -1
- package/lib/zhin-agent/prompt-builder.js +5 -0
- package/lib/zhin-agent/prompt-builder.js.map +1 -1
- package/lib/zhin-agent/prompt.d.ts.map +1 -1
- package/lib/zhin-agent/prompt.js +4 -0
- package/lib/zhin-agent/prompt.js.map +1 -1
- package/lib/zhin-agent/session-io.d.ts +16 -4
- package/lib/zhin-agent/session-io.d.ts.map +1 -1
- package/lib/zhin-agent/session-io.js +46 -13
- package/lib/zhin-agent/session-io.js.map +1 -1
- package/lib/zhin-agent/session-tree-commands.d.ts +5 -0
- package/lib/zhin-agent/session-tree-commands.d.ts.map +1 -0
- package/lib/zhin-agent/session-tree-commands.js +44 -0
- package/lib/zhin-agent/session-tree-commands.js.map +1 -0
- package/lib/zhin-agent/tool-orchestration.d.ts.map +1 -1
- package/lib/zhin-agent/tool-orchestration.js +13 -24
- package/lib/zhin-agent/tool-orchestration.js.map +1 -1
- package/lib/zhin-agent/tool-search-orchestrator.d.ts.map +1 -1
- package/lib/zhin-agent/tool-search-orchestrator.js +13 -0
- package/lib/zhin-agent/tool-search-orchestrator.js.map +1 -1
- package/lib/zhin-agent/turn-pipeline.d.ts +2 -0
- package/lib/zhin-agent/turn-pipeline.d.ts.map +1 -1
- package/lib/zhin-agent/turn-pipeline.js +30 -21
- package/lib/zhin-agent/turn-pipeline.js.map +1 -1
- package/lib/zhin-agent/turn-user-message.d.ts +8 -0
- package/lib/zhin-agent/turn-user-message.d.ts.map +1 -0
- package/lib/zhin-agent/turn-user-message.js +10 -0
- package/lib/zhin-agent/turn-user-message.js.map +1 -0
- package/lib/zhin-agent/zhin-agent-private.d.ts +6 -0
- package/lib/zhin-agent/zhin-agent-private.d.ts.map +1 -1
- package/lib/zhin-agent/zhin-agent-private.js.map +1 -1
- package/package.json +4 -3
|
@@ -0,0 +1,32 @@
|
|
|
1
|
+
import type { AIConfig } from '@zhin.js/core';
|
|
2
|
+
import { McpClientConnection } from '../mcp-client/connection.js';
|
|
3
|
+
import type { McpServerEntry } from './types.js';
|
|
4
|
+
export interface RemoteAgentEntry {
|
|
5
|
+
id: string;
|
|
6
|
+
name: string;
|
|
7
|
+
url: string;
|
|
8
|
+
token?: string;
|
|
9
|
+
roles: string[];
|
|
10
|
+
description: string;
|
|
11
|
+
}
|
|
12
|
+
export interface RemoteAgentHealth {
|
|
13
|
+
id: string;
|
|
14
|
+
healthy: boolean;
|
|
15
|
+
error?: string;
|
|
16
|
+
}
|
|
17
|
+
export declare class RemoteAgentRegistry {
|
|
18
|
+
private entries;
|
|
19
|
+
private health;
|
|
20
|
+
private connections;
|
|
21
|
+
loadFromConfig(config: AIConfig | undefined): void;
|
|
22
|
+
list(): RemoteAgentEntry[];
|
|
23
|
+
get(id: string): RemoteAgentEntry | undefined;
|
|
24
|
+
getHealth(id: string): RemoteAgentHealth | undefined;
|
|
25
|
+
toMcpEntry(agent: RemoteAgentEntry): McpServerEntry;
|
|
26
|
+
getConnection(agentId: string): Promise<McpClientConnection>;
|
|
27
|
+
healthCheckAll(): Promise<RemoteAgentHealth[]>;
|
|
28
|
+
pingAgent(agentId: string): Promise<RemoteAgentHealth>;
|
|
29
|
+
}
|
|
30
|
+
export declare function getRemoteAgentRegistry(): RemoteAgentRegistry;
|
|
31
|
+
export declare function initRemoteAgentRegistry(config: AIConfig | undefined): RemoteAgentRegistry;
|
|
32
|
+
//# sourceMappingURL=remote-agent-registry.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"remote-agent-registry.d.ts","sourceRoot":"","sources":["../../src/orchestrator/remote-agent-registry.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAE9C,OAAO,EAAE,mBAAmB,EAAE,MAAM,6BAA6B,CAAC;AAClE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAIjD,MAAM,WAAW,gBAAgB;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,iBAAiB;IAChC,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAsBD,qBAAa,mBAAmB;IAC9B,OAAO,CAAC,OAAO,CAAuC;IACtD,OAAO,CAAC,MAAM,CAAwC;IACtD,OAAO,CAAC,WAAW,CAA0C;IAE7D,cAAc,CAAC,MAAM,EAAE,QAAQ,GAAG,SAAS,GAAG,IAAI;IAOlD,IAAI,IAAI,gBAAgB,EAAE;IAI1B,GAAG,CAAC,EAAE,EAAE,MAAM,GAAG,gBAAgB,GAAG,SAAS;IAI7C,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,iBAAiB,GAAG,SAAS;IAIpD,UAAU,CAAC,KAAK,EAAE,gBAAgB,GAAG,cAAc;IAa7C,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAc5D,cAAc,IAAI,OAAO,CAAC,iBAAiB,EAAE,CAAC;IAS9C,SAAS,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,CAAC;CAwB7D;AAID,wBAAgB,sBAAsB,IAAI,mBAAmB,CAK5D;AAED,wBAAgB,uBAAuB,CAAC,MAAM,EAAE,QAAQ,GAAG,SAAS,GAAG,mBAAmB,CAIzF"}
|
|
@@ -0,0 +1,120 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* RemoteAgentRegistry — static ai.remoteAgents + health check (Agent Mesh v1).
|
|
3
|
+
*/
|
|
4
|
+
import { Logger } from '@zhin.js/logger';
|
|
5
|
+
import { resolveConfigEnvString } from '../utils/config-env.js';
|
|
6
|
+
import { McpClientConnection } from '../mcp-client/connection.js';
|
|
7
|
+
const logger = new Logger(null, 'RemoteAgentRegistry');
|
|
8
|
+
function parseRemoteAgents(config) {
|
|
9
|
+
const raw = config?.remoteAgents;
|
|
10
|
+
if (!raw?.length)
|
|
11
|
+
return [];
|
|
12
|
+
const out = [];
|
|
13
|
+
for (const item of raw) {
|
|
14
|
+
if (!item?.id?.trim() || !item.url?.trim())
|
|
15
|
+
continue;
|
|
16
|
+
const url = resolveConfigEnvString(item.url) ?? item.url;
|
|
17
|
+
const token = item.token ? (resolveConfigEnvString(item.token) ?? item.token) : undefined;
|
|
18
|
+
out.push({
|
|
19
|
+
id: item.id.trim(),
|
|
20
|
+
name: item.name?.trim() || item.id.trim(),
|
|
21
|
+
url: url.trim(),
|
|
22
|
+
token,
|
|
23
|
+
roles: Array.isArray(item.roles) ? item.roles.map(String) : [],
|
|
24
|
+
description: item.description?.trim() || '',
|
|
25
|
+
});
|
|
26
|
+
}
|
|
27
|
+
return out;
|
|
28
|
+
}
|
|
29
|
+
export class RemoteAgentRegistry {
|
|
30
|
+
entries = new Map();
|
|
31
|
+
health = new Map();
|
|
32
|
+
connections = new Map();
|
|
33
|
+
loadFromConfig(config) {
|
|
34
|
+
this.entries.clear();
|
|
35
|
+
for (const entry of parseRemoteAgents(config)) {
|
|
36
|
+
this.entries.set(entry.id, entry);
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
list() {
|
|
40
|
+
return [...this.entries.values()];
|
|
41
|
+
}
|
|
42
|
+
get(id) {
|
|
43
|
+
return this.entries.get(id);
|
|
44
|
+
}
|
|
45
|
+
getHealth(id) {
|
|
46
|
+
return this.health.get(id);
|
|
47
|
+
}
|
|
48
|
+
toMcpEntry(agent) {
|
|
49
|
+
const headers = {};
|
|
50
|
+
if (agent.token) {
|
|
51
|
+
headers.Authorization = `Bearer ${agent.token}`;
|
|
52
|
+
}
|
|
53
|
+
return {
|
|
54
|
+
name: `remote-agent:${agent.id}`,
|
|
55
|
+
transport: 'streamable-http',
|
|
56
|
+
url: agent.url,
|
|
57
|
+
headers,
|
|
58
|
+
};
|
|
59
|
+
}
|
|
60
|
+
async getConnection(agentId) {
|
|
61
|
+
const agent = this.entries.get(agentId);
|
|
62
|
+
if (!agent)
|
|
63
|
+
throw new Error(`Remote agent "${agentId}" not registered`);
|
|
64
|
+
let conn = this.connections.get(agentId);
|
|
65
|
+
if (!conn) {
|
|
66
|
+
conn = new McpClientConnection(this.toMcpEntry(agent));
|
|
67
|
+
this.connections.set(agentId, conn);
|
|
68
|
+
}
|
|
69
|
+
if (!conn.isConnected) {
|
|
70
|
+
await conn.connect();
|
|
71
|
+
}
|
|
72
|
+
return conn;
|
|
73
|
+
}
|
|
74
|
+
async healthCheckAll() {
|
|
75
|
+
const results = [];
|
|
76
|
+
for (const agent of this.entries.values()) {
|
|
77
|
+
const health = await this.pingAgent(agent.id);
|
|
78
|
+
results.push(health);
|
|
79
|
+
}
|
|
80
|
+
return results;
|
|
81
|
+
}
|
|
82
|
+
async pingAgent(agentId) {
|
|
83
|
+
const agent = this.entries.get(agentId);
|
|
84
|
+
if (!agent) {
|
|
85
|
+
const h = { id: agentId, healthy: false, error: 'not registered' };
|
|
86
|
+
this.health.set(agentId, h);
|
|
87
|
+
return h;
|
|
88
|
+
}
|
|
89
|
+
try {
|
|
90
|
+
const conn = await this.getConnection(agentId);
|
|
91
|
+
await conn.ping();
|
|
92
|
+
const h = { id: agentId, healthy: true };
|
|
93
|
+
this.health.set(agentId, h);
|
|
94
|
+
return h;
|
|
95
|
+
}
|
|
96
|
+
catch (err) {
|
|
97
|
+
const h = {
|
|
98
|
+
id: agentId,
|
|
99
|
+
healthy: false,
|
|
100
|
+
error: err instanceof Error ? err.message : String(err),
|
|
101
|
+
};
|
|
102
|
+
this.health.set(agentId, h);
|
|
103
|
+
logger.debug(`Remote agent ${agentId} health check failed:`, err);
|
|
104
|
+
return h;
|
|
105
|
+
}
|
|
106
|
+
}
|
|
107
|
+
}
|
|
108
|
+
let globalRegistry = null;
|
|
109
|
+
export function getRemoteAgentRegistry() {
|
|
110
|
+
if (!globalRegistry) {
|
|
111
|
+
globalRegistry = new RemoteAgentRegistry();
|
|
112
|
+
}
|
|
113
|
+
return globalRegistry;
|
|
114
|
+
}
|
|
115
|
+
export function initRemoteAgentRegistry(config) {
|
|
116
|
+
const registry = getRemoteAgentRegistry();
|
|
117
|
+
registry.loadFromConfig(config);
|
|
118
|
+
return registry;
|
|
119
|
+
}
|
|
120
|
+
//# sourceMappingURL=remote-agent-registry.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"remote-agent-registry.js","sourceRoot":"","sources":["../../src/orchestrator/remote-agent-registry.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AAEzC,OAAO,EAAE,sBAAsB,EAAE,MAAM,wBAAwB,CAAC;AAChE,OAAO,EAAE,mBAAmB,EAAE,MAAM,6BAA6B,CAAC;AAGlE,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,EAAE,qBAAqB,CAAC,CAAC;AAiBvD,SAAS,iBAAiB,CAAC,MAA4B;IACrD,MAAM,GAAG,GAAG,MAAM,EAAE,YAAY,CAAC;IACjC,IAAI,CAAC,GAAG,EAAE,MAAM;QAAE,OAAO,EAAE,CAAC;IAC5B,MAAM,GAAG,GAAuB,EAAE,CAAC;IACnC,KAAK,MAAM,IAAI,IAAI,GAAG,EAAE,CAAC;QACvB,IAAI,CAAC,IAAI,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,EAAE;YAAE,SAAS;QACrD,MAAM,GAAG,GAAG,sBAAsB,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC;QACzD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC1F,GAAG,CAAC,IAAI,CAAC;YACP,EAAE,EAAE,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE;YAClB,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE;YACzC,GAAG,EAAE,GAAG,CAAC,IAAI,EAAE;YACf,KAAK;YACL,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE;YAC9D,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE;SAC5C,CAAC,CAAC;IACL,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,OAAO,mBAAmB;IACtB,OAAO,GAAG,IAAI,GAAG,EAA4B,CAAC;IAC9C,MAAM,GAAG,IAAI,GAAG,EAA6B,CAAC;IAC9C,WAAW,GAAG,IAAI,GAAG,EAA+B,CAAC;IAE7D,cAAc,CAAC,MAA4B;QACzC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,KAAK,MAAM,KAAK,IAAI,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAED,IAAI;QACF,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IACpC,CAAC;IAED,GAAG,CAAC,EAAU;QACZ,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC9B,CAAC;IAED,SAAS,CAAC,EAAU;QAClB,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC;IAED,UAAU,CAAC,KAAuB;QAChC,MAAM,OAAO,GAA2B,EAAE,CAAC;QAC3C,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,aAAa,GAAG,UAAU,KAAK,CAAC,KAAK,EAAE,CAAC;QAClD,CAAC;QACD,OAAO;YACL,IAAI,EAAE,gBAAgB,KAAK,CAAC,EAAE,EAAE;YAChC,SAAS,EAAE,iBAAiB;YAC5B,GAAG,EAAE,KAAK,CAAC,GAAG;YACd,OAAO;SACR,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,OAAe;QACjC,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACxC,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,iBAAiB,OAAO,kBAAkB,CAAC,CAAC;QACxE,IAAI,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACzC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,IAAI,GAAG,IAAI,mBAAmB,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC;YACvD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACtC,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACvB,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,cAAc;QAClB,MAAM,OAAO,GAAwB,EAAE,CAAC;QACxC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC1C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAC9C,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvB,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,OAAe;QAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACxC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,CAAC,GAAG,EAAE,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,gBAAgB,EAAE,CAAC;YACnE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAC5B,OAAO,CAAC,CAAC;QACX,CAAC;QACD,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;YAC/C,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,MAAM,CAAC,GAAG,EAAE,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YACzC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAC5B,OAAO,CAAC,CAAC;QACX,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,CAAC,GAAG;gBACR,EAAE,EAAE,OAAO;gBACX,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;aACxD,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAC5B,MAAM,CAAC,KAAK,CAAC,gBAAgB,OAAO,uBAAuB,EAAE,GAAG,CAAC,CAAC;YAClE,OAAO,CAAC,CAAC;QACX,CAAC;IACH,CAAC;CACF;AAED,IAAI,cAAc,GAA+B,IAAI,CAAC;AAEtD,MAAM,UAAU,sBAAsB;IACpC,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,cAAc,GAAG,IAAI,mBAAmB,EAAE,CAAC;IAC7C,CAAC;IACD,OAAO,cAAc,CAAC;AACxB,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,MAA4B;IAClE,MAAM,QAAQ,GAAG,sBAAsB,EAAE,CAAC;IAC1C,QAAQ,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;IAChC,OAAO,QAAQ,CAAC;AAClB,CAAC"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
export declare function executeRemoteOrchestrationTask(taskId: string): Promise<{
|
|
2
|
+
ok: boolean;
|
|
3
|
+
message: string;
|
|
4
|
+
}>;
|
|
5
|
+
export declare function pollRemoteTaskStatus(taskId: string): Promise<{
|
|
6
|
+
done: boolean;
|
|
7
|
+
status: string;
|
|
8
|
+
result?: string;
|
|
9
|
+
}>;
|
|
10
|
+
//# sourceMappingURL=remote-task-executor.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"remote-task-executor.d.ts","sourceRoot":"","sources":["../../src/orchestrator/remote-task-executor.ts"],"names":[],"mappings":"AAmBA,wBAAsB,8BAA8B,CAClD,MAAM,EAAE,MAAM,GACb,OAAO,CAAC;IAAE,EAAE,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,CAAC,CAwE3C;AAED,wBAAsB,oBAAoB,CACxC,MAAM,EAAE,MAAM,GACb,OAAO,CAAC;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,MAAM,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CAkD7D"}
|
|
@@ -0,0 +1,142 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Remote task execution via MCP agent.delegate_task (Agent Mesh v1).
|
|
3
|
+
*/
|
|
4
|
+
import { getAgentDispatcher } from './agent-dispatcher.js';
|
|
5
|
+
import { getRemoteAgentRegistry } from './remote-agent-registry.js';
|
|
6
|
+
import { getOrchestrationService } from './orchestration-service.js';
|
|
7
|
+
function extractMcpText(result) {
|
|
8
|
+
if (typeof result === 'string')
|
|
9
|
+
return result;
|
|
10
|
+
const content = result?.content;
|
|
11
|
+
if (Array.isArray(content)) {
|
|
12
|
+
return content
|
|
13
|
+
.filter((c) => c.type === 'text' && c.text)
|
|
14
|
+
.map((c) => c.text)
|
|
15
|
+
.join('\n');
|
|
16
|
+
}
|
|
17
|
+
return JSON.stringify(result);
|
|
18
|
+
}
|
|
19
|
+
export async function executeRemoteOrchestrationTask(taskId) {
|
|
20
|
+
const dispatcher = getAgentDispatcher();
|
|
21
|
+
const orch = getOrchestrationService();
|
|
22
|
+
const task = dispatcher.getTask(taskId);
|
|
23
|
+
if (!task) {
|
|
24
|
+
return { ok: false, message: `任务 ${taskId} 不存在` };
|
|
25
|
+
}
|
|
26
|
+
if (task.executorKind !== 'remote' || !task.remoteAgentId) {
|
|
27
|
+
return { ok: false, message: `任务 ${taskId} 不是远程执行任务` };
|
|
28
|
+
}
|
|
29
|
+
const registry = getRemoteAgentRegistry();
|
|
30
|
+
const agent = registry.get(task.remoteAgentId);
|
|
31
|
+
if (!agent) {
|
|
32
|
+
return { ok: false, message: `远程 Agent ${task.remoteAgentId} 未注册` };
|
|
33
|
+
}
|
|
34
|
+
const promise = (async () => {
|
|
35
|
+
try {
|
|
36
|
+
const conn = await registry.getConnection(task.remoteAgentId);
|
|
37
|
+
const result = await conn.callTool('agent.delegate_task', {
|
|
38
|
+
title: task.name,
|
|
39
|
+
description: task.goal || task.description,
|
|
40
|
+
acceptance_criteria: task.context?.acceptance_criteria ?? '',
|
|
41
|
+
artifacts: task.context?.artifacts ?? [],
|
|
42
|
+
});
|
|
43
|
+
const text = extractMcpText(result);
|
|
44
|
+
let remoteTaskId = taskId;
|
|
45
|
+
try {
|
|
46
|
+
const parsed = JSON.parse(text);
|
|
47
|
+
remoteTaskId = parsed.remote_task_id ?? parsed.task_id ?? taskId;
|
|
48
|
+
}
|
|
49
|
+
catch {
|
|
50
|
+
const match = text.match(/remote_task_id["\s:]+([a-zA-Z0-9-]+)/);
|
|
51
|
+
if (match)
|
|
52
|
+
remoteTaskId = match[1];
|
|
53
|
+
}
|
|
54
|
+
if (orch) {
|
|
55
|
+
await orch.repositoryHandle.updateTaskStatus(taskId, 'running', {
|
|
56
|
+
remote_task_id: remoteTaskId,
|
|
57
|
+
started_at: Date.now(),
|
|
58
|
+
});
|
|
59
|
+
}
|
|
60
|
+
dispatcher.markRunning(taskId, Promise.resolve({
|
|
61
|
+
taskId,
|
|
62
|
+
role: task.role,
|
|
63
|
+
success: true,
|
|
64
|
+
summary: `delegated to ${task.remoteAgentId}:${remoteTaskId}`,
|
|
65
|
+
duration: 0,
|
|
66
|
+
}));
|
|
67
|
+
}
|
|
68
|
+
catch (err) {
|
|
69
|
+
dispatcher.recordResult({
|
|
70
|
+
taskId,
|
|
71
|
+
role: task.role,
|
|
72
|
+
success: false,
|
|
73
|
+
summary: 'remote delegate failed',
|
|
74
|
+
error: err instanceof Error ? err.message : String(err),
|
|
75
|
+
duration: 0,
|
|
76
|
+
});
|
|
77
|
+
}
|
|
78
|
+
})();
|
|
79
|
+
dispatcher.markRunning(taskId, promise.then(() => ({
|
|
80
|
+
taskId,
|
|
81
|
+
role: task.role,
|
|
82
|
+
success: true,
|
|
83
|
+
summary: 'remote delegation started',
|
|
84
|
+
duration: 0,
|
|
85
|
+
})));
|
|
86
|
+
return {
|
|
87
|
+
ok: true,
|
|
88
|
+
message: `远程任务已委托给 ${task.remoteAgentId}(task ${taskId}),将通过轮询同步状态。`,
|
|
89
|
+
};
|
|
90
|
+
}
|
|
91
|
+
export async function pollRemoteTaskStatus(taskId) {
|
|
92
|
+
const dispatcher = getAgentDispatcher();
|
|
93
|
+
const orch = getOrchestrationService();
|
|
94
|
+
const task = dispatcher.getTask(taskId);
|
|
95
|
+
if (!task?.remoteAgentId || !task.remoteTaskId) {
|
|
96
|
+
return { done: false, status: 'unknown' };
|
|
97
|
+
}
|
|
98
|
+
const conn = await getRemoteAgentRegistry().getConnection(task.remoteAgentId);
|
|
99
|
+
const statusResult = await conn.callTool('agent.query_status', { task_id: task.remoteTaskId });
|
|
100
|
+
const statusText = extractMcpText(statusResult);
|
|
101
|
+
let status = 'running';
|
|
102
|
+
try {
|
|
103
|
+
const parsed = JSON.parse(statusText);
|
|
104
|
+
status = parsed.status ?? status;
|
|
105
|
+
}
|
|
106
|
+
catch {
|
|
107
|
+
if (/completed|done|success/i.test(statusText))
|
|
108
|
+
status = 'completed';
|
|
109
|
+
if (/failed|error/i.test(statusText))
|
|
110
|
+
status = 'failed';
|
|
111
|
+
if (/pending/i.test(statusText))
|
|
112
|
+
status = 'pending';
|
|
113
|
+
}
|
|
114
|
+
if (status === 'completed') {
|
|
115
|
+
const resultRaw = await conn.callTool('agent.get_result', { task_id: task.remoteTaskId });
|
|
116
|
+
const resultText = extractMcpText(resultRaw);
|
|
117
|
+
dispatcher.recordResult({
|
|
118
|
+
taskId,
|
|
119
|
+
role: task.role,
|
|
120
|
+
success: true,
|
|
121
|
+
summary: resultText.slice(0, 4000),
|
|
122
|
+
duration: 0,
|
|
123
|
+
});
|
|
124
|
+
return { done: true, status, result: resultText };
|
|
125
|
+
}
|
|
126
|
+
if (status === 'failed' || status === 'cancelled') {
|
|
127
|
+
dispatcher.recordResult({
|
|
128
|
+
taskId,
|
|
129
|
+
role: task.role,
|
|
130
|
+
success: false,
|
|
131
|
+
summary: statusText,
|
|
132
|
+
error: statusText,
|
|
133
|
+
duration: 0,
|
|
134
|
+
});
|
|
135
|
+
return { done: true, status, result: statusText };
|
|
136
|
+
}
|
|
137
|
+
if (orch) {
|
|
138
|
+
await orch.repositoryHandle.updateTaskStatus(taskId, 'running');
|
|
139
|
+
}
|
|
140
|
+
return { done: false, status };
|
|
141
|
+
}
|
|
142
|
+
//# sourceMappingURL=remote-task-executor.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"remote-task-executor.js","sourceRoot":"","sources":["../../src/orchestrator/remote-task-executor.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAC3D,OAAO,EAAE,sBAAsB,EAAE,MAAM,4BAA4B,CAAC;AACpE,OAAO,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAErE,SAAS,cAAc,CAAC,MAAe;IACrC,IAAI,OAAO,MAAM,KAAK,QAAQ;QAAE,OAAO,MAAM,CAAC;IAC9C,MAAM,OAAO,GAAI,MAAgE,EAAE,OAAO,CAAC;IAC3F,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC3B,OAAO,OAAO;aACX,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,CAAC;aAC1C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAK,CAAC;aACnB,IAAI,CAAC,IAAI,CAAC,CAAC;IAChB,CAAC;IACD,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;AAChC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,8BAA8B,CAClD,MAAc;IAEd,MAAM,UAAU,GAAG,kBAAkB,EAAE,CAAC;IACxC,MAAM,IAAI,GAAG,uBAAuB,EAAE,CAAC;IACvC,MAAM,IAAI,GAAG,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACxC,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,MAAM,MAAM,EAAE,CAAC;IACpD,CAAC;IACD,IAAI,IAAI,CAAC,YAAY,KAAK,QAAQ,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;QAC1D,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,MAAM,WAAW,EAAE,CAAC;IACzD,CAAC;IAED,MAAM,QAAQ,GAAG,sBAAsB,EAAE,CAAC;IAC1C,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAC/C,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,YAAY,IAAI,CAAC,aAAa,MAAM,EAAE,CAAC;IACtE,CAAC;IAED,MAAM,OAAO,GAAG,CAAC,KAAK,IAAmB,EAAE;QACzC,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,aAAa,CAAC,IAAI,CAAC,aAAc,CAAC,CAAC;YAC/D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,qBAAqB,EAAE;gBACxD,KAAK,EAAE,IAAI,CAAC,IAAI;gBAChB,WAAW,EAAE,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,WAAW;gBAC1C,mBAAmB,EAAE,IAAI,CAAC,OAAO,EAAE,mBAAmB,IAAI,EAAE;gBAC5D,SAAS,EAAE,IAAI,CAAC,OAAO,EAAE,SAAS,IAAI,EAAE;aACzC,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;YACpC,IAAI,YAAY,GAAG,MAAM,CAAC;YAC1B,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAkD,CAAC;gBACjF,YAAY,GAAG,MAAM,CAAC,cAAc,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC;YACnE,CAAC;YAAC,MAAM,CAAC;gBACP,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,sCAAsC,CAAC,CAAC;gBACjE,IAAI,KAAK;oBAAE,YAAY,GAAG,KAAK,CAAC,CAAC,CAAE,CAAC;YACtC,CAAC;YACD,IAAI,IAAI,EAAE,CAAC;gBACT,MAAM,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,MAAM,EAAE,SAAS,EAAE;oBAC9D,cAAc,EAAE,YAAY;oBAC5B,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE;iBACvB,CAAC,CAAC;YACL,CAAC;YACD,UAAU,CAAC,WAAW,CAAC,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC;gBAC7C,MAAM;gBACN,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,gBAAgB,IAAI,CAAC,aAAa,IAAI,YAAY,EAAE;gBAC7D,QAAQ,EAAE,CAAC;aACZ,CAAC,CAAC,CAAC;QACN,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,UAAU,CAAC,YAAY,CAAC;gBACtB,MAAM;gBACN,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE,wBAAwB;gBACjC,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;gBACvD,QAAQ,EAAE,CAAC;aACZ,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC,EAAE,CAAC;IAEL,UAAU,CAAC,WAAW,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;QACjD,MAAM;QACN,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,OAAO,EAAE,IAAI;QACb,OAAO,EAAE,2BAA2B;QACpC,QAAQ,EAAE,CAAC;KACZ,CAAC,CAAC,CAAC,CAAC;IAEL,OAAO;QACL,EAAE,EAAE,IAAI;QACR,OAAO,EAAE,YAAY,IAAI,CAAC,aAAa,SAAS,MAAM,cAAc;KACrE,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,MAAc;IAEd,MAAM,UAAU,GAAG,kBAAkB,EAAE,CAAC;IACxC,MAAM,IAAI,GAAG,uBAAuB,EAAE,CAAC;IACvC,MAAM,IAAI,GAAG,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACxC,IAAI,CAAC,IAAI,EAAE,aAAa,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;QAC/C,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;IAC5C,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,sBAAsB,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAC9E,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,oBAAoB,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;IAC/F,MAAM,UAAU,GAAG,cAAc,CAAC,YAAY,CAAC,CAAC;IAChD,IAAI,MAAM,GAAG,SAAS,CAAC;IACvB,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAwB,CAAC;QAC7D,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC;IACnC,CAAC;IAAC,MAAM,CAAC;QACP,IAAI,yBAAyB,CAAC,IAAI,CAAC,UAAU,CAAC;YAAE,MAAM,GAAG,WAAW,CAAC;QACrE,IAAI,eAAe,CAAC,IAAI,CAAC,UAAU,CAAC;YAAE,MAAM,GAAG,QAAQ,CAAC;QACxD,IAAI,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC;YAAE,MAAM,GAAG,SAAS,CAAC;IACtD,CAAC;IAED,IAAI,MAAM,KAAK,WAAW,EAAE,CAAC;QAC3B,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,kBAAkB,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;QAC1F,MAAM,UAAU,GAAG,cAAc,CAAC,SAAS,CAAC,CAAC;QAC7C,UAAU,CAAC,YAAY,CAAC;YACtB,MAAM;YACN,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,OAAO,EAAE,IAAI;YACb,OAAO,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC;YAClC,QAAQ,EAAE,CAAC;SACZ,CAAC,CAAC;QACH,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC;IACpD,CAAC;IAED,IAAI,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,WAAW,EAAE,CAAC;QAClD,UAAU,CAAC,YAAY,CAAC;YACtB,MAAM;YACN,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,OAAO,EAAE,KAAK;YACd,OAAO,EAAE,UAAU;YACnB,KAAK,EAAE,UAAU;YACjB,QAAQ,EAAE,CAAC;SACZ,CAAC,CAAC;QACH,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC;IACpD,CAAC;IAED,IAAI,IAAI,EAAE,CAAC;QACT,MAAM,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IAClE,CAAC;IACD,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;AACjC,CAAC"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
export interface RemoteTaskPollerConfig {
|
|
2
|
+
intervalMs?: number;
|
|
3
|
+
enabled?: boolean;
|
|
4
|
+
}
|
|
5
|
+
export declare class RemoteTaskPoller {
|
|
6
|
+
private readonly config;
|
|
7
|
+
private timer;
|
|
8
|
+
private polling;
|
|
9
|
+
constructor(config?: RemoteTaskPollerConfig);
|
|
10
|
+
start(): void;
|
|
11
|
+
stop(): void;
|
|
12
|
+
tick(): Promise<void>;
|
|
13
|
+
}
|
|
14
|
+
export declare function getRemoteTaskPoller(): RemoteTaskPoller;
|
|
15
|
+
export declare function startRemoteTaskPoller(config?: RemoteTaskPollerConfig): RemoteTaskPoller;
|
|
16
|
+
//# sourceMappingURL=remote-task-poller.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"remote-task-poller.d.ts","sourceRoot":"","sources":["../../src/orchestrator/remote-task-poller.ts"],"names":[],"mappings":"AAUA,MAAM,WAAW,sBAAsB;IACrC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED,qBAAa,gBAAgB;IAIf,OAAO,CAAC,QAAQ,CAAC,MAAM;IAHnC,OAAO,CAAC,KAAK,CAA+C;IAC5D,OAAO,CAAC,OAAO,CAAS;gBAEK,MAAM,GAAE,sBAA2B;IAEhE,KAAK,IAAI,IAAI;IAab,IAAI,IAAI,IAAI;IAON,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;CAsB5B;AAID,wBAAgB,mBAAmB,IAAI,gBAAgB,CAKtD;AAED,wBAAgB,qBAAqB,CAAC,MAAM,CAAC,EAAE,sBAAsB,GAAG,gBAAgB,CAIvF"}
|
|
@@ -0,0 +1,75 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* RemoteTaskPoller — 轮询远程 MCP 任务状态(Agent Mesh v1)。
|
|
3
|
+
*/
|
|
4
|
+
import { Logger } from '@zhin.js/logger';
|
|
5
|
+
import { getOrchestrationService } from './orchestration-service.js';
|
|
6
|
+
import { getAgentDispatcher } from './agent-dispatcher.js';
|
|
7
|
+
import { pollRemoteTaskStatus } from './remote-task-executor.js';
|
|
8
|
+
const logger = new Logger(null, 'RemoteTaskPoller');
|
|
9
|
+
export class RemoteTaskPoller {
|
|
10
|
+
config;
|
|
11
|
+
timer = null;
|
|
12
|
+
polling = false;
|
|
13
|
+
constructor(config = {}) {
|
|
14
|
+
this.config = config;
|
|
15
|
+
}
|
|
16
|
+
start() {
|
|
17
|
+
if (this.config.enabled === false)
|
|
18
|
+
return;
|
|
19
|
+
if (this.timer)
|
|
20
|
+
return;
|
|
21
|
+
const intervalMs = this.config.intervalMs ?? 15_000;
|
|
22
|
+
this.timer = setInterval(() => {
|
|
23
|
+
void this.tick();
|
|
24
|
+
}, intervalMs);
|
|
25
|
+
if (this.timer && typeof this.timer === 'object' && 'unref' in this.timer) {
|
|
26
|
+
this.timer.unref();
|
|
27
|
+
}
|
|
28
|
+
logger.debug(`RemoteTaskPoller started (interval=${intervalMs}ms)`);
|
|
29
|
+
}
|
|
30
|
+
stop() {
|
|
31
|
+
if (this.timer) {
|
|
32
|
+
clearInterval(this.timer);
|
|
33
|
+
this.timer = null;
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
async tick() {
|
|
37
|
+
if (this.polling)
|
|
38
|
+
return;
|
|
39
|
+
this.polling = true;
|
|
40
|
+
try {
|
|
41
|
+
const orch = getOrchestrationService();
|
|
42
|
+
if (!orch)
|
|
43
|
+
return;
|
|
44
|
+
const tasks = await orch.repositoryHandle.listActiveRemoteTasks();
|
|
45
|
+
const dispatcher = getAgentDispatcher();
|
|
46
|
+
for (const record of tasks) {
|
|
47
|
+
dispatcher.syncTaskFromRecord(record);
|
|
48
|
+
if (record.remote_task_id) {
|
|
49
|
+
try {
|
|
50
|
+
await pollRemoteTaskStatus(record.id);
|
|
51
|
+
}
|
|
52
|
+
catch (err) {
|
|
53
|
+
logger.debug(`poll failed for ${record.id}:`, err);
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
finally {
|
|
59
|
+
this.polling = false;
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
let globalPoller = null;
|
|
64
|
+
export function getRemoteTaskPoller() {
|
|
65
|
+
if (!globalPoller) {
|
|
66
|
+
globalPoller = new RemoteTaskPoller();
|
|
67
|
+
}
|
|
68
|
+
return globalPoller;
|
|
69
|
+
}
|
|
70
|
+
export function startRemoteTaskPoller(config) {
|
|
71
|
+
globalPoller = new RemoteTaskPoller(config);
|
|
72
|
+
globalPoller.start();
|
|
73
|
+
return globalPoller;
|
|
74
|
+
}
|
|
75
|
+
//# sourceMappingURL=remote-task-poller.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"remote-task-poller.js","sourceRoot":"","sources":["../../src/orchestrator/remote-task-poller.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AACzC,OAAO,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AACrE,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAC3D,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAEjE,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,EAAE,kBAAkB,CAAC,CAAC;AAOpD,MAAM,OAAO,gBAAgB;IAIE;IAHrB,KAAK,GAA0C,IAAI,CAAC;IACpD,OAAO,GAAG,KAAK,CAAC;IAExB,YAA6B,SAAiC,EAAE;QAAnC,WAAM,GAAN,MAAM,CAA6B;IAAG,CAAC;IAEpE,KAAK;QACH,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,KAAK,KAAK;YAAE,OAAO;QAC1C,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO;QACvB,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC;QACpD,IAAI,CAAC,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE;YAC5B,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC;QACnB,CAAC,EAAE,UAAU,CAAC,CAAC;QACf,IAAI,IAAI,CAAC,KAAK,IAAI,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,IAAI,OAAO,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC1E,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACrB,CAAC;QACD,MAAM,CAAC,KAAK,CAAC,sCAAsC,UAAU,KAAK,CAAC,CAAC;IACtE,CAAC;IAED,IAAI;QACF,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QACpB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAI;QACR,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,uBAAuB,EAAE,CAAC;YACvC,IAAI,CAAC,IAAI;gBAAE,OAAO;YAClB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,qBAAqB,EAAE,CAAC;YAClE,MAAM,UAAU,GAAG,kBAAkB,EAAE,CAAC;YACxC,KAAK,MAAM,MAAM,IAAI,KAAK,EAAE,CAAC;gBAC3B,UAAU,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;gBACtC,IAAI,MAAM,CAAC,cAAc,EAAE,CAAC;oBAC1B,IAAI,CAAC;wBACH,MAAM,oBAAoB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;oBACxC,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACb,MAAM,CAAC,KAAK,CAAC,mBAAmB,MAAM,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;oBACrD,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;QACvB,CAAC;IACH,CAAC;CACF;AAED,IAAI,YAAY,GAA4B,IAAI,CAAC;AAEjD,MAAM,UAAU,mBAAmB;IACjC,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,YAAY,GAAG,IAAI,gBAAgB,EAAE,CAAC;IACxC,CAAC;IACD,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,MAA+B;IACnE,YAAY,GAAG,IAAI,gBAAgB,CAAC,MAAM,CAAC,CAAC;IAC5C,YAAY,CAAC,KAAK,EAAE,CAAC;IACrB,OAAO,YAAY,CAAC;AACtB,CAAC"}
|
|
@@ -0,0 +1,32 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Plugin hooks bridged to agentLoop (ADR 0010 D5).
|
|
3
|
+
*/
|
|
4
|
+
import type { AgentMessage, BeforeToolCallResult, ParsedToolCall } from '@zhin.js/ai';
|
|
5
|
+
export interface PluginBeforeToolCallContext {
|
|
6
|
+
toolCall: ParsedToolCall;
|
|
7
|
+
sessionId?: string;
|
|
8
|
+
}
|
|
9
|
+
export interface PluginAfterToolCallContext {
|
|
10
|
+
toolCall: ParsedToolCall;
|
|
11
|
+
result: AgentMessage;
|
|
12
|
+
sessionId?: string;
|
|
13
|
+
}
|
|
14
|
+
export interface PluginTransformContextContext {
|
|
15
|
+
sessionId?: string;
|
|
16
|
+
signal?: AbortSignal;
|
|
17
|
+
}
|
|
18
|
+
export type PluginBeforeToolCallHandler = (ctx: PluginBeforeToolCallContext) => Promise<BeforeToolCallResult | void> | BeforeToolCallResult | void;
|
|
19
|
+
export type PluginAfterToolCallHandler = (ctx: PluginAfterToolCallContext) => Promise<void> | void;
|
|
20
|
+
export type PluginTransformContextHandler = (messages: AgentMessage[], ctx: PluginTransformContextContext) => Promise<AgentMessage[]> | AgentMessage[];
|
|
21
|
+
export declare class PluginAILoopHookRegistry {
|
|
22
|
+
private readonly beforeHandlers;
|
|
23
|
+
private readonly afterHandlers;
|
|
24
|
+
private readonly transformHandlers;
|
|
25
|
+
onBeforeToolCall(handler: PluginBeforeToolCallHandler): () => void;
|
|
26
|
+
onAfterToolCall(handler: PluginAfterToolCallHandler): () => void;
|
|
27
|
+
onTransformContext(handler: PluginTransformContextHandler): () => void;
|
|
28
|
+
runBeforeToolCall(ctx: PluginBeforeToolCallContext): Promise<BeforeToolCallResult | void>;
|
|
29
|
+
runAfterToolCall(ctx: PluginAfterToolCallContext): Promise<void>;
|
|
30
|
+
runTransformContext(messages: AgentMessage[], ctx: PluginTransformContextContext): Promise<AgentMessage[]>;
|
|
31
|
+
}
|
|
32
|
+
//# sourceMappingURL=plugin-loop-hooks.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"plugin-loop-hooks.d.ts","sourceRoot":"","sources":["../src/plugin-loop-hooks.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,KAAK,EAAE,YAAY,EAAE,oBAAoB,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAEtF,MAAM,WAAW,2BAA2B;IAC1C,QAAQ,EAAE,cAAc,CAAC;IACzB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,0BAA0B;IACzC,QAAQ,EAAE,cAAc,CAAC;IACzB,MAAM,EAAE,YAAY,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,6BAA6B;IAC5C,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB;AAED,MAAM,MAAM,2BAA2B,GAAG,CACxC,GAAG,EAAE,2BAA2B,KAC7B,OAAO,CAAC,oBAAoB,GAAG,IAAI,CAAC,GAAG,oBAAoB,GAAG,IAAI,CAAC;AAExE,MAAM,MAAM,0BAA0B,GAAG,CACvC,GAAG,EAAE,0BAA0B,KAC5B,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;AAE1B,MAAM,MAAM,6BAA6B,GAAG,CAC1C,QAAQ,EAAE,YAAY,EAAE,EACxB,GAAG,EAAE,6BAA6B,KAC/B,OAAO,CAAC,YAAY,EAAE,CAAC,GAAG,YAAY,EAAE,CAAC;AAE9C,qBAAa,wBAAwB;IACnC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAqC;IACpE,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAoC;IAClE,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAuC;IAEzE,gBAAgB,CAAC,OAAO,EAAE,2BAA2B,GAAG,MAAM,IAAI;IAQlE,eAAe,CAAC,OAAO,EAAE,0BAA0B,GAAG,MAAM,IAAI;IAQhE,kBAAkB,CAAC,OAAO,EAAE,6BAA6B,GAAG,MAAM,IAAI;IAQhE,iBAAiB,CACrB,GAAG,EAAE,2BAA2B,GAC/B,OAAO,CAAC,oBAAoB,GAAG,IAAI,CAAC;IAgBjC,gBAAgB,CAAC,GAAG,EAAE,0BAA0B,GAAG,OAAO,CAAC,IAAI,CAAC;IAUhE,mBAAmB,CACvB,QAAQ,EAAE,YAAY,EAAE,EACxB,GAAG,EAAE,6BAA6B,GACjC,OAAO,CAAC,YAAY,EAAE,CAAC;CAW3B"}
|
|
@@ -0,0 +1,69 @@
|
|
|
1
|
+
export class PluginAILoopHookRegistry {
|
|
2
|
+
beforeHandlers = [];
|
|
3
|
+
afterHandlers = [];
|
|
4
|
+
transformHandlers = [];
|
|
5
|
+
onBeforeToolCall(handler) {
|
|
6
|
+
this.beforeHandlers.push(handler);
|
|
7
|
+
return () => {
|
|
8
|
+
const i = this.beforeHandlers.indexOf(handler);
|
|
9
|
+
if (i >= 0)
|
|
10
|
+
this.beforeHandlers.splice(i, 1);
|
|
11
|
+
};
|
|
12
|
+
}
|
|
13
|
+
onAfterToolCall(handler) {
|
|
14
|
+
this.afterHandlers.push(handler);
|
|
15
|
+
return () => {
|
|
16
|
+
const i = this.afterHandlers.indexOf(handler);
|
|
17
|
+
if (i >= 0)
|
|
18
|
+
this.afterHandlers.splice(i, 1);
|
|
19
|
+
};
|
|
20
|
+
}
|
|
21
|
+
onTransformContext(handler) {
|
|
22
|
+
this.transformHandlers.push(handler);
|
|
23
|
+
return () => {
|
|
24
|
+
const i = this.transformHandlers.indexOf(handler);
|
|
25
|
+
if (i >= 0)
|
|
26
|
+
this.transformHandlers.splice(i, 1);
|
|
27
|
+
};
|
|
28
|
+
}
|
|
29
|
+
async runBeforeToolCall(ctx) {
|
|
30
|
+
let merged;
|
|
31
|
+
for (const handler of this.beforeHandlers) {
|
|
32
|
+
try {
|
|
33
|
+
const result = await handler(ctx);
|
|
34
|
+
if (result) {
|
|
35
|
+
merged = { ...(merged ?? {}), ...result };
|
|
36
|
+
if (result.allowed === false)
|
|
37
|
+
return merged;
|
|
38
|
+
}
|
|
39
|
+
}
|
|
40
|
+
catch {
|
|
41
|
+
// isolate plugin errors
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
return merged;
|
|
45
|
+
}
|
|
46
|
+
async runAfterToolCall(ctx) {
|
|
47
|
+
for (const handler of this.afterHandlers) {
|
|
48
|
+
try {
|
|
49
|
+
await handler(ctx);
|
|
50
|
+
}
|
|
51
|
+
catch {
|
|
52
|
+
// isolate
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
async runTransformContext(messages, ctx) {
|
|
57
|
+
let out = messages;
|
|
58
|
+
for (const handler of this.transformHandlers) {
|
|
59
|
+
try {
|
|
60
|
+
out = await handler(out, ctx);
|
|
61
|
+
}
|
|
62
|
+
catch {
|
|
63
|
+
// isolate
|
|
64
|
+
}
|
|
65
|
+
}
|
|
66
|
+
return out;
|
|
67
|
+
}
|
|
68
|
+
}
|
|
69
|
+
//# sourceMappingURL=plugin-loop-hooks.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"plugin-loop-hooks.js","sourceRoot":"","sources":["../src/plugin-loop-hooks.ts"],"names":[],"mappings":"AAkCA,MAAM,OAAO,wBAAwB;IAClB,cAAc,GAAkC,EAAE,CAAC;IACnD,aAAa,GAAiC,EAAE,CAAC;IACjD,iBAAiB,GAAoC,EAAE,CAAC;IAEzE,gBAAgB,CAAC,OAAoC;QACnD,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAClC,OAAO,GAAG,EAAE;YACV,MAAM,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC/C,IAAI,CAAC,IAAI,CAAC;gBAAE,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAC/C,CAAC,CAAC;IACJ,CAAC;IAED,eAAe,CAAC,OAAmC;QACjD,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACjC,OAAO,GAAG,EAAE;YACV,MAAM,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC9C,IAAI,CAAC,IAAI,CAAC;gBAAE,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAC9C,CAAC,CAAC;IACJ,CAAC;IAED,kBAAkB,CAAC,OAAsC;QACvD,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrC,OAAO,GAAG,EAAE;YACV,MAAM,CAAC,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAClD,IAAI,CAAC,IAAI,CAAC;gBAAE,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAClD,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,iBAAiB,CACrB,GAAgC;QAEhC,IAAI,MAAwC,CAAC;QAC7C,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YAC1C,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC;gBAClC,IAAI,MAAM,EAAE,CAAC;oBACX,MAAM,GAAG,EAAE,GAAG,CAAC,MAAM,IAAI,EAAE,CAAC,EAAE,GAAG,MAAM,EAAE,CAAC;oBAC1C,IAAI,MAAM,CAAC,OAAO,KAAK,KAAK;wBAAE,OAAO,MAAM,CAAC;gBAC9C,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,wBAAwB;YAC1B,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,GAA+B;QACpD,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACzC,IAAI,CAAC;gBACH,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC;YACrB,CAAC;YAAC,MAAM,CAAC;gBACP,UAAU;YACZ,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,mBAAmB,CACvB,QAAwB,EACxB,GAAkC;QAElC,IAAI,GAAG,GAAG,QAAQ,CAAC;QACnB,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC7C,IAAI,CAAC;gBACH,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAChC,CAAC;YAAC,MAAM,CAAC;gBACP,UAAU;YACZ,CAAC;QACH,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;CACF"}
|
package/lib/service.d.ts
CHANGED
|
@@ -10,6 +10,7 @@ import type { ContextManager, ContextConfig } from '@zhin.js/ai';
|
|
|
10
10
|
import { type NormalizedAiRoutingConfig } from './config/normalize-ai-config.js';
|
|
11
11
|
import { AgentBindingRegistry } from './config/agent-binding-registry.js';
|
|
12
12
|
import type { ToolCallRecord } from './zhin-agent/tool-calls-user-format.js';
|
|
13
|
+
import { PluginAILoopHookRegistry, type PluginAfterToolCallHandler, type PluginBeforeToolCallHandler, type PluginTransformContextHandler } from './plugin-loop-hooks.js';
|
|
13
14
|
/** AIService 程序化 Agent 句柄(agentLoop 隔离上下文,非 legacy `Agent` 类) */
|
|
14
15
|
export interface ServiceAgent {
|
|
15
16
|
run(userInput: string | ContentPart[]): Promise<ServiceAgentResult>;
|
|
@@ -48,6 +49,7 @@ export declare class AIService {
|
|
|
48
49
|
private plugin?;
|
|
49
50
|
private customTools;
|
|
50
51
|
private _modelRegistry;
|
|
52
|
+
readonly loopHooks: PluginAILoopHookRegistry;
|
|
51
53
|
constructor(config?: AIConfig);
|
|
52
54
|
getRoutingConfig(): NormalizedAiRoutingConfig;
|
|
53
55
|
getBindingRegistry(): AgentBindingRegistry;
|
|
@@ -60,6 +62,11 @@ export declare class AIService {
|
|
|
60
62
|
setModelRegistry(registry: ModelRegistry): void;
|
|
61
63
|
getModelRegistry(): ModelRegistry | null;
|
|
62
64
|
registerTool(tool: AgentTool): () => void;
|
|
65
|
+
/** ADR 0010 — bridge plugin beforeToolCall hooks to agentLoop. */
|
|
66
|
+
onBeforeToolCall(handler: PluginBeforeToolCallHandler): () => void;
|
|
67
|
+
onAfterToolCall(handler: PluginAfterToolCallHandler): () => void;
|
|
68
|
+
/** Run after built-in compaction in transformContext chain. */
|
|
69
|
+
onTransformContext(handler: PluginTransformContextHandler): () => void;
|
|
63
70
|
collectAllTools(): AgentTool[];
|
|
64
71
|
/**
|
|
65
72
|
* IM / ZhinAgent 流水线用的常驻 Tool 实例(未经 normalize;与 ToolFeature 工具合并后由 collectRelevantTools 绑定 context)。
|
package/lib/service.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"service.d.ts","sourceRoot":"","sources":["../src/service.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAC7D,OAAO,KAAK,EACV,UAAU,EACV,QAAQ,EAER,qBAAqB,EACrB,sBAAsB,EACtB,mBAAmB,EACnB,SAAS,EACT,IAAI,EACJ,WAAW,EACX,KAAK,EACN,MAAM,eAAe,CAAC;AACvB,OAAO,EACL,cAAc,EAEd,KAAK,uBAAuB,EAC7B,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,KAAK,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAMjE,OAAO,EAA4B,KAAK,yBAAyB,EAAE,MAAM,iCAAiC,CAAC;AAG3G,OAAO,EAAE,oBAAoB,EAAE,MAAM,oCAAoC,CAAC;AAM1E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,wCAAwC,CAAC;
|
|
1
|
+
{"version":3,"file":"service.d.ts","sourceRoot":"","sources":["../src/service.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAC7D,OAAO,KAAK,EACV,UAAU,EACV,QAAQ,EAER,qBAAqB,EACrB,sBAAsB,EACtB,mBAAmB,EACnB,SAAS,EACT,IAAI,EACJ,WAAW,EACX,KAAK,EACN,MAAM,eAAe,CAAC;AACvB,OAAO,EACL,cAAc,EAEd,KAAK,uBAAuB,EAC7B,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,KAAK,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAMjE,OAAO,EAA4B,KAAK,yBAAyB,EAAE,MAAM,iCAAiC,CAAC;AAG3G,OAAO,EAAE,oBAAoB,EAAE,MAAM,oCAAoC,CAAC;AAM1E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,wCAAwC,CAAC;AAC7E,OAAO,EACL,wBAAwB,EACxB,KAAK,0BAA0B,EAC/B,KAAK,2BAA2B,EAChC,KAAK,6BAA6B,EACnC,MAAM,wBAAwB,CAAC;AAEhC,iEAAiE;AACjE,MAAM,WAAW,YAAY;IAC3B,GAAG,CAAC,SAAS,EAAE,MAAM,GAAG,WAAW,EAAE,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC;IACpE,OAAO,IAAI,IAAI,CAAC;CACjB;AAED,MAAM,WAAW,kBAAkB;IACjC,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,cAAc,EAAE,CAAC;IAC5B,KAAK,EAAE,KAAK,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,yBAAyB;IACxC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,KAAK,CAAC,EAAE,SAAS,EAAE,CAAC;IACpB,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,qBAAa,SAAS;IACpB,OAAO,CAAC,SAAS,CAAsC;IACvD,OAAO,CAAC,eAAe,CAAS;IAChC,OAAO,CAAC,OAAO,CAA4B;IAC3C,OAAO,CAAC,eAAe,CAAuB;IACvC,QAAQ,EAAE,cAAc,CAAC;IACzB,cAAc,CAAC,EAAE,cAAc,CAAC;IACvC,OAAO,CAAC,YAAY,CAAe;IACnC,OAAO,CAAC,aAAa,CAA6C;IAClE,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,aAAa,CAAkB;IACvC,OAAO,CAAC,WAAW,CAAoB;IACvC,OAAO,CAAC,qBAAqB,CAAC,CAA0B;IACxD,OAAO,CAAC,MAAM,CAAC,CAAS;IACxB,OAAO,CAAC,WAAW,CAAqC;IACxD,OAAO,CAAC,cAAc,CAA8B;IACpD,QAAQ,CAAC,SAAS,2BAAkC;gBAExC,MAAM,GAAE,QAAa;IA0BjC,gBAAgB,IAAI,yBAAyB;IAI7C,kBAAkB,IAAI,oBAAoB;IAI1C,4BAA4B;IAC5B,mBAAmB,CAAC,SAAS,EAAE,OAAO,uBAAuB,EAAE,SAAS,EAAE,GAAG,IAAI;IAIjF,OAAO,IAAI,OAAO;IAIlB,iBAAiB,CAAC,OAAO,EAAE,cAAc,GAAG,IAAI;IAChD,iBAAiB,CAAC,OAAO,EAAE,cAAc,GAAG,IAAI;IAKhD,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAI/B,gBAAgB,CAAC,QAAQ,EAAE,aAAa,GAAG,IAAI;IAC/C,gBAAgB,IAAI,aAAa,GAAG,IAAI;IACxC,YAAY,CAAC,IAAI,EAAE,SAAS,GAAG,MAAM,IAAI;IAEzC,kEAAkE;IAClE,gBAAgB,CAAC,OAAO,EAAE,2BAA2B,GAAG,MAAM,IAAI;IAIlE,eAAe,CAAC,OAAO,EAAE,0BAA0B,GAAG,MAAM,IAAI;IAIhE,+DAA+D;IAC/D,kBAAkB,CAAC,OAAO,EAAE,6BAA6B,GAAG,MAAM,IAAI;IAItE,eAAe,IAAI,SAAS,EAAE;IAK9B;;OAEG;IACH,uBAAuB,IAAI,IAAI,EAAE;IAMjC,8EAA8E;IAC9E,OAAO,CAAC,wBAAwB;IAMhC,gBAAgB,IAAI,aAAa;IACjC,gBAAgB;qBAvGsB,MAAM;mBAAa,MAAM;;IAwG/D,gBAAgB,IAAI,eAAe;IACnC,iEAAiE;IACjE,cAAc,IAAI,QAAQ,CAAC,OAAO,CAAC;IAEnC,mEAAmE;IACnE,0BAA0B,CAAC,aAAa,EAAE,MAAM,GAAG,uBAAuB;IAK1E,gBAAgB,CAAC,QAAQ,EAAE,UAAU,GAAG,IAAI;IAC5C,WAAW,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,UAAU;IAUtC,aAAa,IAAI,MAAM,EAAE;IAEzB,gCAAgC;IAChC,oBAAoB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO;IAK5C,+EAA+E;IAC/E,qBAAqB,IAAI,IAAI;IAY7B,uBAAuB,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG;QAAE,aAAa,CAAC,EAAE,MAAM,CAAC;QAAC,YAAY,CAAC,EAAE,OAAO,eAAe,EAAE,oBAAoB,CAAA;KAAE;IAQzH,UAAU,CAAC,YAAY,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,EAAE,CAAA;KAAE,EAAE,CAAC;IAapF,IAAI,CAAC,OAAO,EAAE,qBAAqB,EAAE,YAAY,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,sBAAsB,CAAC;IAI3F,UAAU,CAAC,OAAO,EAAE,qBAAqB,EAAE,YAAY,CAAC,EAAE,MAAM,GAAG,aAAa,CAAC,mBAAmB,CAAC;IAItG,GAAG,CACP,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE;QAAE,QAAQ,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,YAAY,CAAC,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAA;KAAO,GAC/F,OAAO,CAAC,MAAM,CAAC;IAclB,OAAO,CAAC,wBAAwB;IAQhC,WAAW,CAAC,OAAO,GAAE,yBAA8B,GAAG,YAAY;IAuB5D,QAAQ,CACZ,IAAI,EAAE,MAAM,GAAG,WAAW,EAAE,EAC5B,OAAO,GAAE,yBAA8B,GACtC,OAAO,CAAC,kBAAkB,CAAC;IASxB,WAAW,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAQrD,OAAO,IAAI,IAAI;CAChB"}
|
package/lib/service.js
CHANGED
|
@@ -14,6 +14,7 @@ import { registerLlmApiFromProviders } from '@zhin.js/ai';
|
|
|
14
14
|
import { AgentBindingRegistry } from './config/agent-binding-registry.js';
|
|
15
15
|
import { DEFAULT_ZHIN_AGENT_NAME } from './config/types.js';
|
|
16
16
|
import { runAgentLoopStandaloneTurn, } from './zhin-agent/agent-loop-standalone.js';
|
|
17
|
+
import { PluginAILoopHookRegistry, } from './plugin-loop-hooks.js';
|
|
17
18
|
export class AIService {
|
|
18
19
|
providers = new Map();
|
|
19
20
|
defaultProvider;
|
|
@@ -30,6 +31,7 @@ export class AIService {
|
|
|
30
31
|
plugin;
|
|
31
32
|
customTools = new Map();
|
|
32
33
|
_modelRegistry = null;
|
|
34
|
+
loopHooks = new PluginAILoopHookRegistry();
|
|
33
35
|
constructor(config = {}) {
|
|
34
36
|
this.routing = normalizeAiRoutingConfig(config);
|
|
35
37
|
const validationErrors = validateAiRoutingConfig(this.routing);
|
|
@@ -80,6 +82,17 @@ export class AIService {
|
|
|
80
82
|
setModelRegistry(registry) { this._modelRegistry = registry; }
|
|
81
83
|
getModelRegistry() { return this._modelRegistry; }
|
|
82
84
|
registerTool(tool) { this.customTools.set(tool.name, tool); return () => { this.customTools.delete(tool.name); }; }
|
|
85
|
+
/** ADR 0010 — bridge plugin beforeToolCall hooks to agentLoop. */
|
|
86
|
+
onBeforeToolCall(handler) {
|
|
87
|
+
return this.loopHooks.onBeforeToolCall(handler);
|
|
88
|
+
}
|
|
89
|
+
onAfterToolCall(handler) {
|
|
90
|
+
return this.loopHooks.onAfterToolCall(handler);
|
|
91
|
+
}
|
|
92
|
+
/** Run after built-in compaction in transformContext chain. */
|
|
93
|
+
onTransformContext(handler) {
|
|
94
|
+
return this.loopHooks.onTransformContext(handler);
|
|
95
|
+
}
|
|
83
96
|
collectAllTools() {
|
|
84
97
|
const tools = [...this.builtinTools, ...this.customTools.values()];
|
|
85
98
|
return tools;
|