From f214f5dbc5dc154f2ea5f970bc9e66ee875119d1 Mon Sep 17 00:00:00 2001
From: vasilije <vas.markovic@gmail.com>
Date: Wed, 13 Aug 2025 21:01:10 +0200
Subject: [PATCH] search router change for UI test

---
 cognee/api/v1/search/routers/get_search_router.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/cognee/api/v1/search/routers/get_search_router.py b/cognee/api/v1/search/routers/get_search_router.py
index 0ceeb1abb..75c67343b 100644
--- a/cognee/api/v1/search/routers/get_search_router.py
+++ b/cognee/api/v1/search/routers/get_search_router.py
@@ -118,6 +118,10 @@ def get_search_router() -> APIRouter:
                 top_k=payload.top_k,
             )
 
+            # Ensure response conforms to response_model=list
+            if not isinstance(results, list):
+                results = [results]
+
             return results
         except PermissionDeniedError:
             return []