Yongkang ZOU commited on
Commit
dfe572b
·
1 Parent(s): 66215da

update agent

Browse files
Files changed (1) hide show
  1. agent.py +18 -2
agent.py CHANGED
@@ -66,8 +66,24 @@ def arvix_search(query: str) -> str:
66
  docs = ArxivLoader(query=query, load_max_docs=3).load()
67
  return "\n\n".join([doc.page_content[:1000] for doc in docs])
68
 
69
-
70
- tools = [multiply, add, subtract, divide, modulus, wiki_search, web_search, arvix_search]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
71
 
72
  # ------------------- SYSTEM PROMPT -------------------
73
  system_prompt_path = "system_prompt.txt"
 
66
  docs = ArxivLoader(query=query, load_max_docs=3).load()
67
  return "\n\n".join([doc.page_content[:1000] for doc in docs])
68
 
69
+ @tool
70
+ def read_excel_file(path: str) -> str:
71
+ """Read an Excel file and return the first few rows of each sheet as text."""
72
+ import pandas as pd
73
+ try:
74
+ xls = pd.ExcelFile(path)
75
+ content = ""
76
+ for sheet in xls.sheet_names:
77
+ df = xls.parse(sheet)
78
+ content += f"Sheet: {sheet}\n"
79
+ content += df.head(5).to_string(index=False) + "\n\n"
80
+ return content.strip()
81
+ except Exception as e:
82
+ return f"Error reading Excel file: {str(e)}"
83
+
84
+
85
+
86
+ tools = [multiply, add, subtract, divide, modulus, wiki_search, web_search, arvix_search, read_excel_file]
87
 
88
  # ------------------- SYSTEM PROMPT -------------------
89
  system_prompt_path = "system_prompt.txt"