mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-07-19 14:11:04 +08:00
refa: resolve tenant model refs consistently (#16744)
This commit is contained in:
@@ -96,6 +96,7 @@ def _load_canvas_runtime(monkeypatch):
|
||||
|
||||
tenant_model_service = ModuleType("api.db.joint_services.tenant_model_service")
|
||||
tenant_model_service.get_tenant_default_model_by_type = lambda *_a, **_kw: None
|
||||
tenant_model_service.resolve_model_config = lambda *_a, **_kw: None
|
||||
monkeypatch.setitem(
|
||||
sys.modules,
|
||||
"api.db.joint_services.tenant_model_service",
|
||||
|
||||
@@ -310,6 +310,7 @@ def _load_chunk_module(monkeypatch):
|
||||
tenant_model_service_mod = ModuleType("api.db.joint_services.tenant_model_service")
|
||||
tenant_model_service_mod.get_model_config_by_id = lambda *_args, **_kwargs: {"llm_name": "embed", "model_type": "embedding"}
|
||||
tenant_model_service_mod.get_model_config_from_provider_instance = lambda *_args, **_kwargs: {"llm_name": "embed", "model_type": "embedding"}
|
||||
tenant_model_service_mod.resolve_model_config = lambda *_args, **_kwargs: {"llm_name": "embed", "model_type": "embedding"}
|
||||
tenant_model_service_mod.get_tenant_default_model_by_type = lambda *_args, **_kwargs: {"llm_name": "chat", "model_type": "chat"}
|
||||
monkeypatch.setitem(sys.modules, "api.db.joint_services.tenant_model_service", tenant_model_service_mod)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user