input fix for assessment submit

Author: kazmer97

lib/idp_common_pkg/idp_common/assessment/strands_tools.py

@@ -12,7 +12,10 @@
 from pydantic import BaseModel, Field
 from strands import Agent, tool
 
-from idp_common.assessment.strands_models import AssessmentOutput, BoundingBox
+from idp_common.assessment.strands_models import (
+    BoundingBox,
+    ConfidenceAssessment,
+)
 from idp_common.utils.grid_overlay import draw_bounding_boxes
 from idp_common.utils.strands_agent_tools.todo_list import (
     create_todo_list,
@@ -36,7 +39,7 @@ class ViewImageInput(BaseModel):
 
 
 @tool
-def submit_assessment(assessment: AssessmentOutput, agent: Agent) -> str:
+def submit_assessment(assessment: ConfidenceAssessment, agent: Agent) -> str:
     """
     Submit your final confidence assessment.
 
@@ -54,7 +57,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.model_validate(assessment)
+    validated_assessment = ConfidenceAssessment.model_validate(assessment)
 
     # Store in agent state
     agent.state.set("assessment_output", validated_assessment.model_dump(mode="json"))