vellum-ai 0.9.2__py3-none-any.whl → 0.9.3__py3-none-any.whl

vellum/__init__.py

@@ -68,6 +68,7 @@ from .types import (
     DeploymentRead,
     DeploymentReleaseTagDeploymentHistoryItem,
     DeploymentReleaseTagRead,
+    DockerServiceToken,
     DocumentDocumentToDocumentIndex,
     DocumentIndexChunking,
     DocumentIndexChunkingRequest,
@@ -572,6 +573,7 @@ __all__ = [
     "DeploymentReleaseTagDeploymentHistoryItem",
     "DeploymentReleaseTagRead",
     "DeploymentsListRequestStatus",
+    "DockerServiceToken",
     "DocumentDocumentToDocumentIndex",
     "DocumentIndexChunking",
     "DocumentIndexChunkingRequest",

vellum/core/client_wrapper.py

@@ -17,7 +17,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "0.9.2",
+            "X-Fern-SDK-Version": "v0.9.3",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

vellum/resources/container_images/client.py

@@ -9,6 +9,7 @@ from json.decoder import JSONDecodeError
 from ...core.api_error import ApiError
 from ...types.container_image_read import ContainerImageRead
 from ...core.jsonable_encoder import jsonable_encoder
+from ...types.docker_service_token import DockerServiceToken
 from ...core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
@@ -131,6 +132,49 @@ class ContainerImagesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def docker_service_token(self, *, request_options: typing.Optional[RequestOptions] = None) -> DockerServiceToken:
+        """
+        An internal-only endpoint that's subject to breaking changes without notice. Not intended for public use.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DockerServiceToken
+
+
+        Examples
+        --------
+        from vellum import Vellum
+
+        client = Vellum(
+            api_key="YOUR_API_KEY",
+        )
+        client.container_images.docker_service_token()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/container-images/docker-service-token",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    DockerServiceToken,
+                    parse_obj_as(
+                        type_=DockerServiceToken,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
     def push_container_image(
         self,
         *,
@@ -330,6 +374,59 @@ class AsyncContainerImagesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def docker_service_token(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> DockerServiceToken:
+        """
+        An internal-only endpoint that's subject to breaking changes without notice. Not intended for public use.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DockerServiceToken
+
+
+        Examples
+        --------
+        import asyncio
+
+        from vellum import AsyncVellum
+
+        client = AsyncVellum(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.container_images.docker_service_token()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/container-images/docker-service-token",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    DockerServiceToken,
+                    parse_obj_as(
+                        type_=DockerServiceToken,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
     async def push_container_image(
         self,
         *,

vellum/resources/workflows/client.py

@@ -2,12 +2,13 @@
 
 import typing
 from ...core.client_wrapper import SyncClientWrapper
-from ...types.workflow_push_exec_config import WorkflowPushExecConfig
 from ...core.request_options import RequestOptions
-from ...types.workflow_push_response import WorkflowPushResponse
-from ...core.pydantic_utilities import parse_obj_as
+from ...core.jsonable_encoder import jsonable_encoder
 from json.decoder import JSONDecodeError
 from ...core.api_error import ApiError
+from ...types.workflow_push_exec_config import WorkflowPushExecConfig
+from ...types.workflow_push_response import WorkflowPushResponse
+from ...core.pydantic_utilities import parse_obj_as
 from ...core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
@@ -18,6 +19,51 @@ class WorkflowsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
+    def pull(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Iterator[bytes]:
+        """
+        An internal-only endpoint that's subject to breaking changes without notice. Not intended for public use.
+
+        Parameters
+        ----------
+        id : str
+            The ID of the Workflow to pull from
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Yields
+        ------
+        typing.Iterator[bytes]
+
+
+        Examples
+        --------
+        from vellum import Vellum
+
+        client = Vellum(
+            api_key="YOUR_API_KEY",
+        )
+        client.workflows.pull(
+            id="string",
+        )
+        """
+        with self._client_wrapper.httpx_client.stream(
+            f"v1/workflows/{jsonable_encoder(id)}/pull",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            request_options=request_options,
+        ) as _response:
+            try:
+                if 200 <= _response.status_code < 300:
+                    for _chunk in _response.iter_bytes():
+                        yield _chunk
+                    return
+                _response.read()
+                _response_json = _response.json()
+            except JSONDecodeError:
+                raise ApiError(status_code=_response.status_code, body=_response.text)
+            raise ApiError(status_code=_response.status_code, body=_response_json)
+
     def push(
         self,
         *,
@@ -88,6 +134,61 @@ class AsyncWorkflowsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
+    async def pull(
+        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.AsyncIterator[bytes]:
+        """
+        An internal-only endpoint that's subject to breaking changes without notice. Not intended for public use.
+
+        Parameters
+        ----------
+        id : str
+            The ID of the Workflow to pull from
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Yields
+        ------
+        typing.AsyncIterator[bytes]
+
+
+        Examples
+        --------
+        import asyncio
+
+        from vellum import AsyncVellum
+
+        client = AsyncVellum(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.workflows.pull(
+                id="string",
+            )
+
+
+        asyncio.run(main())
+        """
+        async with self._client_wrapper.httpx_client.stream(
+            f"v1/workflows/{jsonable_encoder(id)}/pull",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            request_options=request_options,
+        ) as _response:
+            try:
+                if 200 <= _response.status_code < 300:
+                    async for _chunk in _response.aiter_bytes():
+                        yield _chunk
+                    return
+                await _response.aread()
+                _response_json = _response.json()
+            except JSONDecodeError:
+                raise ApiError(status_code=_response.status_code, body=_response.text)
+            raise ApiError(status_code=_response.status_code, body=_response_json)
+
     async def push(
         self,
         *,

vellum/types/__init__.py

@@ -75,6 +75,7 @@ from .deployment_provider_payload_response_payload import DeploymentProviderPayl
 from .deployment_read import DeploymentRead
 from .deployment_release_tag_deployment_history_item import DeploymentReleaseTagDeploymentHistoryItem
 from .deployment_release_tag_read import DeploymentReleaseTagRead
+from .docker_service_token import DockerServiceToken
 from .document_document_to_document_index import DocumentDocumentToDocumentIndex
 from .document_index_chunking import DocumentIndexChunking
 from .document_index_chunking_request import DocumentIndexChunkingRequest
@@ -559,6 +560,7 @@ __all__ = [
     "DeploymentRead",
     "DeploymentReleaseTagDeploymentHistoryItem",
     "DeploymentReleaseTagRead",
+    "DockerServiceToken",
     "DocumentDocumentToDocumentIndex",
     "DocumentIndexChunking",
     "DocumentIndexChunkingRequest",

vellum/types/docker_service_token.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class DockerServiceToken(UniversalBaseModel):
+    access_token: str
+    organization_id: str
+    repository: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/enriched_normalized_completion.py

@@ -39,7 +39,7 @@ class EnrichedNormalizedCompletion(UniversalBaseModel):
     The logprobs of the completion. Only present if specified in the original request options.
     """
 
-    model_version_id: str = pydantic.Field()
+    model_version_id: typing.Optional[str] = pydantic.Field(default=None)
     """
     The ID of the model version used to generate this completion.
     """

{vellum_ai-0.9.2.dist-info → vellum_ai-0.9.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.9.2
+Version: 0.9.3
 Summary: 
 License: MIT
 Requires-Python: >=3.8,<4.0

{vellum_ai-0.9.2.dist-info → vellum_ai-0.9.3.dist-info}/RECORD

@@ -1,8 +1,8 @@
-vellum/__init__.py,sha256=DLO3lFK8O_dYKoxb8rvavpQM33DWDyiUZ7vtj14w41I,34176
+vellum/__init__.py,sha256=uz3omyTboMrMSk6YMbpqA5CI62by2F8tUuu9LWhK66I,34226
 vellum/client.py,sha256=kG4b9g1Jjm6zgzGBXCAYXcM_3xNQfBsa2Xut6F0eHQM,115201
 vellum/core/__init__.py,sha256=SQ85PF84B9MuKnBwHNHWemSGuy-g_515gFYNFhvEE0I,1438
 vellum/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-vellum/core/client_wrapper.py,sha256=U0Ysm0UaPDp8GR9MXU31QAE37RXWmykoO9ek-a5jrwo,1889
+vellum/core/client_wrapper.py,sha256=NgWlRdAr9JcMt5BPf4juPIsbFQfSBTNhjyjHNr8y864,1890
 vellum/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 vellum/core/file.py,sha256=X9IbmkZmB2bB_DpmZAO3crWdXagOakAyn6UCOCImCPg,2322
 vellum/core/http_client.py,sha256=R0pQpCppnEtxccGvXl4uJ76s7ro_65Fo_erlNNLp_AI,19228
@@ -34,7 +34,7 @@ vellum/resources/__init__.py,sha256=oPvhJ3z-7efrRlZ7ILrHQTypPBcXLOtGZdGMmHiVpLU,
 vellum/resources/ad_hoc/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 vellum/resources/ad_hoc/client.py,sha256=maNoWMHH8LSFlr5rDfoJybiPaoWuUiWFkt-IFq8dNMA,17271
 vellum/resources/container_images/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-vellum/resources/container_images/client.py,sha256=UbR8Mf7F0ia_DyFMwiRmHlj1AR9Z4EFCl2oP3h8Iqj0,12276
+vellum/resources/container_images/client.py,sha256=jK1n-NFsdBKCeEKh-EIqvw7R8AG9PP4GxxcoH9F0GYs,15463
 vellum/resources/deployments/__init__.py,sha256=AE0TcFwLrLBljM0ZDX-pPw4Kqt-1f5JDpIok2HS80QI,157
 vellum/resources/deployments/client.py,sha256=tF3llT_g6rfzDHpLhlEoz9gJDy8vIdNGKfICMJp3iEw,29236
 vellum/resources/deployments/types/__init__.py,sha256=IhwnmoXJ0r_QEhh1b2tBcaAm_x3fWMVuIhYmAapp_ZA,183
@@ -64,7 +64,7 @@ vellum/resources/workflow_deployments/types/workflow_deployments_list_request_st
 vellum/resources/workflow_sandboxes/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 vellum/resources/workflow_sandboxes/client.py,sha256=3wVQxkjrJ5bIS8fB5FpKXCP2dX38299ghWrJ8YmXxwQ,7435
 vellum/resources/workflows/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-vellum/resources/workflows/client.py,sha256=y8rjg69GUUiOxpenS8j4S_H7Rhpsn_A-1XV9zw9YTMM,4979
+vellum/resources/workflows/client.py,sha256=lXyxcMpu4pYGd9EmdLHCWTKEwJ2owPOoH_kI7lT5dOQ,8181
 vellum/resources/workspace_secrets/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 vellum/resources/workspace_secrets/client.py,sha256=h7UzXLyTttPq1t-JZGMg1BWxypxJvBGUdqg7KGT7MK4,8027
 vellum/terraform/__init__.py,sha256=CxzT0rkV9cXwAtxZ3gv46DCRt9vBl_Sx1SOj5MJtl0Y,498
@@ -78,7 +78,7 @@ vellum/terraform/ml_model/__init__.py,sha256=I8h1Ru-Rb-Hi_HusK6G7nJQZEKQGsAAHMmw
 vellum/terraform/provider/__init__.py,sha256=-06xKmAmknpohVzw5TD-t1bnUHta8OrQYqvMd04XM-U,12684
 vellum/terraform/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 vellum/terraform/versions.json,sha256=45c7jjRD5i4w9DJQHs5ZqLLVXRnQwP9Rirq3mWY-xEo,56
-vellum/types/__init__.py,sha256=u_3CDRrgpYIHwabMsDv4M_zYjO0gE3hCLGBYt1lc3CM,52006
+vellum/types/__init__.py,sha256=TsOOpzY7n4ut1T1LuZxN54GgthpA1V0pD10bvhXdZt8,52085
 vellum/types/ad_hoc_execute_prompt_event.py,sha256=bCjujA2XsOgyF3bRZbcEqV2rOIymRgsLoIRtZpB14xg,607
 vellum/types/ad_hoc_expand_meta_request.py,sha256=hS8PC3hC--OKvRKi2ZFj_RJPQ1bxo2GXno8qJq1kk28,1338
 vellum/types/ad_hoc_fulfilled_prompt_execution_meta.py,sha256=Bfvf1d_dkmshxRACVM5vcxbH_7AQY23RmrrnPc0ytYY,939
@@ -146,6 +146,7 @@ vellum/types/deployment_provider_payload_response_payload.py,sha256=xHLQnWFN0AZR
 vellum/types/deployment_read.py,sha256=NtXmYsYJOATxkMxeVkSM35XzDVGbva3RWmn5midBd1A,2160
 vellum/types/deployment_release_tag_deployment_history_item.py,sha256=df4qKHT1f-z0jnRS4UmP8MQe6u3PwYej_d8KDF7EL88,631
 vellum/types/deployment_release_tag_read.py,sha256=YlwssIgBd5lKVqelH-gejQXQ7l31vrsRNMJKDGDyTEA,1129
+vellum/types/docker_service_token.py,sha256=T0icNHBKsIs6TrEiDRjckM_f37hcF1DMwEE8161tTvY,614
 vellum/types/document_document_to_document_index.py,sha256=LbXTZyYxA4hxewquf7ZLZgBD9uWGoIs1J64x4fY7bNg,1229
 vellum/types/document_index_chunking.py,sha256=TU0Y7z0Xacm3dhzEDuDIG3ZKJCu3vNURRh3PqEd17mY,356
 vellum/types/document_index_chunking_request.py,sha256=g9BKCsHKg5kzjG7YYeMNQ_5R8TXLeSgumJlMXoSfBcs,435
@@ -154,7 +155,7 @@ vellum/types/document_index_indexing_config_request.py,sha256=Wt-ys1o_acHNyLU0c1
 vellum/types/document_index_read.py,sha256=9btlJ9-H7DwTFfBljT2PuxtyTQjhiWTStvGqVJMwirE,1469
 vellum/types/document_read.py,sha256=heZt7k29GVehbKTofcLjRVe1R_UjbCK5Hcbgga3OODY,1930
 vellum/types/document_status.py,sha256=GD_TSoFmZUBJnPl-chAmaQFzQ2_TYO3PSqi3-9QfEHE,122
-vellum/types/enriched_normalized_completion.py,sha256=_NDkKVJj70uKDtIOsQB7E-dL0mOMPwjwgCzhif0RuJI,1796
+vellum/types/enriched_normalized_completion.py,sha256=qk8o_PCj8rvgxa9omng0Nx5SvGNgzS7wnXMpoH56s8k,1825
 vellum/types/entity_status.py,sha256=bY0jEpISwXqFnbWd3PSb3yXEr-ounPXlAO_fyvHV7l8,158
 vellum/types/entity_visibility.py,sha256=BX1KdYd7dirpv878XDDvtOHkMOqebM8-lkWmLyFLaw4,184
 vellum/types/environment_enum.py,sha256=Wcewxp1cpGAMDIAZbTp4Y0GGfvy2Bq_Qu_67f_wBDGA,179
@@ -554,7 +555,7 @@ vellum/types/workflow_result_event_output_data_string.py,sha256=tM3kgh6tEhD0dFEb
 vellum/types/workflow_stream_event.py,sha256=Wn3Yzuy9MqWAeo8tEaXDTKDEbJoA8DdYdMVq8EKuhu8,361
 vellum/types/workspace_secret_read.py,sha256=3CnHDG72IAY0KRNvc31F0xLmhnpwjQHnDYCfQJzCxI0,714
 vellum/version.py,sha256=jq-1PlAYxN9AXuaZqbYk9ak27SgE2lw9Ia5gx1b1gVI,76
-vellum_ai-0.9.2.dist-info/LICENSE,sha256=CcaljEIoOBaU-wItPH4PmM_mDCGpyuUY0Er1BGu5Ti8,1073
-vellum_ai-0.9.2.dist-info/METADATA,sha256=FFfP07L5xYK0bVhI00nVlnXSKbShSu97EjOb7qZ82y4,4394
-vellum_ai-0.9.2.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-vellum_ai-0.9.2.dist-info/RECORD,,
+vellum_ai-0.9.3.dist-info/LICENSE,sha256=CcaljEIoOBaU-wItPH4PmM_mDCGpyuUY0Er1BGu5Ti8,1073
+vellum_ai-0.9.3.dist-info/METADATA,sha256=D55Hrkl9LmXgxpCVIgO1rfc_G8dFKO3REHEtqj6qE7U,4394
+vellum_ai-0.9.3.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+vellum_ai-0.9.3.dist-info/RECORD,,