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

adding temp fix to mcp

Browse source
This commit is contained in:
vasilije 2025-08-18 17:11:09 +02:00
parent abbce6e8c3
commit e810be136d
1 changed files with 15 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -21,16 +21,16 @@ from cognee.shared.data_models import KnowledgeGraph
from cognee.modules.storage.utils import JSONEncoder
try:
from codingagents.coding_rule_associations import (
add_rule_associations,
get_existing_rules,
)
except ModuleNotFoundError:
from .codingagents.coding_rule_associations import (
add_rule_associations,
get_existing_rules,
)
# try:
# from codingagents.coding_rule_associations import (
# add_rule_associations,
# get_existing_rules,
# )
# except ModuleNotFoundError:
# from .codingagents.coding_rule_associations import (
# add_rule_associations,
# get_existing_rules,
# )
mcp = FastMCP("Cognee")
@ -298,7 +298,7 @@ async def save_interaction(data: str) -> list:
logger.info("Save interaction process finished.")
logger.info("Generating associated rules from interaction data.")
await add_rule_associations(data=data, rules_nodeset_name="coding_agent_rules")
# await add_rule_associations(data=data, rules_nodeset_name="coding_agent_rules")
logger.info("Associated rules generated from interaction data.")
@ -560,8 +560,10 @@ async def get_developer_rules() -> list:
async def fetch_rules_from_cognee() -> str:
"""Collect all developer rules from Cognee"""
with redirect_stdout(sys.stderr):
developer_rules = await get_existing_rules(rules_nodeset_name="coding_agent_rules")
return developer_rules
note = "This is broken in 0.2.2"
return note
# developer_rules = await get_existing_rules(rules_nodeset_name="coding_agent_rules")
# return developer_rules
rules_text = await fetch_rules_from_cognee()