Update app.py
Browse files
app.py
CHANGED
@@ -108,7 +108,7 @@ class BasicAgent:
|
|
108 |
def __init__(self):
|
109 |
|
110 |
self.think_model = OpenAIServerModel(
|
111 |
-
model_id="
|
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=
|
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 |
|