优化默认语言模型获取
This commit is contained in:
@@ -520,25 +520,10 @@ def _parse_extracted_entities_response(text: str) -> tuple[list[dict], list[dict
|
||||
return entities, seg_refs
|
||||
|
||||
|
||||
# 提取步固定锁定豆包 2.0 Pro(与脚本生成同款),不靠 get_default_model 的「最早创建」排序——
|
||||
# 避免不同环境 DB 创建序漂移把提取路由到别的(中转站)推理模型。取不到再回落默认文本模型。
|
||||
EXTRACT_TEXT_MODEL_NAME = "doubao-seed-2-0-pro-260215"
|
||||
|
||||
|
||||
def _resolve_extract_model_config():
|
||||
"""提取实体用的文本模型:优先豆包 2.0 Pro(active 且 provider active),否则回落默认文本模型。"""
|
||||
pinned = (
|
||||
ModelConfig.objects.select_related("provider")
|
||||
.filter(
|
||||
name=EXTRACT_TEXT_MODEL_NAME,
|
||||
capability=ModelConfig.Capability.TEXT,
|
||||
status=ModelConfig.Status.ACTIVE,
|
||||
provider__status="active",
|
||||
)
|
||||
.order_by("created_at")
|
||||
.first()
|
||||
)
|
||||
return pinned or get_default_model(ModelConfig.Capability.TEXT)
|
||||
"""提取实体用平台后台钦定的默认文本模型(ModelConfig.is_default)。
|
||||
未设默认时与 get_default_model 一致:回落最早创建的 active 文本模型。"""
|
||||
return get_default_model(ModelConfig.Capability.TEXT)
|
||||
|
||||
|
||||
def _collect_extract_text(
|
||||
|
||||
Reference in New Issue
Block a user