From 95559551a3f53ba7af2689f78cb19c7e22f030c1 Mon Sep 17 00:00:00 2001 From: trojanhorse47 Date: Fri, 19 Sep 2025 11:39:11 +0200 Subject: [PATCH] =?UTF-8?q?*=20Modificati=20caricamenti=20dei=20provider?= =?UTF-8?q?=20google=20e=20anthropic.=20*=20Modificata=20=5Fpredict=5Fgoog?= =?UTF-8?q?le:=20ora=20=C3=A8=20funzionante.=20+=20Aggiunto=20debug=20per?= =?UTF-8?q?=20provider=20attivi=20trovati.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit # NB: Sia le chiamate che le chiamate agli modelli potrebbero non essere aggiornati (non ho ancora letto la documentazione recente e potrebbero essere da modificare). --- src/app/agents/predictor_agent.py | 24 ++++++++++++++++-------- 1 file changed, 16 insertions(+), 8 deletions(-) diff --git a/src/app/agents/predictor_agent.py b/src/app/agents/predictor_agent.py index f461c83..900539c 100644 --- a/src/app/agents/predictor_agent.py +++ b/src/app/agents/predictor_agent.py @@ -26,13 +26,17 @@ class PredictorAgent: # Anthropic anthropic_key = os.getenv("ANTHROPIC_API_KEY") if anthropic_key: - client = anthropic.Client(api_key=anthropic_key) - self.providers["anthropic"] = {"type": "anthropic", "client": client, "model": "claude-3"} + client = anthropic.Anthropic(api_key=anthropic_key) + self.providers["anthropic"] = { + "type": "anthropic", + "client": client, + "model": "claude-3-haiku-20240307" + } # Google Gemini - google_key = os.getenv("GEMINI_API_KEY") + google_key = os.getenv("GOOGLE_API_KEY") or os.getenv("GEMINI_API_KEY") if google_key: - client = Client(credentials={"api_key": google_key}) + client = Client(api_key=google_key) self.providers["google"] = {"type": "google", "client": client, "model": "gemini-1.5-flash"} # DeepSeek @@ -40,6 +44,8 @@ class PredictorAgent: if deepseek_key: self.providers["deepseek"] = {"type": "deepseek", "api_key": deepseek_key, "model": "deepseek-chat"} + print("✅ Providers attivi:", list(self.providers.keys())) + def predict(self, data, sentiment, style="conservative", provider="mock"): provider = provider.lower() if provider == "mock" or provider not in self.providers: @@ -114,11 +120,13 @@ class PredictorAgent: @staticmethod def _predict_google(prompt, client, model): - response = client.generate_text( + response = client.models.generate_content( model=model, - prompt=prompt, - max_output_tokens=300, - temperature=0.7 + contents=prompt, + config={ + "temperature": 0.7, + "max_output_tokens": 300 + } ) return response.text.strip()