Refactor `agent.py` to enhance `Context` usage, remove unused imports and parameters, and disable parallel tool calls in `gaia_solving_agent`.

src/gaia_solving_agent/agent.py

@@ -1,7 +1,7 @@
 from pathlib import Path
 from typing import Literal
 
-from llama_index.core.agent.workflow import FunctionAgent, AgentWorkflow
+from llama_index.core.agent.workflow import FunctionAgent
 from llama_index.core.prompts import RichPromptTemplate
 from llama_index.llms.nebius import NebiusLLM
 from llama_index.tools.requests import RequestsToolSpec
@@ -108,7 +108,6 @@ Stick strictly to the formatting constraints !
     async def multi_agent_process(self, ctx: Context, ev: QueryEvent) -> AnswerEvent:
         plan = await ctx.get("plan")
         additional_file = await ctx.get("additional_file")
-        additional_file_path = await ctx.get("additional_file_path")
 
         question = await ctx.get("question")
         known_facts = await ctx.get("known_facts")
@@ -127,11 +126,12 @@ The sub-tasks are :
         from llama_index.core.memory import ChatMemoryBuffer
         memory = ChatMemoryBuffer.from_defaults(token_limit=100000)
 
+        agent_ctx = Context(gaia_solving_agent)
+        await agent_ctx.set("additional_file", additional_file)
         agent_output = await gaia_solving_agent.run(
-            user_msg=plan,
+            user_msg=prompt,
             memory=memory,
-            additional_file=additional_file,
-            additional_file_path=additional_file_path,
+            ctx=agent_ctx,
         )
         return AnswerEvent(plan=plan, answer=str(agent_output))
 
@@ -208,7 +208,6 @@ gaia_solving_agent = FunctionAgent(
         tavily_search_web,
         *load_and_search_tools_from_toolspec(WikipediaToolSpec()),
         *simple_web_page_reader_toolspec.to_tool_list(),
-        *RequestsToolSpec().to_tool_list(),
         *youtube_transcript_reader_toolspec.to_tool_list(),
     ],
     llm=get_llm(balanced_model_name),
@@ -228,6 +227,6 @@ gaia_solving_agent = FunctionAgent(
     """,
     name="gaia_solving_agent",
     description="Agent that browse additional information and resources on the web.",
-    num_concurrent_runs=1,
+    allow_parallel_tool_calls=False,
 )