1import {
2	defineCommand,
3	defineQuery,
4} from '@lssm-tech/lib.contracts-spec/operations';
5import { defineSchemaModel, ScalarTypeEnum } from '@lssm-tech/lib.schema';
6import { AgentStatusEnum, ModelProviderEnum } from './agent.enum';
7import { AgentCreatedEvent } from './agent.event';
8import {
9	AgentSummaryModel,
10	AgentWithToolsModel,
11	CreateAgentInputModel,
12	UpdateAgentInputModel,
13} from './agent.schema';
14
15export const UpdateAgentCommand = defineCommand({
16	meta: {
17		key: 'agent-console.agent.update',
18		version: '1.0.0',
19		stability: 'stable',
20		owners: ['@agent-console-team'],
21		tags: ['agent', 'update'],
22		description: 'Updates an existing AI agent configuration.',
23		goal: 'Allow users to modify agent settings and configuration.',
24		context: 'Called from the agent settings UI.',
25	},
26	io: {
27		input: UpdateAgentInputModel,
28		output: defineSchemaModel({
29			name: 'UpdateAgentOutput',
30			fields: {
31				id: { type: ScalarTypeEnum.String_unsecure(), isOptional: false },
32				name: { type: ScalarTypeEnum.NonEmptyString(), isOptional: false },
33				status: { type: AgentStatusEnum, isOptional: false },
34				updatedAt: { type: ScalarTypeEnum.DateTime(), isOptional: false },
35			},
36		}),
37		errors: {
38			AGENT_NOT_FOUND: {
39				description: 'The specified agent does not exist',
40				http: 404,
41				gqlCode: 'AGENT_NOT_FOUND',
42				when: 'Agent ID is invalid',
43			},
44		},
45	},
46	policy: { auth: 'user' },
47	sideEffects: {
48		emits: [
49			{
50				key: 'agent.updated',
51				version: '1.0.0',
52				stability: 'stable',
53				owners: ['@agent-console-team'],
54				tags: ['agent', 'updated'],
55				when: 'Agent is updated',
56				payload: AgentSummaryModel,
57			},
58		],
59		audit: ['agent.updated'],
60	},
61	acceptance: {
62		scenarios: [
63			{
64				key: 'update-agent-happy-path',
65				given: ['Agent exists', 'User owns the agent'],
66				when: ['User submits updated configuration'],
67				then: ['Agent is updated', 'AgentUpdated event is emitted'],
68			},
69			{
70				key: 'update-agent-not-found',
71				given: ['Agent does not exist'],
72				when: ['User attempts to update'],
73				then: ['AGENT_NOT_FOUND error is returned'],
74			},
75		],
76		examples: [
77			{
78				key: 'update-name',
79				input: {
80					agentId: 'agent-123',
81					name: 'Updated Assistant',
82					systemPrompt: 'You are a helpful assistant.',
83				},
84				output: {
85					id: 'agent-123',
86					name: 'Updated Assistant',
87					status: 'draft',
88					updatedAt: '2025-01-01T00:00:00Z',
89				},
90			},
91		],
92	},
93});