diff --git a/graphiti_core/llm_client/openai_client.py b/graphiti_core/llm_client/openai_client.py index e73af497..dd727576 100644 --- a/graphiti_core/llm_client/openai_client.py +++ b/graphiti_core/llm_client/openai_client.py @@ -60,6 +60,5 @@ class OpenAIClient(LLMClient): result = response.choices[0].message.content or '' return json.loads(result) except Exception as e: - print(openai_messages) logger.error(f'Error in generating LLM response: {e}') raise diff --git a/graphiti_core/search/search_utils.py b/graphiti_core/search/search_utils.py index 9937a3cd..ec15966f 100644 --- a/graphiti_core/search/search_utils.py +++ b/graphiti_core/search/search_utils.py @@ -424,7 +424,7 @@ async def node_distance_reranker( records, _, _ = await driver.execute_query( """ MATCH (source:Entity)-[r:RELATES_TO {uuid: $edge_uuid}]->(target:Entity) - MATCH p = SHORTEST 1 (center:Entity)-[:RELATES_TO]-+(n:Entity) + MATCH p = SHORTEST 1 (center:Entity)-[:RELATES_TO*1..10]->(n:Entity) WHERE center.uuid = $center_uuid AND n.uuid IN [source.uuid, target.uuid] RETURN min(length(p)) AS score, source.uuid AS source_uuid, target.uuid AS target_uuid """, diff --git a/graphiti_core/utils/maintenance/edge_operations.py b/graphiti_core/utils/maintenance/edge_operations.py index 4fea6f8b..848f6250 100644 --- a/graphiti_core/utils/maintenance/edge_operations.py +++ b/graphiti_core/utils/maintenance/edge_operations.py @@ -70,7 +70,6 @@ async def extract_edges( } llm_response = await llm_client.generate_response(prompt_library.extract_edges.v2(context)) - print(llm_response) edges_data = llm_response.get('edges', []) end = time()