vellum-ai 0.9.11__py3-none-any.whl → 0.9.13__py3-none-any.whl

vellum/__init__.py

@@ -14,7 +14,7 @@ from .types import (
     ArrayChatMessageContentItem,
     ArrayChatMessageContentItemRequest,
     ArrayChatMessageContentRequest,
-    ArrayInputRequest,
+    ArrayInput,
     ArrayVariableValue,
     ArrayVariableValueItem,
     ArrayVellumValue,
@@ -30,6 +30,7 @@ from .types import (
     BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request,
     BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1,
     BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request,
+    ChatHistoryInput,
     ChatHistoryInputRequest,
     ChatHistoryVariableValue,
     ChatHistoryVellumValue,
@@ -51,11 +52,11 @@ from .types import (
     CodeExecutionNodeResultOutput,
     CodeExecutionNodeSearchResultsResult,
     CodeExecutionNodeStringResult,
-    CodeExecutionPackageRequest,
+    CodeExecutionPackage,
     CodeExecutionRuntime,
-    CodeExecutorInputRequest,
+    CodeExecutorInput,
     CodeExecutorResponse,
-    CodeExecutorSecretInputRequest,
+    CodeExecutorSecretInput,
     CompilePromptDeploymentExpandMetaRequest,
     CompilePromptMeta,
     ComponentsSchemasPdfSearchResultMetaSource,
@@ -77,6 +78,7 @@ from .types import (
     DocumentIndexIndexingConfig,
     DocumentIndexIndexingConfigRequest,
     DocumentIndexRead,
+    DocumentProcessingState,
     DocumentRead,
     DocumentStatus,
     EnrichedNormalizedCompletion,
@@ -85,7 +87,7 @@ from .types import (
     EnvironmentEnum,
     EphemeralPromptCacheConfig,
     EphemeralPromptCacheConfigTypeEnum,
-    ErrorInputRequest,
+    ErrorInput,
     ErrorVariableValue,
     ErrorVellumValue,
     ErrorVellumValueRequest,
@@ -128,12 +130,12 @@ from .types import (
     FunctionCallChatMessageContentRequest,
     FunctionCallChatMessageContentValue,
     FunctionCallChatMessageContentValueRequest,
-    FunctionCallInputRequest,
+    FunctionCallInput,
     FunctionCallRequest,
     FunctionCallVariableValue,
     FunctionCallVellumValue,
     FunctionCallVellumValueRequest,
-    FunctionDefinitionPromptBlock,
+    FunctionDefinition,
     GenerateOptionsRequest,
     GenerateRequest,
     GenerateResponse,
@@ -167,6 +169,7 @@ from .types import (
     InstructorVectorizerConfigRequest,
     IterationStateEnum,
     JinjaPromptBlock,
+    JsonInput,
     JsonInputRequest,
     JsonVariableValue,
     JsonVellumValue,
@@ -182,7 +185,7 @@ from .types import (
     MetadataFilterRuleRequest,
     MetadataFiltersRequest,
     MetricDefinitionExecution,
-    MetricDefinitionInputRequest,
+    MetricDefinitionInput,
     MetricNodeResult,
     MlModelUsage,
     NamedScenarioInputChatHistoryVariableValueRequest,
@@ -227,7 +230,7 @@ from .types import (
     NodeOutputCompiledValue,
     NormalizedLogProbs,
     NormalizedTokenLogProbs,
-    NumberInputRequest,
+    NumberInput,
     NumberVariableValue,
     NumberVellumValue,
     NumberVellumValueRequest,
@@ -254,7 +257,6 @@ from .types import (
     PlainTextPromptBlock,
     Price,
     ProcessingFailureReasonEnum,
-    ProcessingStateEnum,
     PromptBlock,
     PromptBlockState,
     PromptDeploymentExpandMetaRequest,
@@ -302,7 +304,7 @@ from .types import (
     SearchResultMeta,
     SearchResultMetaRequest,
     SearchResultRequest,
-    SearchResultsInputRequest,
+    SearchResultsInput,
     SearchResultsVariableValue,
     SearchResultsVellumValue,
     SearchResultsVellumValueRequest,
@@ -321,6 +323,7 @@ from .types import (
     StreamingWorkflowNodeResultEvent,
     StringChatMessageContent,
     StringChatMessageContentRequest,
+    StringInput,
     StringInputRequest,
     StringVariableValue,
     StringVellumValue,
@@ -520,7 +523,7 @@ __all__ = [
     "ArrayChatMessageContentItem",
     "ArrayChatMessageContentItemRequest",
     "ArrayChatMessageContentRequest",
-    "ArrayInputRequest",
+    "ArrayInput",
     "ArrayVariableValue",
     "ArrayVariableValueItem",
     "ArrayVellumValue",
@@ -538,6 +541,7 @@ __all__ = [
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request",
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1",
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request",
+    "ChatHistoryInput",
     "ChatHistoryInputRequest",
     "ChatHistoryVariableValue",
     "ChatHistoryVellumValue",
@@ -559,11 +563,11 @@ __all__ = [
     "CodeExecutionNodeResultOutput",
     "CodeExecutionNodeSearchResultsResult",
     "CodeExecutionNodeStringResult",
-    "CodeExecutionPackageRequest",
+    "CodeExecutionPackage",
     "CodeExecutionRuntime",
-    "CodeExecutorInputRequest",
+    "CodeExecutorInput",
     "CodeExecutorResponse",
-    "CodeExecutorSecretInputRequest",
+    "CodeExecutorSecretInput",
     "CompilePromptDeploymentExpandMetaRequest",
     "CompilePromptMeta",
     "ComponentsSchemasPdfSearchResultMetaSource",
@@ -587,6 +591,7 @@ __all__ = [
     "DocumentIndexIndexingConfigRequest",
     "DocumentIndexRead",
     "DocumentIndexesListRequestStatus",
+    "DocumentProcessingState",
     "DocumentRead",
     "DocumentStatus",
     "EnrichedNormalizedCompletion",
@@ -595,7 +600,7 @@ __all__ = [
     "EnvironmentEnum",
     "EphemeralPromptCacheConfig",
     "EphemeralPromptCacheConfigTypeEnum",
-    "ErrorInputRequest",
+    "ErrorInput",
     "ErrorVariableValue",
     "ErrorVellumValue",
     "ErrorVellumValueRequest",
@@ -640,12 +645,12 @@ __all__ = [
     "FunctionCallChatMessageContentRequest",
     "FunctionCallChatMessageContentValue",
     "FunctionCallChatMessageContentValueRequest",
-    "FunctionCallInputRequest",
+    "FunctionCallInput",
     "FunctionCallRequest",
     "FunctionCallVariableValue",
     "FunctionCallVellumValue",
     "FunctionCallVellumValueRequest",
-    "FunctionDefinitionPromptBlock",
+    "FunctionDefinition",
     "GenerateOptionsRequest",
     "GenerateRequest",
     "GenerateResponse",
@@ -680,6 +685,7 @@ __all__ = [
     "InternalServerError",
     "IterationStateEnum",
     "JinjaPromptBlock",
+    "JsonInput",
     "JsonInputRequest",
     "JsonVariableValue",
     "JsonVellumValue",
@@ -697,7 +703,7 @@ __all__ = [
     "MetadataFilterRuleRequest",
     "MetadataFiltersRequest",
     "MetricDefinitionExecution",
-    "MetricDefinitionInputRequest",
+    "MetricDefinitionInput",
     "MetricNodeResult",
     "MlModelUsage",
     "NamedScenarioInputChatHistoryVariableValueRequest",
@@ -743,7 +749,7 @@ __all__ = [
     "NormalizedLogProbs",
     "NormalizedTokenLogProbs",
     "NotFoundError",
-    "NumberInputRequest",
+    "NumberInput",
     "NumberVariableValue",
     "NumberVellumValue",
     "NumberVellumValueRequest",
@@ -770,7 +776,6 @@ __all__ = [
     "PlainTextPromptBlock",
     "Price",
     "ProcessingFailureReasonEnum",
-    "ProcessingStateEnum",
     "PromptBlock",
     "PromptBlockState",
     "PromptDeploymentExpandMetaRequest",
@@ -818,7 +823,7 @@ __all__ = [
     "SearchResultMeta",
     "SearchResultMetaRequest",
     "SearchResultRequest",
-    "SearchResultsInputRequest",
+    "SearchResultsInput",
     "SearchResultsVariableValue",
     "SearchResultsVellumValue",
     "SearchResultsVellumValueRequest",
@@ -837,6 +842,7 @@ __all__ = [
     "StreamingWorkflowNodeResultEvent",
     "StringChatMessageContent",
     "StringChatMessageContentRequest",
+    "StringInput",
     "StringInputRequest",
     "StringVariableValue",
     "StringVellumValue",

vellum/client.py

@@ -19,8 +19,8 @@ from .resources.workflow_sandboxes.client import WorkflowSandboxesClient
 from .resources.workflows.client import WorkflowsClient
 from .resources.workspace_secrets.client import WorkspaceSecretsClient
 from .types.code_execution_runtime import CodeExecutionRuntime
-from .types.code_executor_input_request import CodeExecutorInputRequest
-from .types.code_execution_package_request import CodeExecutionPackageRequest
+from .types.code_executor_input import CodeExecutorInput
+from .types.code_execution_package import CodeExecutionPackage
 from .types.vellum_variable_type import VellumVariableType
 from .core.request_options import RequestOptions
 from .types.code_executor_response import CodeExecutorResponse
@@ -145,8 +145,8 @@ class Vellum:
         *,
         code: str,
         runtime: CodeExecutionRuntime,
-        input_values: typing.Sequence[CodeExecutorInputRequest],
-        packages: typing.Sequence[CodeExecutionPackageRequest],
+        input_values: typing.Sequence[CodeExecutorInput],
+        packages: typing.Sequence[CodeExecutionPackage],
         output_type: VellumVariableType,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> CodeExecutorResponse:
@@ -159,9 +159,9 @@ class Vellum:
 
         runtime : CodeExecutionRuntime
 
-        input_values : typing.Sequence[CodeExecutorInputRequest]
+        input_values : typing.Sequence[CodeExecutorInput]
 
-        packages : typing.Sequence[CodeExecutionPackageRequest]
+        packages : typing.Sequence[CodeExecutionPackage]
 
         output_type : VellumVariableType
 
@@ -175,7 +175,7 @@ class Vellum:
 
         Examples
         --------
-        from vellum import CodeExecutionPackageRequest, StringInputRequest, Vellum
+        from vellum import CodeExecutionPackage, StringInput, Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -184,13 +184,13 @@ class Vellum:
             code="code",
             runtime="PYTHON_3_11_6",
             input_values=[
-                StringInputRequest(
+                StringInput(
                     name="name",
                     value="value",
                 )
             ],
             packages=[
-                CodeExecutionPackageRequest(
+                CodeExecutionPackage(
                     version="version",
                     name="name",
                 )
@@ -206,10 +206,10 @@ class Vellum:
                 "code": code,
                 "runtime": runtime,
                 "input_values": convert_and_respect_annotation_metadata(
-                    object_=input_values, annotation=typing.Sequence[CodeExecutorInputRequest], direction="write"
+                    object_=input_values, annotation=typing.Sequence[CodeExecutorInput], direction="write"
                 ),
                 "packages": convert_and_respect_annotation_metadata(
-                    object_=packages, annotation=typing.Sequence[CodeExecutionPackageRequest], direction="write"
+                    object_=packages, annotation=typing.Sequence[CodeExecutionPackage], direction="write"
                 ),
                 "output_type": output_type,
             },
@@ -1464,8 +1464,8 @@ class AsyncVellum:
         *,
         code: str,
         runtime: CodeExecutionRuntime,
-        input_values: typing.Sequence[CodeExecutorInputRequest],
-        packages: typing.Sequence[CodeExecutionPackageRequest],
+        input_values: typing.Sequence[CodeExecutorInput],
+        packages: typing.Sequence[CodeExecutionPackage],
         output_type: VellumVariableType,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> CodeExecutorResponse:
@@ -1478,9 +1478,9 @@ class AsyncVellum:
 
         runtime : CodeExecutionRuntime
 
-        input_values : typing.Sequence[CodeExecutorInputRequest]
+        input_values : typing.Sequence[CodeExecutorInput]
 
-        packages : typing.Sequence[CodeExecutionPackageRequest]
+        packages : typing.Sequence[CodeExecutionPackage]
 
         output_type : VellumVariableType
 
@@ -1496,7 +1496,7 @@ class AsyncVellum:
         --------
         import asyncio
 
-        from vellum import AsyncVellum, CodeExecutionPackageRequest, StringInputRequest
+        from vellum import AsyncVellum, CodeExecutionPackage, StringInput
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -1508,13 +1508,13 @@ class AsyncVellum:
                 code="code",
                 runtime="PYTHON_3_11_6",
                 input_values=[
-                    StringInputRequest(
+                    StringInput(
                         name="name",
                         value="value",
                     )
                 ],
                 packages=[
-                    CodeExecutionPackageRequest(
+                    CodeExecutionPackage(
                         version="version",
                         name="name",
                     )
@@ -1533,10 +1533,10 @@ class AsyncVellum:
                 "code": code,
                 "runtime": runtime,
                 "input_values": convert_and_respect_annotation_metadata(
-                    object_=input_values, annotation=typing.Sequence[CodeExecutorInputRequest], direction="write"
+                    object_=input_values, annotation=typing.Sequence[CodeExecutorInput], direction="write"
                 ),
                 "packages": convert_and_respect_annotation_metadata(
-                    object_=packages, annotation=typing.Sequence[CodeExecutionPackageRequest], direction="write"
+                    object_=packages, annotation=typing.Sequence[CodeExecutionPackage], direction="write"
                 ),
                 "output_type": output_type,
             },

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.11",
+            "X-Fern-SDK-Version": "0.9.13",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

vellum/resources/ad_hoc/client.py

@@ -7,6 +7,7 @@ from ...types.vellum_variable import VellumVariable
 from ...types.prompt_parameters import PromptParameters
 from ...types.prompt_block import PromptBlock
 from ...types.prompt_settings import PromptSettings
+from ...types.function_definition import FunctionDefinition
 from ...types.ad_hoc_expand_meta import AdHocExpandMeta
 from ...core.request_options import RequestOptions
 from ...types.ad_hoc_execute_prompt_event import AdHocExecutePromptEvent
@@ -37,6 +38,7 @@ class AdHocClient:
         parameters: PromptParameters,
         blocks: typing.Sequence[PromptBlock],
         settings: typing.Optional[PromptSettings] = OMIT,
+        functions: typing.Optional[typing.Sequence[FunctionDefinition]] = OMIT,
         expand_meta: typing.Optional[AdHocExpandMeta] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.Iterator[AdHocExecutePromptEvent]:
@@ -57,6 +59,8 @@ class AdHocClient:
 
         settings : typing.Optional[PromptSettings]
 
+        functions : typing.Optional[typing.Sequence[FunctionDefinition]]
+
         expand_meta : typing.Optional[AdHocExpandMeta]
 
         request_options : typing.Optional[RequestOptions]
@@ -72,6 +76,7 @@ class AdHocClient:
         from vellum import (
             AdHocExpandMeta,
             EphemeralPromptCacheConfig,
+            FunctionDefinition,
             JinjaPromptBlock,
             PromptParameters,
             PromptRequestStringInput,
@@ -100,7 +105,7 @@ class AdHocClient:
                     type="STRING",
                     required=True,
                     default=StringVellumValue(
-                        value={"key": "value"},
+                        value="string",
                     ),
                     extensions=VellumVariableExtensions(
                         color={"key": "value"},
@@ -128,6 +133,17 @@ class AdHocClient:
                     template="string",
                 )
             ],
+            functions=[
+                FunctionDefinition(
+                    state="ENABLED",
+                    cache_config=EphemeralPromptCacheConfig(),
+                    name="string",
+                    description="string",
+                    parameters={"string": {"key": "value"}},
+                    forced=True,
+                    strict=True,
+                )
+            ],
             expand_meta=AdHocExpandMeta(
                 cost=True,
                 model_name=True,
@@ -159,6 +175,9 @@ class AdHocClient:
                 "blocks": convert_and_respect_annotation_metadata(
                     object_=blocks, annotation=typing.Sequence[PromptBlock], direction="write"
                 ),
+                "functions": convert_and_respect_annotation_metadata(
+                    object_=functions, annotation=typing.Sequence[FunctionDefinition], direction="write"
+                ),
                 "expand_meta": convert_and_respect_annotation_metadata(
                     object_=expand_meta, annotation=AdHocExpandMeta, direction="write"
                 ),
@@ -232,6 +251,7 @@ class AsyncAdHocClient:
         parameters: PromptParameters,
         blocks: typing.Sequence[PromptBlock],
         settings: typing.Optional[PromptSettings] = OMIT,
+        functions: typing.Optional[typing.Sequence[FunctionDefinition]] = OMIT,
         expand_meta: typing.Optional[AdHocExpandMeta] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.AsyncIterator[AdHocExecutePromptEvent]:
@@ -252,6 +272,8 @@ class AsyncAdHocClient:
 
         settings : typing.Optional[PromptSettings]
 
+        functions : typing.Optional[typing.Sequence[FunctionDefinition]]
+
         expand_meta : typing.Optional[AdHocExpandMeta]
 
         request_options : typing.Optional[RequestOptions]
@@ -270,6 +292,7 @@ class AsyncAdHocClient:
             AdHocExpandMeta,
             AsyncVellum,
             EphemeralPromptCacheConfig,
+            FunctionDefinition,
             JinjaPromptBlock,
             PromptParameters,
             PromptRequestStringInput,
@@ -300,7 +323,7 @@ class AsyncAdHocClient:
                         type="STRING",
                         required=True,
                         default=StringVellumValue(
-                            value={"key": "value"},
+                            value="string",
                         ),
                         extensions=VellumVariableExtensions(
                             color={"key": "value"},
@@ -328,6 +351,17 @@ class AsyncAdHocClient:
                         template="string",
                     )
                 ],
+                functions=[
+                    FunctionDefinition(
+                        state="ENABLED",
+                        cache_config=EphemeralPromptCacheConfig(),
+                        name="string",
+                        description="string",
+                        parameters={"string": {"key": "value"}},
+                        forced=True,
+                        strict=True,
+                    )
+                ],
                 expand_meta=AdHocExpandMeta(
                     cost=True,
                     model_name=True,
@@ -362,6 +396,9 @@ class AsyncAdHocClient:
                 "blocks": convert_and_respect_annotation_metadata(
                     object_=blocks, annotation=typing.Sequence[PromptBlock], direction="write"
                 ),
+                "functions": convert_and_respect_annotation_metadata(
+                    object_=functions, annotation=typing.Sequence[FunctionDefinition], direction="write"
+                ),
                 "expand_meta": convert_and_respect_annotation_metadata(
                     object_=expand_meta, annotation=AdHocExpandMeta, direction="write"
                 ),

vellum/resources/metric_definitions/client.py

@@ -2,7 +2,7 @@
 
 import typing
 from ...core.client_wrapper import SyncClientWrapper
-from ...types.metric_definition_input_request import MetricDefinitionInputRequest
+from ...types.metric_definition_input import MetricDefinitionInput
 from ...core.request_options import RequestOptions
 from ...types.metric_definition_execution import MetricDefinitionExecution
 from ...core.jsonable_encoder import jsonable_encoder
@@ -24,7 +24,7 @@ class MetricDefinitionsClient:
         self,
         id: str,
         *,
-        inputs: typing.Sequence[MetricDefinitionInputRequest],
+        inputs: typing.Sequence[MetricDefinitionInput],
         release_tag: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> MetricDefinitionExecution:
@@ -36,7 +36,7 @@ class MetricDefinitionsClient:
         id : str
             Either the Metric Definition's ID or its unique name
 
-        inputs : typing.Sequence[MetricDefinitionInputRequest]
+        inputs : typing.Sequence[MetricDefinitionInput]
 
         release_tag : typing.Optional[str]
 
@@ -50,7 +50,7 @@ class MetricDefinitionsClient:
 
         Examples
         --------
-        from vellum import StringInputRequest, Vellum
+        from vellum import StringInput, Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -58,7 +58,7 @@ class MetricDefinitionsClient:
         client.metric_definitions.execute_metric_definition(
             id="id",
             inputs=[
-                StringInputRequest(
+                StringInput(
                     name="name",
                     value="value",
                 )
@@ -71,7 +71,7 @@ class MetricDefinitionsClient:
             method="POST",
             json={
                 "inputs": convert_and_respect_annotation_metadata(
-                    object_=inputs, annotation=typing.Sequence[MetricDefinitionInputRequest], direction="write"
+                    object_=inputs, annotation=typing.Sequence[MetricDefinitionInput], direction="write"
                 ),
                 "release_tag": release_tag,
             },
@@ -101,7 +101,7 @@ class AsyncMetricDefinitionsClient:
         self,
         id: str,
         *,
-        inputs: typing.Sequence[MetricDefinitionInputRequest],
+        inputs: typing.Sequence[MetricDefinitionInput],
         release_tag: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> MetricDefinitionExecution:
@@ -113,7 +113,7 @@ class AsyncMetricDefinitionsClient:
         id : str
             Either the Metric Definition's ID or its unique name
 
-        inputs : typing.Sequence[MetricDefinitionInputRequest]
+        inputs : typing.Sequence[MetricDefinitionInput]
 
         release_tag : typing.Optional[str]
 
@@ -129,7 +129,7 @@ class AsyncMetricDefinitionsClient:
         --------
         import asyncio
 
-        from vellum import AsyncVellum, StringInputRequest
+        from vellum import AsyncVellum, StringInput
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -140,7 +140,7 @@ class AsyncMetricDefinitionsClient:
             await client.metric_definitions.execute_metric_definition(
                 id="id",
                 inputs=[
-                    StringInputRequest(
+                    StringInput(
                         name="name",
                         value="value",
                     )
@@ -156,7 +156,7 @@ class AsyncMetricDefinitionsClient:
             method="POST",
             json={
                 "inputs": convert_and_respect_annotation_metadata(
-                    object_=inputs, annotation=typing.Sequence[MetricDefinitionInputRequest], direction="write"
+                    object_=inputs, annotation=typing.Sequence[MetricDefinitionInput], direction="write"
                 ),
                 "release_tag": release_tag,
             },