vellum-ai 1.5.4__py3-none-any.whl → 1.5.6__py3-none-any.whl

vellum/client/resources/integration_auth_configs/client.py

@@ -0,0 +1,186 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.request_options import RequestOptions
+from ...types.paginated_slim_integration_auth_config_read_list import PaginatedSlimIntegrationAuthConfigReadList
+from .raw_client import AsyncRawIntegrationAuthConfigsClient, RawIntegrationAuthConfigsClient
+
+
+class IntegrationAuthConfigsClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawIntegrationAuthConfigsClient(client_wrapper=client_wrapper)
+
+    @property
+    def _client_wrapper(self) -> SyncClientWrapper:
+        return self._raw_client._client_wrapper
+
+    @property
+    def with_raw_response(self) -> RawIntegrationAuthConfigsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawIntegrationAuthConfigsClient
+        """
+        return self._raw_client
+
+    def list_integration_auth_configs(
+        self,
+        *,
+        expand: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        integration_name: typing.Optional[str] = None,
+        integration_provider: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        offset: typing.Optional[int] = None,
+        ordering: typing.Optional[str] = None,
+        search: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> PaginatedSlimIntegrationAuthConfigReadList:
+        """
+        List Integration Auth Configs
+
+        Parameters
+        ----------
+        expand : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            The response fields to expand for more information.
+            - 'integration_credentials' expands the list of integration credentials that the authenticated entity has access to for the auth config.
+
+        integration_name : typing.Optional[str]
+
+        integration_provider : typing.Optional[str]
+
+        limit : typing.Optional[int]
+            Number of results to return per page.
+
+        offset : typing.Optional[int]
+            The initial index from which to return the results.
+
+        ordering : typing.Optional[str]
+            Which field to use when ordering the results.
+
+        search : typing.Optional[str]
+            A search term.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        PaginatedSlimIntegrationAuthConfigReadList
+
+
+        Examples
+        --------
+        from vellum import Vellum
+
+        client = Vellum(
+            api_version="YOUR_API_VERSION",
+            api_key="YOUR_API_KEY",
+        )
+        client.integration_auth_configs.list_integration_auth_configs()
+        """
+        _response = self._raw_client.list_integration_auth_configs(
+            expand=expand,
+            integration_name=integration_name,
+            integration_provider=integration_provider,
+            limit=limit,
+            offset=offset,
+            ordering=ordering,
+            search=search,
+            request_options=request_options,
+        )
+        return _response.data
+
+
+class AsyncIntegrationAuthConfigsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawIntegrationAuthConfigsClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawIntegrationAuthConfigsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawIntegrationAuthConfigsClient
+        """
+        return self._raw_client
+
+    async def list_integration_auth_configs(
+        self,
+        *,
+        expand: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        integration_name: typing.Optional[str] = None,
+        integration_provider: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        offset: typing.Optional[int] = None,
+        ordering: typing.Optional[str] = None,
+        search: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> PaginatedSlimIntegrationAuthConfigReadList:
+        """
+        List Integration Auth Configs
+
+        Parameters
+        ----------
+        expand : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            The response fields to expand for more information.
+            - 'integration_credentials' expands the list of integration credentials that the authenticated entity has access to for the auth config.
+
+        integration_name : typing.Optional[str]
+
+        integration_provider : typing.Optional[str]
+
+        limit : typing.Optional[int]
+            Number of results to return per page.
+
+        offset : typing.Optional[int]
+            The initial index from which to return the results.
+
+        ordering : typing.Optional[str]
+            Which field to use when ordering the results.
+
+        search : typing.Optional[str]
+            A search term.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        PaginatedSlimIntegrationAuthConfigReadList
+
+
+        Examples
+        --------
+        import asyncio
+
+        from vellum import AsyncVellum
+
+        client = AsyncVellum(
+            api_version="YOUR_API_VERSION",
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.integration_auth_configs.list_integration_auth_configs()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.list_integration_auth_configs(
+            expand=expand,
+            integration_name=integration_name,
+            integration_provider=integration_provider,
+            limit=limit,
+            offset=offset,
+            ordering=ordering,
+            search=search,
+            request_options=request_options,
+        )
+        return _response.data

vellum/client/resources/integration_auth_configs/raw_client.py

@@ -0,0 +1,171 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from json.decoder import JSONDecodeError
+
+from ...core.api_error import ApiError
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.http_response import AsyncHttpResponse, HttpResponse
+from ...core.pydantic_utilities import parse_obj_as
+from ...core.request_options import RequestOptions
+from ...types.paginated_slim_integration_auth_config_read_list import PaginatedSlimIntegrationAuthConfigReadList
+
+
+class RawIntegrationAuthConfigsClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def list_integration_auth_configs(
+        self,
+        *,
+        expand: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        integration_name: typing.Optional[str] = None,
+        integration_provider: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        offset: typing.Optional[int] = None,
+        ordering: typing.Optional[str] = None,
+        search: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[PaginatedSlimIntegrationAuthConfigReadList]:
+        """
+        List Integration Auth Configs
+
+        Parameters
+        ----------
+        expand : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            The response fields to expand for more information.
+            - 'integration_credentials' expands the list of integration credentials that the authenticated entity has access to for the auth config.
+
+        integration_name : typing.Optional[str]
+
+        integration_provider : typing.Optional[str]
+
+        limit : typing.Optional[int]
+            Number of results to return per page.
+
+        offset : typing.Optional[int]
+            The initial index from which to return the results.
+
+        ordering : typing.Optional[str]
+            Which field to use when ordering the results.
+
+        search : typing.Optional[str]
+            A search term.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[PaginatedSlimIntegrationAuthConfigReadList]
+
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/integration-auth-configs",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            params={
+                "expand": expand,
+                "integration_name": integration_name,
+                "integration_provider": integration_provider,
+                "limit": limit,
+                "offset": offset,
+                "ordering": ordering,
+                "search": search,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    PaginatedSlimIntegrationAuthConfigReadList,
+                    parse_obj_as(
+                        type_=PaginatedSlimIntegrationAuthConfigReadList,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+
+class AsyncRawIntegrationAuthConfigsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def list_integration_auth_configs(
+        self,
+        *,
+        expand: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        integration_name: typing.Optional[str] = None,
+        integration_provider: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        offset: typing.Optional[int] = None,
+        ordering: typing.Optional[str] = None,
+        search: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[PaginatedSlimIntegrationAuthConfigReadList]:
+        """
+        List Integration Auth Configs
+
+        Parameters
+        ----------
+        expand : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            The response fields to expand for more information.
+            - 'integration_credentials' expands the list of integration credentials that the authenticated entity has access to for the auth config.
+
+        integration_name : typing.Optional[str]
+
+        integration_provider : typing.Optional[str]
+
+        limit : typing.Optional[int]
+            Number of results to return per page.
+
+        offset : typing.Optional[int]
+            The initial index from which to return the results.
+
+        ordering : typing.Optional[str]
+            Which field to use when ordering the results.
+
+        search : typing.Optional[str]
+            A search term.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[PaginatedSlimIntegrationAuthConfigReadList]
+
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/integration-auth-configs",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            params={
+                "expand": expand,
+                "integration_name": integration_name,
+                "integration_provider": integration_provider,
+                "limit": limit,
+                "offset": offset,
+                "ordering": ordering,
+                "search": search,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    PaginatedSlimIntegrationAuthConfigReadList,
+                    parse_obj_as(
+                        type_=PaginatedSlimIntegrationAuthConfigReadList,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

vellum/client/resources/integration_providers/__init__.py

@@ -0,0 +1,4 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+

vellum/client/resources/integration_providers/client.py

@@ -0,0 +1,279 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.request_options import RequestOptions
+from ...types.components_schemas_composio_tool_definition import ComponentsSchemasComposioToolDefinition
+from ...types.paginated_slim_tool_definition_list import PaginatedSlimToolDefinitionList
+from .raw_client import AsyncRawIntegrationProvidersClient, RawIntegrationProvidersClient
+
+
+class IntegrationProvidersClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawIntegrationProvidersClient(client_wrapper=client_wrapper)
+
+    @property
+    def _client_wrapper(self) -> SyncClientWrapper:
+        return self._raw_client._client_wrapper
+
+    @property
+    def with_raw_response(self) -> RawIntegrationProvidersClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawIntegrationProvidersClient
+        """
+        return self._raw_client
+
+    def retrieve_integration_provider_tool_definition(
+        self,
+        integration_name: str,
+        integration_provider: str,
+        tool_name: str,
+        *,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ComponentsSchemasComposioToolDefinition:
+        """
+        Retrieve a specific integration tool definition.
+
+        Parameters
+        ----------
+        integration_name : str
+            The integration name
+
+        integration_provider : str
+            The integration provider name
+
+        tool_name : str
+            The tool's unique name, as specified by the integration provider
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ComponentsSchemasComposioToolDefinition
+
+
+        Examples
+        --------
+        from vellum import Vellum
+
+        client = Vellum(
+            api_version="YOUR_API_VERSION",
+            api_key="YOUR_API_KEY",
+        )
+        client.integration_providers.retrieve_integration_provider_tool_definition(
+            integration_name="integration_name",
+            integration_provider="integration_provider",
+            tool_name="tool_name",
+        )
+        """
+        _response = self._raw_client.retrieve_integration_provider_tool_definition(
+            integration_name, integration_provider, tool_name, request_options=request_options
+        )
+        return _response.data
+
+    def list_integration_tools(
+        self,
+        integration_provider: str,
+        *,
+        integration_name: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        offset: typing.Optional[int] = None,
+        search: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> PaginatedSlimToolDefinitionList:
+        """
+        List all integration tools for a given provider and integration.
+
+        Parameters
+        ----------
+        integration_provider : str
+            The integration provider name
+
+        integration_name : typing.Optional[str]
+            The Vellum Integration name
+
+        limit : typing.Optional[int]
+            Number of results to return per page.
+
+        offset : typing.Optional[int]
+            The initial index from which to return the results.
+
+        search : typing.Optional[str]
+            The search term to filter the tools by
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        PaginatedSlimToolDefinitionList
+
+
+        Examples
+        --------
+        from vellum import Vellum
+
+        client = Vellum(
+            api_version="YOUR_API_VERSION",
+            api_key="YOUR_API_KEY",
+        )
+        client.integration_providers.list_integration_tools(
+            integration_provider="integration_provider",
+        )
+        """
+        _response = self._raw_client.list_integration_tools(
+            integration_provider,
+            integration_name=integration_name,
+            limit=limit,
+            offset=offset,
+            search=search,
+            request_options=request_options,
+        )
+        return _response.data
+
+
+class AsyncIntegrationProvidersClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawIntegrationProvidersClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawIntegrationProvidersClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawIntegrationProvidersClient
+        """
+        return self._raw_client
+
+    async def retrieve_integration_provider_tool_definition(
+        self,
+        integration_name: str,
+        integration_provider: str,
+        tool_name: str,
+        *,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ComponentsSchemasComposioToolDefinition:
+        """
+        Retrieve a specific integration tool definition.
+
+        Parameters
+        ----------
+        integration_name : str
+            The integration name
+
+        integration_provider : str
+            The integration provider name
+
+        tool_name : str
+            The tool's unique name, as specified by the integration provider
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ComponentsSchemasComposioToolDefinition
+
+
+        Examples
+        --------
+        import asyncio
+
+        from vellum import AsyncVellum
+
+        client = AsyncVellum(
+            api_version="YOUR_API_VERSION",
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.integration_providers.retrieve_integration_provider_tool_definition(
+                integration_name="integration_name",
+                integration_provider="integration_provider",
+                tool_name="tool_name",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.retrieve_integration_provider_tool_definition(
+            integration_name, integration_provider, tool_name, request_options=request_options
+        )
+        return _response.data
+
+    async def list_integration_tools(
+        self,
+        integration_provider: str,
+        *,
+        integration_name: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        offset: typing.Optional[int] = None,
+        search: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> PaginatedSlimToolDefinitionList:
+        """
+        List all integration tools for a given provider and integration.
+
+        Parameters
+        ----------
+        integration_provider : str
+            The integration provider name
+
+        integration_name : typing.Optional[str]
+            The Vellum Integration name
+
+        limit : typing.Optional[int]
+            Number of results to return per page.
+
+        offset : typing.Optional[int]
+            The initial index from which to return the results.
+
+        search : typing.Optional[str]
+            The search term to filter the tools by
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        PaginatedSlimToolDefinitionList
+
+
+        Examples
+        --------
+        import asyncio
+
+        from vellum import AsyncVellum
+
+        client = AsyncVellum(
+            api_version="YOUR_API_VERSION",
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.integration_providers.list_integration_tools(
+                integration_provider="integration_provider",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.list_integration_tools(
+            integration_provider,
+            integration_name=integration_name,
+            limit=limit,
+            offset=offset,
+            search=search,
+            request_options=request_options,
+        )
+        return _response.data