feat(ai): Add support for AI features

jira: GDAI-185
risk: low

Author: malmiron

gooddata-sdk/gooddata_sdk/__init__.py

@@ -99,6 +99,10 @@
     CatalogJwkDocument,
     CatalogRsaSpecification,
 )
+from gooddata_sdk.catalog.organization.entity_model.llm_endpoint import (
+    CatalogLlmEndpoint,
+    CatalogLlmEndpointDocument,
+)
 from gooddata_sdk.catalog.organization.entity_model.organization import CatalogOrganization
 from gooddata_sdk.catalog.organization.entity_model.setting import CatalogOrganizationSetting
 from gooddata_sdk.catalog.organization.layout.export_template import (

gooddata-sdk/gooddata_sdk/catalog/organization/entity_model/llm_endpoint.py

@@ -0,0 +1,239 @@
+# (C) 2024 GoodData Corporation
+from __future__ import annotations
+
+from typing import Any, Optional
+
+from attr import define
+from gooddata_api_client.model.json_api_llm_endpoint_in import JsonApiLlmEndpointIn
+from gooddata_api_client.model.json_api_llm_endpoint_in_attributes import JsonApiLlmEndpointInAttributes
+from gooddata_api_client.model.json_api_llm_endpoint_in_document import JsonApiLlmEndpointInDocument
+from gooddata_api_client.model.json_api_llm_endpoint_patch import JsonApiLlmEndpointPatch
+from gooddata_api_client.model.json_api_llm_endpoint_patch_attributes import JsonApiLlmEndpointPatchAttributes
+from gooddata_api_client.model.json_api_llm_endpoint_patch_document import JsonApiLlmEndpointPatchDocument
+
+from gooddata_sdk.catalog.base import Base
+from gooddata_sdk.utils import safeget
+
+
+@define(kw_only=True)
+class CatalogLlmEndpointDocument(Base):
+    data: CatalogLlmEndpoint
+
+    @staticmethod
+    def client_class() -> type[JsonApiLlmEndpointInDocument]:
+        return JsonApiLlmEndpointInDocument
+
+
+@define(kw_only=True)
+class CatalogLlmEndpointPatchDocument(Base):
+    data: CatalogLlmEndpointPatch
+
+    @staticmethod
+    def client_class() -> type[JsonApiLlmEndpointPatchDocument]:
+        return JsonApiLlmEndpointPatchDocument
+
+
+@define(kw_only=True)
+class CatalogLlmEndpoint(Base):
+    id: str
+    attributes: Optional[CatalogLlmEndpointAttributes] = None
+
+    @staticmethod
+    def client_class() -> type[JsonApiLlmEndpointIn]:
+        return JsonApiLlmEndpointIn
+
+    @classmethod
+    def init(
+        cls,
+        id: str,
+        title: str,
+        token: str,
+        description: Optional[str] = None,
+        provider: Optional[str] = None,
+        base_url: Optional[str] = None,
+        llm_organization: Optional[str] = None,
+        llm_model: Optional[str] = None,
+        workspace_ids: Optional[list[str]] = None,
+    ) -> CatalogLlmEndpoint:
+        return cls(
+            id=id,
+            attributes=CatalogLlmEndpointAttributes(
+                title=title,
+                token=token,
+                description=description,
+                provider=provider,
+                base_url=base_url,
+                llm_organization=llm_organization,
+                llm_model=llm_model,
+                workspace_ids=workspace_ids,
+            ),
+        )
+
+    @classmethod
+    def from_api(cls, entity: dict[str, Any]) -> CatalogLlmEndpoint:
+        ea = entity["attributes"]
+        attr = CatalogLlmEndpointAttributes(
+            title=safeget(ea, ["title"]),
+            token=None,  # Token is not returned for security reasons
+            description=safeget(ea, ["description"]),
+            provider=safeget(ea, ["provider"]),
+            base_url=safeget(ea, ["baseUrl"]),
+            llm_organization=safeget(ea, ["llmOrganization"]),
+            llm_model=safeget(ea, ["llmModel"]),
+            workspace_ids=safeget(ea, ["workspaceIds"]),
+        )
+        return cls(
+            id=entity["id"],
+            attributes=attr,
+        )
+
+
+@define(kw_only=True)
+class CatalogLlmEndpointPatch(Base):
+    id: str
+    attributes: Optional[CatalogLlmEndpointPatchAttributes] = None
+
+    @staticmethod
+    def client_class() -> type[JsonApiLlmEndpointPatch]:
+        return JsonApiLlmEndpointPatch
+
+    @classmethod
+    def init(
+        cls,
+        id: str,
+        title: Optional[str] = None,
+        token: Optional[str] = None,
+        description: Optional[str] = None,
+        provider: Optional[str] = None,
+        base_url: Optional[str] = None,
+        llm_organization: Optional[str] = None,
+        llm_model: Optional[str] = None,
+        workspace_ids: Optional[list[str]] = None,
+    ) -> CatalogLlmEndpointPatch:
+        return cls(
+            id=id,
+            attributes=CatalogLlmEndpointPatchAttributes(
+                title=title,
+                token=token,
+                description=description,
+                provider=provider,
+                base_url=base_url,
+                llm_organization=llm_organization,
+                llm_model=llm_model,
+                workspace_ids=workspace_ids,
+            ),
+        )
+
+
+@define(kw_only=True)
+class CatalogLlmEndpointAttributes(Base):
+    title: str
+    token: str
+    description: Optional[str] = None
+    provider: Optional[str] = None
+    base_url: Optional[str] = None
+    llm_organization: Optional[str] = None
+    llm_model: Optional[str] = None
+    workspace_ids: Optional[list[str]] = None
+
+    @staticmethod
+    def client_class() -> type[JsonApiLlmEndpointInAttributes]:
+        return JsonApiLlmEndpointInAttributes
+
+    def to_api(self) -> dict:
+        """Convert to API format (camelCase). Only include non-None attributes."""
+        result = {
+            "title": self.title,
+            "token": self.token,
+        }
+        if self.description is not None:
+            result["description"] = self.description
+        if self.provider is not None:
+            result["provider"] = self.provider
+        if self.base_url is not None:
+            result["baseUrl"] = self.base_url
+        if self.llm_organization is not None:
+            result["llmOrganization"] = self.llm_organization
+        if self.llm_model is not None:
+            result["llmModel"] = self.llm_model
+        if self.workspace_ids is not None:
+            result["workspaceIds"] = self.workspace_ids
+        return result
+
+    def to_dict(self) -> dict:
+        """Convert to dictionary format. Only include non-None attributes."""
+        result = {
+            "title": self.title,
+            "token": self.token,
+        }
+        if self.description is not None:
+            result["description"] = self.description
+        if self.provider is not None:
+            result["provider"] = self.provider
+        if self.base_url is not None:
+            result["baseUrl"] = self.base_url
+        if self.llm_organization is not None:
+            result["llmOrganization"] = self.llm_organization
+        if self.llm_model is not None:
+            result["llmModel"] = self.llm_model
+        if self.workspace_ids is not None:
+            result["workspaceIds"] = self.workspace_ids
+        return result
+
+
+@define(kw_only=True)
+class CatalogLlmEndpointPatchAttributes(Base):
+    title: Optional[str] = None
+    token: Optional[str] = None
+    description: Optional[str] = None
+    provider: Optional[str] = None
+    base_url: Optional[str] = None
+    llm_organization: Optional[str] = None
+    llm_model: Optional[str] = None
+    workspace_ids: Optional[list[str]] = None
+
+    @staticmethod
+    def client_class() -> type[JsonApiLlmEndpointPatchAttributes]:
+        return JsonApiLlmEndpointPatchAttributes
+
+    def to_api(self) -> dict:
+        """Convert to API format (camelCase). Only include non-None attributes."""
+        result = {}
+        if self.title is not None:
+            result["title"] = self.title
+        if self.token is not None:
+            result["token"] = self.token
+        if self.description is not None:
+            result["description"] = self.description
+        if self.provider is not None:
+            result["provider"] = self.provider
+        if self.base_url is not None:
+            result["baseUrl"] = self.base_url
+        if self.llm_organization is not None:
+            result["llmOrganization"] = self.llm_organization
+        if self.llm_model is not None:
+            result["llmModel"] = self.llm_model
+        if self.workspace_ids is not None:
+            result["workspaceIds"] = self.workspace_ids
+        return result
+
+    def to_dict(self) -> dict:
+        """Convert to dictionary format. Only include non-None attributes."""
+        result = {}
+        if self.title is not None:
+            result["title"] = self.title
+        if self.token is not None:
+            result["token"] = self.token
+        if self.description is not None:
+            result["description"] = self.description
+        if self.provider is not None:
+            result["provider"] = self.provider
+        if self.base_url is not None:
+            result["baseUrl"] = self.base_url
+        if self.llm_organization is not None:
+            result["llmOrganization"] = self.llm_organization
+        if self.llm_model is not None:
+            result["llmModel"] = self.llm_model
+        if self.workspace_ids is not None:
+            result["workspaceIds"] = self.workspace_ids
+        return result