ninooo96 commited on
Commit
6366d44
·
1 Parent(s): e04d19e

update model reference in create_agent function

Browse files
Files changed (1) hide show
  1. 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']