ninooo96 commited on
Commit
60a5b7c
·
1 Parent(s): 9aa497d

fix create_agent invocation

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. geminiAgent.py +1 -1
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import os
 
2
  import gradio as gr
3
  import requests
4
  import inspect
 
1
  import os
2
+ from Final_Assignment_Template.geminiAgent import create_agent
3
  import gradio as gr
4
  import requests
5
  import inspect
geminiAgent.py CHANGED
@@ -17,7 +17,7 @@ from PIL import Image
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(), DownloadTaskAttachmentTool()],
21
  add_base_tools=True,
22
  additional_authorized_imports=['pandas','numpy','csv','subprocess', 'exec']
23
  )
 
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(), DownloadTaskAttachmentTool(), ocr_tool, read_csv, read_excel],
21
  add_base_tools=True,
22
  additional_authorized_imports=['pandas','numpy','csv','subprocess', 'exec']
23
  )