Update app.py

app.py

@@ -1,42 +1,21 @@
-import json
-from langchain.chains import SimpleChatChain
-from langchain.prompts import PromptTemplate
-from langchain.llms import OpenAI
+from langchain import hub
+from langchain.agents import AgentExecutor, create_json_chat_agent
+from langchain_community.tools.tavily_search import TavilySearchResults
+from langchain_openai import ChatOpenAI
 
 # Define your API key for OpenAI or any other LLM provider
-openai_api_key = 'sk-nAqoChT9cmkAxALwMLdWT3BIbkFJcNHsH5Z5LN2ixPcDAopT'
+OPENAI_API_KEY = 'sk-nAqoChT9cmkAxALwMLdWT3BIbkFJcNHsH5Z5LN2ixPcDAopT'
+openai.api_key=OPENAI_API_KEY 
 
-# Initialize the LLM
-llm = OpenAI(api_key=openai_api_key)
+tools = [TavilySearchResults(max_results=1)]
 
-# Define a simple chat chain with a prompt template
-chat_chain = SimpleChatChain(
-    llm=llm,
-    prompt_template=PromptTemplate(
-        input_variables=["user_input"],
-        template="User: {user_input}\nBot:"
-    )
-)
-
-def chat_with_json(input_json):
-    # Parse the input JSON
-    input_data = json.loads(input_json)
-    user_input = input_data.get('message', '')
-
-    # Generate a response using the chat chain
-    response = chat_chain.run(user_input)
+# Choose the LLM that will drive the agent
+llm = ChatOpenAI(openai_api_key=openai.api_key,temperature=0)
 
-    # Create the response JSON
-    response_json = json.dumps({'response': response})
-    return response_json
-
-# Example usage
-if __name__ == "__main__":
-    # Simulate a JSON input from the user
-    user_input_json = json.dumps({'message': 'Hello, how are you?'})
-    
-    # Get the response from the chatbot
-    response_json = chat_with_json(user_input_json)
-    
-    # Print the response JSON
-    print(response_json)
+# Construct the JSON agent
+agent = create_json_chat_agent(llm, tools, prompt)
+# Create an agent executor by passing in the agent and tools
+agent_executor = AgentExecutor(
+    agent=agent, tools=tools, verbose=True, handle_parsing_errors=True
+)
+agent_executor.invoke({"input": "what is LangChain?"})