diff --git a/cognee/api/v1/search/search_v2.py b/cognee/api/v1/search/search_v2.py index 516c7b11d..6a5da4648 100644 --- a/cognee/api/v1/search/search_v2.py +++ b/cognee/api/v1/search/search_v2.py @@ -14,13 +14,13 @@ from cognee.modules.users.permissions.methods import get_document_ids_for_user from cognee.tasks.chunks import query_chunks from cognee.tasks.graph import query_graph_connections from cognee.tasks.summarization import query_summaries -from cognee.tasks.compute import query_compute +from cognee.tasks.completion import query_completion class SearchType(Enum): SUMMARIES = "SUMMARIES" INSIGHTS = "INSIGHTS" CHUNKS = "CHUNKS" - COMPUTE = "COMPUTE" + COMPLETION = "COMPLETION" async def search(query_type: SearchType, query_text: str, user: User = None) -> list: if user is None: @@ -52,7 +52,7 @@ async def specific_search(query_type: SearchType, query: str, user) -> list: SearchType.SUMMARIES: query_summaries, SearchType.INSIGHTS: query_graph_connections, SearchType.CHUNKS: query_chunks, - SearchType.COMPUTE: query_compute, + SearchType.COMPLETION: query_completion, } search_task = search_tasks.get(query_type) diff --git a/cognee/tasks/completion/__init__.py b/cognee/tasks/completion/__init__.py new file mode 100644 index 000000000..1bf0fa6bb --- /dev/null +++ b/cognee/tasks/completion/__init__.py @@ -0,0 +1 @@ +from .query_completion import query_completion \ No newline at end of file diff --git a/cognee/tasks/compute/exceptions/__init__.py b/cognee/tasks/completion/exceptions/__init__.py similarity index 100% rename from cognee/tasks/compute/exceptions/__init__.py rename to cognee/tasks/completion/exceptions/__init__.py diff --git a/cognee/tasks/compute/exceptions/exceptions.py b/cognee/tasks/completion/exceptions/exceptions.py similarity index 100% rename from cognee/tasks/compute/exceptions/exceptions.py rename to cognee/tasks/completion/exceptions/exceptions.py diff --git a/cognee/tasks/compute/query_compute.py b/cognee/tasks/completion/query_completion.py similarity index 89% rename from cognee/tasks/compute/query_compute.py rename to cognee/tasks/completion/query_completion.py index cd606ddc5..5324676f8 100644 --- a/cognee/tasks/compute/query_compute.py +++ b/cognee/tasks/completion/query_completion.py @@ -1,10 +1,10 @@ from cognee.infrastructure.databases.vector import get_vector_engine -from cognee.tasks.compute.exceptions import NoRelevantDataFound +from cognee.tasks.completion.exceptions import NoRelevantDataFound from cognee.infrastructure.llm.get_llm_client import get_llm_client from cognee.infrastructure.llm.prompts import read_query_prompt, render_prompt -async def query_compute(query: str) -> list: +async def query_completion(query: str) -> list: """ Parameters: - query (str): The query string to compute. diff --git a/cognee/tasks/compute/__init__.py b/cognee/tasks/compute/__init__.py deleted file mode 100644 index 7f6ed58fd..000000000 --- a/cognee/tasks/compute/__init__.py +++ /dev/null @@ -1 +0,0 @@ -from .query_compute import query_compute \ No newline at end of file