Spaces:
Sleeping
Sleeping
Commit
·
44e8e4f
1
Parent(s):
92c77a0
changes AgentType
Browse files
app.py
CHANGED
|
@@ -856,6 +856,7 @@ llm = HuggingFaceEndpoint(
|
|
| 856 |
)
|
| 857 |
chat_llm = ChatHuggingFace(llm=llm)
|
| 858 |
'''
|
|
|
|
| 859 |
chat_llm = ChatOpenAI(
|
| 860 |
openai_api_key=OPENAI_KEY,
|
| 861 |
model_name=OPENAI_MODEL,
|
|
@@ -863,15 +864,12 @@ chat_llm = ChatOpenAI(
|
|
| 863 |
# max_tokens=10
|
| 864 |
)
|
| 865 |
|
| 866 |
-
#
|
| 867 |
-
# tools = [search_tool, fetch_weather]
|
| 868 |
-
# chat_with_tools = chat.bind_tools(tools)
|
| 869 |
-
|
| 870 |
agent = initialize_agent(
|
| 871 |
tools=tools_list,
|
| 872 |
# llm=llm,
|
| 873 |
llm=chat_llm,
|
| 874 |
-
agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
|
| 875 |
agent_kwargs={"system_message": system_prompt},
|
| 876 |
verbose=True,
|
| 877 |
max_iterations=15, # Increase as needed
|
|
|
|
| 856 |
)
|
| 857 |
chat_llm = ChatHuggingFace(llm=llm)
|
| 858 |
'''
|
| 859 |
+
# Initialize the OpenAI chat model
|
| 860 |
chat_llm = ChatOpenAI(
|
| 861 |
openai_api_key=OPENAI_KEY,
|
| 862 |
model_name=OPENAI_MODEL,
|
|
|
|
| 864 |
# max_tokens=10
|
| 865 |
)
|
| 866 |
|
| 867 |
+
# Initialize the agent with the tools and system prompt
|
|
|
|
|
|
|
|
|
|
| 868 |
agent = initialize_agent(
|
| 869 |
tools=tools_list,
|
| 870 |
# llm=llm,
|
| 871 |
llm=chat_llm,
|
| 872 |
+
agent=AgentType.OPENAI_FUNCTIONS,#AgentType.ZERO_SHOT_REACT_DESCRIPTION,
|
| 873 |
agent_kwargs={"system_message": system_prompt},
|
| 874 |
verbose=True,
|
| 875 |
max_iterations=15, # Increase as needed
|