ama2aifusion commited on
Commit
d6fd324
·
verified ·
1 Parent(s): 28c4815

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +7 -0
agent.py CHANGED
@@ -27,6 +27,7 @@ load_dotenv()
27
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
28
 
29
  ### ======================================== MATHEMATICAL TOOLS ======================================== ###
 
30
  def multiply(a: int, b: int) -> int:
31
  return a * b
32
 
@@ -148,6 +149,7 @@ arvix_search_tool = Tool(
148
 
149
 
150
  ### ======================================== CODE INTERPRETER TOOLS ======================================== ###
 
151
  def execute_code_multilang(code: str, language: str = "python") -> str:
152
  """Execute code in multiple languages (Python, Bash, SQL, C, Java) and return results.
153
  Args:
@@ -221,6 +223,9 @@ execute_code_multilang_tool = Tool(
221
  )
222
 
223
 
 
 
 
224
 
225
  # load the system prompt from the file
226
  with open("system_prompt.txt", "r", encoding="utf-8") as f:
@@ -240,6 +245,8 @@ tools = [
240
  wiki_search_tool,
241
  web_search_tool,
242
  arvix_search_tool,
 
 
243
  ]
244
 
245
  # Build graph function
 
27
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
28
 
29
  ### ======================================== MATHEMATICAL TOOLS ======================================== ###
30
+
31
  def multiply(a: int, b: int) -> int:
32
  return a * b
33
 
 
149
 
150
 
151
  ### ======================================== CODE INTERPRETER TOOLS ======================================== ###
152
+
153
  def execute_code_multilang(code: str, language: str = "python") -> str:
154
  """Execute code in multiple languages (Python, Bash, SQL, C, Java) and return results.
155
  Args:
 
223
  )
224
 
225
 
226
+ ### ======================================== DOCUMENT PROCESSING TOOLS ======================================== ###
227
+
228
+
229
 
230
  # load the system prompt from the file
231
  with open("system_prompt.txt", "r", encoding="utf-8") as f:
 
245
  wiki_search_tool,
246
  web_search_tool,
247
  arvix_search_tool,
248
+ execute_code_multilang_tool,
249
+
250
  ]
251
 
252
  # Build graph function