diff --git a/src/config/types.models.ts b/src/config/types.models.ts index 11b6c64cbc4..3d993cbf610 100644 --- a/src/config/types.models.ts +++ b/src/config/types.models.ts @@ -4,7 +4,8 @@ export type ModelApi = | "anthropic-messages" | "google-generative-ai" | "github-copilot" - | "bedrock-converse-stream"; + | "bedrock-converse-stream" + | "ollama"; export type ModelCompatConfig = { supportsStore?: boolean; diff --git a/src/config/zod-schema.core.ts b/src/config/zod-schema.core.ts index b7da9208a7a..2da5c357cb6 100644 --- a/src/config/zod-schema.core.ts +++ b/src/config/zod-schema.core.ts @@ -9,6 +9,7 @@ export const ModelApiSchema = z.union([ z.literal("google-generative-ai"), z.literal("github-copilot"), z.literal("bedrock-converse-stream"), + z.literal("ollama"), ]); export const ModelCompatSchema = z