fix

cyyeh · cyyeh · commit 5cb313aeb8a5 · 2025-05-15T13:09:15.000+08:00
diff --git a/wren-ai-service/src/pipelines/generation/chart_adjustment.py b/wren-ai-service/src/pipelines/generation/chart_adjustment.py
@@ -111,7 +111,7 @@ def post_process(
 ) -> dict:
     return post_processor.run(
         generate_chart_adjustment.get("replies"),
-        preprocess_data["sample_data"] if data_provided else data["data"],
+        data["data"] if data_provided else preprocess_data["sample_data"],
         remove_data_from_chart_schema=remove_data_from_chart_schema,
     )
 
diff --git a/wren-ai-service/src/pipelines/generation/chart_generation.py b/wren-ai-service/src/pipelines/generation/chart_generation.py
@@ -104,7 +104,7 @@ def post_process(
 ) -> dict:
     return post_processor.run(
         generate_chart.get("replies"),
-        preprocess_data["sample_data"] if data_provided else data["data"],
+        data["data"] if data_provided else preprocess_data["sample_data"],
         remove_data_from_chart_schema=remove_data_from_chart_schema,
     )
 

Original file line number	Diff line number	Diff line change
`@@ -111,7 +111,7 @@ def post_process(`
`111`	`111`	`) -> dict:`
`112`	`112`	`return post_processor.run(`
`113`	`113`	`generate_chart_adjustment.get("replies"),`
`114`		`- preprocess_data["sample_data"] if data_provided else data["data"],`
	`114`	`+ data["data"] if data_provided else preprocess_data["sample_data"],`
`115`	`115`	`remove_data_from_chart_schema=remove_data_from_chart_schema,`
`116`	`116`	`)`
`117`	`117`
Original file line number	Diff line number	Diff line change
`@@ -104,7 +104,7 @@ def post_process(`
`104`	`104`	`) -> dict:`
`105`	`105`	`return post_processor.run(`
`106`	`106`	`generate_chart.get("replies"),`
`107`		`- preprocess_data["sample_data"] if data_provided else data["data"],`
	`107`	`+ data["data"] if data_provided else preprocess_data["sample_data"],`
`108`	`108`	`remove_data_from_chart_schema=remove_data_from_chart_schema,`
`109`	`109`	`)`
`110`	`110`