fix: lint errors

This commit is contained in:
Boris Arzentar 2025-04-23 13:18:04 +02:00
parent bf9678065c
commit c46c80196e
3 changed files with 3 additions and 5 deletions

View file

@ -34,7 +34,6 @@ async def integrate_chunk_graphs(
existing_edges_map = await retrieve_existing_edges(
data_chunks,
chunk_graphs,
graph_engine,
)
graph_nodes, graph_edges = expand_with_nodes_and_edges(

View file

@ -4,9 +4,7 @@ from cognee.modules.graph.utils import deduplicate_nodes_and_edges, get_graph_fr
from distributed.queues import save_data_points_queue
async def save_data_points(
data_points_and_relationships: tuple[list, list]
):
async def save_data_points(data_points_and_relationships: tuple[list, list]):
data_points = data_points_and_relationships[0]
data_point_connections = data_points_and_relationships[1]

View file

@ -8,7 +8,8 @@ from cognee.modules.chunking.models.DocumentChunk import DocumentChunk
async def summarize_text(
data_points_and_relationships: tuple[list[DocumentChunk], list], summarization_model: Type[BaseModel]
data_points_and_relationships: tuple[list[DocumentChunk], list],
summarization_model: Type[BaseModel],
):
data_chunks = data_points_and_relationships[0]
edges = data_points_and_relationships[1]