feat: optimize the prompt for stabiliby

Canner · Oct 3, 2024 · f2e6856 · f2e6856
1 parent 0e5a835
commit f2e6856
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/wren-ai-service/src/pipelines/generation/semantics_description.py b/wren-ai-service/src/pipelines/generation/semantics_description.py
@@ -100,8 +100,9 @@ def wrapper(text: str) -> str:
 ### Output Format:
 
 ```
-[
-    {
+{
+    "models": [
+        {
         "name": "model",
         "columns": [
             {
@@ -124,10 +125,11 @@ def wrapper(text: str) -> str:
             }
         ],
         "properties": {
-            "description": "<description for model>"
+                "description": "<description for model>"
+            }
         }
-    }
-]
+    ]
+}
 ```
 
 Make sure that the descriptions are concise, informative, and contextually appropriate based on the input provided by the user.