vellum-ai 0.9.7__tar.gz → 0.9.9__tar.gz

{vellum_ai-0.9.7 → vellum_ai-0.9.9}/PKG-INFO

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

{vellum_ai-0.9.7 → vellum_ai-0.9.9}/pyproject.toml

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

{vellum_ai-0.9.7 → vellum_ai-0.9.9}/src/vellum/__init__.py

@@ -37,6 +37,7 @@ from .types import (
     ChatMessage,
     ChatMessageContent,
     ChatMessageContentRequest,
+    ChatMessagePromptBlock,
     ChatMessageRequest,
     ChatMessageRole,
     CodeExecutionNodeArrayResult,
@@ -82,6 +83,8 @@ from .types import (
     EntityStatus,
     EntityVisibility,
     EnvironmentEnum,
+    EphemeralPromptCacheConfig,
+    EphemeralPromptCacheConfigTypeEnum,
     ErrorInputRequest,
     ErrorVariableValue,
     ErrorVellumValue,
@@ -130,6 +133,7 @@ from .types import (
     FunctionCallVariableValue,
     FunctionCallVellumValue,
     FunctionCallVellumValueRequest,
+    FunctionDefinitionPromptBlock,
     GenerateOptionsRequest,
     GenerateRequest,
     GenerateResponse,
@@ -162,6 +166,7 @@ from .types import (
     InstructorVectorizerConfig,
     InstructorVectorizerConfigRequest,
     IterationStateEnum,
+    JinjaPromptBlock,
     JsonInputRequest,
     JsonVariableValue,
     JsonVellumValue,
@@ -246,9 +251,12 @@ from .types import (
     PaginatedWorkflowReleaseTagReadList,
     PdfSearchResultMetaSource,
     PdfSearchResultMetaSourceRequest,
+    PlainTextPromptBlock,
     Price,
     ProcessingFailureReasonEnum,
     ProcessingStateEnum,
+    PromptBlock,
+    PromptBlockState,
     PromptDeploymentExpandMetaRequest,
     PromptDeploymentInputRequest,
     PromptExecutionMeta,
@@ -275,6 +283,8 @@ from .types import (
     RejectedWorkflowNodeResultEvent,
     ReleaseTagSource,
     ReplaceTestSuiteTestCaseRequest,
+    RichTextChildBlock,
+    RichTextPromptBlock,
     SandboxScenario,
     ScenarioInput,
     ScenarioInputChatHistoryVariableValue,
@@ -406,6 +416,7 @@ from .types import (
     UnitEnum,
     UploadDocumentResponse,
     UpsertTestSuiteTestCaseRequest,
+    VariablePromptBlock,
     VellumAudio,
     VellumAudioRequest,
     VellumError,
@@ -534,6 +545,7 @@ __all__ = [
     "ChatMessage",
     "ChatMessageContent",
     "ChatMessageContentRequest",
+    "ChatMessagePromptBlock",
     "ChatMessageRequest",
     "ChatMessageRole",
     "CodeExecutionNodeArrayResult",
@@ -581,6 +593,8 @@ __all__ = [
     "EntityStatus",
     "EntityVisibility",
     "EnvironmentEnum",
+    "EphemeralPromptCacheConfig",
+    "EphemeralPromptCacheConfigTypeEnum",
     "ErrorInputRequest",
     "ErrorVariableValue",
     "ErrorVellumValue",
@@ -631,6 +645,7 @@ __all__ = [
     "FunctionCallVariableValue",
     "FunctionCallVellumValue",
     "FunctionCallVellumValueRequest",
+    "FunctionDefinitionPromptBlock",
     "GenerateOptionsRequest",
     "GenerateRequest",
     "GenerateResponse",
@@ -664,6 +679,7 @@ __all__ = [
     "InstructorVectorizerConfigRequest",
     "InternalServerError",
     "IterationStateEnum",
+    "JinjaPromptBlock",
     "JsonInputRequest",
     "JsonVariableValue",
     "JsonVellumValue",
@@ -751,9 +767,12 @@ __all__ = [
     "PaginatedWorkflowReleaseTagReadList",
     "PdfSearchResultMetaSource",
     "PdfSearchResultMetaSourceRequest",
+    "PlainTextPromptBlock",
     "Price",
     "ProcessingFailureReasonEnum",
     "ProcessingStateEnum",
+    "PromptBlock",
+    "PromptBlockState",
     "PromptDeploymentExpandMetaRequest",
     "PromptDeploymentInputRequest",
     "PromptExecutionMeta",
@@ -780,6 +799,8 @@ __all__ = [
     "RejectedWorkflowNodeResultEvent",
     "ReleaseTagSource",
     "ReplaceTestSuiteTestCaseRequest",
+    "RichTextChildBlock",
+    "RichTextPromptBlock",
     "SandboxScenario",
     "ScenarioInput",
     "ScenarioInputChatHistoryVariableValue",
@@ -911,6 +932,7 @@ __all__ = [
     "UnitEnum",
     "UploadDocumentResponse",
     "UpsertTestSuiteTestCaseRequest",
+    "VariablePromptBlock",
     "Vellum",
     "VellumAudio",
     "VellumAudioRequest",

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

{vellum_ai-0.9.7 → vellum_ai-0.9.9}/src/vellum/resources/ad_hoc/client.py

@@ -5,6 +5,7 @@ from ...core.client_wrapper import SyncClientWrapper
 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
@@ -34,7 +35,7 @@ class AdHocClient:
         input_values: typing.Sequence[PromptRequestInput],
         input_variables: typing.Sequence[VellumVariable],
         parameters: PromptParameters,
-        blocks: typing.Sequence[typing.Optional[typing.Any]],
+        blocks: typing.Sequence[PromptBlock],
         settings: typing.Optional[PromptSettings] = OMIT,
         expand_meta: typing.Optional[AdHocExpandMeta] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
@@ -52,7 +53,7 @@ class AdHocClient:
 
         parameters : PromptParameters
 
-        blocks : typing.Sequence[typing.Optional[typing.Any]]
+        blocks : typing.Sequence[PromptBlock]
 
         settings : typing.Optional[PromptSettings]
 
@@ -70,6 +71,8 @@ class AdHocClient:
         --------
         from vellum import (
             AdHocExpandMeta,
+            EphemeralPromptCacheConfig,
+            JinjaPromptBlock,
             PromptParameters,
             PromptRequestStringInput,
             PromptSettings,
@@ -118,7 +121,14 @@ class AdHocClient:
             settings=PromptSettings(
                 timeout=1.1,
             ),
-            blocks=[{"key": "value"}],
+            blocks=[
+                JinjaPromptBlock(
+                    state="ENABLED",
+                    cache_config=EphemeralPromptCacheConfig(),
+                    template="string",
+                    template_type="STRING",
+                )
+            ],
             expand_meta=AdHocExpandMeta(
                 cost=True,
                 model_name=True,
@@ -147,7 +157,9 @@ class AdHocClient:
                 "settings": convert_and_respect_annotation_metadata(
                     object_=settings, annotation=PromptSettings, direction="write"
                 ),
-                "blocks": blocks,
+                "blocks": convert_and_respect_annotation_metadata(
+                    object_=blocks, annotation=typing.Sequence[PromptBlock], direction="write"
+                ),
                 "expand_meta": convert_and_respect_annotation_metadata(
                     object_=expand_meta, annotation=AdHocExpandMeta, direction="write"
                 ),
@@ -219,7 +231,7 @@ class AsyncAdHocClient:
         input_values: typing.Sequence[PromptRequestInput],
         input_variables: typing.Sequence[VellumVariable],
         parameters: PromptParameters,
-        blocks: typing.Sequence[typing.Optional[typing.Any]],
+        blocks: typing.Sequence[PromptBlock],
         settings: typing.Optional[PromptSettings] = OMIT,
         expand_meta: typing.Optional[AdHocExpandMeta] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
@@ -237,7 +249,7 @@ class AsyncAdHocClient:
 
         parameters : PromptParameters
 
-        blocks : typing.Sequence[typing.Optional[typing.Any]]
+        blocks : typing.Sequence[PromptBlock]
 
         settings : typing.Optional[PromptSettings]
 
@@ -258,6 +270,8 @@ class AsyncAdHocClient:
         from vellum import (
             AdHocExpandMeta,
             AsyncVellum,
+            EphemeralPromptCacheConfig,
+            JinjaPromptBlock,
             PromptParameters,
             PromptRequestStringInput,
             PromptSettings,
@@ -308,7 +322,14 @@ class AsyncAdHocClient:
                 settings=PromptSettings(
                     timeout=1.1,
                 ),
-                blocks=[{"key": "value"}],
+                blocks=[
+                    JinjaPromptBlock(
+                        state="ENABLED",
+                        cache_config=EphemeralPromptCacheConfig(),
+                        template="string",
+                        template_type="STRING",
+                    )
+                ],
                 expand_meta=AdHocExpandMeta(
                     cost=True,
                     model_name=True,
@@ -340,7 +361,9 @@ class AsyncAdHocClient:
                 "settings": convert_and_respect_annotation_metadata(
                     object_=settings, annotation=PromptSettings, direction="write"
                 ),
-                "blocks": blocks,
+                "blocks": convert_and_respect_annotation_metadata(
+                    object_=blocks, annotation=typing.Sequence[PromptBlock], direction="write"
+                ),
                 "expand_meta": convert_and_respect_annotation_metadata(
                     object_=expand_meta, annotation=AdHocExpandMeta, direction="write"
                 ),

{vellum_ai-0.9.7 → vellum_ai-0.9.9}/src/vellum/types/__init__.py

@@ -44,6 +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 import ChatMessagePromptBlock
 from .chat_message_request import ChatMessageRequest
 from .chat_message_role import ChatMessageRole
 from .code_execution_node_array_result import CodeExecutionNodeArrayResult
@@ -89,6 +90,8 @@ 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 import EphemeralPromptCacheConfig
+from .ephemeral_prompt_cache_config_type_enum import EphemeralPromptCacheConfigTypeEnum
 from .error_input_request import ErrorInputRequest
 from .error_variable_value import ErrorVariableValue
 from .error_vellum_value import ErrorVellumValue
@@ -137,6 +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 import FunctionDefinitionPromptBlock
 from .generate_options_request import GenerateOptionsRequest
 from .generate_request import GenerateRequest
 from .generate_response import GenerateResponse
@@ -173,6 +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 import JinjaPromptBlock
 from .json_input_request import JsonInputRequest
 from .json_variable_value import JsonVariableValue
 from .json_vellum_value import JsonVellumValue
@@ -257,9 +262,12 @@ 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 import PlainTextPromptBlock
 from .price import Price
 from .processing_failure_reason_enum import ProcessingFailureReasonEnum
 from .processing_state_enum import ProcessingStateEnum
+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
 from .prompt_execution_meta import PromptExecutionMeta
@@ -286,6 +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 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
@@ -421,6 +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 import VariablePromptBlock
 from .vellum_audio import VellumAudio
 from .vellum_audio_request import VellumAudioRequest
 from .vellum_error import VellumError
@@ -519,6 +530,7 @@ __all__ = [
     "ChatMessage",
     "ChatMessageContent",
     "ChatMessageContentRequest",
+    "ChatMessagePromptBlock",
     "ChatMessageRequest",
     "ChatMessageRole",
     "CodeExecutionNodeArrayResult",
@@ -564,6 +576,8 @@ __all__ = [
     "EntityStatus",
     "EntityVisibility",
     "EnvironmentEnum",
+    "EphemeralPromptCacheConfig",
+    "EphemeralPromptCacheConfigTypeEnum",
     "ErrorInputRequest",
     "ErrorVariableValue",
     "ErrorVellumValue",
@@ -612,6 +626,7 @@ __all__ = [
     "FunctionCallVariableValue",
     "FunctionCallVellumValue",
     "FunctionCallVellumValueRequest",
+    "FunctionDefinitionPromptBlock",
     "GenerateOptionsRequest",
     "GenerateRequest",
     "GenerateResponse",
@@ -644,6 +659,7 @@ __all__ = [
     "InstructorVectorizerConfig",
     "InstructorVectorizerConfigRequest",
     "IterationStateEnum",
+    "JinjaPromptBlock",
     "JsonInputRequest",
     "JsonVariableValue",
     "JsonVellumValue",
@@ -728,9 +744,12 @@ __all__ = [
     "PaginatedWorkflowReleaseTagReadList",
     "PdfSearchResultMetaSource",
     "PdfSearchResultMetaSourceRequest",
+    "PlainTextPromptBlock",
     "Price",
     "ProcessingFailureReasonEnum",
     "ProcessingStateEnum",
+    "PromptBlock",
+    "PromptBlockState",
     "PromptDeploymentExpandMetaRequest",
     "PromptDeploymentInputRequest",
     "PromptExecutionMeta",
@@ -757,6 +776,8 @@ __all__ = [
     "RejectedWorkflowNodeResultEvent",
     "ReleaseTagSource",
     "ReplaceTestSuiteTestCaseRequest",
+    "RichTextChildBlock",
+    "RichTextPromptBlock",
     "SandboxScenario",
     "ScenarioInput",
     "ScenarioInputChatHistoryVariableValue",
@@ -888,6 +909,7 @@ __all__ = [
     "UnitEnum",
     "UploadDocumentResponse",
     "UpsertTestSuiteTestCaseRequest",
+    "VariablePromptBlock",
     "VellumAudio",
     "VellumAudioRequest",
     "VellumError",

vellum_ai-0.9.9/src/vellum/types/chat_message_prompt_block.py

@@ -0,0 +1,39 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .prompt_block_state import PromptBlockState
+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 ChatMessagePromptBlock(UniversalBaseModel):
+    """
+    A block that represents a chat message in a prompt template.
+    """
+
+    state: typing.Optional[PromptBlockState] = 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
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+from .prompt_block import PromptBlock  # noqa: E402
+
+update_forward_refs(ChatMessagePromptBlock)

vellum_ai-0.9.9/src/vellum/types/ephemeral_prompt_cache_config.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .ephemeral_prompt_cache_config_type_enum import EphemeralPromptCacheConfigTypeEnum
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class EphemeralPromptCacheConfig(UniversalBaseModel):
+    type: typing.Optional[EphemeralPromptCacheConfigTypeEnum] = 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.9.9/src/vellum/types/ephemeral_prompt_cache_config_type_enum.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+EphemeralPromptCacheConfigTypeEnum = typing.Literal["EPHEMERAL"]

vellum_ai-0.9.9/src/vellum/types/function_definition_prompt_block.py

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .prompt_block_state import PromptBlockState
+from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class FunctionDefinitionPromptBlock(UniversalBaseModel):
+    """
+    A block that represents a function definition in a prompt template.
+    """
+
+    state: typing.Optional[PromptBlockState] = 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
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum_ai-0.9.9/src/vellum/types/jinja_prompt_block.py

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .prompt_block_state import PromptBlockState
+from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
+from .vellum_variable_type import VellumVariableType
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class JinjaPromptBlock(UniversalBaseModel):
+    """
+    A block of Jinja template code that is used to generate a prompt
+    """
+
+    state: typing.Optional[PromptBlockState] = 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
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum_ai-0.9.9/src/vellum/types/plain_text_prompt_block.py

@@ -0,0 +1,28 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .prompt_block_state import PromptBlockState
+from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+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
+
+    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.9.9/src/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_ai-0.9.9/src/vellum/types/prompt_block_state.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+PromptBlockState = typing.Union[typing.Literal["ENABLED", "DISABLED"], typing.Any]

vellum_ai-0.9.9/src/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_ai-0.9.9/src/vellum/types/rich_text_prompt_block.py

@@ -0,0 +1,29 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .prompt_block_state import PromptBlockState
+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 RichTextPromptBlock(UniversalBaseModel):
+    """
+    A block that includes a combination of plain text and variable blocks.
+    """
+
+    state: typing.Optional[PromptBlockState] = 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
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum_ai-0.9.9/src/vellum/types/variable_prompt_block.py

@@ -0,0 +1,28 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .prompt_block_state import PromptBlockState
+from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class VariablePromptBlock(UniversalBaseModel):
+    """
+    A block that represents a variable in a prompt template.
+    """
+
+    state: typing.Optional[PromptBlockState] = None
+    cache_config: typing.Optional[EphemeralPromptCacheConfig] = None
+    block_type: typing.Literal["VARIABLE"] = "VARIABLE"
+    input_variable_id: str
+
+    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