vellum-ai 0.9.16rc4__tar.gz → 0.10.1__tar.gz

{vellum_ai-0.9.16rc4 → vellum_ai-0.10.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.9.16rc4
+Version: 0.10.1
 Summary: 
 License: MIT
 Requires-Python: >=3.9,<4.0
@@ -97,7 +97,7 @@ more below](#workflows-sdk).
 
 ## Client SDK
 The Vellum Client SDK, found within `src/client` is a low-level client used to interact directly with the Vellum API.
-Learn more and get started by visiting the [Vellum Client SDK README](/src/client/README.md).
+Learn more and get started by visiting the [Vellum Client SDK README](/src/vellum/client/README.md).
 
 
 ## Workflows SDK
@@ -109,7 +109,7 @@ The Workflows SDK can be used with or without a Vellum account, but a Vellum acc
 out-of-box nodes and features, including the ability to push and pull your Workflow definition to Vellum for editing
 and debugging via a UI.
 
-To learn more and get started, visit the [Vellum Workflows SDK README](/src/workflows/README.md).
+To learn more and get started, visit the [Vellum Workflows SDK README](/src/vellum/workflows/README.md).
 
 
 ## Open-Source vs. Paid

{vellum_ai-0.9.16rc4 → vellum_ai-0.10.1}/README.md

@@ -57,7 +57,7 @@ more below](#workflows-sdk).
 
 ## Client SDK
 The Vellum Client SDK, found within `src/client` is a low-level client used to interact directly with the Vellum API.
-Learn more and get started by visiting the [Vellum Client SDK README](/src/client/README.md).
+Learn more and get started by visiting the [Vellum Client SDK README](/src/vellum/client/README.md).
 
 
 ## Workflows SDK
@@ -69,7 +69,7 @@ The Workflows SDK can be used with or without a Vellum account, but a Vellum acc
 out-of-box nodes and features, including the ability to push and pull your Workflow definition to Vellum for editing
 and debugging via a UI.
 
-To learn more and get started, visit the [Vellum Workflows SDK README](/src/workflows/README.md).
+To learn more and get started, visit the [Vellum Workflows SDK README](/src/vellum/workflows/README.md).
 
 
 ## Open-Source vs. Paid

{vellum_ai-0.9.16rc4 → vellum_ai-0.10.1}/ee/vellum_ee/workflows/display/utils/vellum.py

@@ -4,11 +4,20 @@ import typing
 from typing import Any, List, Union, cast
 
 from vellum import ChatMessage, SearchResult, SearchResultRequest, VellumVariableType
-
+from vellum.workflows.descriptors.base import BaseDescriptor
+from vellum.workflows.references import OutputReference, WorkflowInputReference
+from vellum.workflows.references.execution_count import ExecutionCountReference
+from vellum.workflows.references.node import NodeReference
+from vellum.workflows.references.vellum_secret import VellumSecretReference
+from vellum.workflows.types.core import VellumValuePrimitive
+from vellum.workflows.utils.vellum_variables import primitive_type_to_vellum_variable_type
+from vellum.workflows.vellum_client import create_vellum_client
 from vellum_ee.workflows.display.types import WorkflowDisplayContext
 from vellum_ee.workflows.display.vellum import (
     ChatHistoryVellumValue,
     ConstantValuePointer,
+    ExecutionCounterData,
+    ExecutionCounterPointer,
     InputVariableData,
     InputVariablePointer,
     JsonVellumValue,
@@ -22,13 +31,6 @@ from vellum_ee.workflows.display.vellum import (
     WorkspaceSecretData,
     WorkspaceSecretPointer,
 )
-from vellum.workflows.descriptors.base import BaseDescriptor
-from vellum.workflows.references import OutputReference, WorkflowInputReference
-from vellum.workflows.references.node import NodeReference
-from vellum.workflows.references.vellum_secret import VellumSecretReference
-from vellum.workflows.types.core import VellumValuePrimitive
-from vellum.workflows.utils.vellum_variables import primitive_type_to_vellum_variable_type
-from vellum.workflows.vellum_client import create_vellum_client
 
 _T = typing.TypeVar("_T")
 
@@ -61,13 +63,12 @@ def create_node_input_value_pointer_rule(
         upstream_node, output_display = display_context.node_output_displays[value]
         upstream_node_display = display_context.node_displays[upstream_node]
         return NodeOutputPointer(
-            type="NODE_OUTPUT",
             data=NodeOutputData(node_id=str(upstream_node_display.node_id), output_id=str(output_display.id)),
         )
     if isinstance(value, WorkflowInputReference):
         workflow_input_display = display_context.workflow_input_displays[value]
         return InputVariablePointer(
-            type="INPUT_VARIABLE", data=InputVariableData(input_variable_id=str(workflow_input_display.id))
+            data=InputVariableData(input_variable_id=str(workflow_input_display.id))
         )
     if isinstance(value, VellumSecretReference):
         # TODO: Pass through the name instead of retrieving the ID
@@ -77,12 +78,16 @@ def create_node_input_value_pointer_rule(
             id=value.name,
         )
         return WorkspaceSecretPointer(
-            type="WORKSPACE_SECRET",
             data=WorkspaceSecretData(
                 type="STRING",
                 workspace_secret_id=str(workspace_secret.id),
             ),
         )
+    if isinstance(value, ExecutionCountReference):
+        node_class_display = display_context.node_displays[value.node_class]
+        return ExecutionCounterPointer(
+            data=ExecutionCounterData(node_id=str(node_class_display.node_id)),
+        )
 
     if not isinstance(value, BaseDescriptor):
         vellum_value = primitive_to_vellum_value(value)

{vellum_ai-0.9.16rc4 → vellum_ai-0.10.1}/ee/vellum_ee/workflows/display/vellum.py

@@ -7,7 +7,6 @@ from pydantic import Field
 
 from vellum import ChatMessage, PromptParameters, SearchResult, SearchResultRequest, VellumVariable, VellumVariableType
 from vellum.core import UniversalBaseModel
-
 from vellum_ee.workflows.display.base import (
     EdgeDisplay,
     EdgeDisplayOverrides,
@@ -211,11 +210,21 @@ class WorkspaceSecretPointer(UniversalBaseModel):
     data: WorkspaceSecretData
 
 
+class ExecutionCounterData(UniversalBaseModel):
+    node_id: str
+
+
+class ExecutionCounterPointer(UniversalBaseModel):
+    type: Literal["EXECUTION_COUNTER"] = "EXECUTION_COUNTER"
+    data: ExecutionCounterData
+
+
 NodeInputValuePointerRule = Union[
     NodeOutputPointer,
     InputVariablePointer,
     ConstantValuePointer,
     WorkspaceSecretPointer,
+    ExecutionCounterPointer,
 ]
 
 

{vellum_ai-0.9.16rc4 → vellum_ai-0.10.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vellum-ai"
-version = "0.9.16-pre4"
+version = "0.10.1"
 description = ""
 readme = "README.md"
 authors = []
@@ -31,7 +31,7 @@ packages = [
 ]
 
 [project.urls]
-Repository = 'https://github.com/vellum-ai/vellum-client-python'
+Repository = 'https://github.com/vellum-ai/vellum-python-sdks'
 
 [tool.poetry.dependencies]
 python = "^3.9"
@@ -60,6 +60,8 @@ python-dateutil = "^2.9.0"
 types-python-dateutil = "^2.9.0.20240316"
 deepdiff = "8.0.1"
 ipython = "8.12.3"
+isort = "5.13.2"
+pre-commit = "3.5.0"
 pytest-mock = "3.10.0"
 requests-mock = "1.11.0"
 ruff = "^0.5.6"
@@ -94,3 +96,18 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry.scripts]
 vellum = "vellum_cli:main"
+
+[tool.isort]
+multi_line_output = 3
+force_grid_wrap = 0
+line_length = 120
+combine_as_imports = true
+force_sort_within_sections = true
+include_trailing_comma = true
+use_parentheses = true
+known_testing = "pytest"
+known_typing = ["typing", "typing_extensions"]
+known_vellum = ["vellum", "vellum_cli", "vellum_ee"]
+default_section = "THIRDPARTY"
+sections = ["FUTURE", "TESTING", "STDLIB", "TYPING", "THIRDPARTY", "VELLUM", "FIRSTPARTY", "LOCALFOLDER"]
+no_lines_before = ["STDLIB", "TYPING"]

{vellum_ai-0.9.16rc4 → vellum_ai-0.10.1}/src/vellum/__init__.py

@@ -66,6 +66,7 @@ from .types import (
     ConditionalNodeResultData,
     ContainerImageRead,
     CreateTestSuiteTestCaseRequest,
+    DeploymentHistoryItem,
     DeploymentProviderPayloadResponse,
     DeploymentProviderPayloadResponsePayload,
     DeploymentRead,
@@ -187,6 +188,7 @@ from .types import (
     MetricDefinitionExecution,
     MetricDefinitionInput,
     MetricNodeResult,
+    MlModelRead,
     MlModelUsage,
     NamedScenarioInputChatHistoryVariableValueRequest,
     NamedScenarioInputJsonVariableValueRequest,
@@ -435,6 +437,7 @@ from .types import (
     VellumVariable,
     VellumVariableExtensions,
     VellumVariableType,
+    WorkflowDeploymentHistoryItem,
     WorkflowDeploymentRead,
     WorkflowEventError,
     WorkflowExecutionActualChatHistoryRequest,
@@ -497,6 +500,7 @@ from .resources import (
     documents,
     folder_entities,
     metric_definitions,
+    ml_models,
     sandboxes,
     test_suite_runs,
     test_suites,
@@ -577,6 +581,7 @@ __all__ = [
     "ConditionalNodeResultData",
     "ContainerImageRead",
     "CreateTestSuiteTestCaseRequest",
+    "DeploymentHistoryItem",
     "DeploymentProviderPayloadResponse",
     "DeploymentProviderPayloadResponsePayload",
     "DeploymentRead",
@@ -705,6 +710,7 @@ __all__ = [
     "MetricDefinitionExecution",
     "MetricDefinitionInput",
     "MetricNodeResult",
+    "MlModelRead",
     "MlModelUsage",
     "NamedScenarioInputChatHistoryVariableValueRequest",
     "NamedScenarioInputJsonVariableValueRequest",
@@ -956,6 +962,7 @@ __all__ = [
     "VellumVariable",
     "VellumVariableExtensions",
     "VellumVariableType",
+    "WorkflowDeploymentHistoryItem",
     "WorkflowDeploymentRead",
     "WorkflowDeploymentsListRequestStatus",
     "WorkflowEventError",
@@ -1011,6 +1018,7 @@ __all__ = [
     "documents",
     "folder_entities",
     "metric_definitions",
+    "ml_models",
     "sandboxes",
     "test_suite_runs",
     "test_suites",

{vellum_ai-0.9.16rc4 → vellum_ai-0.10.1}/src/vellum/client/README.md

@@ -1,6 +1,6 @@
 # Vellum Python Library
 
-[![fern shield](https://img.shields.io/badge/%F0%9F%8C%BF-Built%20with%20Fern-brightgreen)](https://buildwithfern.com?utm_source=github&utm_medium=github&utm_campaign=readme&utm_source=https%3A%2F%2Fgithub.com%2Fvellum-ai%2Fvellum-client-python)
+[![fern shield](https://img.shields.io/badge/%F0%9F%8C%BF-Built%20with%20Fern-brightgreen)](https://buildwithfern.com?utm_source=github&utm_medium=github&utm_campaign=readme&utm_source=https%3A%2F%2Fgithub.com%2Fvellum-ai%2Fvellum-python-sdks)
 [![pypi](https://img.shields.io/pypi/v/vellum-ai)](https://pypi.python.org/pypi/vellum-ai)
 
 The Vellum Python library provides convenient access to the Vellum API from Python.

{vellum_ai-0.9.16rc4 → vellum_ai-0.10.1}/src/vellum/client/__init__.py

@@ -11,6 +11,7 @@ from .resources.document_indexes.client import DocumentIndexesClient
 from .resources.documents.client import DocumentsClient
 from .resources.folder_entities.client import FolderEntitiesClient
 from .resources.metric_definitions.client import MetricDefinitionsClient
+from .resources.ml_models.client import MlModelsClient
 from .resources.sandboxes.client import SandboxesClient
 from .resources.test_suite_runs.client import TestSuiteRunsClient
 from .resources.test_suites.client import TestSuitesClient
@@ -59,6 +60,7 @@ from .resources.document_indexes.client import AsyncDocumentIndexesClient
 from .resources.documents.client import AsyncDocumentsClient
 from .resources.folder_entities.client import AsyncFolderEntitiesClient
 from .resources.metric_definitions.client import AsyncMetricDefinitionsClient
+from .resources.ml_models.client import AsyncMlModelsClient
 from .resources.sandboxes.client import AsyncSandboxesClient
 from .resources.test_suite_runs.client import AsyncTestSuiteRunsClient
 from .resources.test_suites.client import AsyncTestSuitesClient
@@ -132,6 +134,7 @@ class Vellum:
         self.documents = DocumentsClient(client_wrapper=self._client_wrapper)
         self.folder_entities = FolderEntitiesClient(client_wrapper=self._client_wrapper)
         self.metric_definitions = MetricDefinitionsClient(client_wrapper=self._client_wrapper)
+        self.ml_models = MlModelsClient(client_wrapper=self._client_wrapper)
         self.sandboxes = SandboxesClient(client_wrapper=self._client_wrapper)
         self.test_suite_runs = TestSuiteRunsClient(client_wrapper=self._client_wrapper)
         self.test_suites = TestSuitesClient(client_wrapper=self._client_wrapper)
@@ -1451,6 +1454,7 @@ class AsyncVellum:
         self.documents = AsyncDocumentsClient(client_wrapper=self._client_wrapper)
         self.folder_entities = AsyncFolderEntitiesClient(client_wrapper=self._client_wrapper)
         self.metric_definitions = AsyncMetricDefinitionsClient(client_wrapper=self._client_wrapper)
+        self.ml_models = AsyncMlModelsClient(client_wrapper=self._client_wrapper)
         self.sandboxes = AsyncSandboxesClient(client_wrapper=self._client_wrapper)
         self.test_suite_runs = AsyncTestSuiteRunsClient(client_wrapper=self._client_wrapper)
         self.test_suites = AsyncTestSuitesClient(client_wrapper=self._client_wrapper)

{vellum_ai-0.9.16rc4 → vellum_ai-0.10.1}/src/vellum/client/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.17",
+            "X-Fern-SDK-Version": "0.10.1",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

{vellum_ai-0.9.16rc4 → vellum_ai-0.10.1}/src/vellum/client/resources/__init__.py

@@ -8,6 +8,7 @@ from . import (
     documents,
     folder_entities,
     metric_definitions,
+    ml_models,
     sandboxes,
     test_suite_runs,
     test_suites,
@@ -37,6 +38,7 @@ __all__ = [
     "documents",
     "folder_entities",
     "metric_definitions",
+    "ml_models",
     "sandboxes",
     "test_suite_runs",
     "test_suites",

{vellum_ai-0.9.16rc4 → vellum_ai-0.10.1}/src/vellum/client/resources/deployments/client.py

@@ -10,6 +10,7 @@ from json.decoder import JSONDecodeError
 from ...core.api_error import ApiError
 from ...types.deployment_read import DeploymentRead
 from ...core.jsonable_encoder import jsonable_encoder
+from ...types.deployment_history_item import DeploymentHistoryItem
 from .types.list_deployment_release_tags_request_source import ListDeploymentReleaseTagsRequestSource
 from ...types.paginated_deployment_release_tag_read_list import PaginatedDeploymentReleaseTagReadList
 from ...types.deployment_release_tag_read import DeploymentReleaseTagRead
@@ -148,6 +149,60 @@ class DeploymentsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def deployment_history_item_retrieve(
+        self, history_id_or_release_tag: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> DeploymentHistoryItem:
+        """
+        Retrieve a specific Deployment History Item by either its UUID or the name of a Release Tag that points to it.
+
+        Parameters
+        ----------
+        history_id_or_release_tag : str
+            Either the UUID of Deployment History Item you'd like to retrieve, or the name of a Release Tag that's pointing to the Deployment History Item you'd like to retrieve.
+
+        id : str
+            A UUID string identifying this deployment.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DeploymentHistoryItem
+
+
+        Examples
+        --------
+        from vellum import Vellum
+
+        client = Vellum(
+            api_key="YOUR_API_KEY",
+        )
+        client.deployments.deployment_history_item_retrieve(
+            history_id_or_release_tag="history_id_or_release_tag",
+            id="id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/deployments/{jsonable_encoder(id)}/history/{jsonable_encoder(history_id_or_release_tag)}",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    DeploymentHistoryItem,
+                    parse_obj_as(
+                        type_=DeploymentHistoryItem,  # 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 list_deployment_release_tags(
         self,
         id: str,
@@ -614,6 +669,68 @@ class AsyncDeploymentsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def deployment_history_item_retrieve(
+        self, history_id_or_release_tag: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> DeploymentHistoryItem:
+        """
+        Retrieve a specific Deployment History Item by either its UUID or the name of a Release Tag that points to it.
+
+        Parameters
+        ----------
+        history_id_or_release_tag : str
+            Either the UUID of Deployment History Item you'd like to retrieve, or the name of a Release Tag that's pointing to the Deployment History Item you'd like to retrieve.
+
+        id : str
+            A UUID string identifying this deployment.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DeploymentHistoryItem
+
+
+        Examples
+        --------
+        import asyncio
+
+        from vellum import AsyncVellum
+
+        client = AsyncVellum(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.deployments.deployment_history_item_retrieve(
+                history_id_or_release_tag="history_id_or_release_tag",
+                id="id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/deployments/{jsonable_encoder(id)}/history/{jsonable_encoder(history_id_or_release_tag)}",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    DeploymentHistoryItem,
+                    parse_obj_as(
+                        type_=DeploymentHistoryItem,  # 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 list_deployment_release_tags(
         self,
         id: str,

vellum_ai-0.10.1/src/vellum/client/resources/ml_models/client.py

@@ -0,0 +1,125 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ...core.client_wrapper import SyncClientWrapper
+import typing
+from ...core.request_options import RequestOptions
+from ...types.ml_model_read import MlModelRead
+from ...core.jsonable_encoder import jsonable_encoder
+from ...core.pydantic_utilities import parse_obj_as
+from json.decoder import JSONDecodeError
+from ...core.api_error import ApiError
+from ...core.client_wrapper import AsyncClientWrapper
+
+
+class MlModelsClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def retrieve(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> MlModelRead:
+        """
+        Retrieve details about an ML Model
+
+        Parameters
+        ----------
+        id : str
+            Either the ML Model's ID, its unique name, or its ID in the workspace.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        MlModelRead
+
+
+        Examples
+        --------
+        from vellum import Vellum
+
+        client = Vellum(
+            api_key="YOUR_API_KEY",
+        )
+        client.ml_models.retrieve(
+            id="id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/ml-models/{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(
+                    MlModelRead,
+                    parse_obj_as(
+                        type_=MlModelRead,  # 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 AsyncMlModelsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def retrieve(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> MlModelRead:
+        """
+        Retrieve details about an ML Model
+
+        Parameters
+        ----------
+        id : str
+            Either the ML Model's ID, its unique name, or its ID in the workspace.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        MlModelRead
+
+
+        Examples
+        --------
+        import asyncio
+
+        from vellum import AsyncVellum
+
+        client = AsyncVellum(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.ml_models.retrieve(
+                id="id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/ml-models/{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(
+                    MlModelRead,
+                    parse_obj_as(
+                        type_=MlModelRead,  # 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)