revert

Canner · Nov 15, 2024 · e5cf965 · e5cf965
1 parent 2ce0acf
commit e5cf965
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/wren-ai-service/src/pipelines/generation/sql_generation.py b/wren-ai-service/src/pipelines/generation/sql_generation.py
@@ -142,12 +142,11 @@ class GenerationResults(BaseModel):
 
 SQL_GENERATION_MODEL_KWARGS = {
     "response_format": {
-        "type": "text",
-        # "type": "json_schema",
-        # "json_schema": {
-        #     "name": "sql_results",
-        #     "schema": GenerationResults.model_json_schema(),
-        # },
+        "type": "json_schema",
+        "json_schema": {
+            "name": "sql_results",
+            "schema": GenerationResults.model_json_schema(),
+        },
     }
 }