Mehedi2 commited on
Commit
f400f72
·
verified ·
1 Parent(s): ccd0e62

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +6 -2
agent.py CHANGED
@@ -26,9 +26,13 @@ class MultiModelManager:
26
  """Manages multiple Groq models with rotation and fallback."""
27
 
28
  def __init__(self):
29
- # Using only GPT-OSS 120B as requested
 
 
 
 
30
  self.models = [
31
- "openai/gpt-oss-120b", # GPT OSS 120B - Most powerful
32
  ]
33
 
34
  self.api_key = os.getenv("GROQ_API_KEY")
 
26
  """Manages multiple Groq models with rotation and fallback."""
27
 
28
  def __init__(self):
29
+ # Alternative: Use a proven working Groq model
30
+ # If GPT-OSS still has issues, uncomment the line below:
31
+ # self.models = ["groq/llama-3.3-70b-versatile"]
32
+
33
+ # Current: Trying GPT-OSS 120B with groq/ prefix
34
  self.models = [
35
+ "groq/openai/gpt-oss-120b", # GPT OSS 120B via Groq
36
  ]
37
 
38
  self.api_key = os.getenv("GROQ_API_KEY")