BHAVIKBANKER commited on
Commit
80d15c2
·
verified ·
1 Parent(s): e161996

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -3
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.2, timeout=30)
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.2}, timeout=30)
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.2}}, timeout=60)
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",{})