feat: WIP: provider support

Author: csgulati09

portkey_ai/__init__.py

@@ -129,6 +129,8 @@
     AsyncIntegrationsWorkspaces,
     IntegrationsModels,
     AsyncIntegrationsModels,
+    Providers,
+    AsyncProviders,
 )
 
 from portkey_ai.version import VERSION
@@ -277,4 +279,6 @@
     "AsyncIntegrationsWorkspaces",
     "IntegrationsModels",
     "AsyncIntegrationsModels",
+    "Providers",
+    "AsyncProviders",
 ]

portkey_ai/api_resources/__init__.py

@@ -117,6 +117,8 @@
     AsyncIntegrationsWorkspaces,
     IntegrationsModels,
     AsyncIntegrationsModels,
+    Providers,
+    AsyncProviders,
 )
 from .utils import (
     Modes,
@@ -269,4 +271,6 @@
     "AsyncIntegrationsWorkspaces",
     "IntegrationsModels",
     "AsyncIntegrationsModels",
+    "Providers",
+    "AsyncProviders",
 ]

portkey_ai/api_resources/apis/__init__.py

@@ -148,6 +148,8 @@
     AsyncIntegrationsModels,
 )
 
+from .providers import Providers, AsyncProviders
+
 sys.modules["openai"] = vendored_openai  # For pydantic v1 and v2 compatibility
 
 __all__ = [
@@ -278,4 +280,6 @@
     "AsyncIntegrationsWorkspaces",
     "IntegrationsModels",
     "AsyncIntegrationsModels",
+    "Providers",
+    "AsyncProviders",
 ]

portkey_ai/api_resources/apis/providers.py

@@ -0,0 +1,131 @@
+from typing import Any, Dict, Optional
+from portkey_ai.api_resources.base_client import APIClient, AsyncAPIClient
+from urllib.parse import urlencode
+from portkey_ai.api_resources.apis.api_resource import APIResource, AsyncAPIResource
+from portkey_ai.api_resources.utils import GenericResponse
+from portkey_ai.api_resources.utils import PortkeyApiPaths
+
+
+class Providers(APIResource):
+    def __init__(self, client: APIClient) -> None:
+        super().__init__(client)
+
+    def create(
+        self,
+        *,
+        name: Optional[str] = None,
+        description: Optional[str] = None,
+        key: Optional[str] = None,
+        ai_provider_id: Optional[str] = None,
+        workspace_id: Optional[str] = None,
+        slug: Optional[str] = None,
+        organisation_id: Optional[str] = None,
+        note: Optional[str] = None,
+        configuration: Optional[Dict[str, Any]] = None,
+        **kwargs: Any,
+    ) -> GenericResponse:
+        body = {
+            "name": name,
+            "description": description,
+            "key": key,
+            "ai_provider_id": ai_provider_id,
+            "workspace_id": workspace_id,
+            "slug": slug,
+            "organisation_id": organisation_id,
+            "note": note,
+            "configuration": configuration,
+            **kwargs,
+        }
+        return self._post(
+            f"{PortkeyApiPaths.PROVIDERS_API}",
+            body=body,
+            params=None,
+            cast_to=GenericResponse,
+            stream=False,
+            stream_cls=None,
+            headers={},
+        )
+
+    def list(
+        self,
+        *,
+        organisation_id: Optional[str] = None,
+        workspace_id: Optional[str] = None,
+        current_page: Optional[int] = None,
+        page_size: Optional[int] = None,
+    ) -> GenericResponse:
+        query = {
+            "organisation_id": organisation_id,
+            "workspace_id": workspace_id,
+            "current_page": current_page,
+            "page_size": page_size,
+        }
+        filtered_query = {k: v for k, v in query.items() if v is not None}
+        query_string = urlencode(filtered_query)
+        return self._get(
+            f"{PortkeyApiPaths.PROVIDERS_API}?{query_string}",
+            params=None,
+            body=None,
+            cast_to=GenericResponse,
+            stream=False,
+            stream_cls=None,
+            headers={},
+        )
+
+    def retrieve(self, *, integration_id: Optional[str]) -> Any:
+        return self._get(
+            f"{PortkeyApiPaths.PROVIDERS_API}/{integration_id}",
+            params=None,
+            body=None,
+            cast_to=GenericResponse,
+            stream=False,
+            stream_cls=None,
+            headers={},
+        )
+
+    def update(
+        self,
+        *,
+        integration_id: Optional[str] = None,
+        name: Optional[str] = None,
+        description: Optional[str] = None,
+        key: Optional[str] = None,
+        configuration: Optional[Dict[str, Any]] = None,
+        **kwargs: Any,
+    ) -> GenericResponse:
+        body = {
+            "name": name,
+            "description": description,
+            "key": key,
+            "configuration": configuration,
+            **kwargs,
+        }
+        return self._put(
+            f"{PortkeyApiPaths.PROVIDERS_API}/{integration_id}",
+            body=body,
+            params=None,
+            cast_to=GenericResponse,
+            stream=False,
+            stream_cls=None,
+            headers={},
+        )
+
+    def delete(
+        self,
+        *,
+        integration_id: Optional[str],
+    ) -> Any:
+        return self._delete(
+            f"{PortkeyApiPaths.PROVIDERS_API}/{integration_id}",
+            params=None,
+            body=None,
+            cast_to=GenericResponse,
+            stream=False,
+            stream_cls=None,
+            headers={},
+        )
+
+
+class AsyncProviders(AsyncAPIResource):
+    def __init__(self, client: AsyncAPIClient) -> None:
+        super().__init__(client)

portkey_ai/api_resources/client.py

@@ -36,6 +36,7 @@ class Portkey(APIClient):
     labels: apis.Labels
     collections: apis.Collections
     integrations: apis.Integrations
+    providers: apis.Providers
 
     class beta:
         assistants: apis.Assistants
@@ -177,6 +178,7 @@ def __init__(
         self.labels = apis.Labels(self)
         self.collections = apis.Collections(self)
         self.integrations = apis.Integrations(self)
+        self.providers = apis.Providers(self)
         self.beta = self.beta(self)  # type: ignore
 
         if self.instrumentation:
@@ -355,6 +357,7 @@ class AsyncPortkey(AsyncAPIClient):
     labels: apis.AsyncLabels
     collections: apis.AsyncCollections
     integrations: apis.AsyncIntegrations
+    providers: apis.AsyncProviders
 
     class beta:
         assistants: apis.AsyncAssistants
@@ -496,6 +499,7 @@ def __init__(
         self.labels = apis.AsyncLabels(self)
         self.collections = apis.AsyncCollections(self)
         self.integrations = apis.AsyncIntegrations(self)
+        self.providers = apis.AsyncProviders(self)
         self.beta = self.beta(self)  # type: ignore
 
         if self.instrumentation:

portkey_ai/api_resources/utils.py

@@ -130,6 +130,7 @@ class PortkeyApiPaths(str, Enum, metaclass=MetaEnum):
     PROMPTS_API = "/prompts"
     PROMPTS_PARTIALS_API = "/prompts/partials"
     INTEGRATIONS_API = "/integrations"
+    PROVIDERS_API = "/providers"
 
     def __str__(self):
         return self.value