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",