fix: Remove creation of default user during search

cognee/modules/retrieval/context_providers/TripletSearchContextProvider.py

@@ -42,14 +42,12 @@ class TripletSearchContextProvider(BaseContextProvider):
         self,
         entities: List[DataPoint],
         query: str,
-        user: User,
         memory_fragment: CogneeGraph,
     ) -> List:
         """Creates search tasks for valid entities."""
         tasks = [
             brute_force_triplet_search(
                 query=f"{entity_text} {query}",
-                user=user,
                 top_k=self.top_k,
                 collections=self.collections,
                 properties_to_project=self.properties_to_project,
@@ -84,9 +82,8 @@ class TripletSearchContextProvider(BaseContextProvider):
         if not entities:
             return "No entities provided for context search."
 
-        user = await get_default_user()
         memory_fragment = await get_memory_fragment(self.properties_to_project)
-        search_tasks = self._get_search_tasks(entities, query, user, memory_fragment)
+        search_tasks = self._get_search_tasks(entities, query, memory_fragment)
 
         if not search_tasks:
             return "No valid entities found for context search."

cognee/modules/retrieval/graph_completion_retriever.py

@@ -93,11 +93,8 @@ class GraphCompletionRetriever(BaseGraphRetriever):
                         for field_name in index_fields:
                             vector_index_collections.append(f"{subclass.__name__}_{field_name}")
 
-        user = await get_default_user()
-
         found_triplets = await brute_force_triplet_search(
             query,
-            user=user,
             top_k=self.top_k,
             collections=vector_index_collections or None,
             node_type=self.node_type,

cognee/modules/retrieval/utils/brute_force_triplet_search.py

@@ -89,7 +89,6 @@ async def get_memory_fragment(
 
 async def brute_force_triplet_search(
     query: str,
-    user: User,
     top_k: int = 5,
     collections: Optional[List[str]] = None,
     properties_to_project: Optional[List[str]] = None,
@@ -102,7 +101,6 @@ async def brute_force_triplet_search(
 
     Args:
         query (str): The search query.
-        user (User): The user performing the search.
         top_k (int): The number of top results to retrieve.
         collections (Optional[List[str]]): List of collections to query.
         properties_to_project (Optional[List[str]]): List of properties to project.
@@ -139,8 +137,6 @@ async def brute_force_triplet_search(
 
     query_vector = (await vector_engine.embedding_engine.embed_text([query]))[0]
 
-    send_telemetry("cognee.brute_force_triplet_search EXECUTION STARTED", user.id)
-
     async def search_in_collection(collection_name: str):
         try:
             return await vector_engine.search(
@@ -176,20 +172,15 @@ async def brute_force_triplet_search(
 
         results = await memory_fragment.calculate_top_triplet_importances(k=top_k)
 
-        send_telemetry("cognee.brute_force_triplet_search EXECUTION COMPLETED", user.id)
-
         return results
 
     except CollectionNotFoundError:
         return []
     except Exception as error:
         logger.error(
-            "Error during brute force search for user: %s, query: %s. Error: %s",
-            user.id,
+            "Error during brute force search for query: %s. Error: %s",
             query,
             error,
         )
-        send_telemetry(
-            "cognee.brute_force_triplet_search EXECUTION FAILED", user.id, {"error": str(error)}
-        )
+        send_telemetry("cognee.brute_force_triplet_search EXECUTION FAILED", {"error": str(error)})
         raise error

examples/python/graphiti_example.py

@@ -48,7 +48,6 @@ async def main():
     query = "When was Kamala Harris in office?"
     triplets = await brute_force_triplet_search(
         query=query,
-        user=user,
         top_k=3,
         collections=["graphitinode_content", "graphitinode_name", "graphitinode_summary"],
     )