Update app.py

app.py

@@ -36,20 +36,6 @@ class BasicAgent:
             llm=self.llm
         )
 
-        # self.tools = [
-        #     wikipedia_search_tool,
-        #     youtube_transcript_tool,
-        #     youtube_transcript_snippet_tool,
-        #     round_to_two_decimals_tool,
-        #     text_inverter_tool,
-        #     google_web_search_tool
-        # ]
-
-        # self.agent = ReActAgent.from_tools(
-        #     tools=self.tools,
-        #     llm=self.llm,
-        #     verbose=True
-        # )
 
     async def run(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
@@ -62,6 +48,32 @@ class BasicAgent:
         return self.run(question)
         #return self.agent.run(question)
 
+def create_llamaindex_agent() -> ReActAgent:
+    api_key = os.getenv("HF_TOKEN")
+    llm = HuggingFaceInferenceAPI(
+        model_name="deepseek-ai/DeepSeek-R1-0528",
+        token=api_key,
+        provider="auto",
+        max_iterations=10
+    )
+
+    tools = [
+            wikipedia_search_tool,
+            youtube_transcript_tool,
+            youtube_transcript_snippet_tool,
+            round_to_two_decimals_tool,
+            text_inverter_tool,
+            google_web_search_tool
+        ]
+
+    agent = ReActAgent.from_tools(
+        tools=tools,
+        llm=self.llm,
+        verbose=True
+    )
+
+    return agent
+
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
@@ -82,11 +94,12 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     submit_url = f"{api_url}/submit"
 
     # 1. Instantiate Agent ( modify this part to create your agent)
-    try:
-        agent = BasicAgent()
-    except Exception as e:
-        print(f"Error instantiating agent: {e}")
-        return f"Error initializing agent: {e}", None
+    agent = create_llamaindex_agent()
+    # try:
+    #     agent = BasicAgent()
+    # except Exception as e:
+    #     print(f"Error instantiating agent: {e}")
+    #     return f"Error initializing agent: {e}", None
     #In the case of an app running as a hugging Face space, this link points toward your codebase ( usefull for others so please keep it public)
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
     print(agent_code)