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

update model to use deepseek-r1-distill-llama-70b

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="gemini/gemini-2.0-flash-lite-001", api_key=os.getenv("GEMINI_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="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']