jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix(batch_create_segment_to_index_task): count max_position in memory. (#12929)

Browse source
This commit is contained in:
-LAN- 2025-01-22 13:39:02 +08:00 • committed by GitHub
commit f91f5c7401
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 78 additions and 62 deletions
Download patch file Download diff file Expand all files Collapse all files

3
api/models/dataset.py
View file

@ -13,6 +13,7 @@ from typing import Any, cast
from sqlalchemy import func
from sqlalchemy.dialects.postgresql import JSONB
from sqlalchemy.orm import Mapped
from configs import dify_config
from core.rag.retrieval.retrieval_methods import RetrievalMethod
@ -515,7 +516,7 @@ class DocumentSegment(db.Model): # type: ignore[name-defined]
tenant_id = db.Column(StringUUID, nullable=False)
dataset_id = db.Column(StringUUID, nullable=False)
document_id = db.Column(StringUUID, nullable=False)
position = db.Column(db.Integer, nullable=False)
position: Mapped[int]
content = db.Column(db.Text, nullable=False)
answer = db.Column(db.Text, nullable=True)
word_count = db.Column(db.Integer, nullable=False)