Skip to content

Commit c965d3c

Browse files
committed
fix intent classification output
1 parent 3635513 commit c965d3c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

wren-ai-service/src/pipelines/generation/intent_classification_v2.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -355,7 +355,7 @@ def post_process(classify_intent: dict, construct_db_schemas: list[str]) -> dict
355355

356356
class IntentClassificationResult(BaseModel):
357357
rephrased_question: str
358-
results: Literal[
358+
intent: Literal[
359359
"MISLEADING_QUERY",
360360
"TEXT_TO_SQL",
361361
"GENERAL",
@@ -364,6 +364,7 @@ class IntentClassificationResult(BaseModel):
364364
"CHART",
365365
]
366366
reasoning: str
367+
sql: Optional[str] = ""
367368

368369

369370
INTENT_CLASSIFICAION_MODEL_KWARGS = {

0 commit comments

Comments
 (0)