From 5f144a0f922997fa96efea7c7e5684ea8f381a22 Mon Sep 17 00:00:00 2001 From: Boris Date: Tue, 19 Nov 2024 15:30:09 +0100 Subject: [PATCH] fix: make all checks green (#1) --- cognee/modules/pipelines/Pipeline.py | 18 ------------------ docker-compose.yml | 2 +- notebooks/cognee_llama_index.ipynb | 2 +- 3 files changed, 2 insertions(+), 20 deletions(-) delete mode 100644 cognee/modules/pipelines/Pipeline.py diff --git a/cognee/modules/pipelines/Pipeline.py b/cognee/modules/pipelines/Pipeline.py deleted file mode 100644 index 610394174..000000000 --- a/cognee/modules/pipelines/Pipeline.py +++ /dev/null @@ -1,18 +0,0 @@ -from uuid import UUID, uuid4 -from typing import Optional -from pydantic import BaseModel -from .models.Task import Task - -class PipelineConfig(BaseModel): - batch_count: int = 10 - description: Optional[str] = None - -class Pipeline(): - id: UUID = uuid4() - name: str - description: str - tasks: list[Task] = [] - - def __init__(self, name: str, pipeline_config: PipelineConfig): - self.name = name - self.description = pipeline_config.description diff --git a/docker-compose.yml b/docker-compose.yml index 426b178a7..afb216169 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -19,7 +19,7 @@ services: deploy: resources: limits: - cpus: '4.0' + cpus: '2.0' memory: 8GB frontend: diff --git a/notebooks/cognee_llama_index.ipynb b/notebooks/cognee_llama_index.ipynb index 15e17163d..d6f92b2b1 100644 --- a/notebooks/cognee_llama_index.ipynb +++ b/notebooks/cognee_llama_index.ipynb @@ -174,7 +174,7 @@ "\n", "# Query cognee for summaries\n", "search_results = await cognee.search(\n", - " SearchType.SUMMARIES, query=\"What are the main news discussed in the document?\"\n", + " SearchType.SUMMARIES, query_text=\"What are the main news discussed in the document?\"\n", ")\n", "# Display search results\n", "print(\"\\n Summary of main news discussed:\\n\")\n",