Update agent initialization parameters and model reference in app.py
Browse files
app.py
CHANGED
@@ -168,18 +168,18 @@ def generate_audio(prompt: str) -> object:
|
|
168 |
## agent definition
|
169 |
class Agent:
|
170 |
def __init__(self, ):
|
171 |
-
client = HfApiModel("deepseek-ai/DeepSeek-R1
|
172 |
"""client = OpenAIServerModel(
|
173 |
model_id="claude-sonnet-4-20250514",
|
174 |
api_base="https://api.anthropic.com/v1/",
|
175 |
api_key=os.environ["ANTHROPIC_API_KEY"],
|
176 |
-
|
177 |
self.agent = CodeAgent(
|
178 |
model=client,
|
179 |
tools=[DuckDuckGoSearchTool(max_results=5), VisitWebpageTool(max_output_length=20000), generate_image, download_images, transcribe_audio],
|
180 |
additional_authorized_imports=["pandas", "PIL", "io"],
|
181 |
-
planning_interval=
|
182 |
-
max_steps=
|
183 |
stream_outputs=False,
|
184 |
final_answer_checks=[check_format]
|
185 |
)
|
|
|
168 |
## agent definition
|
169 |
class Agent:
|
170 |
def __init__(self, ):
|
171 |
+
client = HfApiModel("deepseek-ai/DeepSeek-R1", provider="nebius", api_key=os.getenv("NEBIUS_API_KEY"))
|
172 |
"""client = OpenAIServerModel(
|
173 |
model_id="claude-sonnet-4-20250514",
|
174 |
api_base="https://api.anthropic.com/v1/",
|
175 |
api_key=os.environ["ANTHROPIC_API_KEY"],
|
176 |
+
)"""
|
177 |
self.agent = CodeAgent(
|
178 |
model=client,
|
179 |
tools=[DuckDuckGoSearchTool(max_results=5), VisitWebpageTool(max_output_length=20000), generate_image, download_images, transcribe_audio],
|
180 |
additional_authorized_imports=["pandas", "PIL", "io"],
|
181 |
+
planning_interval=3,
|
182 |
+
max_steps=6,
|
183 |
stream_outputs=False,
|
184 |
final_answer_checks=[check_format]
|
185 |
)
|