diff --git a/src/config/llm.ts b/src/config/llm.ts index 312fa38b..6ec7736c 100644 --- a/src/config/llm.ts +++ b/src/config/llm.ts @@ -28,5 +28,9 @@ export const llmDefaultConfig = { size: LLMSize.SMALL, temperature: 0.8, } as LLMNodeConfiguration, + orchestrator: { + size: LLMSize.LARGE, + temperature: 0.2, + } as LLMNodeConfiguration, }, }; diff --git a/src/config/schema.ts b/src/config/schema.ts index d94fa421..71175c83 100644 --- a/src/config/schema.ts +++ b/src/config/schema.ts @@ -46,6 +46,10 @@ const llmConfigSchema = z size: z.nativeEnum(LLMSize), temperature: z.number(), }), + orchestrator: z.object({ + size: z.nativeEnum(LLMSize), + temperature: z.number(), + }), }), OPENAI_API_KEY: z.string(), ANTHROPIC_API_KEY: z.string(), @@ -65,6 +69,9 @@ const llmConfigSchema = z data.nodes.response.size === LLMSize.LARGE ? data.configuration.large.provider : data.configuration.small.provider, + data.nodes.orchestrator.size === LLMSize.LARGE + ? data.configuration.large.provider + : data.configuration.small.provider, ]); const missingConfigs = [];