vellum-ai 0.8.21__tar.gz → 0.8.22__tar.gz

{vellum_ai-0.8.21 → vellum_ai-0.8.22}/PKG-INFO

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

{vellum_ai-0.8.21 → vellum_ai-0.8.22}/pyproject.toml

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

{vellum_ai-0.8.21 → vellum_ai-0.8.22}/src/vellum/__init__.py

@@ -407,6 +407,8 @@ from .types import (
     VellumValueLogicalExpressionRequest,
     VellumValueRequest,
     VellumVariable,
+    VellumVariableExtensions,
+    VellumVariableExtensionsRequest,
     VellumVariableRequest,
     VellumVariableType,
     WorkflowDeploymentRead,
@@ -888,6 +890,8 @@ __all__ = [
     "VellumValueLogicalExpressionRequest",
     "VellumValueRequest",
     "VellumVariable",
+    "VellumVariableExtensions",
+    "VellumVariableExtensionsRequest",
     "VellumVariableRequest",
     "VellumVariableType",
     "WorkflowDeploymentRead",

{vellum_ai-0.8.21 → vellum_ai-0.8.22}/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.8.21",
+            "X-Fern-SDK-Version": "0.8.22",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

{vellum_ai-0.8.21 → vellum_ai-0.8.22}/src/vellum/resources/ad_hoc/client.py

@@ -72,7 +72,9 @@ class AdHocClient:
             JinjaPromptBlockRequest,
             PromptParametersRequest,
             PromptRequestStringInputRequest,
+            StringVellumValueRequest,
             Vellum,
+            VellumVariableExtensionsRequest,
             VellumVariableRequest,
         )
 
@@ -92,6 +94,13 @@ class AdHocClient:
                     id="string",
                     key="string",
                     type="STRING",
+                    required=True,
+                    default=StringVellumValueRequest(
+                        value="string",
+                    ),
+                    extensions=VellumVariableExtensionsRequest(
+                        color="string",
+                    ),
                 )
             ],
             parameters=PromptParametersRequest(
@@ -258,6 +267,8 @@ class AsyncAdHocClient:
             JinjaPromptBlockRequest,
             PromptParametersRequest,
             PromptRequestStringInputRequest,
+            StringVellumValueRequest,
+            VellumVariableExtensionsRequest,
             VellumVariableRequest,
         )
 
@@ -280,6 +291,13 @@ class AsyncAdHocClient:
                         id="string",
                         key="string",
                         type="STRING",
+                        required=True,
+                        default=StringVellumValueRequest(
+                            value="string",
+                        ),
+                        extensions=VellumVariableExtensionsRequest(
+                            color="string",
+                        ),
                     )
                 ],
                 parameters=PromptParametersRequest(

{vellum_ai-0.8.21 → vellum_ai-0.8.22}/src/vellum/resources/test_suite_runs/client.py

@@ -26,6 +26,7 @@ class TestSuiteRunsClient:
         *,
         test_suite_id: str,
         exec_config: TestSuiteRunExecConfigRequest,
+        test_suite_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> TestSuiteRunRead:
         """
@@ -34,11 +35,14 @@ class TestSuiteRunsClient:
         Parameters
         ----------
         test_suite_id : str
-            The ID of the Test Suite to run
+            The ID of the Test Suite to run. Must provide either this or test_suite_id.
 
         exec_config : TestSuiteRunExecConfigRequest
             Configuration that defines how the Test Suite should be run
 
+        test_suite_name : typing.Optional[str]
+            The name of the Test Suite to run. Must provide either this or test_suite_id.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -73,6 +77,7 @@ class TestSuiteRunsClient:
             method="POST",
             json={
                 "test_suite_id": test_suite_id,
+                "test_suite_name": test_suite_name,
                 "exec_config": convert_and_respect_annotation_metadata(
                     object_=exec_config, annotation=TestSuiteRunExecConfigRequest, direction="write"
                 ),
@@ -224,6 +229,7 @@ class AsyncTestSuiteRunsClient:
         *,
         test_suite_id: str,
         exec_config: TestSuiteRunExecConfigRequest,
+        test_suite_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> TestSuiteRunRead:
         """
@@ -232,11 +238,14 @@ class AsyncTestSuiteRunsClient:
         Parameters
         ----------
         test_suite_id : str
-            The ID of the Test Suite to run
+            The ID of the Test Suite to run. Must provide either this or test_suite_id.
 
         exec_config : TestSuiteRunExecConfigRequest
             Configuration that defines how the Test Suite should be run
 
+        test_suite_name : typing.Optional[str]
+            The name of the Test Suite to run. Must provide either this or test_suite_id.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -279,6 +288,7 @@ class AsyncTestSuiteRunsClient:
             method="POST",
             json={
                 "test_suite_id": test_suite_id,
+                "test_suite_name": test_suite_name,
                 "exec_config": convert_and_respect_annotation_metadata(
                     object_=exec_config, annotation=TestSuiteRunExecConfigRequest, direction="write"
                 ),

{vellum_ai-0.8.21 → vellum_ai-0.8.22}/src/vellum/types/__init__.py

@@ -422,6 +422,8 @@ from .vellum_value_logical_condition_request import VellumValueLogicalConditionR
 from .vellum_value_logical_expression_request import VellumValueLogicalExpressionRequest
 from .vellum_value_request import VellumValueRequest
 from .vellum_variable import VellumVariable
+from .vellum_variable_extensions import VellumVariableExtensions
+from .vellum_variable_extensions_request import VellumVariableExtensionsRequest
 from .vellum_variable_request import VellumVariableRequest
 from .vellum_variable_type import VellumVariableType
 from .workflow_deployment_read import WorkflowDeploymentRead
@@ -873,6 +875,8 @@ __all__ = [
     "VellumValueLogicalExpressionRequest",
     "VellumValueRequest",
     "VellumVariable",
+    "VellumVariableExtensions",
+    "VellumVariableExtensionsRequest",
     "VellumVariableRequest",
     "VellumVariableType",
     "WorkflowDeploymentRead",

{vellum_ai-0.8.21 → vellum_ai-0.8.22}/src/vellum/types/deployment_read.py

@@ -1,6 +1,8 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
+from .array_vellum_value import ArrayVellumValue
 import datetime as dt
 import pydantic
 import typing
@@ -8,6 +10,7 @@ from .entity_status import EntityStatus
 from .environment_enum import EnvironmentEnum
 from .vellum_variable import VellumVariable
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import update_forward_refs
 
 
 class DeploymentRead(UniversalBaseModel):
@@ -65,3 +68,6 @@ class DeploymentRead(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+update_forward_refs(ArrayVellumValue, DeploymentRead=DeploymentRead)

{vellum_ai-0.8.21 → vellum_ai-0.8.22}/src/vellum/types/paginated_slim_deployment_read_list.py

@@ -1,10 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
+from .array_vellum_value import ArrayVellumValue
 import typing
 from .slim_deployment_read import SlimDeploymentRead
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
+from ..core.pydantic_utilities import update_forward_refs
 
 
 class PaginatedSlimDeploymentReadList(UniversalBaseModel):
@@ -21,3 +24,6 @@ class PaginatedSlimDeploymentReadList(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+update_forward_refs(ArrayVellumValue, PaginatedSlimDeploymentReadList=PaginatedSlimDeploymentReadList)

{vellum_ai-0.8.21 → vellum_ai-0.8.22}/src/vellum/types/paginated_slim_workflow_deployment_list.py

@@ -1,10 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
+from .array_vellum_value import ArrayVellumValue
 import typing
 from .slim_workflow_deployment import SlimWorkflowDeployment
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
+from ..core.pydantic_utilities import update_forward_refs
 
 
 class PaginatedSlimWorkflowDeploymentList(UniversalBaseModel):
@@ -21,3 +24,6 @@ class PaginatedSlimWorkflowDeploymentList(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+update_forward_refs(ArrayVellumValue, PaginatedSlimWorkflowDeploymentList=PaginatedSlimWorkflowDeploymentList)

{vellum_ai-0.8.21 → vellum_ai-0.8.22}/src/vellum/types/slim_deployment_read.py

@@ -1,6 +1,8 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
+from .array_vellum_value import ArrayVellumValue
 import datetime as dt
 import pydantic
 import typing
@@ -8,6 +10,7 @@ from .entity_status import EntityStatus
 from .environment_enum import EnvironmentEnum
 from .vellum_variable import VellumVariable
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import update_forward_refs
 
 
 class SlimDeploymentRead(UniversalBaseModel):
@@ -55,3 +58,6 @@ class SlimDeploymentRead(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+update_forward_refs(ArrayVellumValue, SlimDeploymentRead=SlimDeploymentRead)

{vellum_ai-0.8.21 → vellum_ai-0.8.22}/src/vellum/types/slim_workflow_deployment.py

@@ -1,6 +1,8 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
+from .array_vellum_value import ArrayVellumValue
 import pydantic
 import typing
 from .entity_status import EntityStatus
@@ -8,6 +10,7 @@ from .environment_enum import EnvironmentEnum
 import datetime as dt
 from .vellum_variable import VellumVariable
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import update_forward_refs
 
 
 class SlimWorkflowDeployment(UniversalBaseModel):
@@ -64,3 +67,6 @@ class SlimWorkflowDeployment(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+update_forward_refs(ArrayVellumValue, SlimWorkflowDeployment=SlimWorkflowDeployment)

{vellum_ai-0.8.21 → vellum_ai-0.8.22}/src/vellum/types/vellum_variable.py

@@ -1,16 +1,24 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
+from .array_vellum_value import ArrayVellumValue
 from .vellum_variable_type import VellumVariableType
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+from .vellum_value import VellumValue
+from .vellum_variable_extensions import VellumVariableExtensions
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
+from ..core.pydantic_utilities import update_forward_refs
 
 
 class VellumVariable(UniversalBaseModel):
     id: str
     key: str
     type: VellumVariableType
+    required: typing.Optional[bool] = None
+    default: typing.Optional[VellumValue] = None
+    extensions: typing.Optional[VellumVariableExtensions] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
@@ -20,3 +28,6 @@ class VellumVariable(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+update_forward_refs(ArrayVellumValue, VellumVariable=VellumVariable)

vellum_ai-0.8.22/src/vellum/types/vellum_variable_extensions.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 ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class VellumVariableExtensions(UniversalBaseModel):
+    """
+    A set of fields with additional properties for use in Vellum Variables.
+    """
+
+    color: typing.Optional[str] = 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

vellum_ai-0.8.22/src/vellum/types/vellum_variable_extensions_request.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 ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class VellumVariableExtensionsRequest(UniversalBaseModel):
+    """
+    A set of fields with additional properties for use in Vellum Variables.
+    """
+
+    color: typing.Optional[str] = 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

vellum_ai-0.8.22/src/vellum/types/vellum_variable_request.py

@@ -0,0 +1,33 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+from ..core.pydantic_utilities import UniversalBaseModel
+from .array_vellum_value_request import ArrayVellumValueRequest
+from .vellum_variable_type import VellumVariableType
+import typing
+from .vellum_value_request import VellumValueRequest
+from .vellum_variable_extensions_request import VellumVariableExtensionsRequest
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+from ..core.pydantic_utilities import update_forward_refs
+
+
+class VellumVariableRequest(UniversalBaseModel):
+    id: str
+    key: str
+    type: VellumVariableType
+    required: typing.Optional[bool] = None
+    default: typing.Optional[VellumValueRequest] = None
+    extensions: typing.Optional[VellumVariableExtensionsRequest] = 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
+
+
+update_forward_refs(ArrayVellumValueRequest, VellumVariableRequest=VellumVariableRequest)

{vellum_ai-0.8.21 → vellum_ai-0.8.22}/src/vellum/types/workflow_deployment_read.py

@@ -1,6 +1,8 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
+from .array_vellum_value import ArrayVellumValue
 import pydantic
 import typing
 from .entity_status import EntityStatus
@@ -8,6 +10,7 @@ from .environment_enum import EnvironmentEnum
 import datetime as dt
 from .vellum_variable import VellumVariable
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import update_forward_refs
 
 
 class WorkflowDeploymentRead(UniversalBaseModel):
@@ -69,3 +72,6 @@ class WorkflowDeploymentRead(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+update_forward_refs(ArrayVellumValue, WorkflowDeploymentRead=WorkflowDeploymentRead)

vellum_ai-0.8.21/src/vellum/types/vellum_variable_request.py

@@ -1,22 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from ..core.pydantic_utilities import UniversalBaseModel
-from .vellum_variable_type import VellumVariableType
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import typing
-import pydantic
-
-
-class VellumVariableRequest(UniversalBaseModel):
-    id: str
-    key: str
-    type: VellumVariableType
-
-    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