File size: 2,851 Bytes
4e23b01 | 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 | import {
ErrorCodes,
Error2,
IAgentGoalService,
IAgentLifecycleService,
IAgentPlanService,
IAgentProfileService,
IAgentSwarmService,
IAgentTowerService,
resumeSessionById,
towerEnterFailureMessage,
type PermissionMode,
type Scope,
} from '@moonshot-ai/agent-core-v2';
import type { SessionAgentConfigPartial } from '@moonshot-ai/agent-core-v2/app/sessionLegacy/sessionProtocol';
import { ensureMainAgent } from '../transport/mainAgent';
export async function applySessionAgentConfig(
core: Scope,
sessionId: string,
agentConfig: SessionAgentConfigPartial,
): Promise<void> {
const session = await resumeSessionById(core.accessor, sessionId);
if (session === undefined) {
throw new Error2(ErrorCodes.SESSION_NOT_FOUND, `session ${sessionId} does not exist`);
}
const agent = await ensureMainAgent(session);
const profile = agent.accessor.get(IAgentProfileService);
if (agentConfig.model !== undefined && agentConfig.model !== '') {
await profile.setModel(agentConfig.model);
}
if (agentConfig.thinking !== undefined) {
profile.setThinking(agentConfig.thinking);
}
if (agentConfig.permission_mode !== undefined) {
agent.accessor
.get(IAgentLifecycleService)
.broadcastPermissionMode(agentConfig.permission_mode as PermissionMode);
}
if (agentConfig.plan_mode !== undefined) {
const plan = agent.accessor.get(IAgentPlanService);
const active = (await plan.status()) !== null;
if (active !== agentConfig.plan_mode) {
if (agentConfig.plan_mode) await plan.enter();
else plan.exit();
}
}
if (agentConfig.swarm_mode !== undefined) {
const swarm = agent.accessor.get(IAgentSwarmService);
if (swarm.isActive !== agentConfig.swarm_mode) {
if (agentConfig.swarm_mode) swarm.enter('manual');
else swarm.exit();
}
}
if (agentConfig.tower_mode !== undefined) {
const tower = agent.accessor.get(IAgentTowerService);
if (agentConfig.tower_mode) {
const result = await tower.enter(agentConfig.tower_base);
if (!result.entered) {
throw new Error2(
ErrorCodes.SESSION_TOWER_MODE_INVALID,
towerEnterFailureMessage(result),
);
}
} else {
await tower.exit();
}
}
if (agentConfig.goal_objective !== undefined) {
await agent.accessor
.get(IAgentGoalService)
.createGoal({ objective: agentConfig.goal_objective });
}
if (agentConfig.goal_control !== undefined) {
const goal = agent.accessor.get(IAgentGoalService);
switch (agentConfig.goal_control) {
case 'pause':
await goal.pauseGoal({});
break;
case 'resume':
await goal.resumeGoal({ continueIfPaused: true, continueIfBlocked: true });
break;
case 'cancel':
await goal.cancelGoal({});
break;
}
}
}
|