diff --git a/api/utils/validation_utils.py b/api/utils/validation_utils.py index d60dc5561..3c53a6332 100644 --- a/api/utils/validation_utils.py +++ b/api/utils/validation_utils.py @@ -371,7 +371,7 @@ class ParserConfig(Base): raptor: RaptorConfig | None = None tag_kb_ids: list[str] = Field(default_factory=list) topn_tags: int = Field(default=1, ge=1, le=10) - filename_embd_weight: float | None = Field(default=None, ge=0.0, le=1.0) + filename_embd_weight: float | None = Field(default=0.1, ge=0.0, le=1.0) task_page_size: int | None = Field(default=None, ge=1) pages: list[list[int]] | None = None diff --git a/rag/svr/task_executor.py b/rag/svr/task_executor.py index fbe6f134f..e5fed0bea 100644 --- a/rag/svr/task_executor.py +++ b/rag/svr/task_executor.py @@ -434,7 +434,6 @@ async def embedding(docs, mdl, parser_config=None, callback=None): tk_count += c callback(prog=0.7 + 0.2 * (i + 1) / len(cnts), msg="") cnts = cnts_ - title_w = float(parser_config.get("filename_embd_weight", 0.1)) vects = (title_w * tts + (1 - title_w) * cnts) if len(tts) == len(cnts) else cnts