bwilkie commited on
Commit
5bb3c4f
·
verified ·
1 Parent(s): c99f4b7

Update agent_simple.py

Browse files
Files changed (1) hide show
  1. agent_simple.py +2 -2
agent_simple.py CHANGED
@@ -74,7 +74,7 @@ class GroqRateLimiter:
74
  class GroqWrapper:
75
  """Wrapper for direct Groq API with rate limiting and error handling"""
76
 
77
- def __init__(self, model: str = "llama-3.1-70b-versatile",
78
  rpm: int = 30, tpm: int = 6000):
79
  self.client = Groq(api_key=os.getenv("GROQ_API_KEY"))
80
  self.model = model
@@ -244,7 +244,7 @@ tools = [
244
  arxiv_search,
245
  ]
246
 
247
- def build_graph(provider: str = "direct_groq", model: str = "llama-3.1-70b-versatile"):
248
  """Build the graph with direct Groq API and custom rate limiting"""
249
 
250
  if provider == "direct_groq":
 
74
  class GroqWrapper:
75
  """Wrapper for direct Groq API with rate limiting and error handling"""
76
 
77
+ def __init__(self, model: str = "qwen/qwen3-32b",
78
  rpm: int = 30, tpm: int = 6000):
79
  self.client = Groq(api_key=os.getenv("GROQ_API_KEY"))
80
  self.model = model
 
244
  arxiv_search,
245
  ]
246
 
247
+ def build_graph(provider: str = "direct_groq", model: str = "qwen/qwen3-32b"):
248
  """Build the graph with direct Groq API and custom rate limiting"""
249
 
250
  if provider == "direct_groq":