Bhanu-Chander-ABB commited on
Commit
44e8e4f
·
1 Parent(s): 92c77a0

changes AgentType

Browse files
Files changed (1) hide show
  1. app.py +3 -5
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
- # chat = ChatHuggingFace(llm=llm, verbose=True)
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