diff --git a/api/core/rag/index_processor/processor/paragraph_index_processor.py b/api/core/rag/index_processor/processor/paragraph_index_processor.py index 79c2c16b90..7c2031258e 100644 --- a/api/core/rag/index_processor/processor/paragraph_index_processor.py +++ b/api/core/rag/index_processor/processor/paragraph_index_processor.py @@ -12,6 +12,7 @@ from core.rag.extractor.extract_processor import ExtractProcessor from core.rag.index_processor.index_processor_base import BaseIndexProcessor from core.rag.models.document import Document from core.tools.utils.text_processing_utils import remove_leading_symbols +from core.workflow.nodes.knowledge_index.entities import GeneralStructureChunk from libs import helper from models.dataset import Dataset, DatasetProcessRule from services.entities.knowledge_entities.knowledge_entities import Rule @@ -126,7 +127,6 @@ class ParagraphIndexProcessor(BaseIndexProcessor): docs.append(doc) return docs - def index(self, dataset: Dataset, document: Document, chunks: list[Mapping[str, Any]]): - for chunk in chunks: - GeneralDocument.create( + def index(self, dataset: Dataset, document: Document, chunks: Mapping[str, Any]): + paragraph = GeneralStructureChunk(**chunks) pass