Spaces:
Sleeping
Sleeping
Update temp.py
Browse files
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=
|
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 |
|