Skip to content

Commit fc62fbd

Browse files
committed
strands argument passing update
1 parent 5d4bba5 commit fc62fbd

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

lib/idp_common_pkg/idp_common/assessment/strands_tools.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ def submit_assessment(assessment: AssessmentOutput, agent: Agent) -> str:
5454
Success confirmation message or validation error details
5555
"""
5656
# Validate assessment structure and return helpful errors
57-
validated_assessment = AssessmentOutput(**assessment) # pyright: ignore[reportCallIssue]
57+
validated_assessment = AssessmentOutput.model_validate(assessment)
5858

5959
# Store in agent state
6060
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])
8080
"""
8181

8282
@tool
83-
def view_image(input_data: dict[str, Any], agent: Agent) -> dict:
83+
def view_image(input_data: ViewImageInput, agent: Agent) -> dict:
8484
"""
8585
View a specific page image, optionally highlighting a bounding box area.
8686
@@ -104,7 +104,7 @@ def view_image(input_data: dict[str, Any], agent: Agent) -> dict:
104104
}, agent)
105105
"""
106106
# Validate input - let ValidationError propagate
107-
view_input = ViewImageInput(**input_data)
107+
view_input = ViewImageInput.model_validate(input_data)
108108

109109
# Validate image index exists
110110
if view_input.image_index >= len(page_images):

0 commit comments

Comments
 (0)