small changes

app.py

@@ -15,7 +15,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 from typing import TypedDict, List, Dict, Any, Optional, Annotated
 from langgraph.graph import StateGraph, START, END
-# from langchain_openai import ChatOpenAI
+from langchain_openai import ChatOpenAI
 # from langchain_huggingface.llms import HuggingFaceEndpoint
 from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage
 from langgraph.graph.message import add_messages
@@ -93,17 +93,6 @@ def youtube_transcript(url: str) -> str:
             time.sleep(1) # Import the time module
     return "Failed to retrieve transcript after multiple attempts."
 
-# @tool
-# def divide(a: int, b: int) -> float:
-#     """Divide a and b for occasional calculations.
-#     Args:
-#         a: integer
-#         b: integer
-#     Returns:
-#         A single float containing the result of the division.
-#     """
-#     return a / b
-
 # Equip llm with tools
 tools_list = [
     fetch_website,
@@ -191,8 +180,8 @@ class BasicAgent:
                 content=question
             )
         ]
-        messages = await agent_graph.ainvoke({"messages": messages})
-        response_text = messages['messages'][-1].content
+        response = await agent_graph.ainvoke({"messages": messages})
+        response_text = response['messages'][-1].content
         return response_text.split('</think>')[-1]