Update agent_simple.py
Browse files- 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 = "
|
| 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 = "
|
| 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":
|