fix QTD csv loading a raw file

microsoft · Nov 11, 2024 · 43858d3 · 43858d3
1 parent 001c183
commit 43858d3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/intelligence_toolkit/query_text_data/input_processor.py b/intelligence_toolkit/query_text_data/input_processor.py
@@ -63,7 +63,7 @@ def convert_file_bytes_to_chunks(
 
         if file_name.endswith(".csv"):
             df = pd.read_csv(io.BytesIO(bytes))
-            text_chunks = convert_df_to_chunks(df, file_name)
+            text_to_chunks = convert_df_to_chunks(df, file_name)
         else:
             if file_name.endswith(".pdf"):
                 page_texts = []
@@ -89,7 +89,7 @@ def convert_file_bytes_to_chunks(
                     }
                     text_chunks[index] = dumps(chunk, indent=2, ensure_ascii=False)
 
-        text_to_chunks[file_name] = text_chunks
+            text_to_chunks[file_name] = text_chunks
     return text_to_chunks