fix

agent.py

@@ -78,10 +78,9 @@ class MultiModalAgent:
         llm = self._get_llm(model_name)
         tools = self._get_tools()
         self.agent = create_react_agent(
-            llm,
+            llm=llm,
             tools=tools,
-            state_schema=MultiModalTaskState,
-            system_prompt=MULTIMODAL_TASK_SOLVER_PROMPT,
+            prompt=MULTIMODAL_TASK_SOLVER_PROMPT,
             checkpointer=MemorySaver()
         )
 

tools.py

@@ -270,9 +270,9 @@ async def analyze_image(state: Annotated[dict, InjectedState], image_path: str)
     )
 
     file_agent = create_react_agent(
-        llm,
+        llm=llm,
         tools=[],
-        system_prompt=MULTIMODAL_FILE_ANALYZER_PROMPT,
+        prompt=MULTIMODAL_FILE_ANALYZER_PROMPT
     )
 
     message = HumanMessage(
@@ -318,9 +318,9 @@ async def analyze_excel(state: Annotated[dict, InjectedState], excel_path: str)
     )
 
     file_agent = create_react_agent(
-        llm,
+        llm=llm,
         tools=[],
-        system_prompt=MULTIMODAL_FILE_ANALYZER_PROMPT,
+        prompt=MULTIMODAL_FILE_ANALYZER_PROMPT
     )
 
     message = HumanMessage(
@@ -363,9 +363,9 @@ async def analyze_audio(state: Annotated[dict, InjectedState], audio_path: str)
     )
 
     file_agent = create_react_agent(
-        llm,
+        llm=llm,
         tools=[],
-        system_prompt=MULTIMODAL_FILE_ANALYZER_PROMPT,
+        prompt=MULTIMODAL_FILE_ANALYZER_PROMPT
     )
 
     message = HumanMessage(
@@ -396,9 +396,9 @@ async def analyze_video(state: Annotated[dict, InjectedState], video_url: str) -
     )
 
     file_agent = create_react_agent(
-        llm,
+        llm=llm,
         tools=[],
-        system_prompt=MULTIMODAL_FILE_ANALYZER_PROMPT,
+        prompt=MULTIMODAL_FILE_ANALYZER_PROMPT
     )
 
     message = HumanMessage(