1import {
2	defineCommand,
3	defineQuery,
4} from '@lssm-tech/lib.contracts-spec/operations';
5import { defineSchemaModel, ScalarTypeEnum } from '@lssm-tech/lib.schema';
6import { GranularityEnum, LogLevelEnum, RunStatusEnum } from './run.enum';
7import {
8	RunInputModel,
9	RunLogModel,
10	RunModel,
11	RunStepModel,
12	RunSummaryModel,
13	TimelineDataPointModel,
14} from './run.schema';
15
16export const GetRunMetricsQuery = defineQuery({
17	meta: {
18		key: 'agent.run.getMetrics',
19		version: '1.0.0',
20		stability: 'stable',
21		owners: ['@agent-console-team'],
22		tags: ['run', 'metrics'],
23		description: 'Retrieves aggregated metrics for runs within a time period.',
24		goal: 'Monitor and analyze agent usage.',
25		context: 'Analytics dashboard.',
26	},
27	io: {
28		input: defineSchemaModel({
29			name: 'GetRunMetricsInput',
30			fields: {
31				organizationId: {
32					type: ScalarTypeEnum.String_unsecure(),
33					isOptional: false,
34				},
35				agentId: { type: ScalarTypeEnum.String_unsecure(), isOptional: true },
36				startDate: { type: ScalarTypeEnum.DateTime(), isOptional: false },
37				endDate: { type: ScalarTypeEnum.DateTime(), isOptional: false },
38				granularity: {
39					type: GranularityEnum,
40					isOptional: true,
41					defaultValue: 'day',
42				},
43			},
44		}),
45		output: defineSchemaModel({
46			name: 'GetRunMetricsOutput',
47			fields: {
48				totalRuns: { type: ScalarTypeEnum.Int_unsecure(), isOptional: false },
49				completedRuns: {
50					type: ScalarTypeEnum.Int_unsecure(),
51					isOptional: false,
52				},
53				failedRuns: { type: ScalarTypeEnum.Int_unsecure(), isOptional: false },
54				totalTokens: { type: ScalarTypeEnum.Int_unsecure(), isOptional: false },
55				totalCostUsd: {
56					type: ScalarTypeEnum.Float_unsecure(),
57					isOptional: false,
58				},
59				averageDurationMs: {
60					type: ScalarTypeEnum.Float_unsecure(),
61					isOptional: false,
62				},
63				successRate: {
64					type: ScalarTypeEnum.Float_unsecure(),
65					isOptional: false,
66				},
67				timeline: {
68					type: TimelineDataPointModel,
69					isArray: true,
70					isOptional: false,
71				},
72			},
73		}),
74	},
75	policy: { auth: 'user' },
76	acceptance: {
77		scenarios: [
78			{
79				key: 'get-run-metrics-happy-path',
80				given: ['Organization has run history'],
81				when: ['User requests metrics for date range'],
82				then: ['Aggregated metrics are returned'],
83			},
84		],
85		examples: [
86			{
87				key: 'get-daily-metrics',
88				input: {
89					organizationId: 'org-123',
90					startDate: '2025-01-01',
91					endDate: '2025-01-31',
92					granularity: 'day',
93				},
94				output: {
95					totalRuns: 100,
96					completedRuns: 90,
97					failedRuns: 10,
98					totalTokens: 50000,
99					totalCostUsd: 5.0,
100					averageDurationMs: 2500,
101					successRate: 0.9,
102					timeline: [],
103				},
104			},
105		],
106	},
107});