model.py 532 B

1234567891011121314
  1. from core.model_runtime.entities.provider_entities import ProviderEntity
  2. from core.plugin.manager.base import BasePluginManager
  3. class PluginModelManager(BasePluginManager):
  4. def fetch_model_providers(self, tenant_id: str) -> list[ProviderEntity]:
  5. """
  6. Fetch model providers for the given tenant.
  7. """
  8. response = self._request_with_plugin_daemon_response(
  9. "GET", f"plugin/{tenant_id}/models", list[ProviderEntity], params={"page": 1, "page_size": 256}
  10. )
  11. return response