vellum-ai 0.9.6__tar.gz → 0.9.8__tar.gz

{vellum_ai-0.9.6 → vellum_ai-0.9.8}/PKG-INFO

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

{vellum_ai-0.9.6 → vellum_ai-0.9.8}/pyproject.toml

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

{vellum_ai-0.9.6 → vellum_ai-0.9.8}/src/vellum/__init__.py

@@ -2,7 +2,7 @@
 
 from .types import (
     AdHocExecutePromptEvent,
-    AdHocExpandMetaRequest,
+    AdHocExpandMeta,
     AdHocFulfilledPromptExecutionMeta,
     AdHocInitiatedPromptExecutionMeta,
     AdHocRejectedPromptExecutionMeta,
@@ -37,8 +37,8 @@ from .types import (
     ChatMessage,
     ChatMessageContent,
     ChatMessageContentRequest,
-    ChatMessagePromptBlockPropertiesRequest,
-    ChatMessagePromptBlockRequest,
+    ChatMessagePromptBlock,
+    ChatMessagePromptBlockProperties,
     ChatMessageRequest,
     ChatMessageRole,
     CodeExecutionNodeArrayResult,
@@ -84,7 +84,7 @@ from .types import (
     EntityStatus,
     EntityVisibility,
     EnvironmentEnum,
-    EphemeralPromptCacheConfigRequest,
+    EphemeralPromptCacheConfig,
     EphemeralPromptCacheConfigTypeEnum,
     ErrorInputRequest,
     ErrorVariableValue,
@@ -134,8 +134,8 @@ from .types import (
     FunctionCallVariableValue,
     FunctionCallVellumValue,
     FunctionCallVellumValueRequest,
-    FunctionDefinitionPromptBlockPropertiesRequest,
-    FunctionDefinitionPromptBlockRequest,
+    FunctionDefinitionPromptBlock,
+    FunctionDefinitionPromptBlockProperties,
     GenerateOptionsRequest,
     GenerateRequest,
     GenerateResponse,
@@ -168,8 +168,8 @@ from .types import (
     InstructorVectorizerConfig,
     InstructorVectorizerConfigRequest,
     IterationStateEnum,
-    JinjaPromptBlockPropertiesRequest,
-    JinjaPromptBlockRequest,
+    JinjaPromptBlock,
+    JinjaPromptBlockProperties,
     JsonInputRequest,
     JsonVariableValue,
     JsonVellumValue,
@@ -254,11 +254,11 @@ from .types import (
     PaginatedWorkflowReleaseTagReadList,
     PdfSearchResultMetaSource,
     PdfSearchResultMetaSourceRequest,
-    PlainTextPromptBlockRequest,
+    PlainTextPromptBlock,
     Price,
     ProcessingFailureReasonEnum,
     ProcessingStateEnum,
-    PromptBlockRequest,
+    PromptBlock,
     PromptBlockState,
     PromptDeploymentExpandMetaRequest,
     PromptDeploymentInputRequest,
@@ -267,12 +267,12 @@ from .types import (
     PromptNodeResult,
     PromptNodeResultData,
     PromptOutput,
-    PromptParametersRequest,
-    PromptRequestChatHistoryInputRequest,
-    PromptRequestInputRequest,
-    PromptRequestJsonInputRequest,
-    PromptRequestStringInputRequest,
-    PromptSettingsRequest,
+    PromptParameters,
+    PromptRequestChatHistoryInput,
+    PromptRequestInput,
+    PromptRequestJsonInput,
+    PromptRequestStringInput,
+    PromptSettings,
     RawPromptExecutionOverridesRequest,
     ReductoChunkerConfig,
     ReductoChunkerConfigRequest,
@@ -286,8 +286,8 @@ from .types import (
     RejectedWorkflowNodeResultEvent,
     ReleaseTagSource,
     ReplaceTestSuiteTestCaseRequest,
-    RichTextChildBlockRequest,
-    RichTextPromptBlockRequest,
+    RichTextChildBlock,
+    RichTextPromptBlock,
     SandboxScenario,
     ScenarioInput,
     ScenarioInputChatHistoryVariableValue,
@@ -419,7 +419,7 @@ from .types import (
     UnitEnum,
     UploadDocumentResponse,
     UpsertTestSuiteTestCaseRequest,
-    VariablePromptBlockRequest,
+    VariablePromptBlock,
     VellumAudio,
     VellumAudioRequest,
     VellumError,
@@ -434,8 +434,6 @@ from .types import (
     VellumValueRequest,
     VellumVariable,
     VellumVariableExtensions,
-    VellumVariableExtensionsRequest,
-    VellumVariableRequest,
     VellumVariableType,
     WorkflowDeploymentRead,
     WorkflowEventError,
@@ -513,7 +511,7 @@ from .version import __version__
 
 __all__ = [
     "AdHocExecutePromptEvent",
-    "AdHocExpandMetaRequest",
+    "AdHocExpandMeta",
     "AdHocFulfilledPromptExecutionMeta",
     "AdHocInitiatedPromptExecutionMeta",
     "AdHocRejectedPromptExecutionMeta",
@@ -550,8 +548,8 @@ __all__ = [
     "ChatMessage",
     "ChatMessageContent",
     "ChatMessageContentRequest",
-    "ChatMessagePromptBlockPropertiesRequest",
-    "ChatMessagePromptBlockRequest",
+    "ChatMessagePromptBlock",
+    "ChatMessagePromptBlockProperties",
     "ChatMessageRequest",
     "ChatMessageRole",
     "CodeExecutionNodeArrayResult",
@@ -599,7 +597,7 @@ __all__ = [
     "EntityStatus",
     "EntityVisibility",
     "EnvironmentEnum",
-    "EphemeralPromptCacheConfigRequest",
+    "EphemeralPromptCacheConfig",
     "EphemeralPromptCacheConfigTypeEnum",
     "ErrorInputRequest",
     "ErrorVariableValue",
@@ -651,8 +649,8 @@ __all__ = [
     "FunctionCallVariableValue",
     "FunctionCallVellumValue",
     "FunctionCallVellumValueRequest",
-    "FunctionDefinitionPromptBlockPropertiesRequest",
-    "FunctionDefinitionPromptBlockRequest",
+    "FunctionDefinitionPromptBlock",
+    "FunctionDefinitionPromptBlockProperties",
     "GenerateOptionsRequest",
     "GenerateRequest",
     "GenerateResponse",
@@ -686,8 +684,8 @@ __all__ = [
     "InstructorVectorizerConfigRequest",
     "InternalServerError",
     "IterationStateEnum",
-    "JinjaPromptBlockPropertiesRequest",
-    "JinjaPromptBlockRequest",
+    "JinjaPromptBlock",
+    "JinjaPromptBlockProperties",
     "JsonInputRequest",
     "JsonVariableValue",
     "JsonVellumValue",
@@ -775,11 +773,11 @@ __all__ = [
     "PaginatedWorkflowReleaseTagReadList",
     "PdfSearchResultMetaSource",
     "PdfSearchResultMetaSourceRequest",
-    "PlainTextPromptBlockRequest",
+    "PlainTextPromptBlock",
     "Price",
     "ProcessingFailureReasonEnum",
     "ProcessingStateEnum",
-    "PromptBlockRequest",
+    "PromptBlock",
     "PromptBlockState",
     "PromptDeploymentExpandMetaRequest",
     "PromptDeploymentInputRequest",
@@ -788,12 +786,12 @@ __all__ = [
     "PromptNodeResult",
     "PromptNodeResultData",
     "PromptOutput",
-    "PromptParametersRequest",
-    "PromptRequestChatHistoryInputRequest",
-    "PromptRequestInputRequest",
-    "PromptRequestJsonInputRequest",
-    "PromptRequestStringInputRequest",
-    "PromptSettingsRequest",
+    "PromptParameters",
+    "PromptRequestChatHistoryInput",
+    "PromptRequestInput",
+    "PromptRequestJsonInput",
+    "PromptRequestStringInput",
+    "PromptSettings",
     "RawPromptExecutionOverridesRequest",
     "ReductoChunkerConfig",
     "ReductoChunkerConfigRequest",
@@ -807,8 +805,8 @@ __all__ = [
     "RejectedWorkflowNodeResultEvent",
     "ReleaseTagSource",
     "ReplaceTestSuiteTestCaseRequest",
-    "RichTextChildBlockRequest",
-    "RichTextPromptBlockRequest",
+    "RichTextChildBlock",
+    "RichTextPromptBlock",
     "SandboxScenario",
     "ScenarioInput",
     "ScenarioInputChatHistoryVariableValue",
@@ -940,7 +938,7 @@ __all__ = [
     "UnitEnum",
     "UploadDocumentResponse",
     "UpsertTestSuiteTestCaseRequest",
-    "VariablePromptBlockRequest",
+    "VariablePromptBlock",
     "Vellum",
     "VellumAudio",
     "VellumAudioRequest",
@@ -957,8 +955,6 @@ __all__ = [
     "VellumValueRequest",
     "VellumVariable",
     "VellumVariableExtensions",
-    "VellumVariableExtensionsRequest",
-    "VellumVariableRequest",
     "VellumVariableType",
     "WorkflowDeploymentRead",
     "WorkflowDeploymentsListRequestStatus",

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

{vellum_ai-0.9.6 → vellum_ai-0.9.8}/src/vellum/resources/ad_hoc/client.py

@@ -2,12 +2,12 @@
 
 import typing
 from ...core.client_wrapper import SyncClientWrapper
-from ...types.prompt_request_input_request import PromptRequestInputRequest
-from ...types.vellum_variable_request import VellumVariableRequest
-from ...types.prompt_parameters_request import PromptParametersRequest
-from ...types.prompt_block_request import PromptBlockRequest
-from ...types.prompt_settings_request import PromptSettingsRequest
-from ...types.ad_hoc_expand_meta_request import AdHocExpandMetaRequest
+from ...types.prompt_request_input import PromptRequestInput
+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.ad_hoc_expand_meta import AdHocExpandMeta
 from ...core.request_options import RequestOptions
 from ...types.ad_hoc_execute_prompt_event import AdHocExecutePromptEvent
 from ...core.serialization import convert_and_respect_annotation_metadata
@@ -32,12 +32,12 @@ class AdHocClient:
         self,
         *,
         ml_model: str,
-        input_values: typing.Sequence[PromptRequestInputRequest],
-        input_variables: typing.Sequence[VellumVariableRequest],
-        parameters: PromptParametersRequest,
-        blocks: typing.Sequence[PromptBlockRequest],
-        settings: typing.Optional[PromptSettingsRequest] = OMIT,
-        expand_meta: typing.Optional[AdHocExpandMetaRequest] = OMIT,
+        input_values: typing.Sequence[PromptRequestInput],
+        input_variables: typing.Sequence[VellumVariable],
+        parameters: PromptParameters,
+        blocks: typing.Sequence[PromptBlock],
+        settings: typing.Optional[PromptSettings] = OMIT,
+        expand_meta: typing.Optional[AdHocExpandMeta] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.Iterator[AdHocExecutePromptEvent]:
         """
@@ -47,17 +47,17 @@ class AdHocClient:
         ----------
         ml_model : str
 
-        input_values : typing.Sequence[PromptRequestInputRequest]
+        input_values : typing.Sequence[PromptRequestInput]
 
-        input_variables : typing.Sequence[VellumVariableRequest]
+        input_variables : typing.Sequence[VellumVariable]
 
-        parameters : PromptParametersRequest
+        parameters : PromptParameters
 
-        blocks : typing.Sequence[PromptBlockRequest]
+        blocks : typing.Sequence[PromptBlock]
 
-        settings : typing.Optional[PromptSettingsRequest]
+        settings : typing.Optional[PromptSettings]
 
-        expand_meta : typing.Optional[AdHocExpandMetaRequest]
+        expand_meta : typing.Optional[AdHocExpandMeta]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -70,17 +70,17 @@ class AdHocClient:
         Examples
         --------
         from vellum import (
-            AdHocExpandMetaRequest,
-            EphemeralPromptCacheConfigRequest,
-            JinjaPromptBlockPropertiesRequest,
-            JinjaPromptBlockRequest,
-            PromptParametersRequest,
-            PromptRequestStringInputRequest,
-            PromptSettingsRequest,
-            StringVellumValueRequest,
+            AdHocExpandMeta,
+            EphemeralPromptCacheConfig,
+            JinjaPromptBlock,
+            JinjaPromptBlockProperties,
+            PromptParameters,
+            PromptRequestStringInput,
+            PromptSettings,
+            StringVellumValue,
             Vellum,
-            VellumVariableExtensionsRequest,
-            VellumVariableRequest,
+            VellumVariable,
+            VellumVariableExtensions,
         )
 
         client = Vellum(
@@ -89,26 +89,26 @@ class AdHocClient:
         response = client.ad_hoc.adhoc_execute_prompt_stream(
             ml_model="string",
             input_values=[
-                PromptRequestStringInputRequest(
+                PromptRequestStringInput(
                     key="string",
                     value="string",
                 )
             ],
             input_variables=[
-                VellumVariableRequest(
+                VellumVariable(
                     id="string",
                     key="string",
                     type="STRING",
                     required=True,
-                    default=StringVellumValueRequest(
-                        value="string",
+                    default=StringVellumValue(
+                        value={"key": "value"},
                     ),
-                    extensions=VellumVariableExtensionsRequest(
-                        color="string",
+                    extensions=VellumVariableExtensions(
+                        color={"key": "value"},
                     ),
                 )
             ],
-            parameters=PromptParametersRequest(
+            parameters=PromptParameters(
                 stop=["string"],
                 temperature=1.1,
                 max_tokens=1,
@@ -119,22 +119,20 @@ class AdHocClient:
                 logit_bias={"string": {"key": "value"}},
                 custom_parameters={"string": {"key": "value"}},
             ),
-            settings=PromptSettingsRequest(
+            settings=PromptSettings(
                 timeout=1.1,
             ),
             blocks=[
-                JinjaPromptBlockRequest(
+                JinjaPromptBlock(
                     state="ENABLED",
-                    cache_config=EphemeralPromptCacheConfigRequest(
-                        type={"key": "value"},
-                    ),
-                    properties=JinjaPromptBlockPropertiesRequest(
+                    cache_config=EphemeralPromptCacheConfig(),
+                    properties=JinjaPromptBlockProperties(
                         template="string",
                         template_type="STRING",
                     ),
                 )
             ],
-            expand_meta=AdHocExpandMetaRequest(
+            expand_meta=AdHocExpandMeta(
                 cost=True,
                 model_name=True,
                 usage=True,
@@ -151,22 +149,22 @@ class AdHocClient:
             json={
                 "ml_model": ml_model,
                 "input_values": convert_and_respect_annotation_metadata(
-                    object_=input_values, annotation=typing.Sequence[PromptRequestInputRequest], direction="write"
+                    object_=input_values, annotation=typing.Sequence[PromptRequestInput], direction="write"
                 ),
                 "input_variables": convert_and_respect_annotation_metadata(
-                    object_=input_variables, annotation=typing.Sequence[VellumVariableRequest], direction="write"
+                    object_=input_variables, annotation=typing.Sequence[VellumVariable], direction="write"
                 ),
                 "parameters": convert_and_respect_annotation_metadata(
-                    object_=parameters, annotation=PromptParametersRequest, direction="write"
+                    object_=parameters, annotation=PromptParameters, direction="write"
                 ),
                 "settings": convert_and_respect_annotation_metadata(
-                    object_=settings, annotation=PromptSettingsRequest, direction="write"
+                    object_=settings, annotation=PromptSettings, direction="write"
                 ),
                 "blocks": convert_and_respect_annotation_metadata(
-                    object_=blocks, annotation=typing.Sequence[PromptBlockRequest], direction="write"
+                    object_=blocks, annotation=typing.Sequence[PromptBlock], direction="write"
                 ),
                 "expand_meta": convert_and_respect_annotation_metadata(
-                    object_=expand_meta, annotation=AdHocExpandMetaRequest, direction="write"
+                    object_=expand_meta, annotation=AdHocExpandMeta, direction="write"
                 ),
             },
             request_options=request_options,
@@ -233,12 +231,12 @@ class AsyncAdHocClient:
         self,
         *,
         ml_model: str,
-        input_values: typing.Sequence[PromptRequestInputRequest],
-        input_variables: typing.Sequence[VellumVariableRequest],
-        parameters: PromptParametersRequest,
-        blocks: typing.Sequence[PromptBlockRequest],
-        settings: typing.Optional[PromptSettingsRequest] = OMIT,
-        expand_meta: typing.Optional[AdHocExpandMetaRequest] = OMIT,
+        input_values: typing.Sequence[PromptRequestInput],
+        input_variables: typing.Sequence[VellumVariable],
+        parameters: PromptParameters,
+        blocks: typing.Sequence[PromptBlock],
+        settings: typing.Optional[PromptSettings] = OMIT,
+        expand_meta: typing.Optional[AdHocExpandMeta] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.AsyncIterator[AdHocExecutePromptEvent]:
         """
@@ -248,17 +246,17 @@ class AsyncAdHocClient:
         ----------
         ml_model : str
 
-        input_values : typing.Sequence[PromptRequestInputRequest]
+        input_values : typing.Sequence[PromptRequestInput]
 
-        input_variables : typing.Sequence[VellumVariableRequest]
+        input_variables : typing.Sequence[VellumVariable]
 
-        parameters : PromptParametersRequest
+        parameters : PromptParameters
 
-        blocks : typing.Sequence[PromptBlockRequest]
+        blocks : typing.Sequence[PromptBlock]
 
-        settings : typing.Optional[PromptSettingsRequest]
+        settings : typing.Optional[PromptSettings]
 
-        expand_meta : typing.Optional[AdHocExpandMetaRequest]
+        expand_meta : typing.Optional[AdHocExpandMeta]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -273,17 +271,17 @@ class AsyncAdHocClient:
         import asyncio
 
         from vellum import (
-            AdHocExpandMetaRequest,
+            AdHocExpandMeta,
             AsyncVellum,
-            EphemeralPromptCacheConfigRequest,
-            JinjaPromptBlockPropertiesRequest,
-            JinjaPromptBlockRequest,
-            PromptParametersRequest,
-            PromptRequestStringInputRequest,
-            PromptSettingsRequest,
-            StringVellumValueRequest,
-            VellumVariableExtensionsRequest,
-            VellumVariableRequest,
+            EphemeralPromptCacheConfig,
+            JinjaPromptBlock,
+            JinjaPromptBlockProperties,
+            PromptParameters,
+            PromptRequestStringInput,
+            PromptSettings,
+            StringVellumValue,
+            VellumVariable,
+            VellumVariableExtensions,
         )
 
         client = AsyncVellum(
@@ -295,26 +293,26 @@ class AsyncAdHocClient:
             response = await client.ad_hoc.adhoc_execute_prompt_stream(
                 ml_model="string",
                 input_values=[
-                    PromptRequestStringInputRequest(
+                    PromptRequestStringInput(
                         key="string",
                         value="string",
                     )
                 ],
                 input_variables=[
-                    VellumVariableRequest(
+                    VellumVariable(
                         id="string",
                         key="string",
                         type="STRING",
                         required=True,
-                        default=StringVellumValueRequest(
-                            value="string",
+                        default=StringVellumValue(
+                            value={"key": "value"},
                         ),
-                        extensions=VellumVariableExtensionsRequest(
-                            color="string",
+                        extensions=VellumVariableExtensions(
+                            color={"key": "value"},
                         ),
                     )
                 ],
-                parameters=PromptParametersRequest(
+                parameters=PromptParameters(
                     stop=["string"],
                     temperature=1.1,
                     max_tokens=1,
@@ -325,22 +323,20 @@ class AsyncAdHocClient:
                     logit_bias={"string": {"key": "value"}},
                     custom_parameters={"string": {"key": "value"}},
                 ),
-                settings=PromptSettingsRequest(
+                settings=PromptSettings(
                     timeout=1.1,
                 ),
                 blocks=[
-                    JinjaPromptBlockRequest(
+                    JinjaPromptBlock(
                         state="ENABLED",
-                        cache_config=EphemeralPromptCacheConfigRequest(
-                            type={"key": "value"},
-                        ),
-                        properties=JinjaPromptBlockPropertiesRequest(
+                        cache_config=EphemeralPromptCacheConfig(),
+                        properties=JinjaPromptBlockProperties(
                             template="string",
                             template_type="STRING",
                         ),
                     )
                 ],
-                expand_meta=AdHocExpandMetaRequest(
+                expand_meta=AdHocExpandMeta(
                     cost=True,
                     model_name=True,
                     usage=True,
@@ -360,22 +356,22 @@ class AsyncAdHocClient:
             json={
                 "ml_model": ml_model,
                 "input_values": convert_and_respect_annotation_metadata(
-                    object_=input_values, annotation=typing.Sequence[PromptRequestInputRequest], direction="write"
+                    object_=input_values, annotation=typing.Sequence[PromptRequestInput], direction="write"
                 ),
                 "input_variables": convert_and_respect_annotation_metadata(
-                    object_=input_variables, annotation=typing.Sequence[VellumVariableRequest], direction="write"
+                    object_=input_variables, annotation=typing.Sequence[VellumVariable], direction="write"
                 ),
                 "parameters": convert_and_respect_annotation_metadata(
-                    object_=parameters, annotation=PromptParametersRequest, direction="write"
+                    object_=parameters, annotation=PromptParameters, direction="write"
                 ),
                 "settings": convert_and_respect_annotation_metadata(
-                    object_=settings, annotation=PromptSettingsRequest, direction="write"
+                    object_=settings, annotation=PromptSettings, direction="write"
                 ),
                 "blocks": convert_and_respect_annotation_metadata(
-                    object_=blocks, annotation=typing.Sequence[PromptBlockRequest], direction="write"
+                    object_=blocks, annotation=typing.Sequence[PromptBlock], direction="write"
                 ),
                 "expand_meta": convert_and_respect_annotation_metadata(
-                    object_=expand_meta, annotation=AdHocExpandMetaRequest, direction="write"
+                    object_=expand_meta, annotation=AdHocExpandMeta, direction="write"
                 ),
             },
             request_options=request_options,