mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-08-12 03:33:33 +08:00
Feat: v0.27.0 model provider (#16604)
This commit is contained in:
@@ -113,9 +113,15 @@ class TaskDict(TypedDict, total=False):
|
||||
llm_id: str
|
||||
"""LLM model identifier."""
|
||||
|
||||
tenant_llm_id: str | None
|
||||
"""Tenant model ID for LLM (id in tenant_model table)."""
|
||||
|
||||
embd_id: str
|
||||
"""Embedding model identifier."""
|
||||
|
||||
tenant_embd_id: str | None
|
||||
"""Tenant model ID for embedding (id in tenant_model table)."""
|
||||
|
||||
from_page: int
|
||||
"""Starting page number for processing (0-based)."""
|
||||
|
||||
@@ -227,7 +233,9 @@ class TaskContext:
|
||||
"kb_parser_config": {},
|
||||
"language": "Chinese",
|
||||
"llm_id": "",
|
||||
"tenant_llm_id": "",
|
||||
"embd_id": "",
|
||||
"tenant_embd_id": "",
|
||||
"from_page": 0,
|
||||
"to_page": -1,
|
||||
"task_type": "",
|
||||
@@ -361,11 +369,21 @@ class TaskContext:
|
||||
"""LLM model identifier."""
|
||||
return self._task.get("llm_id", self._DEFAULTS["llm_id"])
|
||||
|
||||
@property
|
||||
def tenant_llm_id(self) -> str | None:
|
||||
"""Tenant model ID for LLM (id in tenant_model table)."""
|
||||
return self._task.get("tenant_llm_id", self._DEFAULTS["tenant_llm_id"]) or None
|
||||
|
||||
@property
|
||||
def embd_id(self) -> str:
|
||||
"""Embedding model identifier."""
|
||||
return self._task.get("embd_id", self._DEFAULTS["embd_id"])
|
||||
|
||||
@property
|
||||
def tenant_embd_id(self) -> str | None:
|
||||
"""Tenant model ID for embedding (id in tenant_model table)."""
|
||||
return self._task.get("tenant_embd_id", self._DEFAULTS["tenant_embd_id"]) or None
|
||||
|
||||
# =========================================================================
|
||||
# Page range properties
|
||||
# =========================================================================
|
||||
|
||||
Reference in New Issue
Block a user