File size: 3,823 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 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 | import { randomUUID } from 'node:crypto';
import { LifecycleScope } from '#/app/scopes';
import { ScopeActivation, registerScopedService } from '#/_base/di/scope';
import { IEventService } from '#/app/event/event';
import { ErrorCodes, Error2 } from '#/errors';
import { IAgentScopeContext } from '#/agent/scopeContext/scopeContext';
import { MAIN_AGENT_ID } from '#/session/agentLifecycle/agentLifecycle';
import { expandCommandArguments } from '#/app/plugin/commands';
import { IPluginService } from '#/app/plugin/plugin';
import { IAgentLoopService } from '#/agent/loop/loop';
import { promptMetadataTextFromText } from '#/agent/prompt/promptMetadataText';
import { ISessionMetadata } from '#/session/sessionMetadata/sessionMetadata';
import { ISessionContext } from '#/session/sessionContext/sessionContext';
import { applyPromptMetadataUpdate } from '#/session/sessionMetadata/promptMetadata';
import { IEventDispatcher } from '#/state/eventDispatcher';
import {
IAgentPluginCommandService,
PluginCommandActivated,
type ActivatePluginCommandPayload,
} from './pluginCommand';
export class AgentPluginCommandService implements IAgentPluginCommandService {
declare readonly _serviceBrand: undefined;
constructor(
@IPluginService private readonly plugins: IPluginService,
@IAgentLoopService private readonly loop: IAgentLoopService,
@IEventDispatcher private readonly dispatcher: IEventDispatcher,
@ISessionMetadata private readonly metadata: ISessionMetadata,
@IEventService private readonly eventService: IEventService,
@ISessionContext private readonly sessionContext: ISessionContext,
@IAgentScopeContext private readonly scopeContext: IAgentScopeContext,
) { }
async activate(payload: ActivatePluginCommandPayload): Promise<void> {
const commands = await this.plugins.listPluginCommands();
const def = commands.find(
(command) => command.pluginId === payload.pluginId && command.name === payload.commandName,
);
if (def === undefined) {
throw new Error2(
ErrorCodes.REQUEST_INVALID,
`Plugin command "${payload.pluginId}:${payload.commandName}" was not found`,
);
}
const commandArgs = payload.args ?? '';
const expanded = expandCommandArguments(def.body, commandArgs);
const origin = {
kind: 'plugin_command' as const,
activationId: randomUUID(),
pluginId: payload.pluginId,
commandName: payload.commandName,
commandArgs: payload.args,
trigger: 'user-slash' as const,
};
await this.dispatcher.dispatch(
new PluginCommandActivated({
agentId: this.scopeContext.agentId,
activationId: origin.activationId,
pluginId: origin.pluginId,
commandName: origin.commandName,
commandArgs: origin.commandArgs,
trigger: origin.trigger,
}),
);
this.loop.submit({
message: { role: 'user', content: [{ type: 'text', text: expanded }] },
meta: { origin, tracked: true },
});
if (this.scopeContext.agentId === MAIN_AGENT_ID) {
await applyPromptMetadataUpdate(
{
metadata: this.metadata,
eventService: this.eventService,
sessionId: this.sessionContext.sessionId,
},
promptMetadataTextFromPluginCommand(payload),
);
}
}
}
function promptMetadataTextFromPluginCommand(
payload: ActivatePluginCommandPayload,
): string | undefined {
const args = payload.args?.trim();
const command = `/${payload.pluginId}:${payload.commandName}`;
return promptMetadataTextFromText(
args === undefined || args.length === 0 ? command : `${command} ${args}`,
);
}
registerScopedService(
LifecycleScope.Agent,
IAgentPluginCommandService,
AgentPluginCommandService,
ScopeActivation.OnScopeCreated,
'pluginCommand',
);
|