openai

app.py

@@ -8,6 +8,7 @@ from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
 from langchain.agents import initialize_agent, AgentType
 from bs4 import BeautifulSoup
 import base64
+from langchain_openai import ChatOpenAI
 
 ## # Load environment variables from .env file
 # --- Constants ---
@@ -16,7 +17,8 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # Load the environment variables
 HF_ACCESS_KEY = os.getenv('HF_ACCESS_KEY')
 WEATHER_API_KEY = os.getenv('WEATHER_API_KEY')
-
+OPENAI_KEY = os.getenv('OPENAI_KEY')
+OPENAI_MODEL = os.getenv ('OPENAI_MODEL')
 
 ########## ----- DEFINING TOOLS -----##########
 
@@ -329,7 +331,7 @@ Instructions:
 
 ## --- Initialize Hugging Face Model ---
 # Generate the chat interface, including the tools
-
+'''
 llm = HuggingFaceEndpoint(
     repo_id="meta-llama/Llama-3.3-70B-Instruct",
     # repo_id="Qwen/Qwen2.5-32B-Instruct",
@@ -339,12 +341,12 @@ llm = HuggingFaceEndpoint(
 )
 chat_llm = ChatHuggingFace(llm=llm)
 '''
-llm = ChatOpenAI(
+chat_llm = ChatOpenAI(
     openai_api_key=OPENAI_KEY,
     model_name=OPENAI_MODEL,
     temperature=0.1
 )
-'''
+
 # chat = ChatHuggingFace(llm=llm, verbose=True)
 # tools = [search_tool, fetch_weather]
 # chat_with_tools = chat.bind_tools(tools)
@@ -354,7 +356,7 @@ agent = initialize_agent(
     # llm=llm,
     llm=chat_llm,
     agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
-    # agent_kwargs={"system_message": system_prompt},
+    agent_kwargs={"system_message": system_prompt},
     verbose=True,
     handle_parsing_errors=True
 )
@@ -424,9 +426,9 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            full_prompt = f"{system_prompt}\n Input Question: {question_text}"
-            submitted_answer = agent.run(full_prompt)
-            # submitted_answer = agent.run(question_text)
+            # full_prompt = f"{system_prompt}\n Input Question: {question_text}"
+            # submitted_answer = agent.run(full_prompt)
+            submitted_answer = agent.run(question_text)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e: