apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-04-25 00:36:31 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

feat: use batch embedding in create_document_chunks

Browse source
This commit is contained in:
CREDO23 2026-03-09 16:21:14 +02:00
parent 929445afd9
commit c4f2e9a3a5
1 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
surfsense_backend/app/utils/document_converters.py
View file

@ -223,12 +223,11 @@ async def create_document_chunks(content: str) -> list[Chunk]:
Returns:
List of Chunk objects with embeddings
"""
chunk_texts = [c.text for c in config.chunker_instance.chunk(content)]
chunk_embeddings = embed_texts(chunk_texts)
return [
Chunk(
content=chunk.text,
embedding=embed_text(chunk.text),
)
for chunk in config.chunker_instance.chunk(content)
Chunk(content=text, embedding=emb)
for text, emb in zip(chunk_texts, chunk_embeddings)
]