vellum-ai 0.9.5__py3-none-any.whl → 0.9.9__py3-none-any.whl

vellum/types/__init__.py

@@ -1,7 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .ad_hoc_execute_prompt_event import AdHocExecutePromptEvent
-from .ad_hoc_expand_meta_request import AdHocExpandMetaRequest
+from .ad_hoc_expand_meta import AdHocExpandMeta
 from .ad_hoc_fulfilled_prompt_execution_meta import AdHocFulfilledPromptExecutionMeta
 from .ad_hoc_initiated_prompt_execution_meta import AdHocInitiatedPromptExecutionMeta
 from .ad_hoc_rejected_prompt_execution_meta import AdHocRejectedPromptExecutionMeta
@@ -44,8 +44,7 @@ from .chat_history_vellum_value_request import ChatHistoryVellumValueRequest
 from .chat_message import ChatMessage
 from .chat_message_content import ChatMessageContent
 from .chat_message_content_request import ChatMessageContentRequest
-from .chat_message_prompt_block_properties_request import ChatMessagePromptBlockPropertiesRequest
-from .chat_message_prompt_block_request import ChatMessagePromptBlockRequest
+from .chat_message_prompt_block import ChatMessagePromptBlock
 from .chat_message_request import ChatMessageRequest
 from .chat_message_role import ChatMessageRole
 from .code_execution_node_array_result import CodeExecutionNodeArrayResult
@@ -91,7 +90,7 @@ from .enriched_normalized_completion import EnrichedNormalizedCompletion
 from .entity_status import EntityStatus
 from .entity_visibility import EntityVisibility
 from .environment_enum import EnvironmentEnum
-from .ephemeral_prompt_cache_config_request import EphemeralPromptCacheConfigRequest
+from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
 from .ephemeral_prompt_cache_config_type_enum import EphemeralPromptCacheConfigTypeEnum
 from .error_input_request import ErrorInputRequest
 from .error_variable_value import ErrorVariableValue
@@ -141,8 +140,7 @@ from .function_call_request import FunctionCallRequest
 from .function_call_variable_value import FunctionCallVariableValue
 from .function_call_vellum_value import FunctionCallVellumValue
 from .function_call_vellum_value_request import FunctionCallVellumValueRequest
-from .function_definition_prompt_block_properties_request import FunctionDefinitionPromptBlockPropertiesRequest
-from .function_definition_prompt_block_request import FunctionDefinitionPromptBlockRequest
+from .function_definition_prompt_block import FunctionDefinitionPromptBlock
 from .generate_options_request import GenerateOptionsRequest
 from .generate_request import GenerateRequest
 from .generate_response import GenerateResponse
@@ -179,8 +177,7 @@ from .initiated_workflow_node_result_event import InitiatedWorkflowNodeResultEve
 from .instructor_vectorizer_config import InstructorVectorizerConfig
 from .instructor_vectorizer_config_request import InstructorVectorizerConfigRequest
 from .iteration_state_enum import IterationStateEnum
-from .jinja_prompt_block_properties_request import JinjaPromptBlockPropertiesRequest
-from .jinja_prompt_block_request import JinjaPromptBlockRequest
+from .jinja_prompt_block import JinjaPromptBlock
 from .json_input_request import JsonInputRequest
 from .json_variable_value import JsonVariableValue
 from .json_vellum_value import JsonVellumValue
@@ -265,11 +262,11 @@ from .paginated_test_suite_test_case_list import PaginatedTestSuiteTestCaseList
 from .paginated_workflow_release_tag_read_list import PaginatedWorkflowReleaseTagReadList
 from .pdf_search_result_meta_source import PdfSearchResultMetaSource
 from .pdf_search_result_meta_source_request import PdfSearchResultMetaSourceRequest
-from .plain_text_prompt_block_request import PlainTextPromptBlockRequest
+from .plain_text_prompt_block import PlainTextPromptBlock
 from .price import Price
 from .processing_failure_reason_enum import ProcessingFailureReasonEnum
 from .processing_state_enum import ProcessingStateEnum
-from .prompt_block_request import PromptBlockRequest
+from .prompt_block import PromptBlock
 from .prompt_block_state import PromptBlockState
 from .prompt_deployment_expand_meta_request import PromptDeploymentExpandMetaRequest
 from .prompt_deployment_input_request import PromptDeploymentInputRequest
@@ -278,12 +275,12 @@ from .prompt_node_execution_meta import PromptNodeExecutionMeta
 from .prompt_node_result import PromptNodeResult
 from .prompt_node_result_data import PromptNodeResultData
 from .prompt_output import PromptOutput
-from .prompt_parameters_request import PromptParametersRequest
-from .prompt_request_chat_history_input_request import PromptRequestChatHistoryInputRequest
-from .prompt_request_input_request import PromptRequestInputRequest
-from .prompt_request_json_input_request import PromptRequestJsonInputRequest
-from .prompt_request_string_input_request import PromptRequestStringInputRequest
-from .prompt_settings_request import PromptSettingsRequest
+from .prompt_parameters import PromptParameters
+from .prompt_request_chat_history_input import PromptRequestChatHistoryInput
+from .prompt_request_input import PromptRequestInput
+from .prompt_request_json_input import PromptRequestJsonInput
+from .prompt_request_string_input import PromptRequestStringInput
+from .prompt_settings import PromptSettings
 from .raw_prompt_execution_overrides_request import RawPromptExecutionOverridesRequest
 from .reducto_chunker_config import ReductoChunkerConfig
 from .reducto_chunker_config_request import ReductoChunkerConfigRequest
@@ -297,8 +294,8 @@ from .rejected_prompt_execution_meta import RejectedPromptExecutionMeta
 from .rejected_workflow_node_result_event import RejectedWorkflowNodeResultEvent
 from .release_tag_source import ReleaseTagSource
 from .replace_test_suite_test_case_request import ReplaceTestSuiteTestCaseRequest
-from .rich_text_child_block_request import RichTextChildBlockRequest
-from .rich_text_prompt_block_request import RichTextPromptBlockRequest
+from .rich_text_child_block import RichTextChildBlock
+from .rich_text_prompt_block import RichTextPromptBlock
 from .sandbox_scenario import SandboxScenario
 from .scenario_input import ScenarioInput
 from .scenario_input_chat_history_variable_value import ScenarioInputChatHistoryVariableValue
@@ -434,7 +431,7 @@ from .token_overlapping_window_chunking_request import TokenOverlappingWindowChu
 from .unit_enum import UnitEnum
 from .upload_document_response import UploadDocumentResponse
 from .upsert_test_suite_test_case_request import UpsertTestSuiteTestCaseRequest
-from .variable_prompt_block_request import VariablePromptBlockRequest
+from .variable_prompt_block import VariablePromptBlock
 from .vellum_audio import VellumAudio
 from .vellum_audio_request import VellumAudioRequest
 from .vellum_error import VellumError
@@ -449,8 +446,6 @@ from .vellum_value_logical_expression_request import VellumValueLogicalExpressio
 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
 from .workflow_event_error import WorkflowEventError
@@ -500,7 +495,7 @@ from .workspace_secret_read import WorkspaceSecretRead
 
 __all__ = [
     "AdHocExecutePromptEvent",
-    "AdHocExpandMetaRequest",
+    "AdHocExpandMeta",
     "AdHocFulfilledPromptExecutionMeta",
     "AdHocInitiatedPromptExecutionMeta",
     "AdHocRejectedPromptExecutionMeta",
@@ -535,8 +530,7 @@ __all__ = [
     "ChatMessage",
     "ChatMessageContent",
     "ChatMessageContentRequest",
-    "ChatMessagePromptBlockPropertiesRequest",
-    "ChatMessagePromptBlockRequest",
+    "ChatMessagePromptBlock",
     "ChatMessageRequest",
     "ChatMessageRole",
     "CodeExecutionNodeArrayResult",
@@ -582,7 +576,7 @@ __all__ = [
     "EntityStatus",
     "EntityVisibility",
     "EnvironmentEnum",
-    "EphemeralPromptCacheConfigRequest",
+    "EphemeralPromptCacheConfig",
     "EphemeralPromptCacheConfigTypeEnum",
     "ErrorInputRequest",
     "ErrorVariableValue",
@@ -632,8 +626,7 @@ __all__ = [
     "FunctionCallVariableValue",
     "FunctionCallVellumValue",
     "FunctionCallVellumValueRequest",
-    "FunctionDefinitionPromptBlockPropertiesRequest",
-    "FunctionDefinitionPromptBlockRequest",
+    "FunctionDefinitionPromptBlock",
     "GenerateOptionsRequest",
     "GenerateRequest",
     "GenerateResponse",
@@ -666,8 +659,7 @@ __all__ = [
     "InstructorVectorizerConfig",
     "InstructorVectorizerConfigRequest",
     "IterationStateEnum",
-    "JinjaPromptBlockPropertiesRequest",
-    "JinjaPromptBlockRequest",
+    "JinjaPromptBlock",
     "JsonInputRequest",
     "JsonVariableValue",
     "JsonVellumValue",
@@ -752,11 +744,11 @@ __all__ = [
     "PaginatedWorkflowReleaseTagReadList",
     "PdfSearchResultMetaSource",
     "PdfSearchResultMetaSourceRequest",
-    "PlainTextPromptBlockRequest",
+    "PlainTextPromptBlock",
     "Price",
     "ProcessingFailureReasonEnum",
     "ProcessingStateEnum",
-    "PromptBlockRequest",
+    "PromptBlock",
     "PromptBlockState",
     "PromptDeploymentExpandMetaRequest",
     "PromptDeploymentInputRequest",
@@ -765,12 +757,12 @@ __all__ = [
     "PromptNodeResult",
     "PromptNodeResultData",
     "PromptOutput",
-    "PromptParametersRequest",
-    "PromptRequestChatHistoryInputRequest",
-    "PromptRequestInputRequest",
-    "PromptRequestJsonInputRequest",
-    "PromptRequestStringInputRequest",
-    "PromptSettingsRequest",
+    "PromptParameters",
+    "PromptRequestChatHistoryInput",
+    "PromptRequestInput",
+    "PromptRequestJsonInput",
+    "PromptRequestStringInput",
+    "PromptSettings",
     "RawPromptExecutionOverridesRequest",
     "ReductoChunkerConfig",
     "ReductoChunkerConfigRequest",
@@ -784,8 +776,8 @@ __all__ = [
     "RejectedWorkflowNodeResultEvent",
     "ReleaseTagSource",
     "ReplaceTestSuiteTestCaseRequest",
-    "RichTextChildBlockRequest",
-    "RichTextPromptBlockRequest",
+    "RichTextChildBlock",
+    "RichTextPromptBlock",
     "SandboxScenario",
     "ScenarioInput",
     "ScenarioInputChatHistoryVariableValue",
@@ -917,7 +909,7 @@ __all__ = [
     "UnitEnum",
     "UploadDocumentResponse",
     "UpsertTestSuiteTestCaseRequest",
-    "VariablePromptBlockRequest",
+    "VariablePromptBlock",
     "VellumAudio",
     "VellumAudioRequest",
     "VellumError",
@@ -932,8 +924,6 @@ __all__ = [
     "VellumValueRequest",
     "VellumVariable",
     "VellumVariableExtensions",
-    "VellumVariableExtensionsRequest",
-    "VellumVariableRequest",
     "VellumVariableType",
     "WorkflowDeploymentRead",
     "WorkflowEventError",

vellum/types/{ad_hoc_expand_meta_request.py → ad_hoc_expand_meta.py}

@@ -6,7 +6,7 @@ import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class AdHocExpandMetaRequest(UniversalBaseModel):
+class AdHocExpandMeta(UniversalBaseModel):
     cost: typing.Optional[bool] = pydantic.Field(default=None)
     """
     If enabled, the response will include model host cost tracking. This may increase latency for some model hosts.

vellum/types/{chat_message_prompt_block_request.py → chat_message_prompt_block.py}

@@ -4,22 +4,25 @@ from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
 import typing
 from .prompt_block_state import PromptBlockState
-from .ephemeral_prompt_cache_config_request import EphemeralPromptCacheConfigRequest
+from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
+from .chat_message_role import ChatMessageRole
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 from ..core.pydantic_utilities import update_forward_refs
 
 
-class ChatMessagePromptBlockRequest(UniversalBaseModel):
+class ChatMessagePromptBlock(UniversalBaseModel):
     """
     A block that represents a chat message in a prompt template.
     """
 
-    block_type: typing.Literal["CHAT_MESSAGE"] = "CHAT_MESSAGE"
-    properties: "ChatMessagePromptBlockPropertiesRequest"
-    id: str
     state: typing.Optional[PromptBlockState] = None
-    cache_config: typing.Optional[EphemeralPromptCacheConfigRequest] = None
+    cache_config: typing.Optional[EphemeralPromptCacheConfig] = None
+    block_type: typing.Literal["CHAT_MESSAGE"] = "CHAT_MESSAGE"
+    chat_role: ChatMessageRole
+    chat_source: typing.Optional[str] = None
+    chat_message_unterminated: typing.Optional[bool] = None
+    blocks: typing.List["PromptBlock"]
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
@@ -31,9 +34,6 @@ class ChatMessagePromptBlockRequest(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-from .chat_message_prompt_block_properties_request import ChatMessagePromptBlockPropertiesRequest  # noqa: E402
+from .prompt_block import PromptBlock  # noqa: E402
 
-update_forward_refs(
-    ChatMessagePromptBlockPropertiesRequest, ChatMessagePromptBlockRequest=ChatMessagePromptBlockRequest
-)
-update_forward_refs(ChatMessagePromptBlockRequest)
+update_forward_refs(ChatMessagePromptBlock)

vellum/types/{ephemeral_prompt_cache_config_request.py → ephemeral_prompt_cache_config.py}

@@ -7,7 +7,7 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class EphemeralPromptCacheConfigRequest(UniversalBaseModel):
+class EphemeralPromptCacheConfig(UniversalBaseModel):
     type: typing.Optional[EphemeralPromptCacheConfigTypeEnum] = None
 
     if IS_PYDANTIC_V2:

vellum/types/{function_definition_prompt_block_request.py → function_definition_prompt_block.py}

@@ -2,23 +2,23 @@
 
 from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .function_definition_prompt_block_properties_request import FunctionDefinitionPromptBlockPropertiesRequest
 from .prompt_block_state import PromptBlockState
-from .ephemeral_prompt_cache_config_request import EphemeralPromptCacheConfigRequest
+from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class FunctionDefinitionPromptBlockRequest(UniversalBaseModel):
+class FunctionDefinitionPromptBlock(UniversalBaseModel):
     """
     A block that represents a function definition in a prompt template.
     """
 
-    block_type: typing.Literal["FUNCTION_DEFINITION"] = "FUNCTION_DEFINITION"
-    properties: FunctionDefinitionPromptBlockPropertiesRequest
-    id: str
     state: typing.Optional[PromptBlockState] = None
-    cache_config: typing.Optional[EphemeralPromptCacheConfigRequest] = None
+    cache_config: typing.Optional[EphemeralPromptCacheConfig] = None
+    block_type: typing.Literal["FUNCTION_DEFINITION"] = "FUNCTION_DEFINITION"
+    function_name: typing.Optional[str] = None
+    function_description: typing.Optional[str] = None
+    function_parameters: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/types/{jinja_prompt_block_request.py → jinja_prompt_block.py}

@@ -2,23 +2,23 @@
 
 from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .jinja_prompt_block_properties_request import JinjaPromptBlockPropertiesRequest
 from .prompt_block_state import PromptBlockState
-from .ephemeral_prompt_cache_config_request import EphemeralPromptCacheConfigRequest
+from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
+from .vellum_variable_type import VellumVariableType
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class JinjaPromptBlockRequest(UniversalBaseModel):
+class JinjaPromptBlock(UniversalBaseModel):
     """
     A block of Jinja template code that is used to generate a prompt
     """
 
-    block_type: typing.Literal["JINJA"] = "JINJA"
-    properties: JinjaPromptBlockPropertiesRequest
-    id: str
     state: typing.Optional[PromptBlockState] = None
-    cache_config: typing.Optional[EphemeralPromptCacheConfigRequest] = None
+    cache_config: typing.Optional[EphemeralPromptCacheConfig] = None
+    block_type: typing.Literal["JINJA"] = "JINJA"
+    template: str
+    template_type: VellumVariableType
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/types/{plain_text_prompt_block_request.py → plain_text_prompt_block.py}

@@ -3,21 +3,20 @@
 from ..core.pydantic_utilities import UniversalBaseModel
 import typing
 from .prompt_block_state import PromptBlockState
-from .ephemeral_prompt_cache_config_request import EphemeralPromptCacheConfigRequest
+from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class PlainTextPromptBlockRequest(UniversalBaseModel):
+class PlainTextPromptBlock(UniversalBaseModel):
     """
     A block that holds a plain text string value.
     """
 
+    state: typing.Optional[PromptBlockState] = None
+    cache_config: typing.Optional[EphemeralPromptCacheConfig] = None
     block_type: typing.Literal["PLAIN_TEXT"] = "PLAIN_TEXT"
     text: str
-    id: str
-    state: typing.Optional[PromptBlockState] = None
-    cache_config: typing.Optional[EphemeralPromptCacheConfigRequest] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/types/prompt_block.py

@@ -0,0 +1,15 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+import typing
+from .jinja_prompt_block import JinjaPromptBlock
+from .function_definition_prompt_block import FunctionDefinitionPromptBlock
+from .variable_prompt_block import VariablePromptBlock
+from .rich_text_prompt_block import RichTextPromptBlock
+import typing
+
+if typing.TYPE_CHECKING:
+    from .chat_message_prompt_block import ChatMessagePromptBlock
+PromptBlock = typing.Union[
+    JinjaPromptBlock, "ChatMessagePromptBlock", FunctionDefinitionPromptBlock, VariablePromptBlock, RichTextPromptBlock
+]

vellum/types/{prompt_parameters_request.py → prompt_parameters.py}

@@ -6,7 +6,7 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class PromptParametersRequest(UniversalBaseModel):
+class PromptParameters(UniversalBaseModel):
     stop: typing.Optional[typing.List[str]] = None
     temperature: typing.Optional[float] = None
     max_tokens: typing.Optional[int] = None

vellum/types/{prompt_request_chat_history_input_request.py → prompt_request_chat_history_input.py}

@@ -3,18 +3,18 @@
 from ..core.pydantic_utilities import UniversalBaseModel
 import pydantic
 import typing
-from .chat_message_request import ChatMessageRequest
+from .chat_message import ChatMessage
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class PromptRequestChatHistoryInputRequest(UniversalBaseModel):
+class PromptRequestChatHistoryInput(UniversalBaseModel):
     key: str = pydantic.Field()
     """
     The variable's name, as defined in the Prompt.
     """
 
     type: typing.Literal["CHAT_HISTORY"] = "CHAT_HISTORY"
-    value: typing.List[ChatMessageRequest]
+    value: typing.List[ChatMessage]
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/types/prompt_request_input.py

@@ -0,0 +1,8 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from .prompt_request_string_input import PromptRequestStringInput
+from .prompt_request_json_input import PromptRequestJsonInput
+from .prompt_request_chat_history_input import PromptRequestChatHistoryInput
+
+PromptRequestInput = typing.Union[PromptRequestStringInput, PromptRequestJsonInput, PromptRequestChatHistoryInput]

vellum/types/{prompt_request_json_input_request.py → prompt_request_json_input.py}

@@ -6,7 +6,7 @@ import typing
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class PromptRequestJsonInputRequest(UniversalBaseModel):
+class PromptRequestJsonInput(UniversalBaseModel):
     key: str = pydantic.Field()
     """
     The variable's name, as defined in the Prompt.

vellum/types/{prompt_request_string_input_request.py → prompt_request_string_input.py}

@@ -6,7 +6,7 @@ import typing
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class PromptRequestStringInputRequest(UniversalBaseModel):
+class PromptRequestStringInput(UniversalBaseModel):
     key: str = pydantic.Field()
     """
     The variable's name, as defined in the Prompt.

vellum/types/{prompt_settings_request.py → prompt_settings.py}

@@ -6,7 +6,7 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class PromptSettingsRequest(UniversalBaseModel):
+class PromptSettings(UniversalBaseModel):
     timeout: typing.Optional[float] = None
 
     if IS_PYDANTIC_V2:

vellum/types/rich_text_child_block.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from .variable_prompt_block import VariablePromptBlock
+from .plain_text_prompt_block import PlainTextPromptBlock
+
+RichTextChildBlock = typing.Union[VariablePromptBlock, PlainTextPromptBlock]

vellum/types/{rich_text_prompt_block_request.py → rich_text_prompt_block.py}

@@ -2,23 +2,22 @@
 
 from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .rich_text_child_block_request import RichTextChildBlockRequest
 from .prompt_block_state import PromptBlockState
-from .ephemeral_prompt_cache_config_request import EphemeralPromptCacheConfigRequest
+from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
+from .rich_text_child_block import RichTextChildBlock
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class RichTextPromptBlockRequest(UniversalBaseModel):
+class RichTextPromptBlock(UniversalBaseModel):
     """
     A block that includes a combination of plain text and variable blocks.
     """
 
-    block_type: typing.Literal["RICH_TEXT"] = "RICH_TEXT"
-    blocks: typing.List[RichTextChildBlockRequest]
-    id: str
     state: typing.Optional[PromptBlockState] = None
-    cache_config: typing.Optional[EphemeralPromptCacheConfigRequest] = None
+    cache_config: typing.Optional[EphemeralPromptCacheConfig] = None
+    block_type: typing.Literal["RICH_TEXT"] = "RICH_TEXT"
+    blocks: typing.List[RichTextChildBlock]
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/types/{variable_prompt_block_request.py → variable_prompt_block.py}

@@ -3,20 +3,19 @@
 from ..core.pydantic_utilities import UniversalBaseModel
 import typing
 from .prompt_block_state import PromptBlockState
-from .ephemeral_prompt_cache_config_request import EphemeralPromptCacheConfigRequest
+from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class VariablePromptBlockRequest(UniversalBaseModel):
+class VariablePromptBlock(UniversalBaseModel):
     """
     A block that represents a variable in a prompt template.
     """
 
-    block_type: typing.Literal["VARIABLE"] = "VARIABLE"
-    id: str
     state: typing.Optional[PromptBlockState] = None
-    cache_config: typing.Optional[EphemeralPromptCacheConfigRequest] = None
+    cache_config: typing.Optional[EphemeralPromptCacheConfig] = None
+    block_type: typing.Literal["VARIABLE"] = "VARIABLE"
     input_variable_id: str
 
     if IS_PYDANTIC_V2:

{vellum_ai-0.9.5.dist-info → vellum_ai-0.9.9.dist-info}/METADATA

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