Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -62,7 +62,7 @@ def guest_search(query: str) -> str:
|
|
| 62 |
def build_agent(use_case=False):
|
| 63 |
token=os.environ.get('HF_TOKEN')
|
| 64 |
if not token: raise RuntimeError('Add HF_TOKEN to Space secrets.')
|
| 65 |
-
model=InferenceClientModel(token=token,provider='
|
| 66 |
tools=[web_search,visit_webpage]
|
| 67 |
if use_case: tools.append(guest_search)
|
| 68 |
return CodeAgent(tools=tools,model=model,max_steps=8)
|
|
|
|
| 62 |
def build_agent(use_case=False):
|
| 63 |
token=os.environ.get('HF_TOKEN')
|
| 64 |
if not token: raise RuntimeError('Add HF_TOKEN to Space secrets.')
|
| 65 |
+
model=InferenceClientModel(token=token,provider='cerebras',max_tokens=4096,model_id=os.environ.get('MODEL_ID','Qwen/Qwen3-235B-A22B-Instruct-2507'))
|
| 66 |
tools=[web_search,visit_webpage]
|
| 67 |
if use_case: tools.append(guest_search)
|
| 68 |
return CodeAgent(tools=tools,model=model,max_steps=8)
|