anthropic_provider.py 792 B

1234567891011121314151617181920212223
  1. from typing import Optional
  2. from core.llm.provider.base import BaseProvider
  3. from models.provider import ProviderName
  4. class AnthropicProvider(BaseProvider):
  5. def get_models(self, model_id: Optional[str] = None) -> list[dict]:
  6. credentials = self.get_credentials(model_id)
  7. # todo
  8. return []
  9. def get_credentials(self, model_id: Optional[str] = None) -> dict:
  10. """
  11. Returns the API credentials for Azure OpenAI as a dictionary, for the given tenant_id.
  12. The dictionary contains keys: azure_api_type, azure_api_version, azure_api_base, and azure_api_key.
  13. """
  14. return {
  15. 'anthropic_api_key': self.get_provider_api_key(model_id=model_id)
  16. }
  17. def get_provider_name(self):
  18. return ProviderName.ANTHROPIC