소스 검색

add child chunk preview number limit (#12309)

tags/0.15.0
Jyong 10 달 전
부모
커밋
b873e6349c
No account linked to committer's email address
2개의 변경된 파일10개의 추가작업 그리고 0개의 파일을 삭제
  1. 5
    0
      api/configs/feature/__init__.py
  2. 5
    0
      api/core/rag/index_processor/processor/parent_child_index_processor.py

+ 5
- 0
api/configs/feature/__init__.py 파일 보기

@@ -667,6 +667,11 @@ class IndexingConfig(BaseSettings):
default=4000,
)

CHILD_CHUNKS_PREVIEW_NUMBER: PositiveInt = Field(
description="Maximum number of child chunks to preview",
default=50,
)


class MultiModalTransferConfig(BaseSettings):
MULTIMODAL_SEND_FORMAT: Literal["base64", "url"] = Field(

+ 5
- 0
api/core/rag/index_processor/processor/parent_child_index_processor.py 파일 보기

@@ -3,6 +3,7 @@
import uuid
from typing import Optional

from configs import dify_config
from core.model_manager import ModelInstance
from core.rag.cleaner.clean_processor import CleanProcessor
from core.rag.datasource.retrieval_service import RetrievalService
@@ -80,6 +81,10 @@ class ParentChildIndexProcessor(BaseIndexProcessor):
child_nodes = self._split_child_nodes(
document, rules, process_rule.get("mode"), kwargs.get("embedding_model_instance")
)
if kwargs.get("preview"):
if len(child_nodes) > dify_config.CHILD_CHUNKS_PREVIEW_NUMBER:
child_nodes = child_nodes[: dify_config.CHILD_CHUNKS_PREVIEW_NUMBER]

document.children = child_nodes
doc_id = str(uuid.uuid4())
hash = helper.generate_text_hash(document.page_content)

Loading…
취소
저장