edithram23 commited on
Commit
1b14c2f
·
verified ·
1 Parent(s): 2930959

Update temp.py

Browse files
Files changed (1) hide show
  1. temp.py +1 -1
temp.py CHANGED
@@ -22,7 +22,7 @@ class Script:
22
  self.groq = ChatGroq(model='llama3-70b-8192')
23
  self.db = SQLDatabase.from_uri("sqlite:///sample_database.db")
24
  self.toolkit = SQLDatabaseToolkit(db=self.db, llm=self.model)
25
- search = TavilySearchResults(max_results=2)
26
  tools = self.toolkit.get_tools()+[search]
27
  self.agent = create_react_agent(self.model, tools)
28
 
 
22
  self.groq = ChatGroq(model='llama3-70b-8192')
23
  self.db = SQLDatabase.from_uri("sqlite:///sample_database.db")
24
  self.toolkit = SQLDatabaseToolkit(db=self.db, llm=self.model)
25
+ search = TavilySearchResults(name='Search-Tool',max_results=1,max_retries=3, description='A Search Tool that can be used to get zip codes, city, state, etc.')
26
  tools = self.toolkit.get_tools()+[search]
27
  self.agent = create_react_agent(self.model, tools)
28