SystemReset commited on
Commit
f86e825
·
verified ·
1 Parent(s): 0a2c30f

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
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='deepinfra',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)
 
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)