Update app.py

app.py

@@ -13,6 +13,7 @@ from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
 from youtube_tool import youtube_transcript_tool, youtube_transcript_snippet_tool
 from multiple_tools import round_to_two_decimals_tool, text_inverter_tool, google_web_search_tool, wikipedia_search_tool
 from agent import smart_agent
+from llama_index.llms.openai import OpenAI
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -24,7 +25,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized. . . .")
-        self.agent = smart_agent()
+        # self.agent = smart_agent()
         # self.api_key = os.getenv("HF_TOKEN")
         # self.llm = HuggingFaceInferenceAPI(
         #     model_name="deepseek-ai/DeepSeek-R1-0528",
@@ -37,6 +38,12 @@ class BasicAgent:
         #      round_to_two_decimals_tool, text_inverter_tool, google_web_search_tool],
         #     llm=self.llm
         # )
+        OpenAI_key = os.getenv("OPEN_AI_TOKEN")
+        self.llm = OpenAI(model="gpt-4o-mini", temperature=0.2, api_key=OpenAI_key)
+        self.agent = AgentWorkflow.from_tools_or_functions(
+            [wikipedia_search_tool, youtube_transcript_tool, youtube_transcript_snippet_tool, round_to_two_decimals_tool, text_inverter_tool, google_web_search_tool],
+            llm=self.llm,
+        )
 
     async def run(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")