vellum-ai 1.11.2__py3-none-any.whl → 1.13.5__py3-none-any.whl

vellum/client/resources/container_images/client.py

@@ -4,6 +4,7 @@ import typing
 
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
+from ...types.code_execution_package_request import CodeExecutionPackageRequest
 from ...types.container_image_read import ContainerImageRead
 from ...types.docker_service_token import DockerServiceToken
 from ...types.paginated_container_image_read_list import PaginatedContainerImageReadList
@@ -77,6 +78,76 @@ class ContainerImagesClient:
         )
         return _response.data
 
+    def create_container_image(
+        self,
+        *,
+        name: str,
+        packages: typing.Sequence[CodeExecutionPackageRequest],
+        tag: str,
+        user_script: typing.Optional[str] = OMIT,
+        is_hotswappable: typing.Optional[bool] = OMIT,
+        server_version: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ContainerImageRead:
+        """
+        Create a new Container Image.
+
+        Parameters
+        ----------
+        name : str
+
+        packages : typing.Sequence[CodeExecutionPackageRequest]
+
+        tag : str
+
+        user_script : typing.Optional[str]
+
+        is_hotswappable : typing.Optional[bool]
+
+        server_version : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ContainerImageRead
+
+
+        Examples
+        --------
+        from vellum import CodeExecutionPackageRequest, Vellum
+
+        client = Vellum(
+            api_version="YOUR_API_VERSION",
+            api_key="YOUR_API_KEY",
+        )
+        client.container_images.create_container_image(
+            name="x",
+            packages=[
+                CodeExecutionPackageRequest(
+                    version="x",
+                    name="x",
+                ),
+                CodeExecutionPackageRequest(
+                    version="x",
+                    name="x",
+                ),
+            ],
+            tag="x",
+        )
+        """
+        _response = self._raw_client.create_container_image(
+            name=name,
+            packages=packages,
+            tag=tag,
+            user_script=user_script,
+            is_hotswappable=is_hotswappable,
+            server_version=server_version,
+            request_options=request_options,
+        )
+        return _response.data
+
     def retrieve(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> ContainerImageRead:
         """
         Retrieve a Container Image by its ID or name.
@@ -109,6 +180,77 @@ class ContainerImagesClient:
         _response = self._raw_client.retrieve(id, request_options=request_options)
         return _response.data
 
+    def update_container_image(
+        self,
+        id: str,
+        *,
+        packages: typing.Sequence[CodeExecutionPackageRequest],
+        tag: str,
+        user_script: typing.Optional[str] = OMIT,
+        is_hotswappable: typing.Optional[bool] = OMIT,
+        server_version: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ContainerImageRead:
+        """
+        Update an existing Container Image.
+
+        Parameters
+        ----------
+        id : str
+            A UUID string identifying this container image.
+
+        packages : typing.Sequence[CodeExecutionPackageRequest]
+
+        tag : str
+
+        user_script : typing.Optional[str]
+
+        is_hotswappable : typing.Optional[bool]
+
+        server_version : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ContainerImageRead
+
+
+        Examples
+        --------
+        from vellum import CodeExecutionPackageRequest, Vellum
+
+        client = Vellum(
+            api_version="YOUR_API_VERSION",
+            api_key="YOUR_API_KEY",
+        )
+        client.container_images.update_container_image(
+            id="id",
+            packages=[
+                CodeExecutionPackageRequest(
+                    version="x",
+                    name="x",
+                ),
+                CodeExecutionPackageRequest(
+                    version="x",
+                    name="x",
+                ),
+            ],
+            tag="x",
+        )
+        """
+        _response = self._raw_client.update_container_image(
+            id,
+            packages=packages,
+            tag=tag,
+            user_script=user_script,
+            is_hotswappable=is_hotswappable,
+            server_version=server_version,
+            request_options=request_options,
+        )
+        return _response.data
+
     def docker_service_token(self, *, request_options: typing.Optional[RequestOptions] = None) -> DockerServiceToken:
         """
         Parameters
@@ -250,6 +392,84 @@ class AsyncContainerImagesClient:
         )
         return _response.data
 
+    async def create_container_image(
+        self,
+        *,
+        name: str,
+        packages: typing.Sequence[CodeExecutionPackageRequest],
+        tag: str,
+        user_script: typing.Optional[str] = OMIT,
+        is_hotswappable: typing.Optional[bool] = OMIT,
+        server_version: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ContainerImageRead:
+        """
+        Create a new Container Image.
+
+        Parameters
+        ----------
+        name : str
+
+        packages : typing.Sequence[CodeExecutionPackageRequest]
+
+        tag : str
+
+        user_script : typing.Optional[str]
+
+        is_hotswappable : typing.Optional[bool]
+
+        server_version : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ContainerImageRead
+
+
+        Examples
+        --------
+        import asyncio
+
+        from vellum import AsyncVellum, CodeExecutionPackageRequest
+
+        client = AsyncVellum(
+            api_version="YOUR_API_VERSION",
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.container_images.create_container_image(
+                name="x",
+                packages=[
+                    CodeExecutionPackageRequest(
+                        version="x",
+                        name="x",
+                    ),
+                    CodeExecutionPackageRequest(
+                        version="x",
+                        name="x",
+                    ),
+                ],
+                tag="x",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.create_container_image(
+            name=name,
+            packages=packages,
+            tag=tag,
+            user_script=user_script,
+            is_hotswappable=is_hotswappable,
+            server_version=server_version,
+            request_options=request_options,
+        )
+        return _response.data
+
     async def retrieve(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> ContainerImageRead:
         """
         Retrieve a Container Image by its ID or name.
@@ -290,6 +510,85 @@ class AsyncContainerImagesClient:
         _response = await self._raw_client.retrieve(id, request_options=request_options)
         return _response.data
 
+    async def update_container_image(
+        self,
+        id: str,
+        *,
+        packages: typing.Sequence[CodeExecutionPackageRequest],
+        tag: str,
+        user_script: typing.Optional[str] = OMIT,
+        is_hotswappable: typing.Optional[bool] = OMIT,
+        server_version: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ContainerImageRead:
+        """
+        Update an existing Container Image.
+
+        Parameters
+        ----------
+        id : str
+            A UUID string identifying this container image.
+
+        packages : typing.Sequence[CodeExecutionPackageRequest]
+
+        tag : str
+
+        user_script : typing.Optional[str]
+
+        is_hotswappable : typing.Optional[bool]
+
+        server_version : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ContainerImageRead
+
+
+        Examples
+        --------
+        import asyncio
+
+        from vellum import AsyncVellum, CodeExecutionPackageRequest
+
+        client = AsyncVellum(
+            api_version="YOUR_API_VERSION",
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.container_images.update_container_image(
+                id="id",
+                packages=[
+                    CodeExecutionPackageRequest(
+                        version="x",
+                        name="x",
+                    ),
+                    CodeExecutionPackageRequest(
+                        version="x",
+                        name="x",
+                    ),
+                ],
+                tag="x",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.update_container_image(
+            id,
+            packages=packages,
+            tag=tag,
+            user_script=user_script,
+            is_hotswappable=is_hotswappable,
+            server_version=server_version,
+            request_options=request_options,
+        )
+        return _response.data
+
     async def docker_service_token(
         self, *, request_options: typing.Optional[RequestOptions] = None
     ) -> DockerServiceToken:

vellum/client/resources/container_images/raw_client.py

@@ -9,6 +9,8 @@ from ...core.http_response import AsyncHttpResponse, HttpResponse
 from ...core.jsonable_encoder import jsonable_encoder
 from ...core.pydantic_utilities import parse_obj_as
 from ...core.request_options import RequestOptions
+from ...core.serialization import convert_and_respect_annotation_metadata
+from ...types.code_execution_package_request import CodeExecutionPackageRequest
 from ...types.container_image_read import ContainerImageRead
 from ...types.docker_service_token import DockerServiceToken
 from ...types.paginated_container_image_read_list import PaginatedContainerImageReadList
@@ -77,6 +79,77 @@ class RawContainerImagesClient:
             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)
 
+    def create_container_image(
+        self,
+        *,
+        name: str,
+        packages: typing.Sequence[CodeExecutionPackageRequest],
+        tag: str,
+        user_script: typing.Optional[str] = OMIT,
+        is_hotswappable: typing.Optional[bool] = OMIT,
+        server_version: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[ContainerImageRead]:
+        """
+        Create a new Container Image.
+
+        Parameters
+        ----------
+        name : str
+
+        packages : typing.Sequence[CodeExecutionPackageRequest]
+
+        tag : str
+
+        user_script : typing.Optional[str]
+
+        is_hotswappable : typing.Optional[bool]
+
+        server_version : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[ContainerImageRead]
+
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/container-images",
+            base_url=self._client_wrapper.get_environment().default,
+            method="POST",
+            json={
+                "name": name,
+                "packages": convert_and_respect_annotation_metadata(
+                    object_=packages, annotation=typing.Sequence[CodeExecutionPackageRequest], direction="write"
+                ),
+                "tag": tag,
+                "user_script": user_script,
+                "is_hotswappable": is_hotswappable,
+                "server_version": server_version,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    ContainerImageRead,
+                    parse_obj_as(
+                        type_=ContainerImageRead,  # 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)
+
     def retrieve(
         self, id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> HttpResponse[ContainerImageRead]:
@@ -117,6 +190,77 @@ class RawContainerImagesClient:
             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)
 
+    def update_container_image(
+        self,
+        id: str,
+        *,
+        packages: typing.Sequence[CodeExecutionPackageRequest],
+        tag: str,
+        user_script: typing.Optional[str] = OMIT,
+        is_hotswappable: typing.Optional[bool] = OMIT,
+        server_version: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[ContainerImageRead]:
+        """
+        Update an existing Container Image.
+
+        Parameters
+        ----------
+        id : str
+            A UUID string identifying this container image.
+
+        packages : typing.Sequence[CodeExecutionPackageRequest]
+
+        tag : str
+
+        user_script : typing.Optional[str]
+
+        is_hotswappable : typing.Optional[bool]
+
+        server_version : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[ContainerImageRead]
+
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/container-images/{jsonable_encoder(id)}",
+            base_url=self._client_wrapper.get_environment().default,
+            method="PUT",
+            json={
+                "packages": convert_and_respect_annotation_metadata(
+                    object_=packages, annotation=typing.Sequence[CodeExecutionPackageRequest], direction="write"
+                ),
+                "tag": tag,
+                "user_script": user_script,
+                "is_hotswappable": is_hotswappable,
+                "server_version": server_version,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    ContainerImageRead,
+                    parse_obj_as(
+                        type_=ContainerImageRead,  # 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)
+
     def docker_service_token(
         self, *, request_options: typing.Optional[RequestOptions] = None
     ) -> HttpResponse[DockerServiceToken]:
@@ -272,6 +416,77 @@ class AsyncRawContainerImagesClient:
             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)
 
+    async def create_container_image(
+        self,
+        *,
+        name: str,
+        packages: typing.Sequence[CodeExecutionPackageRequest],
+        tag: str,
+        user_script: typing.Optional[str] = OMIT,
+        is_hotswappable: typing.Optional[bool] = OMIT,
+        server_version: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[ContainerImageRead]:
+        """
+        Create a new Container Image.
+
+        Parameters
+        ----------
+        name : str
+
+        packages : typing.Sequence[CodeExecutionPackageRequest]
+
+        tag : str
+
+        user_script : typing.Optional[str]
+
+        is_hotswappable : typing.Optional[bool]
+
+        server_version : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[ContainerImageRead]
+
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/container-images",
+            base_url=self._client_wrapper.get_environment().default,
+            method="POST",
+            json={
+                "name": name,
+                "packages": convert_and_respect_annotation_metadata(
+                    object_=packages, annotation=typing.Sequence[CodeExecutionPackageRequest], direction="write"
+                ),
+                "tag": tag,
+                "user_script": user_script,
+                "is_hotswappable": is_hotswappable,
+                "server_version": server_version,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    ContainerImageRead,
+                    parse_obj_as(
+                        type_=ContainerImageRead,  # 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)
+
     async def retrieve(
         self, id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> AsyncHttpResponse[ContainerImageRead]:
@@ -312,6 +527,77 @@ class AsyncRawContainerImagesClient:
             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)
 
+    async def update_container_image(
+        self,
+        id: str,
+        *,
+        packages: typing.Sequence[CodeExecutionPackageRequest],
+        tag: str,
+        user_script: typing.Optional[str] = OMIT,
+        is_hotswappable: typing.Optional[bool] = OMIT,
+        server_version: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[ContainerImageRead]:
+        """
+        Update an existing Container Image.
+
+        Parameters
+        ----------
+        id : str
+            A UUID string identifying this container image.
+
+        packages : typing.Sequence[CodeExecutionPackageRequest]
+
+        tag : str
+
+        user_script : typing.Optional[str]
+
+        is_hotswappable : typing.Optional[bool]
+
+        server_version : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[ContainerImageRead]
+
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/container-images/{jsonable_encoder(id)}",
+            base_url=self._client_wrapper.get_environment().default,
+            method="PUT",
+            json={
+                "packages": convert_and_respect_annotation_metadata(
+                    object_=packages, annotation=typing.Sequence[CodeExecutionPackageRequest], direction="write"
+                ),
+                "tag": tag,
+                "user_script": user_script,
+                "is_hotswappable": is_hotswappable,
+                "server_version": server_version,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    ContainerImageRead,
+                    parse_obj_as(
+                        type_=ContainerImageRead,  # 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)
+
     async def docker_service_token(
         self, *, request_options: typing.Optional[RequestOptions] = None
     ) -> AsyncHttpResponse[DockerServiceToken]: