cognee 0.5.3.dev0__tar.gz → 0.5.3.dev1__tar.gz

{cognee-0.5.3.dev0 → cognee-0.5.3.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cognee
-Version: 0.5.3.dev0
+Version: 0.5.3.dev1
 Summary: Cognee - is a library for enriching LLM context with a semantic layer for better understanding and reasoning.
 Project-URL: Homepage, https://www.cognee.ai
 Project-URL: Repository, https://github.com/topoteretes/cognee

cognee-0.5.3.dev1/cognee/alembic/versions/18f01b0a0b4c_create_principal_configuration_table.py

@@ -0,0 +1,54 @@
+"""create principal configuration table
+
+Revision ID: 18f01b0a0b4c
+Revises: 84e5d08260d6
+Create Date: 2026-02-17 12:49:02.909973
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = "18f01b0a0b4c"
+down_revision: Union[str, None] = "84e5d08260d6"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    table_name = "principal_configuration"
+    connection = op.get_bind()
+    inspector = sa.inspect(connection)
+
+    if table_name not in inspector.get_table_names():
+        op.create_table(
+            table_name,
+            sa.Column(
+                "owner_id",
+                sa.UUID(),
+                sa.ForeignKey("principals.id", ondelete="CASCADE"),
+                primary_key=True,
+                index=True,
+            ),
+            sa.Column("name", sa.String(), unique=False, nullable=False),
+            sa.Column("configuration", sa.JSON()),
+            sa.Column("created_at", sa.DateTime(timezone=True), nullable=True),
+            sa.Column("updated_at", sa.DateTime(timezone=True), nullable=True),
+        )
+    else:
+        print(f"{table_name} table already exists, skipping creation")
+
+
+def downgrade() -> None:
+    table_name = "principal_configuration"
+    connection = op.get_bind()
+    inspector = sa.inspect(connection)
+
+    if table_name in inspector.get_table_names():
+        op.drop_table(table_name)
+    else:
+        print(f"{table_name} table doesn't exist, skipping downgrade")

{cognee-0.5.3.dev0 → cognee-0.5.3.dev1}/cognee/api/client.py

@@ -15,7 +15,6 @@ from fastapi.openapi.utils import get_openapi
 
 from cognee.exceptions import CogneeApiError
 from cognee.shared.logging_utils import get_logger, setup_logging
-from cognee.api.health import health_checker, HealthStatus
 from cognee.api.v1.cloud.routers import get_checks_router
 from cognee.api.v1.notebooks.routers import get_notebooks_router
 from cognee.api.v1.permissions.routers import get_permissions_router
@@ -29,6 +28,7 @@ from cognee.api.v1.add.routers import get_add_router
 from cognee.api.v1.delete.routers import get_delete_router
 from cognee.api.v1.responses.routers import get_responses_router
 from cognee.api.v1.sync.routers import get_sync_router
+from cognee.api.v1.health.routers import get_health_router
 from cognee.api.v1.update.routers import get_update_router
 from cognee.api.v1.users.routers import (
     get_auth_router,
@@ -37,6 +37,7 @@ from cognee.api.v1.users.routers import (
     get_verify_router,
     get_users_router,
     get_visualize_router,
+    get_configuration_router,
 )
 from cognee.modules.users.methods.get_authenticated_user import REQUIRE_AUTHENTICATION
 
@@ -175,59 +176,6 @@ async def exception_handler(_: Request, exc: CogneeApiError) -> JSONResponse:
     return JSONResponse(status_code=status_code, content={"detail": detail["message"]})
 
 
-@app.get("/")
-async def root():
-    """
-    Root endpoint that returns a welcome message.
-    """
-    return {"message": "Hello, World, I am alive!"}
-
-
-@app.get("/health")
-async def health_check():
-    """
-    Health check endpoint for liveness/readiness probes.
-    """
-    try:
-        health_status = await health_checker.get_health_status(detailed=False)
-        status_code = 503 if health_status.status == HealthStatus.UNHEALTHY else 200
-
-        return JSONResponse(
-            status_code=status_code,
-            content={
-                "status": "ready" if status_code == 200 else "not ready",
-                "health": health_status.status,
-                "version": health_status.version,
-            },
-        )
-    except Exception as e:
-        return JSONResponse(
-            status_code=503,
-            content={"status": "not ready", "reason": f"health check failed: {str(e)}"},
-        )
-
-
-@app.get("/health/detailed")
-async def detailed_health_check():
-    """
-    Comprehensive health status with component details.
-    """
-    try:
-        health_status = await health_checker.get_health_status(detailed=True)
-        status_code = 200
-        if health_status.status == HealthStatus.UNHEALTHY:
-            status_code = 503
-        elif health_status.status == HealthStatus.DEGRADED:
-            status_code = 200  # Degraded is still operational
-
-        return JSONResponse(status_code=status_code, content=health_status.model_dump())
-    except Exception as e:
-        return JSONResponse(
-            status_code=503,
-            content={"status": "unhealthy", "error": f"Health check system failure: {str(e)}"},
-        )
-
-
 app.include_router(get_auth_router(), prefix="/api/v1/auth", tags=["auth"])
 
 app.include_router(
@@ -270,6 +218,12 @@ app.include_router(get_settings_router(), prefix="/api/v1/settings", tags=["sett
 
 app.include_router(get_visualize_router(), prefix="/api/v1/visualize", tags=["visualize"])
 
+app.include_router(
+    get_configuration_router(),
+    prefix="/api/v1/configuration",
+    tags=["configuration"],
+)
+
 app.include_router(get_delete_router(), prefix="/api/v1/delete", tags=["delete"])
 
 app.include_router(get_update_router(), prefix="/api/v1/update", tags=["update"])
@@ -296,6 +250,20 @@ app.include_router(
     tags=["checks"],
 )
 
+app.include_router(
+    get_health_router(),
+    prefix="/health",
+    tags=["health"],
+)
+
+
+@app.get("/")
+async def root():
+    """
+    Root endpoint that returns a welcome message.
+    """
+    return {"message": "Hello, World, I am alive!"}
+
 
 def start_api_server(host: str = "0.0.0.0", port: int = 8000):
     """

cognee-0.5.3.dev1/cognee/api/v1/health/__init__.py

@@ -0,0 +1 @@
+from .health import health_checker, HealthStatus

{cognee-0.5.3.dev0/cognee/api → cognee-0.5.3.dev1/cognee/api/v1/health}/health.py

@@ -3,6 +3,7 @@
 from io import BytesIO
 import time
 import asyncio
+from sqlalchemy import text
 from datetime import datetime, timezone
 from typing import Dict
 from enum import Enum
@@ -51,9 +52,11 @@ class HealthChecker:
             config = get_relational_config()
             engine = get_relational_engine()
 
-            # Test connection by creating a session
-            session = engine.get_session()
-            if session:
+            try:
+                async with engine.get_async_session() as session:
+                    # This works for both SQLite and PostgreSQL
+                    await session.execute(text("SELECT 1"))
+            finally:
                 session.close()
 
             response_time = int((time.time() - start_time) * 1000)

cognee-0.5.3.dev1/cognee/api/v1/health/routers/__init__.py

@@ -0,0 +1 @@
+from .get_health_router import get_health_router

cognee-0.5.3.dev1/cognee/api/v1/health/routers/get_health_router.py

@@ -0,0 +1,53 @@
+from fastapi import APIRouter
+from fastapi.responses import JSONResponse
+
+from cognee.api.v1.health import health_checker, HealthStatus
+
+
+def get_health_router():
+    health_router = APIRouter()
+
+    @health_router.get("", response_model=dict)
+    async def health_check():
+        """
+        Health check endpoint for liveness/readiness probes.
+        """
+        try:
+            health_status = await health_checker.get_health_status(detailed=False)
+            status_code = 503 if health_status.status == HealthStatus.UNHEALTHY else 200
+
+            return JSONResponse(
+                status_code=status_code,
+                content={
+                    "status": "ready" if status_code == 200 else "not ready",
+                    "health": health_status.status,
+                    "version": health_status.version,
+                },
+            )
+        except Exception as e:
+            return JSONResponse(
+                status_code=503,
+                content={"status": "not ready", "reason": f"health check failed: {str(e)}"},
+            )
+
+    @health_router.get("/detailed", response_model=dict)
+    async def detailed_health_check():
+        """
+        Comprehensive health status with component details.
+        """
+        try:
+            health_status = await health_checker.get_health_status(detailed=True)
+            status_code = 200
+            if health_status.status == HealthStatus.UNHEALTHY:
+                status_code = 503
+            elif health_status.status == HealthStatus.DEGRADED:
+                status_code = 503
+
+            return JSONResponse(status_code=status_code, content=health_status.model_dump())
+        except Exception as e:
+            return JSONResponse(
+                status_code=503,
+                content={"status": "unhealthy", "error": f"Health check system failure: {str(e)}"},
+            )
+
+    return health_router

{cognee-0.5.3.dev0 → cognee-0.5.3.dev1}/cognee/api/v1/users/routers/__init__.py

@@ -4,3 +4,4 @@ from .get_reset_password_router import get_reset_password_router
 from .get_users_router import get_users_router
 from .get_verify_router import get_verify_router
 from .get_visualize_router import get_visualize_router
+from .get_configuration_router import get_configuration_router

cognee-0.5.3.dev1/cognee/api/v1/users/routers/get_configuration_router.py

@@ -0,0 +1,46 @@
+from fastapi import APIRouter
+from fastapi import Form, Depends
+
+from cognee.api.DTO import InDTO
+from cognee.modules.users.models import User
+from cognee.modules.users.methods import get_authenticated_user
+from cognee.shared.logging_utils import get_logger
+from cognee.shared.usage_logger import log_usage
+from cognee.modules.users.methods import (
+    store_principal_configuration as method_store_principal_configuration,
+)
+from cognee.modules.users.methods import (
+    get_principal_configuration as method_get_principal_configuration,
+)
+
+logger = get_logger()
+
+
+class StorePrincipalConfigurationPayloadDTO(InDTO):
+    name: str = (Form(..., description="Name of the configuration to store"),)
+    config: dict = (Form(..., description="The configuration data to store as a JSON object"),)
+
+
+def get_configuration_router() -> APIRouter:
+    router = APIRouter()
+
+    @router.post("/store_user_configuration", response_model=None)
+    @log_usage(
+        function_name="POST /v1/configuration/store_user_configuration", log_type="api_endpoint"
+    )
+    async def store_user_configuration(
+        payload: StorePrincipalConfigurationPayloadDTO,
+        user: User = Depends(get_authenticated_user),
+    ):
+        await method_store_principal_configuration(
+            principal_id=user.id, name=payload.name, configuration=payload.config
+        )
+
+    @router.get("/get_user_configuration/{name}", response_model=dict)
+    async def get_user_configuration(
+        name: str,
+        user: User = Depends(get_authenticated_user),
+    ):
+        return await method_get_principal_configuration(principal_id=user.id, name=name)
+
+    return router

{cognee-0.5.3.dev0 → cognee-0.5.3.dev1}/cognee/infrastructure/databases/graph/get_graph_engine.py

@@ -99,6 +99,8 @@ def _create_graph_engine(
             graph_database_url=graph_database_url,
             graph_database_username=graph_database_username,
             graph_database_password=graph_database_password,
+            graph_database_port=graph_database_port,
+            graph_database_key=graph_database_key,
             database_name=graph_database_name,
         )
 

{cognee-0.5.3.dev0 → cognee-0.5.3.dev1}/cognee/infrastructure/databases/vector/chromadb/ChromaDBAdapter.py

@@ -3,6 +3,7 @@ import asyncio
 from uuid import UUID
 from typing import List, Optional
 from chromadb import AsyncHttpClient, Settings
+from pydantic import BaseModel
 
 from cognee.shared.logging_utils import get_logger
 from cognee.modules.storage.utils import get_own_properties
@@ -361,6 +362,7 @@ class ChromaDBAdapter(VectorDBInterface):
         with_vector: bool = False,
         normalized: bool = True,
         include_payload: bool = False,  # TODO: Add support for this parameter when set to False
+        node_name: Optional[List[str]] = None,  # TODO: Add support/functionality for this parameter
     ):
         """
         Search for items in a collection using either a text or a vector query.

{cognee-0.5.3.dev0 → cognee-0.5.3.dev1}/cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py

@@ -35,6 +35,7 @@ class IndexSchema(DataPoint):
     text: str
 
     metadata: dict = {"index_fields": ["text"]}
+    belongs_to_set: List[str] = []
 
 
 class LanceDBAdapter(VectorDBInterface):
@@ -237,6 +238,7 @@ class LanceDBAdapter(VectorDBInterface):
         with_vector: bool = False,
         normalized: bool = True,
         include_payload: bool = False,
+        node_name: Optional[List[str]] = None,
     ):
         if query_text is None and query_vector is None:
             raise MissingQueryParameterError()
@@ -259,12 +261,26 @@ class LanceDBAdapter(VectorDBInterface):
             if include_payload
             else ["id", "vector", "_distance"]
         )
-        result_values = (
-            await collection.vector_search(query_vector)
-            .select(select_columns)
-            .limit(limit)
-            .to_list()
-        )
+        if node_name:
+            # Escape quotes to make this input safer, since it's coming from the user
+            # At the time of writing this, no specific binding instructions found on LanceDB docs
+            escaped_node_names = [name.replace("'", "''") for name in node_name]
+            literal_node_names = "[" + ", ".join(f"'{name}'" for name in escaped_node_names) + "]"
+
+            result_values = (
+                await collection.vector_search(query_vector)
+                .where(f"array_has_any(payload.belongs_to_set, {literal_node_names})")
+                .select(select_columns)
+                .limit(limit)
+                .to_list()
+            )
+        else:
+            result_values = (
+                await collection.vector_search(query_vector)
+                .select(select_columns)
+                .limit(limit)
+                .to_list()
+            )
 
         if not result_values:
             return []
@@ -286,6 +302,7 @@ class LanceDBAdapter(VectorDBInterface):
         limit: Optional[int] = None,
         with_vectors: bool = False,
         include_payload: bool = False,
+        node_name: Optional[List[str]] = None,
     ):
         query_vectors = await self.embedding_engine.embed_text(query_texts)
 
@@ -297,6 +314,7 @@ class LanceDBAdapter(VectorDBInterface):
                     limit=limit,
                     with_vector=with_vectors,
                     include_payload=include_payload,
+                    node_name=node_name,
                 )
                 for query_vector in query_vectors
             ]
@@ -327,6 +345,7 @@ class LanceDBAdapter(VectorDBInterface):
                 IndexSchema(
                     id=str(data_point.id),
                     text=getattr(data_point, data_point.metadata["index_fields"][0]),
+                    belongs_to_set=(data_point.belongs_to_set or []),
                 )
                 for data_point in data_points
             ],

{cognee-0.5.3.dev0 → cognee-0.5.3.dev1}/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py

@@ -43,6 +43,7 @@ class IndexSchema(DataPoint):
     text: str
 
     metadata: dict = {"index_fields": ["text"]}
+    belongs_to_set: List[str] = []
 
 
 class PGVectorAdapter(SQLAlchemyAdapter, VectorDBInterface):
@@ -258,6 +259,7 @@ class PGVectorAdapter(SQLAlchemyAdapter, VectorDBInterface):
                 IndexSchema(
                     id=data_point.id,
                     text=DataPoint.get_embeddable_data(data_point),
+                    belongs_to_set=(data_point.belongs_to_set or []),
                 )
                 for data_point in data_points
             ],
@@ -307,6 +309,7 @@ class PGVectorAdapter(SQLAlchemyAdapter, VectorDBInterface):
         limit: Optional[int] = 15,
         with_vector: bool = False,
         include_payload: bool = False,
+        node_name: Optional[List[str]] = None,
     ) -> List[ScoredResult]:
         if query_text is None and query_vector is None:
             raise MissingQueryParameterError()
@@ -338,10 +341,30 @@ class PGVectorAdapter(SQLAlchemyAdapter, VectorDBInterface):
         )
         # Use async session to connect to the database
         async with self.get_async_session() as session:
-            query = select(
-                *select_columns,
-                PGVectorDataPoint.c.vector.cosine_distance(query_vector).label("similarity"),
-            ).order_by("similarity")
+            if node_name:
+                from sqlalchemy import cast, bindparam
+                from sqlalchemy.dialects.postgresql import JSONB, ARRAY, TEXT
+
+                target = bindparam("target", value=node_name, type_=ARRAY(TEXT()))
+                query = (
+                    select(
+                        *select_columns,
+                        PGVectorDataPoint.c.vector.cosine_distance(query_vector).label(
+                            "similarity"
+                        ),
+                    )
+                    .where(
+                        cast(PGVectorDataPoint.c.payload, JSONB)
+                        .op("->")("belongs_to_set")
+                        .op("?|")(target)
+                    )
+                    .order_by("similarity")
+                )
+            else:
+                query = select(
+                    *select_columns,
+                    PGVectorDataPoint.c.vector.cosine_distance(query_vector).label("similarity"),
+                ).order_by("similarity")
 
             if limit > 0:
                 query = query.limit(limit)
@@ -386,6 +409,7 @@ class PGVectorAdapter(SQLAlchemyAdapter, VectorDBInterface):
         limit: int = None,
         with_vectors: bool = False,
         include_payload: bool = False,
+        node_name: Optional[List[str]] = None,
     ):
         query_vectors = await self.embedding_engine.embed_text(query_texts)
 
@@ -397,6 +421,7 @@ class PGVectorAdapter(SQLAlchemyAdapter, VectorDBInterface):
                     limit=limit,
                     with_vector=with_vectors,
                     include_payload=include_payload,
+                    node_name=node_name,
                 )
                 for query_vector in query_vectors
             ]

{cognee-0.5.3.dev0 → cognee-0.5.3.dev1}/cognee/infrastructure/databases/vector/vector_db_interface.py

@@ -88,6 +88,7 @@ class VectorDBInterface(Protocol):
         limit: Optional[int],
         with_vector: bool = False,
         include_payload: bool = False,
+        node_name: Optional[List[str]] = None,
     ):
         """
         Perform a search in the specified collection using either a text query or a vector
@@ -107,6 +108,8 @@ class VectorDBInterface(Protocol):
             - include_payload (bool): Whether to include the payload data with search. Search is faster when set to False.
               Payload contains metadata about the data point, useful for searches that are only based on embedding distances
               like the RAG_COMPLETION search type, but not needed when search also contains graph data.
+            - node_name (Optional[List[str]]): An optional list of names of nodes. Search results are filtered
+              based on these, and only data which has at least one of the names in its "belongs_to_set" field is returned.
         """
         raise NotImplementedError
 
@@ -118,6 +121,7 @@ class VectorDBInterface(Protocol):
         limit: Optional[int],
         with_vectors: bool = False,
         include_payload: bool = False,
+        node_name: Optional[List[str]] = None,
     ):
         """
         Perform a batch search using multiple text queries against a collection.

{cognee-0.5.3.dev0 → cognee-0.5.3.dev1}/cognee/infrastructure/engine/models/DataPoint.py

@@ -46,7 +46,7 @@ class DataPoint(BaseModel):
     topological_rank: Optional[int] = 0
     metadata: Optional[MetaData] = {"index_fields": []}
     type: str = Field(default_factory=lambda: DataPoint.__name__)
-    belongs_to_set: Optional[List["DataPoint"]] = None
+    belongs_to_set: Optional[List["DataPoint"] | List[str]] = None
     source_pipeline: Optional[str] = None
     source_task: Optional[str] = None
     source_note_set: Optional[str] = None

{cognee-0.5.3.dev0 → cognee-0.5.3.dev1}/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/ollama/adapter.py

@@ -126,7 +126,7 @@ class OllamaAPIAdapter(LLMInterface):
         before_sleep=before_sleep_log(logger, logging.DEBUG),
         reraise=True,
     )
-    async def create_transcript(self, input_file: str, **kwargs) -> str:
+    async def create_transcript(self, input: str, **kwargs) -> str:
         """
         Generate an audio transcript from a user query.
 
@@ -137,7 +137,7 @@ class OllamaAPIAdapter(LLMInterface):
         Parameters:
         -----------
 
-            - input_file (str): The path to the audio file to be transcribed.
+            - input (str): The path to the audio file to be transcribed.
 
         Returns:
         --------
@@ -145,7 +145,7 @@ class OllamaAPIAdapter(LLMInterface):
             - str: The transcription of the audio as a string.
         """
 
-        async with open_data_file(input_file, mode="rb") as audio_file:
+        async with open_data_file(input, mode="rb") as audio_file:
             transcription = self.aclient.audio.transcriptions.create(
                 model="whisper-1",  # Ensure the correct model for transcription
                 file=audio_file,
@@ -167,7 +167,7 @@ class OllamaAPIAdapter(LLMInterface):
         before_sleep=before_sleep_log(logger, logging.DEBUG),
         reraise=True,
     )
-    async def transcribe_image(self, input_file: str, **kwargs) -> str:
+    async def transcribe_image(self, input: str, **kwargs) -> str:
         """
         Transcribe content from an image using base64 encoding.
 
@@ -179,7 +179,7 @@ class OllamaAPIAdapter(LLMInterface):
         Parameters:
         -----------
 
-            - input_file (str): The path to the image file to be transcribed.
+            - input (str): The path to the image file to be transcribed.
 
         Returns:
         --------
@@ -187,7 +187,7 @@ class OllamaAPIAdapter(LLMInterface):
             - str: The transcription of the image's content as a string.
         """
 
-        async with open_data_file(input_file, mode="rb") as image_file:
+        async with open_data_file(input, mode="rb") as image_file:
             encoded_image = base64.b64encode(image_file.read()).decode("utf-8")
 
         response = self.aclient.chat.completions.create(

{cognee-0.5.3.dev0 → cognee-0.5.3.dev1}/cognee/infrastructure/llm/utils.py

@@ -6,6 +6,7 @@ from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.ll
     get_llm_client,
 )
 from cognee.shared.logging_utils import get_logger
+from cognee.infrastructure.llm.LLMGateway import LLMGateway
 
 logger = get_logger()
 
@@ -74,25 +75,12 @@ def get_model_max_completion_tokens(model_name: str):
 async def test_llm_connection():
     """
     Test connectivity to the LLM endpoint using a simple completion call.
-
-    Uses a lightweight litellm.acompletion() call instead of the full instructor
-    structured output pipeline to avoid failures with local models that cannot
-    produce instructor-compatible structured JSON. The call is wrapped in a timeout
-    to prevent indefinite hangs.
     """
-    from cognee.infrastructure.llm.config import get_llm_config
-
-    config = get_llm_config()
     try:
-        await asyncio.wait_for(
-            litellm.acompletion(
-                model=config.llm_model,
-                messages=[{"role": "user", "content": "Say ok"}],
-                api_key=config.llm_api_key,
-                api_base=config.llm_endpoint,
-                max_tokens=5,
-            ),
-            timeout=CONNECTION_TEST_TIMEOUT_SECONDS,
+        await LLMGateway.acreate_structured_output(
+            text_input="test",
+            system_prompt='Respond to me with the following string: "test"',
+            response_model=str,
         )
     except asyncio.TimeoutError:
         msg = (

{cognee-0.5.3.dev0 → cognee-0.5.3.dev1}/cognee/modules/graph/utils/get_graph_from_model.py

@@ -102,7 +102,8 @@ def _process_datapoint_field(
     excluded_properties: set,
 ) -> None:
     """Process a field containing DataPoints, always working with lists."""
-    excluded_properties.add(field_name)
+    if field_name != "belongs_to_set":
+        excluded_properties.add(field_name)
 
     for edge_metadata, datapoints in edge_datapoint_pairs:
         relationship_key = _get_relationship_key(field_name, edge_metadata)
@@ -207,6 +208,13 @@ async def get_graph_from_model(
                 excluded_properties,
             )
 
+            # We want to enable nodeset filtering on the vector database side
+            if field_name == "belongs_to_set":
+                node_set_names = []
+                for node_set in field_value:
+                    node_set_names.append(node_set.name)
+                data_point_properties[field_name] = node_set_names
+
     # Create node for current DataPoint if needed
     if include_root and data_point_id not in added_nodes:
         SimpleDataPointModel = copy_model(

{cognee-0.5.3.dev0 → cognee-0.5.3.dev1}/cognee/modules/retrieval/utils/brute_force_triplet_search.py

@@ -178,6 +178,7 @@ async def brute_force_triplet_search(
             query_batch=query_batch if query_list_length else None,
             collections=collections,
             wide_search_limit=wide_search_limit,
+            node_name=node_name,
         )
 
         if not vector_search.has_results():