Update app.py

app.py

@@ -11,7 +11,6 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 import os
-
 from typing import TypedDict, Annotated
 
 from duckduckgo_search import DDGS
@@ -22,16 +21,13 @@ from langgraph.graph import START, StateGraph
 from langgraph.graph.message import add_messages
 from langgraph.prebuilt import ToolNode, tools_condition
 
-from langchain_huggingface import (
-    HuggingFaceEndpoint,
-    ChatHuggingFace,
-)
+from langchain_openai import ChatOpenAI
 
 # =====================================================
 # HF TOKEN FROM SPACE SECRET
 # =====================================================
 
-HF_TOKEN = os.environ["HF_TOKEN"]
+FM_TOKEN = os.environ["FM_TOKEN"]
 
 # =====================================================
 # WEB SEARCH TOOL
@@ -71,17 +67,15 @@ URL: {r.get("href","")}
 # LLM
 # =====================================================
 
-llm = HuggingFaceEndpoint(
-    repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-    huggingfacehub_api_token=HF_TOKEN,
+llm = ChatOpenAI(
+    base_url="https://api.freemodel.dev",
+    api_key=FM_TOKEN,
+    model="gpt-5.4",
     temperature=0.0,
-    max_new_tokens=1024,
+    max_tokens=1024,
 )
 
-chat = ChatHuggingFace(
-    llm=llm,
-    verbose=False,
-)
+chat = llm
 
 tools = [web_search]