Spaces:
Sleeping
Sleeping
acecalisto3
commited on
Update app.py
Browse files
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=
|
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 |
)
|