feat(gooddata-sdk): [AUTO] Add Anthropic provider config to LLM provider config

Author: yenkins-admin

packages/gooddata-sdk/src/gooddata_sdk/__init__.py

@@ -116,6 +116,8 @@
     CatalogRsaSpecification,
 )
 from gooddata_sdk.catalog.organization.entity_model.llm_provider import (
+    CatalogAnthropicApiKeyAuth,
+    CatalogAnthropicProviderConfig,
     CatalogAwsBedrockProviderConfig,
     CatalogAzureFoundryApiKeyAuth,
     CatalogAzureFoundryProviderConfig,

packages/gooddata-sdk/src/gooddata_sdk/catalog/organization/entity_model/llm_provider.py

@@ -75,6 +75,24 @@ def client_class() -> type[AzureFoundryProviderAuth]:
 
 CatalogAzureFoundryAuth = Union[CatalogAzureFoundryApiKeyAuth]
 
+# --- Anthropic auth ---
+
+
+@define(kw_only=True)
+class CatalogAnthropicApiKeyAuth(Base):
+    """API key authentication for the Anthropic provider."""
+
+    api_key: str | None = None
+    type: str = "API_KEY"
+
+    @staticmethod
+    def client_class() -> type[OpenAiProviderAuth]:
+        # Stand-in: AnthropicProviderAuth not yet present in the generated client.
+        return OpenAiProviderAuth
+
+
+CatalogAnthropicAuth = Union[CatalogAnthropicApiKeyAuth]
+
 # --- Provider config types ---
 
 
@@ -118,10 +136,25 @@ def client_class() -> type[AzureFoundryProviderConfig]:
         return AzureFoundryProviderConfig
 
 
+@define(kw_only=True)
+class CatalogAnthropicProviderConfig(Base):
+    """Anthropic provider configuration."""
+
+    auth: CatalogAnthropicAuth | None = None
+    base_url: str | None = None
+    type: str = "ANTHROPIC"
+
+    @staticmethod
+    def client_class() -> type[OpenAIProviderConfig]:
+        # Stand-in: AnthropicProviderConfig not yet present in the generated client.
+        return OpenAIProviderConfig
+
+
 CatalogLlmProviderConfig = Union[
     CatalogOpenAiProviderConfig,
     CatalogAwsBedrockProviderConfig,
     CatalogAzureFoundryProviderConfig,
+    CatalogAnthropicProviderConfig,
 ]
 
 
@@ -157,6 +190,16 @@ def _azure_foundry_auth_from_api(data: dict[str, Any]) -> CatalogAzureFoundryAut
     raise ValueError(f"Unknown Azure Foundry auth type: {auth_type}")
 
 
+def _anthropic_auth_from_api(data: dict[str, Any]) -> CatalogAnthropicAuth:
+    auth_type = safeget(data, ["type"]) or "API_KEY"
+    if auth_type == "API_KEY":
+        return CatalogAnthropicApiKeyAuth(
+            api_key="",  # Credentials are not returned for security reasons
+            type=auth_type,
+        )
+    raise ValueError(f"Unknown Anthropic auth type: {auth_type}")
+
+
 def _provider_config_from_api(data: dict[str, Any]) -> CatalogLlmProviderConfig:
     provider_type = safeget(data, ["type"]) or "OPENAI"
     auth_data = safeget(data, ["auth"])
@@ -173,6 +216,12 @@ def _provider_config_from_api(data: dict[str, Any]) -> CatalogLlmProviderConfig:
             endpoint=safeget(data, ["endpoint"]),
         )
 
+    if provider_type == "ANTHROPIC":
+        return CatalogAnthropicProviderConfig(
+            auth=_anthropic_auth_from_api(auth_data) if auth_data is not None else None,
+            base_url=safeget(data, ["baseUrl"]),
+        )
+
     # Default: OpenAI
     return CatalogOpenAiProviderConfig(
         auth=_openai_auth_from_api(auth_data) if auth_data is not None else None,

packages/gooddata-sdk/tests/catalog/test_catalog_organization.py

@@ -5,9 +5,13 @@
 
 from gooddata_api_client.exceptions import NotFoundException
 from gooddata_sdk import (
+    CatalogAnthropicApiKeyAuth,
+    CatalogAnthropicProviderConfig,
     CatalogCspDirective,
     CatalogDeclarativeNotificationChannel,
     CatalogJwk,
+    CatalogLlmProvider,
+    CatalogLlmProviderModel,
     CatalogOrganization,
     CatalogOrganizationSetting,
     CatalogRsaSpecification,
@@ -334,6 +338,29 @@ def test_layout_notification_channels(test_config, snapshot_notification_channel
     # snapshot_notification_channels fixture restores original state in teardown
 
 
+@gd_vcr.use_cassette(str(_fixtures_dir / "test_anthropic_llm_provider.yaml"))
+def test_create_anthropic_llm_provider(test_config):
+    sdk = GoodDataSdk.create(host_=test_config["host"], token_=test_config["token"])
+    provider_id = "test-anthropic-provider"
+    new_provider = CatalogLlmProvider.init(
+        id=provider_id,
+        models=[CatalogLlmProviderModel(id="claude-3-5-sonnet-20241022", family="claude")],
+        provider_config=CatalogAnthropicProviderConfig(
+            auth=CatalogAnthropicApiKeyAuth(api_key="sk-ant-test-key"),
+            base_url="https://api.anthropic.com",
+        ),
+        name="Test Anthropic Provider",
+    )
+    try:
+        created = sdk.catalog_organization.create_llm_provider(new_provider)
+        assert created.id == provider_id
+        fetched = sdk.catalog_organization.get_llm_provider(provider_id)
+        assert fetched.id == provider_id
+        assert isinstance(fetched.attributes.provider_config, CatalogAnthropicProviderConfig)
+    finally:
+        safe_delete(sdk.catalog_organization.delete_llm_provider, provider_id)
+
+
 #
 # The following tests are commented out as they require the organization to have the FEDERATED_IDENTITY_MANAGEMENT
 # entitlement enabled which cannot be done via SDK and must be done by GoodData support.