Spaces:
Running
Running
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -214,7 +214,7 @@ def _parse(raw: str) -> dict:
|
|
| 214 |
def _groq(client, prompt):
|
| 215 |
try:
|
| 216 |
r = client.chat.completions.create(model=GROQ_MODEL,
|
| 217 |
-
messages=[{"role":"user","content":prompt}], temperature=0
|
| 218 |
return _parse(r.choices[0].message.content)
|
| 219 |
except Exception as e: logger.warning("Groq: %s", e); return {}
|
| 220 |
|
|
@@ -224,7 +224,7 @@ def _mistral(prompt, key):
|
|
| 224 |
r = requests.post("https://api.mistral.ai/v1/chat/completions",
|
| 225 |
headers={"Authorization":f"Bearer {key}","Content-Type":"application/json"},
|
| 226 |
json={"model":MISTRAL_MODEL,"messages":[{"role":"user","content":prompt}],
|
| 227 |
-
"temperature":0
|
| 228 |
return _parse(r.json()["choices"][0]["message"]["content"])
|
| 229 |
except Exception as e: logger.warning("Mistral: %s", e); return {}
|
| 230 |
|
|
@@ -238,7 +238,7 @@ def _gemini(prompt, key):
|
|
| 238 |
f"{model}:generateContent?key={key}",
|
| 239 |
headers={"Content-Type":"application/json"},
|
| 240 |
json={"contents":[{"parts":[{"text":prompt}]}],
|
| 241 |
-
"generationConfig":{"temperature":0
|
| 242 |
d = r.json()
|
| 243 |
if "candidates" not in d:
|
| 244 |
err = d.get("error",{})
|
|
|
|
| 214 |
def _groq(client, prompt):
|
| 215 |
try:
|
| 216 |
r = client.chat.completions.create(model=GROQ_MODEL,
|
| 217 |
+
messages=[{"role":"user","content":prompt}], temperature=0, timeout=30)
|
| 218 |
return _parse(r.choices[0].message.content)
|
| 219 |
except Exception as e: logger.warning("Groq: %s", e); return {}
|
| 220 |
|
|
|
|
| 224 |
r = requests.post("https://api.mistral.ai/v1/chat/completions",
|
| 225 |
headers={"Authorization":f"Bearer {key}","Content-Type":"application/json"},
|
| 226 |
json={"model":MISTRAL_MODEL,"messages":[{"role":"user","content":prompt}],
|
| 227 |
+
"temperature":0}, timeout=30)
|
| 228 |
return _parse(r.json()["choices"][0]["message"]["content"])
|
| 229 |
except Exception as e: logger.warning("Mistral: %s", e); return {}
|
| 230 |
|
|
|
|
| 238 |
f"{model}:generateContent?key={key}",
|
| 239 |
headers={"Content-Type":"application/json"},
|
| 240 |
json={"contents":[{"parts":[{"text":prompt}]}],
|
| 241 |
+
"generationConfig":{"temperature":0}}, timeout=60)
|
| 242 |
d = r.json()
|
| 243 |
if "candidates" not in d:
|
| 244 |
err = d.get("error",{})
|