From 46f1463310cb4e666c54cba688e2fcb361311735 Mon Sep 17 00:00:00 2001
From: Vishesh Arora <>
Date: Wed, 7 Feb 2024 22:59:49 +0800
Subject: [PATCH] Flatten IDs

---
 backend/main.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/backend/main.py b/backend/main.py
index d57596a..aa868bc 100644
--- 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