vellum-ai 0.9.0__tar.gz → 0.9.1__tar.gz

{vellum_ai-0.9.0 → vellum_ai-0.9.1}/PKG-INFO

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

{vellum_ai-0.9.0 → vellum_ai-0.9.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vellum-ai"
-version = "0.9.0"
+version = "v0.9.1"
 description = ""
 readme = "README.md"
 authors = []

{vellum_ai-0.9.0 → vellum_ai-0.9.1}/src/vellum/__init__.py

@@ -59,6 +59,7 @@ from .types import (
     ConditionCombinator,
     ConditionalNodeResult,
     ConditionalNodeResultData,
+    ContainerImageRead,
     CreateTestSuiteTestCaseRequest,
     DeploymentProviderPayloadResponse,
     DeploymentProviderPayloadResponsePayload,
@@ -75,6 +76,7 @@ from .types import (
     DocumentStatus,
     EnrichedNormalizedCompletion,
     EntityStatus,
+    EntityVisibility,
     EnvironmentEnum,
     EphemeralPromptCacheConfigRequest,
     EphemeralPromptCacheConfigTypeEnum,
@@ -234,6 +236,7 @@ from .types import (
     OpenAiVectorizerTextEmbedding3SmallRequest,
     OpenAiVectorizerTextEmbeddingAda002,
     OpenAiVectorizerTextEmbeddingAda002Request,
+    PaginatedContainerImageReadList,
     PaginatedDocumentIndexReadList,
     PaginatedFolderEntityList,
     PaginatedSlimDeploymentReadList,
@@ -476,6 +479,7 @@ from .resources import (
     FolderEntitiesListRequestEntityStatus,
     WorkflowDeploymentsListRequestStatus,
     ad_hoc,
+    container_images,
     deployments,
     document_indexes,
     documents,
@@ -554,6 +558,7 @@ __all__ = [
     "ConditionCombinator",
     "ConditionalNodeResult",
     "ConditionalNodeResultData",
+    "ContainerImageRead",
     "CreateTestSuiteTestCaseRequest",
     "DeploymentProviderPayloadResponse",
     "DeploymentProviderPayloadResponsePayload",
@@ -572,6 +577,7 @@ __all__ = [
     "DocumentStatus",
     "EnrichedNormalizedCompletion",
     "EntityStatus",
+    "EntityVisibility",
     "EnvironmentEnum",
     "EphemeralPromptCacheConfigRequest",
     "EphemeralPromptCacheConfigTypeEnum",
@@ -735,6 +741,7 @@ __all__ = [
     "OpenAiVectorizerTextEmbedding3SmallRequest",
     "OpenAiVectorizerTextEmbeddingAda002",
     "OpenAiVectorizerTextEmbeddingAda002Request",
+    "PaginatedContainerImageReadList",
     "PaginatedDocumentIndexReadList",
     "PaginatedFolderEntityList",
     "PaginatedSlimDeploymentReadList",
@@ -974,6 +981,7 @@ __all__ = [
     "WorkspaceSecretRead",
     "__version__",
     "ad_hoc",
+    "container_images",
     "deployments",
     "document_indexes",
     "documents",

{vellum_ai-0.9.0 → vellum_ai-0.9.1}/src/vellum/client.py

@@ -5,6 +5,7 @@ from .environment import VellumEnvironment
 import httpx
 from .core.client_wrapper import SyncClientWrapper
 from .resources.ad_hoc.client import AdHocClient
+from .resources.container_images.client import ContainerImagesClient
 from .resources.deployments.client import DeploymentsClient
 from .resources.document_indexes.client import DocumentIndexesClient
 from .resources.documents.client import DocumentsClient
@@ -52,6 +53,7 @@ from .types.submit_completion_actual_request import SubmitCompletionActualReques
 from .types.submit_workflow_execution_actual_request import SubmitWorkflowExecutionActualRequest
 from .core.client_wrapper import AsyncClientWrapper
 from .resources.ad_hoc.client import AsyncAdHocClient
+from .resources.container_images.client import AsyncContainerImagesClient
 from .resources.deployments.client import AsyncDeploymentsClient
 from .resources.document_indexes.client import AsyncDocumentIndexesClient
 from .resources.documents.client import AsyncDocumentsClient
@@ -124,6 +126,7 @@ class Vellum:
             timeout=_defaulted_timeout,
         )
         self.ad_hoc = AdHocClient(client_wrapper=self._client_wrapper)
+        self.container_images = ContainerImagesClient(client_wrapper=self._client_wrapper)
         self.deployments = DeploymentsClient(client_wrapper=self._client_wrapper)
         self.document_indexes = DocumentIndexesClient(client_wrapper=self._client_wrapper)
         self.documents = DocumentsClient(client_wrapper=self._client_wrapper)
@@ -1442,6 +1445,7 @@ class AsyncVellum:
             timeout=_defaulted_timeout,
         )
         self.ad_hoc = AsyncAdHocClient(client_wrapper=self._client_wrapper)
+        self.container_images = AsyncContainerImagesClient(client_wrapper=self._client_wrapper)
         self.deployments = AsyncDeploymentsClient(client_wrapper=self._client_wrapper)
         self.document_indexes = AsyncDocumentIndexesClient(client_wrapper=self._client_wrapper)
         self.documents = AsyncDocumentsClient(client_wrapper=self._client_wrapper)

{vellum_ai-0.9.0 → vellum_ai-0.9.1}/src/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.0",
+            "X-Fern-SDK-Version": "v0.9.1",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

{vellum_ai-0.9.0 → vellum_ai-0.9.1}/src/vellum/resources/__init__.py

@@ -2,6 +2,7 @@
 
 from . import (
     ad_hoc,
+    container_images,
     deployments,
     document_indexes,
     documents,
@@ -26,6 +27,7 @@ __all__ = [
     "FolderEntitiesListRequestEntityStatus",
     "WorkflowDeploymentsListRequestStatus",
     "ad_hoc",
+    "container_images",
     "deployments",
     "document_indexes",
     "documents",

vellum_ai-0.9.1/src/vellum/resources/container_images/client.py

@@ -0,0 +1,405 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from ...core.client_wrapper import SyncClientWrapper
+from ...core.request_options import RequestOptions
+from ...types.paginated_container_image_read_list import PaginatedContainerImageReadList
+from ...core.pydantic_utilities import parse_obj_as
+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 ...core.client_wrapper import AsyncClientWrapper
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class ContainerImagesClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def list(
+        self,
+        *,
+        limit: typing.Optional[int] = None,
+        offset: typing.Optional[int] = None,
+        ordering: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> PaginatedContainerImageReadList:
+        """
+        Retrieve a list of container images for the organization.
+
+        Parameters
+        ----------
+        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.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        PaginatedContainerImageReadList
+
+
+        Examples
+        --------
+        from vellum import Vellum
+
+        client = Vellum(
+            api_key="YOUR_API_KEY",
+        )
+        client.container_images.list()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/container-images",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            params={
+                "limit": limit,
+                "offset": offset,
+                "ordering": ordering,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    PaginatedContainerImageReadList,
+                    parse_obj_as(
+                        type_=PaginatedContainerImageReadList,  # 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 retrieve(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> ContainerImageRead:
+        """
+        Retrieve a Container Image by its ID or name.
+
+        Parameters
+        ----------
+        id : str
+            Either the Container Image's ID or its unique name
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ContainerImageRead
+
+
+        Examples
+        --------
+        from vellum import Vellum
+
+        client = Vellum(
+            api_key="YOUR_API_KEY",
+        )
+        client.container_images.retrieve(
+            id="id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/container-images/{jsonable_encoder(id)}",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    ContainerImageRead,
+                    parse_obj_as(
+                        type_=ContainerImageRead,  # 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,
+        *,
+        name: str,
+        sha: str,
+        tags: typing.Sequence[str],
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ContainerImageRead:
+        """
+        An internal-only endpoint that's subject to breaking changes without notice. Not intended for public use.
+
+        Parameters
+        ----------
+        name : str
+
+        sha : str
+
+        tags : typing.Sequence[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ContainerImageRead
+
+
+        Examples
+        --------
+        from vellum import Vellum
+
+        client = Vellum(
+            api_key="YOUR_API_KEY",
+        )
+        client.container_images.push_container_image(
+            name="name",
+            sha="sha",
+            tags=["tags"],
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/container-images/push",
+            base_url=self._client_wrapper.get_environment().default,
+            method="POST",
+            json={
+                "name": name,
+                "sha": sha,
+                "tags": tags,
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    ContainerImageRead,
+                    parse_obj_as(
+                        type_=ContainerImageRead,  # 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)
+
+
+class AsyncContainerImagesClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def list(
+        self,
+        *,
+        limit: typing.Optional[int] = None,
+        offset: typing.Optional[int] = None,
+        ordering: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> PaginatedContainerImageReadList:
+        """
+        Retrieve a list of container images for the organization.
+
+        Parameters
+        ----------
+        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.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        PaginatedContainerImageReadList
+
+
+        Examples
+        --------
+        import asyncio
+
+        from vellum import AsyncVellum
+
+        client = AsyncVellum(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.container_images.list()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/container-images",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            params={
+                "limit": limit,
+                "offset": offset,
+                "ordering": ordering,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    PaginatedContainerImageReadList,
+                    parse_obj_as(
+                        type_=PaginatedContainerImageReadList,  # 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 retrieve(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> ContainerImageRead:
+        """
+        Retrieve a Container Image by its ID or name.
+
+        Parameters
+        ----------
+        id : str
+            Either the Container Image's ID or its unique name
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ContainerImageRead
+
+
+        Examples
+        --------
+        import asyncio
+
+        from vellum import AsyncVellum
+
+        client = AsyncVellum(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.container_images.retrieve(
+                id="id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/container-images/{jsonable_encoder(id)}",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    ContainerImageRead,
+                    parse_obj_as(
+                        type_=ContainerImageRead,  # 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,
+        *,
+        name: str,
+        sha: str,
+        tags: typing.Sequence[str],
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ContainerImageRead:
+        """
+        An internal-only endpoint that's subject to breaking changes without notice. Not intended for public use.
+
+        Parameters
+        ----------
+        name : str
+
+        sha : str
+
+        tags : typing.Sequence[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ContainerImageRead
+
+
+        Examples
+        --------
+        import asyncio
+
+        from vellum import AsyncVellum
+
+        client = AsyncVellum(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.container_images.push_container_image(
+                name="name",
+                sha="sha",
+                tags=["tags"],
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/container-images/push",
+            base_url=self._client_wrapper.get_environment().default,
+            method="POST",
+            json={
+                "name": name,
+                "sha": sha,
+                "tags": tags,
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    ContainerImageRead,
+                    parse_obj_as(
+                        type_=ContainerImageRead,  # 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)

vellum_ai-0.9.1/src/vellum/resources/workspace_secrets/__init__.py

@@ -0,0 +1,2 @@
+# This file was auto-generated by Fern from our API Definition.
+

{vellum_ai-0.9.0 → vellum_ai-0.9.1}/src/vellum/types/__init__.py

@@ -66,6 +66,7 @@ from .components_schemas_pdf_search_result_meta_source_request import Components
 from .condition_combinator import ConditionCombinator
 from .conditional_node_result import ConditionalNodeResult
 from .conditional_node_result_data import ConditionalNodeResultData
+from .container_image_read import ContainerImageRead
 from .create_test_suite_test_case_request import CreateTestSuiteTestCaseRequest
 from .deployment_provider_payload_response import DeploymentProviderPayloadResponse
 from .deployment_provider_payload_response_payload import DeploymentProviderPayloadResponsePayload
@@ -82,6 +83,7 @@ from .document_read import DocumentRead
 from .document_status import DocumentStatus
 from .enriched_normalized_completion import EnrichedNormalizedCompletion
 from .entity_status import EntityStatus
+from .entity_visibility import EntityVisibility
 from .environment_enum import EnvironmentEnum
 from .ephemeral_prompt_cache_config_request import EphemeralPromptCacheConfigRequest
 from .ephemeral_prompt_cache_config_type_enum import EphemeralPromptCacheConfigTypeEnum
@@ -245,6 +247,7 @@ from .open_ai_vectorizer_text_embedding_3_small import OpenAiVectorizerTextEmbed
 from .open_ai_vectorizer_text_embedding_3_small_request import OpenAiVectorizerTextEmbedding3SmallRequest
 from .open_ai_vectorizer_text_embedding_ada_002 import OpenAiVectorizerTextEmbeddingAda002
 from .open_ai_vectorizer_text_embedding_ada_002_request import OpenAiVectorizerTextEmbeddingAda002Request
+from .paginated_container_image_read_list import PaginatedContainerImageReadList
 from .paginated_document_index_read_list import PaginatedDocumentIndexReadList
 from .paginated_folder_entity_list import PaginatedFolderEntityList
 from .paginated_slim_deployment_read_list import PaginatedSlimDeploymentReadList
@@ -543,6 +546,7 @@ __all__ = [
     "ConditionCombinator",
     "ConditionalNodeResult",
     "ConditionalNodeResultData",
+    "ContainerImageRead",
     "CreateTestSuiteTestCaseRequest",
     "DeploymentProviderPayloadResponse",
     "DeploymentProviderPayloadResponsePayload",
@@ -559,6 +563,7 @@ __all__ = [
     "DocumentStatus",
     "EnrichedNormalizedCompletion",
     "EntityStatus",
+    "EntityVisibility",
     "EnvironmentEnum",
     "EphemeralPromptCacheConfigRequest",
     "EphemeralPromptCacheConfigTypeEnum",
@@ -718,6 +723,7 @@ __all__ = [
     "OpenAiVectorizerTextEmbedding3SmallRequest",
     "OpenAiVectorizerTextEmbeddingAda002",
     "OpenAiVectorizerTextEmbeddingAda002Request",
+    "PaginatedContainerImageReadList",
     "PaginatedDocumentIndexReadList",
     "PaginatedFolderEntityList",
     "PaginatedSlimDeploymentReadList",

vellum_ai-0.9.1/src/vellum/types/container_image_read.py

@@ -0,0 +1,28 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .entity_visibility import EntityVisibility
+import datetime as dt
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class ContainerImageRead(UniversalBaseModel):
+    id: str
+    name: str
+    visibility: EntityVisibility
+    created: dt.datetime
+    modified: dt.datetime
+    repository: str
+    sha: str
+    tags: typing.List[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_ai-0.9.1/src/vellum/types/entity_visibility.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+EntityVisibility = typing.Union[typing.Literal["DEFAULT", "PUBLIC", "PRIVATE", "DISABLED"], typing.Any]

vellum_ai-0.9.1/src/vellum/types/paginated_container_image_read_list.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .container_image_read import ContainerImageRead
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class PaginatedContainerImageReadList(UniversalBaseModel):
+    count: typing.Optional[int] = None
+    next: typing.Optional[str] = None
+    previous: typing.Optional[str] = None
+    results: typing.Optional[typing.List[ContainerImageRead]] = None
+
+    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