opensource/dify
3
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-09-24 23:22:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: handle Xinference model credential context (#38348)

Browse Source
This commit is contained in:
AsperforMias
2026-07-03 04:12:00 +00:00
committed by GitHub
parent f4ec608ef4
commit 1bc279e7d4
8 changed files with 177 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
api/tests/unit_tests/core/entities/test_entities_provider_configuration.py
+23
View File
@@ -1337,6 +1337,29 @@ def test_create_update_delete_custom_model_credential_flow() -> None:
assert provider_model_record.credential_id is None
assert mock_cache.return_value.delete.call_count == 2
session = Mock()
mismatched_credential_record = SimpleNamespace(
id="cred-2",
model_name="stored-model",
model_type=ModelType.TEXT_EMBEDDING,
)
provider_model_record = SimpleNamespace(id="model-2", credential_id="cred-2", updated_at=None)
session.execute.side_effect = [
_exec_result(scalar_one_or_none=None),
_exec_result(scalar_one_or_none=mismatched_credential_record),
_exec_result(scalars_all=[]),
_exec_result(scalar=1),
]
with _patched_session(session):
with patch.object(
ProviderConfiguration,
"_get_custom_model_record",
return_value=provider_model_record,
) as mock_get_model:
configuration.delete_custom_model_credential(ModelType.LLM, "request-model", "cred-2")
mock_get_model.assert_called_once_with(ModelType.TEXT_EMBEDDING, "stored-model", session=session)
session.delete.assert_any_call(mismatched_credential_record)
def test_add_model_credential_to_model_and_switch_custom_model_credential() -> None:
configuration = _build_provider_configuration()