vellum-ai 0.7.1__py3-none-any.whl → 0.7.2__py3-none-any.whl

vellum/__init__.py

@@ -23,6 +23,13 @@ from .types import (
     ArrayVariableValueItem_Number,
     ArrayVariableValueItem_String,
     ArrayVellumValueItem,
+    ArrayVellumValueItemRequest,
+    ArrayVellumValueItemRequest_Error,
+    ArrayVellumValueItemRequest_FunctionCall,
+    ArrayVellumValueItemRequest_Image,
+    ArrayVellumValueItemRequest_Json,
+    ArrayVellumValueItemRequest_Number,
+    ArrayVellumValueItemRequest_String,
     ArrayVellumValueItem_Error,
     ArrayVellumValueItem_FunctionCall,
     ArrayVellumValueItem_Image,
@@ -101,6 +108,7 @@ from .types import (
     ErrorEnum,
     ErrorVariableValue,
     ErrorVellumValue,
+    ErrorVellumValueRequest,
     ExecutePromptApiErrorResponse,
     ExecutePromptEvent,
     ExecutePromptEvent_Fulfilled,
@@ -151,6 +159,7 @@ from .types import (
     FunctionCallRequest,
     FunctionCallVariableValue,
     FunctionCallVellumValue,
+    FunctionCallVellumValueRequest,
     GenerateErrorResponse,
     GenerateOptionsRequest,
     GenerateRequest,
@@ -169,6 +178,7 @@ from .types import (
     ImageEnum,
     ImageVariableValue,
     ImageVellumValue,
+    ImageVellumValueRequest,
     IndexingConfigVectorizer,
     IndexingConfigVectorizerRequest,
     IndexingConfigVectorizerRequest_HkunlpInstructorXl,
@@ -198,6 +208,7 @@ from .types import (
     JsonInputRequest,
     JsonVariableValue,
     JsonVellumValue,
+    JsonVellumValueRequest,
     LogicalOperator,
     LogprobsEnum,
     MapEnum,
@@ -217,6 +228,8 @@ from .types import (
     NamedScenarioInputRequest_ChatHistory,
     NamedScenarioInputRequest_String,
     NamedScenarioInputStringVariableValueRequest,
+    NamedTestCaseArrayVariableValue,
+    NamedTestCaseArrayVariableValueRequest,
     NamedTestCaseChatHistoryVariableValue,
     NamedTestCaseChatHistoryVariableValueRequest,
     NamedTestCaseErrorVariableValue,
@@ -233,6 +246,7 @@ from .types import (
     NamedTestCaseStringVariableValueRequest,
     NamedTestCaseVariableValue,
     NamedTestCaseVariableValueRequest,
+    NamedTestCaseVariableValueRequest_Array,
     NamedTestCaseVariableValueRequest_ChatHistory,
     NamedTestCaseVariableValueRequest_Error,
     NamedTestCaseVariableValueRequest_FunctionCall,
@@ -240,6 +254,7 @@ from .types import (
     NamedTestCaseVariableValueRequest_Number,
     NamedTestCaseVariableValueRequest_SearchResults,
     NamedTestCaseVariableValueRequest_String,
+    NamedTestCaseVariableValue_Array,
     NamedTestCaseVariableValue_ChatHistory,
     NamedTestCaseVariableValue_Error,
     NamedTestCaseVariableValue_FunctionCall,
@@ -286,6 +301,7 @@ from .types import (
     NumberEnum,
     NumberVariableValue,
     NumberVellumValue,
+    NumberVellumValueRequest,
     OpenAiVectorizerConfig,
     OpenAiVectorizerConfigRequest,
     OpenAiVectorizerTextEmbedding3Large,
@@ -311,6 +327,7 @@ from .types import (
     PromptDeploymentInputRequest_Json,
     PromptDeploymentInputRequest_String,
     PromptExecutionMeta,
+    PromptNodeExecutionMeta,
     PromptNodeResult,
     PromptNodeResultData,
     PromptOutput,
@@ -379,6 +396,7 @@ from .types import (
     StringInputRequest,
     StringVariableValue,
     StringVellumValue,
+    StringVellumValueRequest,
     SubmitCompletionActualRequest,
     SubmitCompletionActualsErrorResponse,
     SubmitWorkflowExecutionActualRequest,
@@ -426,6 +444,7 @@ from .types import (
     TerminalNodeResultOutput_String,
     TerminalNodeSearchResultsResult,
     TerminalNodeStringResult,
+    TestCaseArrayVariableValue,
     TestCaseChatHistoryVariableValue,
     TestCaseErrorVariableValue,
     TestCaseFunctionCallVariableValue,
@@ -434,6 +453,7 @@ from .types import (
     TestCaseSearchResultsVariableValue,
     TestCaseStringVariableValue,
     TestCaseVariableValue,
+    TestCaseVariableValue_Array,
     TestCaseVariableValue_ChatHistory,
     TestCaseVariableValue_Error,
     TestCaseVariableValue_FunctionCall,
@@ -548,6 +568,7 @@ from .types import (
     WorkflowExecutionEventType,
     WorkflowExecutionNodeResultEvent,
     WorkflowExecutionWorkflowResultEvent,
+    WorkflowExpandMetaRequest,
     WorkflowNodeResultData,
     WorkflowNodeResultData_Api,
     WorkflowNodeResultData_CodeExecution,
@@ -659,6 +680,13 @@ __all__ = [
     "ArrayVariableValueItem_Number",
     "ArrayVariableValueItem_String",
     "ArrayVellumValueItem",
+    "ArrayVellumValueItemRequest",
+    "ArrayVellumValueItemRequest_Error",
+    "ArrayVellumValueItemRequest_FunctionCall",
+    "ArrayVellumValueItemRequest_Image",
+    "ArrayVellumValueItemRequest_Json",
+    "ArrayVellumValueItemRequest_Number",
+    "ArrayVellumValueItemRequest_String",
     "ArrayVellumValueItem_Error",
     "ArrayVellumValueItem_FunctionCall",
     "ArrayVellumValueItem_Image",
@@ -740,6 +768,7 @@ __all__ = [
     "ErrorEnum",
     "ErrorVariableValue",
     "ErrorVellumValue",
+    "ErrorVellumValueRequest",
     "ExecutePromptApiErrorResponse",
     "ExecutePromptEvent",
     "ExecutePromptEvent_Fulfilled",
@@ -791,6 +820,7 @@ __all__ = [
     "FunctionCallRequest",
     "FunctionCallVariableValue",
     "FunctionCallVellumValue",
+    "FunctionCallVellumValueRequest",
     "GenerateErrorResponse",
     "GenerateOptionsRequest",
     "GenerateRequest",
@@ -809,6 +839,7 @@ __all__ = [
     "ImageEnum",
     "ImageVariableValue",
     "ImageVellumValue",
+    "ImageVellumValueRequest",
     "IndexingConfigVectorizer",
     "IndexingConfigVectorizerRequest",
     "IndexingConfigVectorizerRequest_HkunlpInstructorXl",
@@ -839,6 +870,7 @@ __all__ = [
     "JsonInputRequest",
     "JsonVariableValue",
     "JsonVellumValue",
+    "JsonVellumValueRequest",
     "LogicalOperator",
     "LogprobsEnum",
     "MapEnum",
@@ -858,6 +890,8 @@ __all__ = [
     "NamedScenarioInputRequest_ChatHistory",
     "NamedScenarioInputRequest_String",
     "NamedScenarioInputStringVariableValueRequest",
+    "NamedTestCaseArrayVariableValue",
+    "NamedTestCaseArrayVariableValueRequest",
     "NamedTestCaseChatHistoryVariableValue",
     "NamedTestCaseChatHistoryVariableValueRequest",
     "NamedTestCaseErrorVariableValue",
@@ -874,6 +908,7 @@ __all__ = [
     "NamedTestCaseStringVariableValueRequest",
     "NamedTestCaseVariableValue",
     "NamedTestCaseVariableValueRequest",
+    "NamedTestCaseVariableValueRequest_Array",
     "NamedTestCaseVariableValueRequest_ChatHistory",
     "NamedTestCaseVariableValueRequest_Error",
     "NamedTestCaseVariableValueRequest_FunctionCall",
@@ -881,6 +916,7 @@ __all__ = [
     "NamedTestCaseVariableValueRequest_Number",
     "NamedTestCaseVariableValueRequest_SearchResults",
     "NamedTestCaseVariableValueRequest_String",
+    "NamedTestCaseVariableValue_Array",
     "NamedTestCaseVariableValue_ChatHistory",
     "NamedTestCaseVariableValue_Error",
     "NamedTestCaseVariableValue_FunctionCall",
@@ -928,6 +964,7 @@ __all__ = [
     "NumberEnum",
     "NumberVariableValue",
     "NumberVellumValue",
+    "NumberVellumValueRequest",
     "OpenAiVectorizerConfig",
     "OpenAiVectorizerConfigRequest",
     "OpenAiVectorizerTextEmbedding3Large",
@@ -953,6 +990,7 @@ __all__ = [
     "PromptDeploymentInputRequest_Json",
     "PromptDeploymentInputRequest_String",
     "PromptExecutionMeta",
+    "PromptNodeExecutionMeta",
     "PromptNodeResult",
     "PromptNodeResultData",
     "PromptOutput",
@@ -1021,6 +1059,7 @@ __all__ = [
     "StringInputRequest",
     "StringVariableValue",
     "StringVellumValue",
+    "StringVellumValueRequest",
     "SubmitCompletionActualRequest",
     "SubmitCompletionActualsErrorResponse",
     "SubmitWorkflowExecutionActualRequest",
@@ -1068,6 +1107,7 @@ __all__ = [
     "TerminalNodeResultOutput_String",
     "TerminalNodeSearchResultsResult",
     "TerminalNodeStringResult",
+    "TestCaseArrayVariableValue",
     "TestCaseChatHistoryVariableValue",
     "TestCaseErrorVariableValue",
     "TestCaseFunctionCallVariableValue",
@@ -1076,6 +1116,7 @@ __all__ = [
     "TestCaseSearchResultsVariableValue",
     "TestCaseStringVariableValue",
     "TestCaseVariableValue",
+    "TestCaseVariableValue_Array",
     "TestCaseVariableValue_ChatHistory",
     "TestCaseVariableValue_Error",
     "TestCaseVariableValue_FunctionCall",
@@ -1192,6 +1233,7 @@ __all__ = [
     "WorkflowExecutionEventType",
     "WorkflowExecutionNodeResultEvent",
     "WorkflowExecutionWorkflowResultEvent",
+    "WorkflowExpandMetaRequest",
     "WorkflowNodeResultData",
     "WorkflowNodeResultData_Api",
     "WorkflowNodeResultData_CodeExecution",

vellum/client.py

@@ -42,6 +42,7 @@ from .types.search_response import SearchResponse
 from .types.submit_completion_actual_request import SubmitCompletionActualRequest
 from .types.submit_workflow_execution_actual_request import SubmitWorkflowExecutionActualRequest
 from .types.workflow_execution_event_type import WorkflowExecutionEventType
+from .types.workflow_expand_meta_request import WorkflowExpandMetaRequest
 from .types.workflow_request_input_request import WorkflowRequestInputRequest
 from .types.workflow_stream_event import WorkflowStreamEvent
 
@@ -164,11 +165,11 @@ class Vellum:
             external_id="string",
             expand_meta=PromptDeploymentExpandMetaRequestRequest(
                 model_name=True,
+                usage=True,
+                finish_reason=True,
                 latency=True,
                 deployment_release_tag=True,
                 prompt_version_id=True,
-                finish_reason=True,
-                usage=True,
             ),
             raw_overrides=RawPromptExecutionOverridesRequest(
                 body={"string": {"key": "value"}},
@@ -299,11 +300,11 @@ class Vellum:
             external_id="string",
             expand_meta=PromptDeploymentExpandMetaRequestRequest(
                 model_name=True,
+                usage=True,
+                finish_reason=True,
                 latency=True,
                 deployment_release_tag=True,
                 prompt_version_id=True,
-                finish_reason=True,
-                usage=True,
             ),
             raw_overrides=RawPromptExecutionOverridesRequest(
                 body={"string": {"key": "value"}},
@@ -382,6 +383,7 @@ class Vellum:
         self,
         *,
         inputs: typing.Sequence[WorkflowRequestInputRequest],
+        expand_meta: typing.Optional[WorkflowExpandMetaRequest] = OMIT,
         workflow_deployment_id: typing.Optional[str] = OMIT,
         workflow_deployment_name: typing.Optional[str] = OMIT,
         release_tag: typing.Optional[str] = OMIT,
@@ -394,6 +396,8 @@ class Vellum:
         Parameters:
             - inputs: typing.Sequence[WorkflowRequestInputRequest]. The list of inputs defined in the Workflow's Deployment with their corresponding values.
 
+            - expand_meta: typing.Optional[WorkflowExpandMetaRequest]. An optionally specified configuration used to opt in to including additional metadata about this workflow execution in the API response. Corresponding values will be returned under the `execution_meta` key within NODE events in the response stream.
+
             - workflow_deployment_id: typing.Optional[str]. The ID of the Workflow Deployment. Must provide either this or workflow_deployment_name.
 
             - workflow_deployment_name: typing.Optional[str]. The name of the Workflow Deployment. Must provide either this or workflow_deployment_id.
@@ -404,7 +408,7 @@ class Vellum:
 
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
-        from vellum import WorkflowRequestInputRequest_String
+        from vellum import WorkflowExpandMetaRequest, WorkflowRequestInputRequest_String
         from vellum.client import Vellum
 
         client = Vellum(
@@ -417,6 +421,9 @@ class Vellum:
                     value="string",
                 )
             ],
+            expand_meta=WorkflowExpandMetaRequest(
+                usage=True,
+            ),
             workflow_deployment_id="string",
             workflow_deployment_name="string",
             release_tag="string",
@@ -424,6 +431,8 @@ class Vellum:
         )
         """
         _request: typing.Dict[str, typing.Any] = {"inputs": inputs}
+        if expand_meta is not OMIT:
+            _request["expand_meta"] = expand_meta
         if workflow_deployment_id is not OMIT:
             _request["workflow_deployment_id"] = workflow_deployment_id
         if workflow_deployment_name is not OMIT:
@@ -476,6 +485,7 @@ class Vellum:
         self,
         *,
         inputs: typing.Sequence[WorkflowRequestInputRequest],
+        expand_meta: typing.Optional[WorkflowExpandMetaRequest] = OMIT,
         workflow_deployment_id: typing.Optional[str] = OMIT,
         workflow_deployment_name: typing.Optional[str] = OMIT,
         release_tag: typing.Optional[str] = OMIT,
@@ -489,6 +499,8 @@ class Vellum:
         Parameters:
             - inputs: typing.Sequence[WorkflowRequestInputRequest]. The list of inputs defined in the Workflow's Deployment with their corresponding values.
 
+            - expand_meta: typing.Optional[WorkflowExpandMetaRequest]. An optionally specified configuration used to opt in to including additional metadata about this workflow execution in the API response. Corresponding values will be returned under the `execution_meta` key within NODE events in the response stream.
+
             - workflow_deployment_id: typing.Optional[str]. The ID of the Workflow Deployment. Must provide either this or workflow_deployment_name.
 
             - workflow_deployment_name: typing.Optional[str]. The name of the Workflow Deployment. Must provide either this or workflow_deployment_id.
@@ -501,7 +513,7 @@ class Vellum:
 
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
-        from vellum import WorkflowRequestInputRequest_String
+        from vellum import WorkflowExpandMetaRequest, WorkflowRequestInputRequest_String
         from vellum.client import Vellum
 
         client = Vellum(
@@ -514,6 +526,9 @@ class Vellum:
                     value="string",
                 )
             ],
+            expand_meta=WorkflowExpandMetaRequest(
+                usage=True,
+            ),
             workflow_deployment_id="string",
             workflow_deployment_name="string",
             release_tag="string",
@@ -522,6 +537,8 @@ class Vellum:
         )
         """
         _request: typing.Dict[str, typing.Any] = {"inputs": inputs}
+        if expand_meta is not OMIT:
+            _request["expand_meta"] = expand_meta
         if workflow_deployment_id is not OMIT:
             _request["workflow_deployment_id"] = workflow_deployment_id
         if workflow_deployment_name is not OMIT:
@@ -1125,11 +1142,11 @@ class AsyncVellum:
             external_id="string",
             expand_meta=PromptDeploymentExpandMetaRequestRequest(
                 model_name=True,
+                usage=True,
+                finish_reason=True,
                 latency=True,
                 deployment_release_tag=True,
                 prompt_version_id=True,
-                finish_reason=True,
-                usage=True,
             ),
             raw_overrides=RawPromptExecutionOverridesRequest(
                 body={"string": {"key": "value"}},
@@ -1260,11 +1277,11 @@ class AsyncVellum:
             external_id="string",
             expand_meta=PromptDeploymentExpandMetaRequestRequest(
                 model_name=True,
+                usage=True,
+                finish_reason=True,
                 latency=True,
                 deployment_release_tag=True,
                 prompt_version_id=True,
-                finish_reason=True,
-                usage=True,
             ),
             raw_overrides=RawPromptExecutionOverridesRequest(
                 body={"string": {"key": "value"}},
@@ -1343,6 +1360,7 @@ class AsyncVellum:
         self,
         *,
         inputs: typing.Sequence[WorkflowRequestInputRequest],
+        expand_meta: typing.Optional[WorkflowExpandMetaRequest] = OMIT,
         workflow_deployment_id: typing.Optional[str] = OMIT,
         workflow_deployment_name: typing.Optional[str] = OMIT,
         release_tag: typing.Optional[str] = OMIT,
@@ -1355,6 +1373,8 @@ class AsyncVellum:
         Parameters:
             - inputs: typing.Sequence[WorkflowRequestInputRequest]. The list of inputs defined in the Workflow's Deployment with their corresponding values.
 
+            - expand_meta: typing.Optional[WorkflowExpandMetaRequest]. An optionally specified configuration used to opt in to including additional metadata about this workflow execution in the API response. Corresponding values will be returned under the `execution_meta` key within NODE events in the response stream.
+
             - workflow_deployment_id: typing.Optional[str]. The ID of the Workflow Deployment. Must provide either this or workflow_deployment_name.
 
             - workflow_deployment_name: typing.Optional[str]. The name of the Workflow Deployment. Must provide either this or workflow_deployment_id.
@@ -1365,7 +1385,7 @@ class AsyncVellum:
 
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
-        from vellum import WorkflowRequestInputRequest_String
+        from vellum import WorkflowExpandMetaRequest, WorkflowRequestInputRequest_String
         from vellum.client import AsyncVellum
 
         client = AsyncVellum(
@@ -1378,6 +1398,9 @@ class AsyncVellum:
                     value="string",
                 )
             ],
+            expand_meta=WorkflowExpandMetaRequest(
+                usage=True,
+            ),
             workflow_deployment_id="string",
             workflow_deployment_name="string",
             release_tag="string",
@@ -1385,6 +1408,8 @@ class AsyncVellum:
         )
         """
         _request: typing.Dict[str, typing.Any] = {"inputs": inputs}
+        if expand_meta is not OMIT:
+            _request["expand_meta"] = expand_meta
         if workflow_deployment_id is not OMIT:
             _request["workflow_deployment_id"] = workflow_deployment_id
         if workflow_deployment_name is not OMIT:
@@ -1437,6 +1462,7 @@ class AsyncVellum:
         self,
         *,
         inputs: typing.Sequence[WorkflowRequestInputRequest],
+        expand_meta: typing.Optional[WorkflowExpandMetaRequest] = OMIT,
         workflow_deployment_id: typing.Optional[str] = OMIT,
         workflow_deployment_name: typing.Optional[str] = OMIT,
         release_tag: typing.Optional[str] = OMIT,
@@ -1450,6 +1476,8 @@ class AsyncVellum:
         Parameters:
             - inputs: typing.Sequence[WorkflowRequestInputRequest]. The list of inputs defined in the Workflow's Deployment with their corresponding values.
 
+            - expand_meta: typing.Optional[WorkflowExpandMetaRequest]. An optionally specified configuration used to opt in to including additional metadata about this workflow execution in the API response. Corresponding values will be returned under the `execution_meta` key within NODE events in the response stream.
+
             - workflow_deployment_id: typing.Optional[str]. The ID of the Workflow Deployment. Must provide either this or workflow_deployment_name.
 
             - workflow_deployment_name: typing.Optional[str]. The name of the Workflow Deployment. Must provide either this or workflow_deployment_id.
@@ -1462,7 +1490,7 @@ class AsyncVellum:
 
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
-        from vellum import WorkflowRequestInputRequest_String
+        from vellum import WorkflowExpandMetaRequest, WorkflowRequestInputRequest_String
         from vellum.client import AsyncVellum
 
         client = AsyncVellum(
@@ -1475,6 +1503,9 @@ class AsyncVellum:
                     value="string",
                 )
             ],
+            expand_meta=WorkflowExpandMetaRequest(
+                usage=True,
+            ),
             workflow_deployment_id="string",
             workflow_deployment_name="string",
             release_tag="string",
@@ -1483,6 +1514,8 @@ class AsyncVellum:
         )
         """
         _request: typing.Dict[str, typing.Any] = {"inputs": inputs}
+        if expand_meta is not OMIT:
+            _request["expand_meta"] = expand_meta
         if workflow_deployment_id is not OMIT:
             _request["workflow_deployment_id"] = workflow_deployment_id
         if workflow_deployment_name is not OMIT:

vellum/core/client_wrapper.py

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