minor enforcement of CSV length for internal processing (#3109)

danswer-ai · Nov 23, 2024 · c93e3fb · c93e3fb
1 parent 7edbd72
commit c93e3fb
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/backend/danswer/llm/utils.py b/backend/danswer/llm/utils.py
@@ -136,9 +136,11 @@ def translate_history_to_basemessages(
     return history_basemessages, history_token_counts
 
 
-def _process_csv_file(file: InMemoryChatFile) -> str:
+#  Processes CSV files to show the first 5 rows and max_columns (default 40) columns
+def _process_csv_file(file: InMemoryChatFile, max_columns: int = 40) -> str:
     df = pd.read_csv(io.StringIO(file.content.decode("utf-8")))
-    csv_preview = df.head().to_string()
+
+    csv_preview = df.head().to_string(max_cols=max_columns)
 
     file_name_section = (
         f"CSV FILE NAME: {file.filename}\n"