From 94c892d4b5de51aef993d5ebf1480a161e903978 Mon Sep 17 00:00:00 2001 From: Igor Ilic <30923996+dexters1@users.noreply.github.com> Date: Wed, 23 Apr 2025 15:58:49 +0200 Subject: [PATCH] fix: return appropriate result type for chunks search in mcp (#777) ## Description Add CHUNKS search handling and better unknown search type return value ## DCO Affirmation I affirm that all code in every commit of this pull request conforms to the terms of the Topoteretes Developer Certificate of Origin. --- cognee-mcp/src/server.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/cognee-mcp/src/server.py b/cognee-mcp/src/server.py index 4b658d1e3..f4c9cc142 100755 --- a/cognee-mcp/src/server.py +++ b/cognee-mcp/src/server.py @@ -196,9 +196,13 @@ async def search(search_query: str, search_type: str) -> str: return json.dumps(search_results, cls=JSONEncoder) elif search_type.upper() == "GRAPH_COMPLETION" or search_type.upper() == "RAG_COMPLETION": return search_results[0] - else: + elif search_type.upper() == "CHUNKS": + return str(search_results) + elif search_type.upper() == "INSIGHTS": results = retrieved_edges_to_string(search_results) return results + else: + return str(search_results) async def prune():