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
|