File size: 4,037 Bytes
9853396 | 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 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 | import { z } from 'zod';
import { pageInfoSchema } from '@/gql/pagination';
import { apiKeySchema } from '@/features/apikeys/data/schema';
import { channelSchema } from '@/features/channels/data';
import { usageLogSchema } from '@/features/usage-logs/data/schema';
// Request Status
export const requestStatusSchema = z.enum(['pending', 'processing', 'completed', 'failed', 'canceled']);
export type RequestStatus = z.infer<typeof requestStatusSchema>;
// Request Source
export const requestSourceSchema = z.enum(['api', 'playground', 'test']);
export type RequestSource = z.infer<typeof requestSourceSchema>;
// Request Execution Status
export const requestExecutionStatusSchema = z.enum(['pending', 'processing', 'completed', 'failed', 'canceled']);
export type RequestExecutionStatus = z.infer<typeof requestExecutionStatusSchema>;
// Request Execution
export const requestExecutionSchema = z.object({
id: z.string(),
createdAt: z.coerce.date(),
updatedAt: z.coerce.date(),
// userID: z.number(),
// requestID: z.string(),
// channelID: z.number(),
channel: channelSchema.partial().nullable().optional(),
modelID: z.string(),
requestHeaders: z.any().nullable().optional(),
requestBody: z.any(), // JSONRawMessage
responseBody: z.any().nullable(), // JSONRawMessage
responseChunks: z.array(z.any()).nullable(), // [JSONRawMessage!]
errorMessage: z.string().nullable(),
status: requestExecutionStatusSchema,
format: z.string().optional(),
metricsLatencyMs: z.number().nullable().optional(),
metricsFirstTokenLatencyMs: z.number().nullable().optional(),
});
export type RequestExecution = z.infer<typeof requestExecutionSchema>;
// Request
export const requestSchema = z.object({
id: z.string(),
createdAt: z.coerce.date(),
updatedAt: z.coerce.date(),
// apiKeyID: z.string().optional().nullable(),
apiKey: apiKeySchema.partial().nullable().optional(),
// channelID: z.string().optional().nullable(),
channel: channelSchema.partial().nullable().optional(),
source: requestSourceSchema,
modelID: z.string(),
requestHeaders: z.any().nullable().optional(),
requestBody: z.any().nullable().optional(), // JSONRawMessage
responseBody: z.any().nullable().optional(), // JSONRawMessage
responseChunks: z.array(z.any()).nullable().optional(), // [JSONRawMessage!]
status: requestStatusSchema,
format: z.string().optional(),
clientIP: z.string().nullable().optional(),
stream: z.boolean().nullable(),
metricsLatencyMs: z.number().nullable().optional(),
metricsFirstTokenLatencyMs: z.number().nullable().optional(),
executions: z
.object({
edges: z.array(
z.object({
node: requestExecutionSchema.partial().nullable().optional(),
cursor: z.string(),
})
),
pageInfo: pageInfoSchema,
totalCount: z.number(),
})
.optional(),
usageLogs: z
.object({
edges: z
.array(
z.object({
node: usageLogSchema.partial().nullable().optional(),
cursor: z.string().optional(),
})
)
.optional(),
pageInfo: pageInfoSchema.optional(),
})
.optional()
.nullable(),
});
export type Request = z.infer<typeof requestSchema>;
// Request Connection (for pagination)
export const requestConnectionSchema = z.object({
edges: z.array(
z.object({
node: requestSchema,
cursor: z.string(),
})
),
pageInfo: pageInfoSchema,
totalCount: z.number(),
});
export type RequestConnection = z.infer<typeof requestConnectionSchema>;
// Request Execution Connection (for pagination)
export const requestExecutionConnectionSchema = z.object({
edges: z.array(
z.object({
node: requestExecutionSchema,
cursor: z.string(),
})
),
pageInfo: pageInfoSchema,
totalCount: z.number(),
});
export type RequestExecutionConnection = z.infer<typeof requestExecutionConnectionSchema>;
|