2024-09-23 13:09:46 +08:00
|
|
|
from core.model_runtime.entities.provider_entities import ProviderEntity
|
2024-09-20 13:55:09 +08:00
|
|
|
from core.plugin.manager.base import BasePluginManager
|
|
|
|
|
|
|
|
|
|
|
|
class PluginModelManager(BasePluginManager):
|
2024-09-23 13:09:46 +08:00
|
|
|
def fetch_model_providers(self, tenant_id: str) -> list[ProviderEntity]:
|
|
|
|
"""
|
|
|
|
Fetch model providers for the given tenant.
|
|
|
|
"""
|
|
|
|
response = self._request_with_plugin_daemon_response(
|
|
|
|
"GET", f"plugin/{tenant_id}/models", list[ProviderEntity], params={"page": 1, "page_size": 256}
|
|
|
|
)
|
|
|
|
return response
|