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 AssignToolToAgentCommand = defineCommand({
16	meta: {
17		key: 'agent-console.agent.assignTool',
18		version: '1.0.0',
19		stability: 'stable',
20		owners: ['@agent-console-team'],
21		tags: ['agent', 'tool', 'assign'],
22		description: 'Assigns a tool to an agent with optional configuration.',
23		goal: 'Enable agents to use specific tools.',
24		context: 'Agent configuration UI - tool selection.',
25	},
26	io: {
27		input: defineSchemaModel({
28			name: 'AssignToolToAgentInput',
29			fields: {
30				agentId: { type: ScalarTypeEnum.String_unsecure(), isOptional: false },
31				toolId: { type: ScalarTypeEnum.String_unsecure(), isOptional: false },
32				config: { type: ScalarTypeEnum.JSONObject(), isOptional: true },
33				order: { type: ScalarTypeEnum.Int_unsecure(), isOptional: true },
34			},
35		}),
36		output: defineSchemaModel({
37			name: 'AssignToolToAgentOutput',
38			fields: {
39				agentToolId: {
40					type: ScalarTypeEnum.String_unsecure(),
41					isOptional: false,
42				},
43				agentId: { type: ScalarTypeEnum.String_unsecure(), isOptional: false },
44				toolId: { type: ScalarTypeEnum.String_unsecure(), isOptional: false },
45			},
46		}),
47		errors: {
48			TOOL_ALREADY_ASSIGNED: {
49				description: 'This tool is already assigned to the agent',
50				http: 409,
51				gqlCode: 'TOOL_ALREADY_ASSIGNED',
52				when: 'Tool assignment already exists',
53			},
54		},
55	},
56	policy: { auth: 'user' },
57	sideEffects: { audit: ['agent.tool.assigned'] },
58	acceptance: {
59		scenarios: [
60			{
61				key: 'assign-tool-happy-path',
62				given: ['Agent exists', 'Tool exists and is not assigned'],
63				when: ['User assigns tool to agent'],
64				then: ['Tool is assigned', 'Assignment ID is returned'],
65			},
66			{
67				key: 'assign-tool-already-assigned',
68				given: ['Tool is already assigned to agent'],
69				when: ['User attempts to assign again'],
70				then: ['TOOL_ALREADY_ASSIGNED error is returned'],
71			},
72		],
73		examples: [
74			{
75				key: 'assign-basic',
76				input: { agentId: 'agent-123', toolId: 'tool-456' },
77				output: {
78					agentToolId: 'at-789',
79					agentId: 'agent-123',
80					toolId: 'tool-456',
81				},
82			},
83		],
84	},
85});