wip: fix chart

Canner · Dec 18, 2024 · 9ee90e0 · 9ee90e0
1 parent e4790b6
commit 9ee90e0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wren-ai-service/src/pipelines/generation/chart_generation.py b/wren-ai-service/src/pipelines/generation/chart_generation.py
@@ -70,7 +70,7 @@ def run(
         try:
             generation_result = orjson.loads(replies[0])
             reasoning = generation_result.get("reasoning", "")
-            if chart_schema := generation_result.get("chart_schema", {}):
+            if chart_schema := orjson.loads(generation_result.get("chart_schema", {})):
                 # validate(chart_schema, schema=vega_schema)
                 chart_schema["data"]["values"] = []
                 return {