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

Remove passing of metadata_id to DocumentChunk

Browse source
This commit is contained in:
Leon Luithlen 2024-11-27 12:22:28 +01:00
parent 80517f5117
commit aacba555c9
1 changed files with 3 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
cognee/modules/chunking/TextChunker.py
View file

@ -34,8 +34,7 @@ class TextChunker():
word_count = chunk_data["word_count"], word_count = chunk_data["word_count"],
is_part_of = self.document, is_part_of = self.document,
chunk_index = self.chunk_index, chunk_index = self.chunk_index,
cut_type = chunk_data["cut_type"], cut_type = chunk_data["cut_type"]
metadata_id = self.document.metadata_id
) )
paragraph_chunks = [] paragraph_chunks = []
self.chunk_size = 0 self.chunk_size = 0
@ -48,8 +47,7 @@ class TextChunker():
word_count = self.chunk_size, word_count = self.chunk_size,
is_part_of = self.document, is_part_of = self.document,
chunk_index = self.chunk_index, chunk_index = self.chunk_index,
cut_type = paragraph_chunks[len(paragraph_chunks) - 1]["cut_type"], cut_type = paragraph_chunks[len(paragraph_chunks) - 1]["cut_type"]
metadata_id = self.document.metadata_id
) )
except Exception as e: except Exception as e:
print(e) print(e)
@ -66,8 +64,7 @@ class TextChunker():
word_count = self.chunk_size, word_count = self.chunk_size,
is_part_of = self.document, is_part_of = self.document,
chunk_index = self.chunk_index, chunk_index = self.chunk_index,
cut_type = paragraph_chunks[len(paragraph_chunks) - 1]["cut_type"], cut_type = paragraph_chunks[len(paragraph_chunks) - 1]["cut_type"]
metadata_id = self.document.metadata_id
) )
except Exception as e: except Exception as e:
print(e) print(e)