Merge pull request #341 from aws-samples/v1.8.0_dev

supinyu · web-flow · commit 50a0a3ed439c · 2024-09-19T16:36:58.000+08:00
fix search_data data type
diff --git a/application/utils/llm.py b/application/utils/llm.py
@@ -355,6 +355,7 @@ def data_visualization(model_id, search_box, search_data, prompt_map):
     else:
         all_columns_data = search_data
         columns = all_columns_data[0]
+        search_data = pd.DataFrame(search_data[1:], columns=search_data[0])
     all_columns_data = convert_timestamps_to_str(all_columns_data)
     try:
         if len(all_columns_data) < 1:
