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():