Spaces:
Sleeping
Sleeping
update model reference in create_agent function
Browse files- geminiAgent.py +1 -1
geminiAgent.py
CHANGED
|
@@ -16,7 +16,7 @@ from PIL import Image
|
|
| 16 |
|
| 17 |
def create_agent():
|
| 18 |
return CodeAgent(
|
| 19 |
-
model=LiteLLMModel(model_id="deepseek/deepseek-r1-distill-llama-70b:free", api_key=os.getenv("DEEPSEEK_KEY")),
|
| 20 |
tools=[DuckDuckGoSearchTool(), WikipediaSearchTool(), VisitWebpageTool(), ocr_tool, read_csv, read_excel],
|
| 21 |
add_base_tools=True,
|
| 22 |
additional_authorized_imports=['pandas','numpy','csv','subprocess', 'exec']
|
|
|
|
| 16 |
|
| 17 |
def create_agent():
|
| 18 |
return CodeAgent(
|
| 19 |
+
model=LiteLLMModel(model_id="openrouter/deepseek/deepseek-r1-distill-llama-70b:free", api_key=os.getenv("DEEPSEEK_KEY")),
|
| 20 |
tools=[DuckDuckGoSearchTool(), WikipediaSearchTool(), VisitWebpageTool(), ocr_tool, read_csv, read_excel],
|
| 21 |
add_base_tools=True,
|
| 22 |
additional_authorized_imports=['pandas','numpy','csv','subprocess', 'exec']
|