Update `requirements.txt` to include new dependencies for enhanced functionality and compatibility

src/gaia_solving_agent/agent.py

@@ -52,15 +52,15 @@ class AnswerEvent(Event):
 class GaiaWorkflow(Workflow):
     @step
     async def setup(self, ctx: Context, ev: StartEvent) -> PlanEvent:
-        await ctx.set("user_msg", ev.user_msg)
-        await ctx.set("additional_file", ev.additional_file)
-        await ctx.set("additional_file_path", ev.additional_file_path)
+        await ctx.store.set("user_msg", ev.user_msg)
+        await ctx.store.set("additional_file", ev.additional_file)
+        await ctx.store.set("additional_file_path", ev.additional_file_path)
         return PlanEvent()
 
     @step
     async def make_plan(self, ctx: Context, ev: PlanEvent) -> PlanEvent | QueryEvent:
-        additional_file_path = await ctx.get("additional_file_path")
-        user_msg = await ctx.get("user_msg")
+        additional_file_path = await ctx.store.get("additional_file_path")
+        user_msg = await ctx.store.get("user_msg")
 
         llm = get_llm(reasoning_model_name)
         prompt_template = RichPromptTemplate(PLANING_PROMPT)
@@ -87,7 +87,7 @@ Stick strictly to the formatting constraints !
 """
 
         plan = llm.complete(prompt)
-        await ctx.set("plan", plan.text)
+        await ctx.store.set("plan", plan.text)
 
         question = extract_pattern(pattern=r"<Question> :\s*([\s\S]*?)\s*</Question>", text=plan.text)
         known_facts = extract_pattern(pattern=r"<Known facts> :\s*([\s\S]*?)\s*</Known facts>", text=plan.text)
@@ -98,20 +98,20 @@ Stick strictly to the formatting constraints !
         ):
             return PlanEvent(to_do="Format", plan=plan.text)
         else:
-            await ctx.set("question", question if question is not None else "")
-            await ctx.set("known_facts", known_facts if known_facts is not None else "")
-            await ctx.set("sub_tasks", sub_tasks if sub_tasks is not None else "")
+            await ctx.store.set("question", question if question is not None else "")
+            await ctx.store.set("known_facts", known_facts if known_facts is not None else "")
+            await ctx.store.set("sub_tasks", sub_tasks if sub_tasks is not None else "")
 
             return QueryEvent()
 
     @step()
     async def multi_agent_process(self, ctx: Context, ev: QueryEvent) -> AnswerEvent:
-        plan = await ctx.get("plan")
-        additional_file = await ctx.get("additional_file")
+        plan = await ctx.store.get("plan")
+        additional_file = await ctx.store.get("additional_file")
 
-        question = await ctx.get("question")
-        known_facts = await ctx.get("known_facts")
-        sub_tasks = await ctx.get("sub_tasks")
+        question = await ctx.store.get("question")
+        known_facts = await ctx.store.get("known_facts")
+        sub_tasks = await ctx.store.get("sub_tasks")
         prompt = f"""
 The question is : {question}
 
@@ -127,7 +127,7 @@ The sub-tasks are :
         memory = ChatMemoryBuffer.from_defaults(token_limit=100000)
 
         agent_ctx = Context(gaia_solving_agent)
-        await agent_ctx.set("additional_file", additional_file)
+        await agent_ctx.store.set("additional_file", additional_file)
         agent_output = await gaia_solving_agent.run(
             user_msg=prompt,
             memory=memory,
@@ -139,7 +139,7 @@ The sub-tasks are :
     async def parse_answer(self, ctx: Context, ev: AnswerEvent) -> StopEvent:
         llm = get_llm(balanced_model_name)
         prompt_template = RichPromptTemplate(FORMAT_ANSWER)
-        question = await ctx.get("question")
+        question = await ctx.store.get("question")
         prompt = prompt_template.format(question=question, answer=ev.answer)
         result = llm.complete(prompt)
 

src/gaia_solving_agent/tools.py

@@ -45,7 +45,7 @@ async def vllm_ask_image_tool(ctx: Context, query: str) -> str:
         The result or response to the provided query based on the processed
         image content.
     """
-    images = await ctx.get("additional_file")
+    images = await ctx.store.get("additional_file")
     return await vllm_ask_image(query=query, images=images)