strands argument passing update

Author: kazmer97

lib/idp_common_pkg/idp_common/assessment/strands_tools.py

@@ -54,7 +54,7 @@ def submit_assessment(assessment: AssessmentOutput, agent: Agent) -> str:
         Success confirmation message or validation error details
     """
     # Validate assessment structure and return helpful errors
-    validated_assessment = AssessmentOutput(**assessment)  # pyright: ignore[reportCallIssue]
+    validated_assessment = AssessmentOutput.model_validate(assessment)
 
     # Store in agent state
     agent.state.set("assessment_output", validated_assessment.model_dump())
@@ -80,7 +80,7 @@ def create_view_image_tool(page_images: list[bytes], sorted_page_ids: list[str])
     """
 
     @tool
-    def view_image(input_data: dict[str, Any], agent: Agent) -> dict:
+    def view_image(input_data: ViewImageInput, agent: Agent) -> dict:
         """
         View a specific page image, optionally highlighting a bounding box area.
 
@@ -104,7 +104,7 @@ def view_image(input_data: dict[str, Any], agent: Agent) -> dict:
             }, agent)
         """
         # Validate input - let ValidationError propagate
-        view_input = ViewImageInput(**input_data)
+        view_input = ViewImageInput.model_validate(input_data)
 
         # Validate image index exists
         if view_input.image_index >= len(page_images):

lib/idp_common_pkg/idp_common/extraction/agentic_idp.py

@@ -223,7 +223,9 @@ def extraction_tool(
         This tool needs to be Successfully invoked before the patch tool can be used."""
 
         logger.info("extraction_tool called", extra={"models_extraction": extraction})
-        extraction_model = model_class(**extraction)  # pyright: ignore[reportAssignmentType]
+        # Note: The @tool decorator passes data as a dict, not as a model instance
+        # We need to validate it manually using the Pydantic model
+        extraction_model = model_class.model_validate(extraction)  # pyright: ignore[reportAssignmentType]
         extraction_dict = extraction_model.model_dump()
         agent.state.set(key="current_extraction", value=extraction_dict)
         logger.debug(
@@ -263,7 +265,9 @@ def apply_json_patches(
 
     @tool
     def make_buffer_data_final_extraction(agent: Agent) -> str:
-        valid_extraction = model_class(**agent.state.get("intermediate_extraction"))
+        valid_extraction = model_class.model_validate(
+            agent.state.get("intermediate_extraction")
+        )
 
         agent.state.set("current_extraction", valid_extraction.model_dump())