Procházet zdrojové kódy

changes for gemini model

VISHAL BHANUSHALI před 3 měsíci
rodič
revize
ed6f1a5ea4
1 změnil soubory, kde provedl 2 přidání a 2 odebrání
  1. 2 2
      core/services/gemini_service.py

+ 2 - 2
core/services/gemini_service.py

@@ -469,7 +469,7 @@ class GeminiAttributeService:
         if not api_key:
             raise ValueError("GEMINI_API_KEY not found in settings")
         genai.configure(api_key=api_key)
-        self.model = genai.GenerativeModel('gemini-2.5-flash-exp')
+        self.model = genai.GenerativeModel('gemini-2.5-flash')
     
     @retry(
         stop=stop_after_attempt(3),
@@ -510,7 +510,7 @@ class GeminiAttributeService:
             
             prompt = self._build_comprehensive_prompt(product, limited_issues, category_rules, scores)
             response = self._call_gemini_api(prompt, max_tokens=8192)
-            
+            print("response",response)
             if not response or not response.candidates:
                 logger.error(f"No candidates returned for SKU: {product.get('sku')}")
                 return {