gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: mcp CODE search error (#495)

Browse source 
<!-- .github/pull_request_template.md -->

## Description
<!-- Provide a clear description of the changes in this PR -->

## 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


<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->

## Summary by CodeRabbit

- **New Features**
- Enhanced search functionality now returns JSON formatted results for
code-based queries for clearer response presentation.
  
- **Refactor**
- Updated the search logic to differentiate result handling based on
query type, maintaining string output for non-code queries.

<!-- end of auto-generated comment: release notes by coderabbit.ai -->
This commit is contained in:
Hande 2025-02-04 19:32:04 +01:00 • committed by GitHub
parent 1260fc7db0
commit 690d028928
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
cognee-mcp/src/server.py
View file

@ -1,5 +1,5 @@
import json
import os import os
import cognee import cognee
import logging import logging
import importlib.util import importlib.util
@ -12,6 +12,7 @@ from mcp.server.models import InitializationOptions
from cognee.api.v1.cognify.code_graph_pipeline import run_code_graph_pipeline from cognee.api.v1.cognify.code_graph_pipeline import run_code_graph_pipeline
from cognee.modules.search.types import SearchType from cognee.modules.search.types import SearchType
from cognee.shared.data_models import KnowledgeGraph from cognee.shared.data_models import KnowledgeGraph
from cognee.modules.storage.utils import JSONEncoder
mcp = Server("cognee") mcp = Server("cognee")
@ -143,9 +144,11 @@ async def search(search_query: str, search_type: str) -> str:
query_type=SearchType[search_type.upper()], query_text=search_query query_type=SearchType[search_type.upper()], query_text=search_query
) )
results = retrieved_edges_to_string(search_results) if search_type.upper() == "CODE":
return json.dumps(search_results, cls=JSONEncoder)
return results else:
results = retrieved_edges_to_string(search_results)
return results
async def prune(): async def prune():