mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-22 02:15:22 -04:00
Maries
569e68ac62
* refactor: update model provider to use ModelFactory for instance management * refactor: integrate ModelFactory for model instance mapping in tests * refactor: remove unused imports and clean up test files
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
from unittest.mock import MagicMock
|
|
|
|
from dify_plugin.core.model_factory import ModelFactory
|
|
from dify_plugin.entities import I18nObject
|
|
from dify_plugin.entities.model import ModelType
|
|
from dify_plugin.entities.model.provider import ProviderEntity
|
|
from dify_plugin.interfaces.model import ModelProvider
|
|
|
|
|
|
def test_construct_model_provider():
|
|
"""
|
|
Ensure ModelProvider constructor is intact and usable.
|
|
This guards against overriding or changing __init__ signature.
|
|
"""
|
|
|
|
class ProviderImpl(ModelProvider):
|
|
def validate_provider_credentials(self, credentials: dict) -> None:
|
|
pass
|
|
|
|
provider_schema = ProviderEntity(
|
|
provider="test",
|
|
label=I18nObject(en_US="test"),
|
|
supported_model_types=[ModelType.LLM],
|
|
configurate_methods=[],
|
|
)
|
|
|
|
model_factory = MagicMock(spec=ModelFactory)
|
|
|
|
provider = ProviderImpl(provider_schemas=provider_schema, model_factory=model_factory)
|
|
|
|
assert provider is not None
|
|
assert provider.get_provider_schema() == provider_schema
|
|
assert provider.model_factory is model_factory
|