fix: xls file support

GooeyAI · Dec 27, 2024 · 8c5e85c · 8c5e85c
1 parent 8be0079
commit 8c5e85c
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 1 deletion.
diff --git a/daras_ai_v2/vector_search.py b/daras_ai_v2/vector_search.py
@@ -802,7 +802,7 @@ def pdf_or_tabular_bytes_to_text_pages_or_df(
     if "sections" in df.columns or "snippet" in df.columns:
         return df
     else:
-        df.columns = [THEAD + col + THEAD for col in df.columns]
+        df.columns = [THEAD + str(col) + THEAD for col in df.columns]
         return pd.DataFrame(["csv=" + df.to_csv(index=False)], columns=["sections"])
 
 

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -90,6 +90,7 @@ django-csp = "^3.8"
 python-pptx = "^1.0.2"
 azure-identity = "^1.19.0"
 azure-keyvault-secrets = "^4.9.0"
+xlrd = "^2.0.1"
 
 [tool.poetry.group.dev.dependencies]
 watchdog = "^2.1.9"