gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

ruff fix

Browse source
This commit is contained in:
hajdul88 2025-09-18 18:25:00 +02:00
parent 7c33418ae9
commit 9ee93a4260
2 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
cognee/tasks/graph/extract_graph_from_data.py
View file

@ -29,20 +29,20 @@ async def integrate_chunk_graphs(
ontology_resolver: BaseOntologyResolver, ontology_resolver: BaseOntologyResolver,
) -> List[DocumentChunk]: ) -> List[DocumentChunk]:
"""Integrate chunk graphs with ontology validation and store in databases. """Integrate chunk graphs with ontology validation and store in databases.
This function processes document chunks and their associated knowledge graphs, This function processes document chunks and their associated knowledge graphs,
validates entities against an ontology resolver, and stores the integrated validates entities against an ontology resolver, and stores the integrated
data points and edges in the configured databases. data points and edges in the configured databases.
Args: Args:
data_chunks: List of document chunks containing source data data_chunks: List of document chunks containing source data
chunk_graphs: List of knowledge graphs corresponding to each chunk chunk_graphs: List of knowledge graphs corresponding to each chunk
graph_model: Pydantic model class for graph data validation graph_model: Pydantic model class for graph data validation
ontology_resolver: Resolver for validating entities against ontology ontology_resolver: Resolver for validating entities against ontology
Returns: Returns:
List of updated DocumentChunk objects with integrated data List of updated DocumentChunk objects with integrated data
Raises: Raises:
InvalidChunkGraphInputError: If input validation fails InvalidChunkGraphInputError: If input validation fails
InvalidGraphModelError: If graph model validation fails InvalidGraphModelError: If graph model validation fails

6
cognee/tasks/graph/extract_graph_from_data_v2.py
View file

@ -20,15 +20,15 @@ async def extract_graph_from_data(
ontology_adapter: BaseOntologyResolver = None, ontology_adapter: BaseOntologyResolver = None,
) -> List[DocumentChunk]: ) -> List[DocumentChunk]:
"""Extract and update graph data from document chunks using cascade extraction. """Extract and update graph data from document chunks using cascade extraction.
This function performs multi-step graph extraction from document chunks, This function performs multi-step graph extraction from document chunks,
using cascade extraction techniques to build comprehensive knowledge graphs. using cascade extraction techniques to build comprehensive knowledge graphs.
Args: Args:
data_chunks: List of document chunks to process data_chunks: List of document chunks to process
n_rounds: Number of extraction rounds to perform (default: 2) n_rounds: Number of extraction rounds to perform (default: 2)
ontology_adapter: Resolver for validating entities against ontology ontology_adapter: Resolver for validating entities against ontology
Returns: Returns:
List of updated DocumentChunk objects with extracted graph data List of updated DocumentChunk objects with extracted graph data
""" """