Update functions_hf.py

functions_hf.py

@@ -5,8 +5,23 @@ from langchain_core.output_parsers import StrOutputParser
 from langchain_core.runnables import RunnablePassthrough
 import re
 from typing import Dict, List
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+
+from langchain_openai import ChatOpenAI
+
+llm = ChatOpenAI(
+    model="gpt-4o",
+    temperature=0,
+    max_tokens=None,
+    timeout=None,
+    max_retries=2,
+    api_key=OPENAI_API_KEY,  # if you prefer to pass api key in directly instaed of using env vars
+)
 
-llm = HuggingFaceHub(repo_id="Qwen/Qwen2.5-72B-Instruct", model_kwargs={"temperature":0, "max_length":64})
 
 # Initialize database connection
 db = SQLDatabase.from_uri("sqlite:///Spring_2025_courses.db", sample_rows_in_table_info=0)