fix compass params (#716)

fix
cohere-ai · Aug 20, 2024 · a16fa2e · a16fa2e
1 parent 424ef27
commit a16fa2e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/backend/services/file.py b/src/backend/services/file.py
@@ -487,6 +487,7 @@ async def consolidate_agent_files_in_compass(
                     "index": agent_id,
                     "file_id": file_id,
                     "file_bytes": fetched_doc["text"],
+                    "filename": fetched_doc["file_name"],
                     "custom_context": {
                         "file_id": file_id,
                         "file_name": fetched_doc["file_name"],
@@ -569,6 +570,7 @@ async def insert_files_in_compass(
                     "index": new_file_id if index is None else index,
                     "file_id": new_file_id,
                     "file_bytes": file_bytes,
+                    "filename": filename,
                     "custom_context": {
                         "file_id": new_file_id,
                         "file_name": filename,