feat: integration workspace support added

Author: csgulati09

portkey_ai/__init__.py

@@ -125,6 +125,8 @@
     AsyncContent,
     Integrations,
     AsyncIntegrations,
+    IntegrationsWorkspaces,
+    AsyncIntegrationsWorkspaces,
 )
 
 from portkey_ai.version import VERSION
@@ -269,4 +271,6 @@
     "AsyncContent",
     "Integrations",
     "AsyncIntegrations",
+    "IntegrationsWorkspaces",
+    "AsyncIntegrationsWorkspaces",
 ]

portkey_ai/api_resources/__init__.py

@@ -113,6 +113,8 @@
     AsyncContent,
     Integrations,
     AsyncIntegrations,
+    IntegrationsWorkspaces,
+    AsyncIntegrationsWorkspaces,
 )
 from .utils import (
     Modes,
@@ -261,4 +263,6 @@
     "AsyncContent",
     "Integrations",
     "AsyncIntegrations",
+    "IntegrationsWorkspaces",
+    "AsyncIntegrationsWorkspaces",
 ]

portkey_ai/api_resources/apis/__init__.py

@@ -139,7 +139,12 @@
 
 from .collections import Collections, AsyncCollections
 
-from .integrations import Integrations, AsyncIntegrations
+from .integrations import (
+    Integrations,
+    AsyncIntegrations,
+    IntegrationsWorkspaces,
+    AsyncIntegrationsWorkspaces,
+)
 
 sys.modules["openai"] = vendored_openai  # For pydantic v1 and v2 compatibility
 
@@ -267,4 +272,6 @@
     "AsyncContent",
     "Integrations",
     "AsyncIntegrations",
+    "IntegrationsWorkspaces",
+    "AsyncIntegrationsWorkspaces",
 ]

portkey_ai/api_resources/apis/integrations.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, Optional
+from typing import Any, Dict, List, 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
@@ -9,6 +9,7 @@
 class Integrations(APIResource):
     def __init__(self, client: APIClient) -> None:
         super().__init__(client)
+        self.workspaces = IntegrationsWorkspaces(client)
 
     def create(
         self,
@@ -126,9 +127,57 @@ def delete(
         )
 
 
+class IntegrationsWorkspaces(APIResource):
+    def __init__(self, client: APIClient) -> None:
+        super().__init__(client)
+
+    def list(
+        self,
+        *,
+        provider_integration_id: Optional[str] = None,
+    ) -> GenericResponse:
+        return self._get(
+            f"{PortkeyApiPaths.INTEGRATIONS_API}/{provider_integration_id}/workspaces",
+            params=None,
+            body=None,
+            cast_to=GenericResponse,
+            stream=False,
+            stream_cls=None,
+            headers={},
+        )
+
+    def update(
+        self,
+        *,
+        provider_integration_id: Optional[str] = None,
+        global_workspace_access: Optional[Dict[str, Any]] = None,
+        workspace_ids: Optional[List[str]] = None,
+        override_existing_workspaces_access: Optional[bool] = None,
+        workspaces: Optional[List[Dict[str, Any]]] = None,
+        **kwargs: Any,
+    ) -> GenericResponse:
+        body = {
+            "global_workspace_access": global_workspace_access,
+            "workspace_ids": workspace_ids,
+            "override_existing_workspaces_access": override_existing_workspaces_access,
+            "workspaces": workspaces,
+            **kwargs,
+        }
+        return self._put(
+            f"{PortkeyApiPaths.INTEGRATIONS_API}/{provider_integration_id}/workspaces",
+            body=body,
+            params=None,
+            cast_to=GenericResponse,
+            stream=False,
+            stream_cls=None,
+            headers={},
+        )
+
+
 class AsyncIntegrations(AsyncAPIResource):
     def __init__(self, client: AsyncAPIClient) -> None:
         super().__init__(client)
+        self.workspaces = AsyncIntegrationsWorkspaces(client)
 
     async def create(
         self,
@@ -244,3 +293,50 @@ async def delete(
             stream_cls=None,
             headers={},
         )
+
+
+class AsyncIntegrationsWorkspaces(AsyncAPIResource):
+    def __init__(self, client: AsyncAPIClient) -> None:
+        super().__init__(client)
+
+    async def list(
+        self,
+        *,
+        provider_integration_id: Optional[str] = None,
+    ) -> GenericResponse:
+        return await self._get(
+            f"{PortkeyApiPaths.INTEGRATIONS_API}/{provider_integration_id}/workspaces",
+            params=None,
+            body=None,
+            cast_to=GenericResponse,
+            stream=False,
+            stream_cls=None,
+            headers={},
+        )
+
+    async def update(
+        self,
+        *,
+        provider_integration_id: Optional[str] = None,
+        global_workspace_access: Optional[Dict[str, Any]] = None,
+        workspace_ids: Optional[List[str]] = None,
+        override_existing_workspaces_access: Optional[bool] = None,
+        workspaces: Optional[List[Dict[str, Any]]] = None,
+        **kwargs: Any,
+    ) -> GenericResponse:
+        body = {
+            "global_workspace_access": global_workspace_access,
+            "workspace_ids": workspace_ids,
+            "override_existing_workspaces_access": override_existing_workspaces_access,
+            "workspaces": workspaces,
+            **kwargs,
+        }
+        return await self._put(
+            f"{PortkeyApiPaths.INTEGRATIONS_API}/{provider_integration_id}/workspaces",
+            body=body,
+            params=None,
+            cast_to=GenericResponse,
+            stream=False,
+            stream_cls=None,
+            headers={},
+        )