优化默认语言模型获取

This commit is contained in:
Azmat@qq.com
2026-09-03 16:00:14 +08:00
parent 6a628b0ca7
commit f53b6e1035
6 changed files with 49 additions and 28 deletions
+3 -18
View File
@@ -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(