add orchestrator node type to LLM options

autonomys · Jan 20, 2025 · ba4e232 · ba4e232
1 parent 8d8dd70
commit ba4e232
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git 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
@@ -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 = [];