diff --git a/backend/app/services/knowledge_base_service.py b/backend/app/services/knowledge_base_service.py index c58d90a..bfcabaa 100644 --- a/backend/app/services/knowledge_base_service.py +++ b/backend/app/services/knowledge_base_service.py @@ -292,7 +292,7 @@ class KnowledgeBaseService: prompt = prompt_template.format(combined_markdown=combined_markdown) logger.info(f"[KB_SERVICE] Sending {len(combined_markdown)} chars to Gemini for distillation") - response = await self.gemini.client.aio.models.generate_content( + response = await self.gemini.primary_client.aio.models.generate_content( model=self.gemini.model, contents=prompt, )