aleixlopezpascual commited on
Commit
6233ccc
·
1 Parent(s): f1879ec
Files changed (2) hide show
  1. core_agent.py +1 -1
  2. requirements.txt +3 -1
core_agent.py CHANGED
@@ -319,7 +319,7 @@ class GAIAAgent:
319
  print(f"Initialized with {len(self.tools)} tools")
320
 
321
  # Setup imports allowed
322
- self.imports = ["pandas", "numpy", "datetime", "json", "re", "math", "os", "requests", "csv", "urllib"]
323
  if additional_imports:
324
  self.imports.extend(additional_imports)
325
 
 
319
  print(f"Initialized with {len(self.tools)} tools")
320
 
321
  # Setup imports allowed
322
+ self.imports = ["pandas", "numpy", "datetime", "json", "re", "math", "os", "io", "requests", "csv", "urllib", "PyPDF2", "pdfplumber", "bs4", "openpyxl"]
323
  if additional_imports:
324
  self.imports.extend(additional_imports)
325
 
requirements.txt CHANGED
@@ -5,4 +5,6 @@ python-dotenv
5
  pandas
6
  numpy
7
  openai
8
- duckduckgo-search
 
 
 
5
  pandas
6
  numpy
7
  openai
8
+ duckduckgo-search
9
+ Pillow
10
+ pytesseract