Fix: Restore ai_agent and add Gemini Env support

This commit is contained in:
Marcos
2026-03-22 10:19:54 -03:00
parent 6589c62b18
commit 8e5258d804
2 changed files with 158 additions and 223 deletions

View File

@@ -8,7 +8,7 @@ from config import get_config
def get_llm_response(prompt: str, provider: str, cfg: dict) -> str:
"""Invoca o provedor de LLM configurado."""
if provider == "gemini":
api_key = cfg.get("gemini_api_key")
api_key = cfg.get("gemini_api_key") or os.getenv("GEMINI_API_KEY")
url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent?key={api_key}"
payload = {"contents": [{"parts": [{"text": prompt}]}]}
res = requests.post(url, json=payload)