Update app.py

app.py

@@ -12,6 +12,7 @@ from langgraph.prebuilt import tools_condition
 from langgraph.graph import StateGraph, START, END
 from langchain_community.tools import DuckDuckGoSearchRun
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
+from langchain.chat_models import ChatOpenAI
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -26,9 +27,16 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         
-        llm = HuggingFaceEndpoint(
-        repo_id="mistralai/Mistral-7B-v0.1",
-        huggingfacehub_api_token=os.environ.get("HUGGINGFACEHUB_API_TOKEN"),
+        # llm = HuggingFaceEndpoint(
+        # repo_id="mistralai/Mistral-7B-v0.1",
+        # huggingfacehub_api_token=os.environ.get("HUGGINGFACEHUB_API_TOKEN"),
+        # )
+
+        llm = ChatOpenAI(
+            model="gpt-4",
+            temperature=0.0,
+            openai_api_key=os.environ.get("OPENAI_API_KEY"),  # use OpenAI API key securely from environment variable
+            verbose=True
         )
 
         chat = ChatHuggingFace(llm=llm, verbose=True)