| |
| |
| |
| |
| |
| |
| |
| |
|
|
| import { z } from 'zod'; |
|
|
| import type { ServiceContract, StreamingProcedureContract } from '../types.js'; |
|
|
| export const modelCatalogItemSchema = z.object({ |
| provider: z.string(), |
| model: z.string(), |
| display_name: z.string().optional(), |
| max_context_size: z.number(), |
| capabilities: z.array(z.string()).optional(), |
| support_efforts: z.array(z.string()).optional(), |
| default_effort: z.string().optional(), |
| }); |
|
|
| export const providerCatalogStatusSchema = z.enum(['connected', 'error', 'unconfigured']); |
|
|
| export const providerCatalogItemSchema = z.object({ |
| id: z.string(), |
| type: z.string(), |
| base_url: z.string().optional(), |
| default_model: z.string().optional(), |
| has_api_key: z.boolean(), |
| status: providerCatalogStatusSchema, |
| models: z.array(z.string()).optional(), |
| }); |
|
|
| export const setDefaultModelResponseSchema = z.object({ |
| default_model: z.string(), |
| model: modelCatalogItemSchema, |
| }); |
|
|
| const generateInputSchema = z.object({ |
| systemPrompt: z.string(), |
| messages: z.array(z.unknown()), |
| tools: z.array(z.unknown()).optional(), |
| responseFormat: z.unknown().optional(), |
| }); |
|
|
| const generateParamsSchema = z.object({ |
| cacheKey: z.string().optional(), |
| temperature: z.number().optional(), |
| topP: z.number().optional(), |
| thinkingEffort: z.string().optional(), |
| maxCompletionTokens: z.number().optional(), |
| }).optional(); |
|
|
| const generateEventSchema = z.object({ |
| type: z.string(), |
| }).passthrough(); |
|
|
| export const catalogContract = { |
| listModels: { input: z.tuple([]), output: z.array(modelCatalogItemSchema) }, |
| listProviders: { input: z.tuple([]), output: z.array(providerCatalogItemSchema) }, |
| getProvider: { input: z.tuple([z.string()]), output: providerCatalogItemSchema }, |
| setDefaultModel: { input: z.tuple([z.string()]), output: setDefaultModelResponseSchema }, |
| generate: { |
| input: z.tuple([z.string(), generateInputSchema, generateParamsSchema]), |
| chunk: generateEventSchema, |
| streaming: true, |
| } as StreamingProcedureContract, |
| } satisfies ServiceContract; |
|
|
