kuzumab commited on
Commit
0649f54
·
verified ·
1 Parent(s): d5251ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -108,7 +108,7 @@ class BasicAgent:
108
  def __init__(self):
109
 
110
  self.think_model = OpenAIServerModel(
111
- model_id="Qwen/QwQ-32B-Preview",
112
  api_base="https://api.siliconflow.cn/v1/",
113
  api_key=os.getenv('MODEL_TOKEN'),
114
  )
@@ -125,8 +125,8 @@ class BasicAgent:
125
  attachment_tool=AttachmentDownloadTool()
126
  image_tool=ImageCaptionTool()
127
  audio_tool=AudioToTextTool()
128
- # self.tools = [attachment_tool,image_tool,audio_tool,DuckDuckGoSearchTool(), VisitWebpageTool()]
129
  self.tools = [attachment_tool,image_tool,audio_tool,DuckDuckGoSearchTool(), VisitWebpageTool()]
 
130
  web_agent = CodeAgent(
131
  tools=[DuckDuckGoSearchTool(), VisitWebpageTool()],
132
  model=self.base_model,
@@ -140,7 +140,7 @@ class BasicAgent:
140
  model=self.think_model,
141
  # managed_agents=[web_agent],
142
  additional_authorized_imports=["time", "numpy", "pandas"],
143
- max_steps=20
144
  )
145
  print("BasicAgent initialized.")
146
 
 
108
  def __init__(self):
109
 
110
  self.think_model = OpenAIServerModel(
111
+ model_id="deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
112
  api_base="https://api.siliconflow.cn/v1/",
113
  api_key=os.getenv('MODEL_TOKEN'),
114
  )
 
125
  attachment_tool=AttachmentDownloadTool()
126
  image_tool=ImageCaptionTool()
127
  audio_tool=AudioToTextTool()
 
128
  self.tools = [attachment_tool,image_tool,audio_tool,DuckDuckGoSearchTool(), VisitWebpageTool()]
129
+ # self.tools = [attachment_tool,image_tool,audio_tool]
130
  web_agent = CodeAgent(
131
  tools=[DuckDuckGoSearchTool(), VisitWebpageTool()],
132
  model=self.base_model,
 
140
  model=self.think_model,
141
  # managed_agents=[web_agent],
142
  additional_authorized_imports=["time", "numpy", "pandas"],
143
+ max_steps=10
144
  )
145
  print("BasicAgent initialized.")
146