Daniel Amendoeira commited on
Commit
766af75
·
verified ·
1 Parent(s): 3ee8738

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -1
agent.py CHANGED
@@ -1,7 +1,9 @@
1
  import os
2
 
3
  from langchain_openai import ChatOpenAI
4
- from langchain_community.tools import BraveSearch
 
 
5
 
6
  # gpt-4.1-nano (cheaper for debugging) with temperature 0 for less randomness
7
  # for benchmarking use gpt-4.1-mini or 04-mini
 
1
  import os
2
 
3
  from langchain_openai import ChatOpenAI
4
+ from langchain_community.tools import BraveSearch # web search
5
+ from langchain.tools import Calculator # for basic math
6
+ from langchain.tools.python.tool import PythonAstREPLTool # for logic/math problems
7
 
8
  # gpt-4.1-nano (cheaper for debugging) with temperature 0 for less randomness
9
  # for benchmarking use gpt-4.1-mini or 04-mini