Update kig_core/planner.py

kig_core/planner.py

@@ -118,6 +118,12 @@ class KeyIssue(BaseModel):
 class KeyIssueList(BaseModel):
     key_issues: List[KeyIssue] = Field(description="List of key issues")
 
+class KeyIssueInvoke(BaseModel):
+    id: int
+    title: str
+    description: str
+    challenges: List[str]
+    potential_impact: Optional[str] = None
 
 def generate_structured_issues(state: PlannerState) -> Dict[str, Any]:
     """Generates the final structured Key Issues based on all gathered context."""
@@ -157,6 +163,7 @@ def generate_structured_issues(state: PlannerState) -> Dict[str, Any]:
         })
         print(f"structured_issues_obj => type : {type(structured_issues_obj)}, value : {structured_issues_obj}")
         structured_issues = structured_issues_obj #.key_issues
+        key_issues_list = [KeyIssueInvoke(**issue_dict) for issue_dict in structured_issues_obj]
         # Ensure IDs are sequential if the LLM didn't assign them correctly
         for i, issue in enumerate(structured_issues):
             issue.id = i + 1