Spaces:
Running
Running
Update kig_core/planner.py
Browse files- kig_core/planner.py +3 -3
kig_core/planner.py
CHANGED
@@ -133,16 +133,16 @@ def generate_structured_issues(state: PlannerState) -> Dict[str, Any]:
|
|
133 |
output_parser = JsonOutputParser(pydantic_object=List[KeyIssue])
|
134 |
|
135 |
prompt = KEY_ISSUE_STRUCTURING_PROMPT.partial(
|
136 |
-
|
137 |
)
|
138 |
|
139 |
chain = prompt | issue_llm | output_parser
|
140 |
|
141 |
try:
|
142 |
-
structured_issues = chain.invoke({
|
143 |
"user_query": user_query,
|
144 |
"context": full_context
|
145 |
-
}
|
146 |
|
147 |
# Ensure IDs are sequential if the LLM didn't assign them correctly
|
148 |
for i, issue in enumerate(structured_issues):
|
|
|
133 |
output_parser = JsonOutputParser(pydantic_object=List[KeyIssue])
|
134 |
|
135 |
prompt = KEY_ISSUE_STRUCTURING_PROMPT.partial(
|
136 |
+
schema=output_parser.get_format_instructions(), # Inject schema instructions if needed by prompt
|
137 |
)
|
138 |
|
139 |
chain = prompt | issue_llm | output_parser
|
140 |
|
141 |
try:
|
142 |
+
structured_issues = chain.invoke({
|
143 |
"user_query": user_query,
|
144 |
"context": full_context
|
145 |
+
})
|
146 |
|
147 |
# Ensure IDs are sequential if the LLM didn't assign them correctly
|
148 |
for i, issue in enumerate(structured_issues):
|