Flatten IDs

adidoesnt · Feb 7, 2024 · 46f1463 · 46f1463
1 parent 6e7dccb
commit 46f1463
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/backend/main.py b/backend/main.py
@@ -78,6 +78,7 @@ async def search_messages(search_string: str = Query(..., min_length=1), chat_id
         n_results=5
         )
 
+    results["ids"] = results["ids"][0]
     results["documents"] = results["documents"][0]
 
     return results