diff --git a/cognee/__init__.py b/cognee/__init__.py index 7aa6388d9..be5a16b3b 100644 --- a/cognee/__init__.py +++ b/cognee/__init__.py @@ -18,6 +18,7 @@ logger = setup_logging() from .api.v1.add import add from .api.v1.delete import delete from .api.v1.cognify import cognify +from .modules.memify import memify from .api.v1.config.config import config from .api.v1.datasets.datasets import datasets from .api.v1.prune import prune diff --git a/cognee/modules/memify/__init__.py b/cognee/modules/memify/__init__.py new file mode 100644 index 000000000..90aaa8404 --- /dev/null +++ b/cognee/modules/memify/__init__.py @@ -0,0 +1 @@ +from .memify import memify diff --git a/cognee/api/v1/cognify/memify.py b/cognee/modules/memify/memify.py similarity index 100% rename from cognee/api/v1/cognify/memify.py rename to cognee/modules/memify/memify.py diff --git a/examples/python/memify_coding_agent_example.py b/examples/python/memify_coding_agent_example.py index 0238cf775..17bf8fc0e 100644 --- a/examples/python/memify_coding_agent_example.py +++ b/examples/python/memify_coding_agent_example.py @@ -3,15 +3,12 @@ import pathlib import os import cognee +from cognee import memify from cognee.api.v1.visualize.visualize import visualize_graph from cognee.shared.logging_utils import setup_logging, ERROR -from cognee.api.v1.cognify.memify import memify from cognee.modules.pipelines.tasks.task import Task from cognee.tasks.memify.extract_subgraph_chunks import extract_subgraph_chunks -from cognee.tasks.codingagents.coding_rule_associations import ( - add_rule_associations, - get_existing_rules, -) +from cognee.tasks.codingagents.coding_rule_associations import add_rule_associations # Prerequisites: # 1. Copy `.env.template` and rename it to `.env`.