File size: 2,399 Bytes
4e23b01 | 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 | /**
* `modelResolver` — the engine's `IModelCatalog`: materialized model lookup
* plus the read-only catalog enumeration over configured providers and model
* aliases, and the global default-model selection. Mirrors
* `agent-core-v2/llm-adapter/model/catalog.ts`; wire shapes mirror
* `protocol/src/modelCatalog.ts` and `protocol/src/rest/modelCatalog.ts`
* (snake_case fields).
*/
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;
|