acecalisto3 commited on
Commit
a044e48
·
verified ·
1 Parent(s): 2619fce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -11,6 +11,7 @@ from langchain.llms import HuggingFaceHub
11
  from langchain.agents import initialize_agent, AgentType
12
  from langchain.chains import LLMChain
13
  from langchain.prompts import PromptTemplate
 
14
 
15
  # Load environment variables
16
  load_dotenv()
@@ -165,9 +166,14 @@ class Agent:
165
  model_kwargs={"temperature": config['temperature']},
166
  huggingfacehub_api_token=config['api_key']
167
  )
 
 
 
 
 
168
  self.agent = initialize_agent(
169
  llm=self.llm,
170
- tools=self.tools,
171
  agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
172
  verbose=config['verbose']
173
  )
 
11
  from langchain.agents import initialize_agent, AgentType
12
  from langchain.chains import LLMChain
13
  from langchain.prompts import PromptTemplate
14
+ from langchain.tools import Tool
15
 
16
  # Load environment variables
17
  load_dotenv()
 
166
  model_kwargs={"temperature": config['temperature']},
167
  huggingfacehub_api_token=config['api_key']
168
  )
169
+ # Convert tools to dictionaries
170
+ tools_dict = [
171
+ {"name": tool.name, "description": tool.description, "func": tool.run}
172
+ for tool in self.tools
173
+ ]
174
  self.agent = initialize_agent(
175
  llm=self.llm,
176
+ tools=tools_dict,
177
  agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
178
  verbose=config['verbose']
179
  )