added fix
This commit is contained in:
parent
68a650f05e
commit
0bc02a522a
1 changed files with 15 additions and 29 deletions
|
|
@ -21,16 +21,16 @@ from cognee.shared.data_models import KnowledgeGraph
|
||||||
from cognee.modules.storage.utils import JSONEncoder
|
from cognee.modules.storage.utils import JSONEncoder
|
||||||
|
|
||||||
|
|
||||||
try:
|
# try:
|
||||||
from codingagents.coding_rule_associations import (
|
# from codingagents.coding_rule_associations import (
|
||||||
add_rule_associations,
|
# add_rule_associations,
|
||||||
get_existing_rules,
|
# get_existing_rules,
|
||||||
)
|
# )
|
||||||
except ModuleNotFoundError:
|
# except ModuleNotFoundError:
|
||||||
from .codingagents.coding_rule_associations import (
|
# from .codingagents.coding_rule_associations import (
|
||||||
add_rule_associations,
|
# add_rule_associations,
|
||||||
get_existing_rules,
|
# get_existing_rules,
|
||||||
)
|
# )
|
||||||
|
|
||||||
|
|
||||||
mcp = FastMCP("Cognee")
|
mcp = FastMCP("Cognee")
|
||||||
|
|
@ -221,14 +221,6 @@ async def cognify(data: str, graph_model_file: str = None, graph_model_name: str
|
||||||
- The actual cognify process may take significant time depending on text length
|
- The actual cognify process may take significant time depending on text length
|
||||||
- Use the cognify_status tool to check the progress of the operation
|
- Use the cognify_status tool to check the progress of the operation
|
||||||
|
|
||||||
Raises
|
|
||||||
------
|
|
||||||
InvalidValueError
|
|
||||||
If LLM_API_KEY is not set
|
|
||||||
ValueError
|
|
||||||
If chunks exceed max token limits (reduce chunk_size)
|
|
||||||
DatabaseNotCreatedError
|
|
||||||
If databases are not properly initialized
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
async def cognify_task(
|
async def cognify_task(
|
||||||
|
|
@ -306,7 +298,7 @@ async def save_interaction(data: str) -> list:
|
||||||
logger.info("Save interaction process finished.")
|
logger.info("Save interaction process finished.")
|
||||||
logger.info("Generating associated rules from interaction data.")
|
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.")
|
logger.info("Associated rules generated from interaction data.")
|
||||||
|
|
||||||
|
|
@ -512,14 +504,6 @@ async def search(search_query: str, search_type: str) -> list:
|
||||||
- Different search types produce different output formats
|
- Different search types produce different output formats
|
||||||
- The function handles the conversion between Cognee's internal result format and MCP's output format
|
- The function handles the conversion between Cognee's internal result format and MCP's output format
|
||||||
|
|
||||||
Raises
|
|
||||||
------
|
|
||||||
InvalidValueError
|
|
||||||
If LLM_API_KEY is not set (for LLM-based search types)
|
|
||||||
ValueError
|
|
||||||
If query_text is empty or search parameters are invalid
|
|
||||||
NoDataError
|
|
||||||
If no relevant data found for the search query
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
async def search_task(search_query: str, search_type: str) -> str:
|
async def search_task(search_query: str, search_type: str) -> str:
|
||||||
|
|
@ -576,8 +560,10 @@ async def get_developer_rules() -> list:
|
||||||
async def fetch_rules_from_cognee() -> str:
|
async def fetch_rules_from_cognee() -> str:
|
||||||
"""Collect all developer rules from Cognee"""
|
"""Collect all developer rules from Cognee"""
|
||||||
with redirect_stdout(sys.stderr):
|
with redirect_stdout(sys.stderr):
|
||||||
developer_rules = await get_existing_rules(rules_nodeset_name="coding_agent_rules")
|
note = "This is broken in 0.2.2"
|
||||||
return developer_rules
|
return note
|
||||||
|
# developer_rules = await get_existing_rules(rules_nodeset_name="coding_agent_rules")
|
||||||
|
# return developer_rules
|
||||||
|
|
||||||
rules_text = await fetch_rules_from_cognee()
|
rules_text = await fetch_rules_from_cognee()
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue