fix: align model default handling (#16782)

This commit is contained in:
buua436
2026-07-10 10:34:19 +08:00
committed by GitHub
parent d48a5622df
commit 74bbbba3e0
15 changed files with 169 additions and 71 deletions

View File

@@ -18,6 +18,7 @@
import importlib.util
import sys
from pathlib import Path
from enum import IntEnum
from types import ModuleType, SimpleNamespace
from unittest.mock import MagicMock
@@ -26,6 +27,16 @@ import pytest
pytestmark = pytest.mark.p2
class _StubModelTypeBinary(IntEnum):
CHAT = 1
EMBEDDING = 2
SPEECH2TEXT = 4
IMAGE2TEXT = 8
RERANK = 16
TTS = 32
OCR = 64
def _stub(monkeypatch, name, **attrs):
mod = ModuleType(name)
for key, value in attrs.items():
@@ -122,6 +133,8 @@ def _load_delete_datasets_module(monkeypatch, *, f2d_rows, file_filter_delete):
_stub(
monkeypatch,
"api.db.db_models",
DB=SimpleNamespace(connection_context=lambda: lambda func: func),
TenantModel=SimpleNamespace(),
File=SimpleNamespace(source_type="source_type", id="id", type="type", name="name"),
)
_stub(
@@ -130,8 +143,18 @@ def _load_delete_datasets_module(monkeypatch, *, f2d_rows, file_filter_delete):
PAGERANK_FLD="pagerank",
TAG_FLD="tag",
FileSource=SimpleNamespace(KNOWLEDGEBASE="knowledgebase"),
PipelineTaskType=SimpleNamespace(
PARSE="parse",
DOWNLOAD="download",
RAPTOR="raptor",
GRAPH_RAG="graph_rag",
MINDMAP="mindmap",
ARTIFACT="artifact",
SKILL="skill",
),
StatusEnum=SimpleNamespace(),
LLMType=SimpleNamespace(),
ModelTypeBinary=_StubModelTypeBinary,
)
_stub(
monkeypatch,

View File

@@ -25,6 +25,7 @@ import sys
from pathlib import Path
from enum import IntEnum
from types import ModuleType, SimpleNamespace
from unittest.mock import MagicMock
import pytest
@@ -141,6 +142,7 @@ def _load_module(monkeypatch, *, tenant_model_records, factory_llm_infos=None):
ensure_mineru_from_env=lambda *a, **kw: None,
ensure_paddleocr_from_env=lambda *a, **kw: None,
ensure_opendataloader_from_env=lambda *a, **kw: None,
resolve_model_id=MagicMock(),
)
_stub(
monkeypatch,