1 files changed, 8 insertions, 2 deletions
diff --git a/packages/server/src/core/client.ts b/packages/server/src/core/client.ts
index fc6080ab..3b5f7a99 100644
--- a/packages/server/src/core/client.ts
+++ b/packages/server/src/core/client.ts
@@ -193,12 +193,18 @@ export class GeminiClient {
   async generateJson(
     contents: Content[],
     schema: SchemaUnion,
+    model: string = 'gemini-2.0-flash',
+    config: GenerateContentConfig = {},
   ): Promise<Record<string, unknown>> {
     try {
+      const requestConfig = {
+        ...this.generateContentConfig,
+        ...config,
+      };
       const result = await this.client.models.generateContent({
-        model: 'gemini-2.0-flash',
+        model,
         config: {
-          ...this.generateContentConfig,
+          ...requestConfig,
           systemInstruction: getCoreSystemPrompt(),
           responseSchema: schema,
           responseMimeType: 'application/json',