vellum-ai 0.9.7__py3-none-any.whl → 0.9.8__py3-none-any.whl

vellum/__init__.py

@@ -37,6 +37,8 @@ from .types import (
     ChatMessage,
     ChatMessageContent,
     ChatMessageContentRequest,
+    ChatMessagePromptBlock,
+    ChatMessagePromptBlockProperties,
     ChatMessageRequest,
     ChatMessageRole,
     CodeExecutionNodeArrayResult,
@@ -82,6 +84,8 @@ from .types import (
     EntityStatus,
     EntityVisibility,
     EnvironmentEnum,
+    EphemeralPromptCacheConfig,
+    EphemeralPromptCacheConfigTypeEnum,
     ErrorInputRequest,
     ErrorVariableValue,
     ErrorVellumValue,
@@ -130,6 +134,8 @@ from .types import (
     FunctionCallVariableValue,
     FunctionCallVellumValue,
     FunctionCallVellumValueRequest,
+    FunctionDefinitionPromptBlock,
+    FunctionDefinitionPromptBlockProperties,
     GenerateOptionsRequest,
     GenerateRequest,
     GenerateResponse,
@@ -162,6 +168,8 @@ from .types import (
     InstructorVectorizerConfig,
     InstructorVectorizerConfigRequest,
     IterationStateEnum,
+    JinjaPromptBlock,
+    JinjaPromptBlockProperties,
     JsonInputRequest,
     JsonVariableValue,
     JsonVellumValue,
@@ -246,9 +254,12 @@ from .types import (
     PaginatedWorkflowReleaseTagReadList,
     PdfSearchResultMetaSource,
     PdfSearchResultMetaSourceRequest,
+    PlainTextPromptBlock,
     Price,
     ProcessingFailureReasonEnum,
     ProcessingStateEnum,
+    PromptBlock,
+    PromptBlockState,
     PromptDeploymentExpandMetaRequest,
     PromptDeploymentInputRequest,
     PromptExecutionMeta,
@@ -275,6 +286,8 @@ from .types import (
     RejectedWorkflowNodeResultEvent,
     ReleaseTagSource,
     ReplaceTestSuiteTestCaseRequest,
+    RichTextChildBlock,
+    RichTextPromptBlock,
     SandboxScenario,
     ScenarioInput,
     ScenarioInputChatHistoryVariableValue,
@@ -406,6 +419,7 @@ from .types import (
     UnitEnum,
     UploadDocumentResponse,
     UpsertTestSuiteTestCaseRequest,
+    VariablePromptBlock,
     VellumAudio,
     VellumAudioRequest,
     VellumError,
@@ -534,6 +548,8 @@ __all__ = [
     "ChatMessage",
     "ChatMessageContent",
     "ChatMessageContentRequest",
+    "ChatMessagePromptBlock",
+    "ChatMessagePromptBlockProperties",
     "ChatMessageRequest",
     "ChatMessageRole",
     "CodeExecutionNodeArrayResult",
@@ -581,6 +597,8 @@ __all__ = [
     "EntityStatus",
     "EntityVisibility",
     "EnvironmentEnum",
+    "EphemeralPromptCacheConfig",
+    "EphemeralPromptCacheConfigTypeEnum",
     "ErrorInputRequest",
     "ErrorVariableValue",
     "ErrorVellumValue",
@@ -631,6 +649,8 @@ __all__ = [
     "FunctionCallVariableValue",
     "FunctionCallVellumValue",
     "FunctionCallVellumValueRequest",
+    "FunctionDefinitionPromptBlock",
+    "FunctionDefinitionPromptBlockProperties",
     "GenerateOptionsRequest",
     "GenerateRequest",
     "GenerateResponse",
@@ -664,6 +684,8 @@ __all__ = [
     "InstructorVectorizerConfigRequest",
     "InternalServerError",
     "IterationStateEnum",
+    "JinjaPromptBlock",
+    "JinjaPromptBlockProperties",
     "JsonInputRequest",
     "JsonVariableValue",
     "JsonVellumValue",
@@ -751,9 +773,12 @@ __all__ = [
     "PaginatedWorkflowReleaseTagReadList",
     "PdfSearchResultMetaSource",
     "PdfSearchResultMetaSourceRequest",
+    "PlainTextPromptBlock",
     "Price",
     "ProcessingFailureReasonEnum",
     "ProcessingStateEnum",
+    "PromptBlock",
+    "PromptBlockState",
     "PromptDeploymentExpandMetaRequest",
     "PromptDeploymentInputRequest",
     "PromptExecutionMeta",
@@ -780,6 +805,8 @@ __all__ = [
     "RejectedWorkflowNodeResultEvent",
     "ReleaseTagSource",
     "ReplaceTestSuiteTestCaseRequest",
+    "RichTextChildBlock",
+    "RichTextPromptBlock",
     "SandboxScenario",
     "ScenarioInput",
     "ScenarioInputChatHistoryVariableValue",
@@ -911,6 +938,7 @@ __all__ = [
     "UnitEnum",
     "UploadDocumentResponse",
     "UpsertTestSuiteTestCaseRequest",
+    "VariablePromptBlock",
     "Vellum",
     "VellumAudio",
     "VellumAudioRequest",

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.8",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

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,9 @@ class AdHocClient:
         --------
         from vellum import (
             AdHocExpandMeta,
+            EphemeralPromptCacheConfig,
+            JinjaPromptBlock,
+            JinjaPromptBlockProperties,
             PromptParameters,
             PromptRequestStringInput,
             PromptSettings,
@@ -118,7 +122,16 @@ class AdHocClient:
             settings=PromptSettings(
                 timeout=1.1,
             ),
-            blocks=[{"key": "value"}],
+            blocks=[
+                JinjaPromptBlock(
+                    state="ENABLED",
+                    cache_config=EphemeralPromptCacheConfig(),
+                    properties=JinjaPromptBlockProperties(
+                        template="string",
+                        template_type="STRING",
+                    ),
+                )
+            ],
             expand_meta=AdHocExpandMeta(
                 cost=True,
                 model_name=True,
@@ -147,7 +160,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 +234,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 +252,7 @@ class AsyncAdHocClient:
 
         parameters : PromptParameters
 
-        blocks : typing.Sequence[typing.Optional[typing.Any]]
+        blocks : typing.Sequence[PromptBlock]
 
         settings : typing.Optional[PromptSettings]
 
@@ -258,6 +273,9 @@ class AsyncAdHocClient:
         from vellum import (
             AdHocExpandMeta,
             AsyncVellum,
+            EphemeralPromptCacheConfig,
+            JinjaPromptBlock,
+            JinjaPromptBlockProperties,
             PromptParameters,
             PromptRequestStringInput,
             PromptSettings,
@@ -308,7 +326,16 @@ class AsyncAdHocClient:
                 settings=PromptSettings(
                     timeout=1.1,
                 ),
-                blocks=[{"key": "value"}],
+                blocks=[
+                    JinjaPromptBlock(
+                        state="ENABLED",
+                        cache_config=EphemeralPromptCacheConfig(),
+                        properties=JinjaPromptBlockProperties(
+                            template="string",
+                            template_type="STRING",
+                        ),
+                    )
+                ],
                 expand_meta=AdHocExpandMeta(
                     cost=True,
                     model_name=True,
@@ -340,7 +367,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/types/__init__.py

@@ -44,6 +44,8 @@ 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_prompt_block_properties import ChatMessagePromptBlockProperties
 from .chat_message_request import ChatMessageRequest
 from .chat_message_role import ChatMessageRole
 from .code_execution_node_array_result import CodeExecutionNodeArrayResult
@@ -89,6 +91,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 +141,8 @@ 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 .function_definition_prompt_block_properties import FunctionDefinitionPromptBlockProperties
 from .generate_options_request import GenerateOptionsRequest
 from .generate_request import GenerateRequest
 from .generate_response import GenerateResponse
@@ -173,6 +179,8 @@ 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 .jinja_prompt_block_properties import JinjaPromptBlockProperties
 from .json_input_request import JsonInputRequest
 from .json_variable_value import JsonVariableValue
 from .json_vellum_value import JsonVellumValue
@@ -257,9 +265,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 +297,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 +434,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 +533,8 @@ __all__ = [
     "ChatMessage",
     "ChatMessageContent",
     "ChatMessageContentRequest",
+    "ChatMessagePromptBlock",
+    "ChatMessagePromptBlockProperties",
     "ChatMessageRequest",
     "ChatMessageRole",
     "CodeExecutionNodeArrayResult",
@@ -564,6 +580,8 @@ __all__ = [
     "EntityStatus",
     "EntityVisibility",
     "EnvironmentEnum",
+    "EphemeralPromptCacheConfig",
+    "EphemeralPromptCacheConfigTypeEnum",
     "ErrorInputRequest",
     "ErrorVariableValue",
     "ErrorVellumValue",
@@ -612,6 +630,8 @@ __all__ = [
     "FunctionCallVariableValue",
     "FunctionCallVellumValue",
     "FunctionCallVellumValueRequest",
+    "FunctionDefinitionPromptBlock",
+    "FunctionDefinitionPromptBlockProperties",
     "GenerateOptionsRequest",
     "GenerateRequest",
     "GenerateResponse",
@@ -644,6 +664,8 @@ __all__ = [
     "InstructorVectorizerConfig",
     "InstructorVectorizerConfigRequest",
     "IterationStateEnum",
+    "JinjaPromptBlock",
+    "JinjaPromptBlockProperties",
     "JsonInputRequest",
     "JsonVariableValue",
     "JsonVellumValue",
@@ -728,9 +750,12 @@ __all__ = [
     "PaginatedWorkflowReleaseTagReadList",
     "PdfSearchResultMetaSource",
     "PdfSearchResultMetaSourceRequest",
+    "PlainTextPromptBlock",
     "Price",
     "ProcessingFailureReasonEnum",
     "ProcessingStateEnum",
+    "PromptBlock",
+    "PromptBlockState",
     "PromptDeploymentExpandMetaRequest",
     "PromptDeploymentInputRequest",
     "PromptExecutionMeta",
@@ -757,6 +782,8 @@ __all__ = [
     "RejectedWorkflowNodeResultEvent",
     "ReleaseTagSource",
     "ReplaceTestSuiteTestCaseRequest",
+    "RichTextChildBlock",
+    "RichTextPromptBlock",
     "SandboxScenario",
     "ScenarioInput",
     "ScenarioInputChatHistoryVariableValue",
@@ -888,6 +915,7 @@ __all__ = [
     "UnitEnum",
     "UploadDocumentResponse",
     "UpsertTestSuiteTestCaseRequest",
+    "VariablePromptBlock",
     "VellumAudio",
     "VellumAudioRequest",
     "VellumError",

vellum/types/chat_message_prompt_block.py

@@ -0,0 +1,36 @@
+# 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 ..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"
+    properties: "ChatMessagePromptBlockProperties"
+
+    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 .chat_message_prompt_block_properties import ChatMessagePromptBlockProperties  # noqa: E402
+
+update_forward_refs(ChatMessagePromptBlockProperties, ChatMessagePromptBlock=ChatMessagePromptBlock)
+update_forward_refs(ChatMessagePromptBlock)

vellum/types/chat_message_prompt_block_properties.py

@@ -0,0 +1,36 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+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 ChatMessagePromptBlockProperties(UniversalBaseModel):
+    """
+    The properties of a ChatMessagePromptTemplateBlock
+    """
+
+    blocks: typing.List["PromptBlock"]
+    chat_role: typing.Optional[ChatMessageRole] = None
+    chat_source: typing.Optional[str] = None
+    chat_message_unterminated: typing.Optional[bool] = 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
+
+
+from .chat_message_prompt_block import ChatMessagePromptBlock  # noqa: E402
+from .prompt_block import PromptBlock  # noqa: E402
+
+update_forward_refs(ChatMessagePromptBlock, ChatMessagePromptBlockProperties=ChatMessagePromptBlockProperties)
+update_forward_refs(ChatMessagePromptBlockProperties)

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/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/types/function_definition_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 .function_definition_prompt_block_properties import FunctionDefinitionPromptBlockProperties
+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"
+    properties: FunctionDefinitionPromptBlockProperties
+
+    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/types/function_definition_prompt_block_properties.py

@@ -0,0 +1,42 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class FunctionDefinitionPromptBlockProperties(UniversalBaseModel):
+    function_name: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The name identifying the function.
+    """
+
+    function_description: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    A description to help guide the model when to invoke this function.
+    """
+
+    function_parameters: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
+    """
+    An OpenAPI specification of parameters that are supported by this function.
+    """
+
+    function_forced: typing.Optional[bool] = pydantic.Field(default=None)
+    """
+    Set this option to true to force the model to return a function call of this function.
+    """
+
+    function_strict: typing.Optional[bool] = pydantic.Field(default=None)
+    """
+    Set this option to use strict schema decoding when available.
+    """
+
+    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/types/jinja_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 .jinja_prompt_block_properties import JinjaPromptBlockProperties
+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"
+    properties: JinjaPromptBlockProperties
+
+    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/types/jinja_prompt_block_properties.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .vellum_variable_type import VellumVariableType
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class JinjaPromptBlockProperties(UniversalBaseModel):
+    template: typing.Optional[str] = None
+    template_type: typing.Optional[VellumVariableType] = 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/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/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_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/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.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/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

{vellum_ai-0.9.7.dist-info → vellum_ai-0.9.8.dist-info}/METADATA

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

{vellum_ai-0.9.7.dist-info → vellum_ai-0.9.8.dist-info}/RECORD

@@ -1,8 +1,8 @@
-vellum/__init__.py,sha256=XLTmPODTz4SX5gIYT6qkg5WfBvqH1Jq2h5w9dZGF0_s,33632
+vellum/__init__.py,sha256=ymG1XhLJf664BANGlkkkn9WQYO3OBdOq9Hmmey5IQDA,34482
 vellum/client.py,sha256=kG4b9g1Jjm6zgzGBXCAYXcM_3xNQfBsa2Xut6F0eHQM,115201
 vellum/core/__init__.py,sha256=SQ85PF84B9MuKnBwHNHWemSGuy-g_515gFYNFhvEE0I,1438
 vellum/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-vellum/core/client_wrapper.py,sha256=HejxYCB47RVZhppXHM51_7mVBa8ycn6A6-5EmrNTnok,1889
+vellum/core/client_wrapper.py,sha256=59_RRff6AtadFRoSTsynbNZ3BIbnSEkZepjV_9c7XIQ,1889
 vellum/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 vellum/core/file.py,sha256=X9IbmkZmB2bB_DpmZAO3crWdXagOakAyn6UCOCImCPg,2322
 vellum/core/http_client.py,sha256=R0pQpCppnEtxccGvXl4uJ76s7ro_65Fo_erlNNLp_AI,19228
@@ -32,7 +32,7 @@ vellum/lib/utils/uuid.py,sha256=nedyhTNQDS2YvrU5gL3PtvG9cgGH87yKOcpGDJAe44E,214
 vellum/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/resources/__init__.py,sha256=6tqe3AwLJGLW38iua0Tje0n3uz3a4vkqMFxbUJGRs98,1346
 vellum/resources/ad_hoc/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-vellum/resources/ad_hoc/client.py,sha256=JOdjssV2rLBHHOO0wVn5fFQ1-1-mfQZ4mQHCt_iGr7I,15291
+vellum/resources/ad_hoc/client.py,sha256=IMQyFBK007oQDfGion1rcSVjX0dIhFeDGxI6H5HjgJA,16467
 vellum/resources/container_images/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 vellum/resources/container_images/client.py,sha256=jK1n-NFsdBKCeEKh-EIqvw7R8AG9PP4GxxcoH9F0GYs,15463
 vellum/resources/deployments/__init__.py,sha256=m64MNuPx3qVazOnTNwOY8oEeDrAkNwMJvUEe5xoMDvs,239
@@ -82,7 +82,7 @@ vellum/terraform/ml_model/__init__.py,sha256=I8h1Ru-Rb-Hi_HusK6G7nJQZEKQGsAAHMmw
 vellum/terraform/provider/__init__.py,sha256=-06xKmAmknpohVzw5TD-t1bnUHta8OrQYqvMd04XM-U,12684
 vellum/terraform/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 vellum/terraform/versions.json,sha256=45c7jjRD5i4w9DJQHs5ZqLLVXRnQwP9Rirq3mWY-xEo,56
-vellum/types/__init__.py,sha256=S1Z3m1icSICPDg0gPStafY6z_drQo2nByLdngrmg1MQ,50767
+vellum/types/__init__.py,sha256=PzW_-1tQv-UQHJB9_s4EXd2Jis0h3ptz-hBVvnWAt1U,52111
 vellum/types/ad_hoc_execute_prompt_event.py,sha256=bCjujA2XsOgyF3bRZbcEqV2rOIymRgsLoIRtZpB14xg,607
 vellum/types/ad_hoc_expand_meta.py,sha256=1gv-NCsy_6xBYupLvZH979yf2VMdxAU-l0y0ynMKZaw,1331
 vellum/types/ad_hoc_fulfilled_prompt_execution_meta.py,sha256=Bfvf1d_dkmshxRACVM5vcxbH_7AQY23RmrrnPc0ytYY,939
@@ -119,6 +119,8 @@ vellum/types/chat_history_vellum_value_request.py,sha256=HzAiysncG5unJ-tlb43HhGZ
 vellum/types/chat_message.py,sha256=EOA8v5Ebx2KS9BtwBBGbuvSK-pn4xWYZiioHuuPWvzw,916
 vellum/types/chat_message_content.py,sha256=DQLB5rG40qLRLsmKWWo-XKa4rhk9TGQs_eFTFow2zEM,607
 vellum/types/chat_message_content_request.py,sha256=iFT_PmN6sUjeN1_fZXr2ePJEbSq_GZcClBvtu8SdVmQ,724
+vellum/types/chat_message_prompt_block.py,sha256=fnr8atRGphzLeevqrTl-dxvuCPcvaQZY6kF7TligUD4,1326
+vellum/types/chat_message_prompt_block_properties.py,sha256=_XukG09TPcw7kYiXHZrp9vdYAmVVtNwACiiddOql378,1268
 vellum/types/chat_message_request.py,sha256=r2EW1pfnvNYx2fo6mBqU5HQrUzp67WXuE5G-XK281E4,945
 vellum/types/chat_message_role.py,sha256=-i0Jrcbwf72MkMoaFTGyxRduvlN7f5Y9ULhCXR5KNdA,182
 vellum/types/code_execution_node_array_result.py,sha256=KCdbmjXjReO-hPPpBsSR17h_roDUpc4R-92cmIn59ck,952
@@ -164,6 +166,8 @@ vellum/types/enriched_normalized_completion.py,sha256=qk8o_PCj8rvgxa9omng0Nx5SvG
 vellum/types/entity_status.py,sha256=bY0jEpISwXqFnbWd3PSb3yXEr-ounPXlAO_fyvHV7l8,158
 vellum/types/entity_visibility.py,sha256=BX1KdYd7dirpv878XDDvtOHkMOqebM8-lkWmLyFLaw4,184
 vellum/types/environment_enum.py,sha256=Wcewxp1cpGAMDIAZbTp4Y0GGfvy2Bq_Qu_67f_wBDGA,179
+vellum/types/ephemeral_prompt_cache_config.py,sha256=PK26yfHCLPAnXd89r8gz0rCD9hRR5pjKrE-o6-Mk-dI,712
+vellum/types/ephemeral_prompt_cache_config_type_enum.py,sha256=houFyNbNr9r2uXTBoRTA0eZJaBDe0CFTIUbsDyWK2e4,145
 vellum/types/error_input_request.py,sha256=iWHzxp5DBkVPsinv8sDkf5m51YxzNCyQzRoO-ULaSz0,815
 vellum/types/error_variable_value.py,sha256=ULT_D9h8dKeHTnXEDv6UHxNlpmwQ3iDHTGU6Zg8a6H0,676
 vellum/types/error_vellum_value.py,sha256=Y7yO8Y1eOlbRyOqbOh6MyPRNHYKH48gIhlZIRcw1UaA,726
@@ -212,6 +216,8 @@ vellum/types/function_call_request.py,sha256=eJBIN-wLkkkDUIwAy1nMeWHu3MZ5aJpOXyW
 vellum/types/function_call_variable_value.py,sha256=VQKCiEtJsmIK3i7CtFV_2ZpxeX70rqpUViXIvAci8L0,702
 vellum/types/function_call_vellum_value.py,sha256=lLJb-S_-S_UXm6una1BMyCbqLpMhbbMcaVIYNO45h5o,759
 vellum/types/function_call_vellum_value_request.py,sha256=oUteuCfWcj7UJbSE_Vywmmva9kyTaeL9iv5WJHabDVs,788
+vellum/types/function_definition_prompt_block.py,sha256=vz3ywLJEuBs5NEmTNcS6L0k-Wezc-5zT0avQ0DWzdOg,1118
+vellum/types/function_definition_prompt_block_properties.py,sha256=0D1yrSnCKQoJQlYu8Kk9AHItuc4LOIuz9LUoB4msTcw,1416
 vellum/types/generate_options_request.py,sha256=TUDqsH0tiPWDZH4T-p5gsvKvwVHEVZ_k6oI3qsjlsk4,782
 vellum/types/generate_request.py,sha256=gL6ywAJe6YCJ5oKbtYwL2H_TMdC_6PJZAI7-P3UOF3I,1286
 vellum/types/generate_response.py,sha256=QJmSRsYhZhtDmk2xpE9ueQEkHyXmYsaEQqqlKl9-bS4,699
@@ -244,6 +250,8 @@ vellum/types/initiated_workflow_node_result_event.py,sha256=Nu1J4iQYsW2HHjQFzQq-
 vellum/types/instructor_vectorizer_config.py,sha256=7udlosXv4CUWTW_Q9m0mz3VRi1FKSbBhDIOhtxRd0-U,731
 vellum/types/instructor_vectorizer_config_request.py,sha256=6LGFFQKntMfX7bdetUqEMVdr3KJHEps0oDp2bNmqWbM,738
 vellum/types/iteration_state_enum.py,sha256=83JSh842OJgQiLtNn1KMimy6RlEYRVH3mDmYWS6Ewzo,180
+vellum/types/jinja_prompt_block.py,sha256=_0muggk56IM0SfzUmvqt71WvLNumn1UQ-HdJnYLsQqY,1034
+vellum/types/jinja_prompt_block_properties.py,sha256=zC0AfT86xaLVVX8WkWLADoMonegv1jQvpked0DIhlEs,712
 vellum/types/json_input_request.py,sha256=x5sA-VXxF4QH-98xRcIKPZhsMVbnJNUQofiUQqyfGk4,768
 vellum/types/json_variable_value.py,sha256=X7eBEWxuozfvIdqD5sIZ5L-L77Ou6IIsZaQVNXh5G2k,634
 vellum/types/json_vellum_value.py,sha256=8irlw6NkRRVafysfTc1Q5BFFhRrWJYzdwrDYTdJK4JY,689
@@ -328,9 +336,12 @@ vellum/types/paginated_test_suite_test_case_list.py,sha256=9KrCCQKy0egMmVx5U2k6o
 vellum/types/paginated_workflow_release_tag_read_list.py,sha256=dH24ESWyAMVtyHsBkxG8kJ9oORY04Wn3IN-7jvV7Lu4,818
 vellum/types/pdf_search_result_meta_source.py,sha256=EMVhqdN1bwE6Ujdx4VhlmKQtJvitN-57kY8oZPxh9dI,1126
 vellum/types/pdf_search_result_meta_source_request.py,sha256=nUhaD2Kw1paGC6O_ICVNu3R0e1SVgTshRTkGNgmcjXo,1133
+vellum/types/plain_text_prompt_block.py,sha256=GlzovdRbT6mhunMvIpF94_0apjh_0SZtkdlx5HFrE0k,930
 vellum/types/price.py,sha256=ewzXDBVLaleuXMVQ-gQ3G1Nl5J2OWOVEMEFfnQIpiTk,610
 vellum/types/processing_failure_reason_enum.py,sha256=R_KIW7TcQejhc-vLhtNf9SdkYADgoZCn4ch4_RRIvsI,195
 vellum/types/processing_state_enum.py,sha256=lIEunnCpgYQExm2bGyTb12KyjQ3O7XOx636aWXb_Iwo,190
+vellum/types/prompt_block.py,sha256=qJISp3lSZ0EVrTuqs9306KO14mZPQmj-yeghBqqfBao,606
+vellum/types/prompt_block_state.py,sha256=BRAzTYARoSU36IVZGWMeeqhl5fgFMXCyhJ8rCbfB-f0,163
 vellum/types/prompt_deployment_expand_meta_request.py,sha256=agsiAaHB6lDoZPlnfJ2nmhB4Ud4EiJJTX05YmduyCPo,1910
 vellum/types/prompt_deployment_input_request.py,sha256=KrT4-Ew2VvTWXEkYQz2oyHn5EDOgrMW7FzRFaPH3ARg,353
 vellum/types/prompt_execution_meta.py,sha256=3hhMZgdAR5mKfnh2e_eVN3oKfT0E9w26khVPrpjn7jk,1141
@@ -357,6 +368,8 @@ vellum/types/rejected_prompt_execution_meta.py,sha256=4Oidh3XBBHzgjOys2kuETxqWc_
 vellum/types/rejected_workflow_node_result_event.py,sha256=o9AUc9hT60F8ckMkCx8HtKxtj4y82LBoJduB9u8w0TM,1319
 vellum/types/release_tag_source.py,sha256=YavosOXZ976yfXTNWRTZwh2HhRiYmSDk0bQCkl-jCoQ,158
 vellum/types/replace_test_suite_test_case_request.py,sha256=c1GT1RUCei1yWxyZy4Gv40PkXYisvK5OkzlqQ6WeBYA,1906
+vellum/types/rich_text_child_block.py,sha256=X_ACKFKSUx5SXT1cLp0Y5-7VrNxcGOggPm67Lk2442U,270
+vellum/types/rich_text_prompt_block.py,sha256=60Bg_1wuXAd5eBiO9xiN8NjMvYE67ZLTtAb1A3N-7oI,1036
 vellum/types/sandbox_scenario.py,sha256=f4S0tDxmPYHIrD_BMjRL3XZGcGxlWy9apfI64hV7MBE,844
 vellum/types/scenario_input.py,sha256=caC8p5ZnuXhv2ZiHvzTvKL3Ebtr4NXP9Q8UcJEVB8-U,476
 vellum/types/scenario_input_chat_history_variable_value.py,sha256=ihgJktmZpz_12wx2yPtyrSrBjcBcSg9fby7h_eLoXgI,835
@@ -488,6 +501,7 @@ vellum/types/token_overlapping_window_chunking_request.py,sha256=IjCs9UDrwBT6tnf
 vellum/types/unit_enum.py,sha256=BKWRVp2WfHtGK4D6TsolhNJHGHfExzrRHkFn8H8QkwQ,113
 vellum/types/upload_document_response.py,sha256=6_5Cm4yBPq5nD-rEql6GsmrAtSVVtNRczOL5YwsBVMI,649
 vellum/types/upsert_test_suite_test_case_request.py,sha256=iB38vx4mo4yNLV5XTeXMGR-PJLOQPloWQOAAi7PDpM0,2079
+vellum/types/variable_prompt_block.py,sha256=cQezrQTHZRN4anIZDv21a6VEUfjYdCkDLqA7OrNGSBs,949
 vellum/types/vellum_audio.py,sha256=oPm1bcxk7fTfWfHWOPSLvrZrRBjCyPDVDRMACPoWmMI,721
 vellum/types/vellum_audio_request.py,sha256=y9CZgQ1TteW0AHNk8GuAZLNVFa981rh7P9vyV8bfgys,728
 vellum/types/vellum_error.py,sha256=jCKfuCkDTiyFb1-QyP2cg0wReja6wMuooKPAjNhBA0M,643
@@ -549,7 +563,7 @@ vellum/types/workflow_result_event_output_data_string.py,sha256=tM3kgh6tEhD0dFEb
 vellum/types/workflow_stream_event.py,sha256=Wn3Yzuy9MqWAeo8tEaXDTKDEbJoA8DdYdMVq8EKuhu8,361
 vellum/types/workspace_secret_read.py,sha256=3CnHDG72IAY0KRNvc31F0xLmhnpwjQHnDYCfQJzCxI0,714
 vellum/version.py,sha256=jq-1PlAYxN9AXuaZqbYk9ak27SgE2lw9Ia5gx1b1gVI,76
-vellum_ai-0.9.7.dist-info/LICENSE,sha256=CcaljEIoOBaU-wItPH4PmM_mDCGpyuUY0Er1BGu5Ti8,1073
-vellum_ai-0.9.7.dist-info/METADATA,sha256=hNHe7CoBAqYIH-ArQUtMpzL0iiyA_nHyJVyI-aNRlD0,4394
-vellum_ai-0.9.7.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-vellum_ai-0.9.7.dist-info/RECORD,,
+vellum_ai-0.9.8.dist-info/LICENSE,sha256=CcaljEIoOBaU-wItPH4PmM_mDCGpyuUY0Er1BGu5Ti8,1073
+vellum_ai-0.9.8.dist-info/METADATA,sha256=XQ67tx6y1xy6HGrnZfvJsudcGjCZDEPjC-52aBSuZdA,4394
+vellum_ai-0.9.8.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+vellum_ai-0.9.8.dist-info/RECORD,,