Romain Lembo commited on
Commit
cb49713
·
1 Parent(s): 35d86b1

Refactor tool imports and update tools list to include analyze_csv_file, analyze_excel_file, and get_image_properties

Browse files
Files changed (1) hide show
  1. agent.py +14 -12
agent.py CHANGED
@@ -14,23 +14,24 @@ from supabase.client import Client, create_client
14
  from langchain_litellm import ChatLiteLLM
15
 
16
  # Tools
17
- from tools.wiki_search import wiki_search
18
- from tools.web_search import web_search
19
- from tools.arvix_search import arvix_search
20
- from tools.python_code_parser import python_code_parser
21
- from tools.multiply import multiply
22
  from tools.add import add
 
 
 
23
  from tools.divide import divide
 
 
 
 
24
  from tools.modulus import modulus
 
25
  from tools.power import power
26
- from tools.subtract import subtract
27
- from tools.square_root import square_root
28
- from tools.execute_code_multilang import execute_code_multilang
29
  from tools.save_read_file import save_read_file
30
- from tools.download_file import download_file
31
- from tools.extract_text_from_image import extract_text_from_image
32
- from tools.analyze_csv_file import analyze_csv_file
33
- from tools.analyze_excel_file import analyze_excel_file
34
 
35
  load_dotenv()
36
 
@@ -76,6 +77,7 @@ tools = [
76
  extract_text_from_image,
77
  analyze_csv_file,
78
  analyze_excel_file,
 
79
  ]
80
 
81
  # Build graph function
 
14
  from langchain_litellm import ChatLiteLLM
15
 
16
  # Tools
 
 
 
 
 
17
  from tools.add import add
18
+ from tools.analyze_csv_file import analyze_csv_file
19
+ from tools.analyze_excel_file import analyze_excel_file
20
+ from tools.arvix_search import arvix_search
21
  from tools.divide import divide
22
+ from tools.download_file import download_file
23
+ from tools.execute_code_multilang import execute_code_multilang
24
+ from tools.extract_text_from_image import extract_text_from_image
25
+ from tools.get_image_properties import get_image_properties
26
  from tools.modulus import modulus
27
+ from tools.multiply import multiply
28
  from tools.power import power
29
+ from tools.python_code_parser import python_code_parser
 
 
30
  from tools.save_read_file import save_read_file
31
+ from tools.square_root import square_root
32
+ from tools.subtract import subtract
33
+ from tools.web_search import web_search
34
+ from tools.wiki_search import wiki_search
35
 
36
  load_dotenv()
37
 
 
77
  extract_text_from_image,
78
  analyze_csv_file,
79
  analyze_excel_file,
80
+ get_image_properties,
81
  ]
82
 
83
  # Build graph function