diff --git a/cognee/modules/retrieval/context_providers/TripletSearchContextProvider.py b/cognee/modules/retrieval/context_providers/TripletSearchContextProvider.py index b539055fa..6cdd57eec 100644 --- a/cognee/modules/retrieval/context_providers/TripletSearchContextProvider.py +++ b/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." diff --git a/cognee/modules/retrieval/graph_completion_retriever.py b/cognee/modules/retrieval/graph_completion_retriever.py index cd4944ca9..1772c6300 100644 --- a/cognee/modules/retrieval/graph_completion_retriever.py +++ b/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, diff --git a/cognee/modules/retrieval/utils/brute_force_triplet_search.py b/cognee/modules/retrieval/utils/brute_force_triplet_search.py index aa3233776..172c45d6d 100644 --- a/cognee/modules/retrieval/utils/brute_force_triplet_search.py +++ b/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 diff --git a/examples/python/graphiti_example.py b/examples/python/graphiti_example.py index f3bd86847..ece9c452b 100644 --- a/examples/python/graphiti_example.py +++ b/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"], )