File size: 2,212 Bytes
68d7816 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 | import { toDisposable, type IDisposable } from "#/_base/di/lifecycle";
import { LifecycleScope } from '#/app/scopes';
import { ScopeActivation, registerScopedService } from '#/_base/di/scope';
import type {
ExecutableTool,
ToolDisclosure,
ToolInfo,
ToolSource,
} from '#/tool/toolContract';
import {
IAgentToolRegistryService,
type ToolReference,
type ToolRegistrationOptions,
} from './toolRegistry';
import './builtinToolAssemblyService';
interface ToolEntry {
readonly tool: ExecutableTool;
readonly source: ToolSource;
readonly disclosure?: ToolDisclosure;
}
export class AgentToolRegistryService implements IAgentToolRegistryService {
declare readonly _serviceBrand: undefined;
private readonly tools = new Map<string, ToolEntry>();
register(tool: ExecutableTool, options: ToolRegistrationOptions = {}): IDisposable {
const source = options.source ?? 'builtin';
const entry: ToolEntry = { tool, source, disclosure: options.disclosure };
this.unregisterTool(tool.name);
this.tools.set(tool.name, entry);
return toDisposable(() => {
const current = this.tools.get(tool.name);
if (current !== entry) return;
this.unregisterTool(tool.name);
});
}
list(): readonly ToolInfo[] {
return [...this.tools.values()]
.map(({ tool, source, disclosure }) => ({
name: tool.name,
description: tool.description,
parameters: tool.parameters,
source,
disclosure,
}))
.toSorted((a, b) => a.name.localeCompare(b.name));
}
listReferences(): readonly ToolReference[] {
return [...this.tools.entries()]
.map(([name, { source }]) => ({ name, source }))
.toSorted((a, b) => a.name.localeCompare(b.name));
}
resolve(name: string): ExecutableTool | undefined {
return this.tools.get(name)?.tool;
}
private unregisterTool(name: string): ToolEntry | undefined {
const entry = this.tools.get(name);
if (entry === undefined) return undefined;
this.tools.delete(name);
return entry;
}
}
registerScopedService(
LifecycleScope.Agent,
IAgentToolRegistryService,
AgentToolRegistryService,
ScopeActivation.OnScopeCreated,
'toolRegistry',
);
|