vellum-ai 0.3.23__py3-none-any.whl → 0.5.0__py3-none-any.whl

vellum/types/prompt_template_block.py

@@ -1,41 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from __future__ import annotations
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .block_type_enum import BlockTypeEnum
-from .prompt_template_block_state import PromptTemplateBlockState
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class PromptTemplateBlock(pydantic.BaseModel):
-    id: str
-    block_type: BlockTypeEnum
-    properties: PromptTemplateBlockProperties
-    state: typing.Optional[PromptTemplateBlockState] = None
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
-
-from .prompt_template_block_properties import PromptTemplateBlockProperties  # noqa: E402
-
-PromptTemplateBlock.update_forward_refs()

vellum/types/prompt_template_block_data.py

@@ -1,31 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .prompt_template_block import PromptTemplateBlock
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class PromptTemplateBlockData(pydantic.BaseModel):
-    version: int
-    blocks: typing.List[PromptTemplateBlock]
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/prompt_template_block_data_request.py

@@ -1,31 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .prompt_template_block_request import PromptTemplateBlockRequest
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class PromptTemplateBlockDataRequest(pydantic.BaseModel):
-    version: int
-    blocks: typing.List[PromptTemplateBlockRequest]
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/prompt_template_block_properties.py

@@ -1,47 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from __future__ import annotations
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .chat_message_role import ChatMessageRole
-from .vellum_variable_type import VellumVariableType
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class PromptTemplateBlockProperties(pydantic.BaseModel):
-    chat_role: typing.Optional[ChatMessageRole] = None
-    chat_message_unterminated: typing.Optional[bool] = None
-    chat_source: typing.Optional[str] = None
-    template: typing.Optional[str] = None
-    template_type: typing.Optional[VellumVariableType] = None
-    function_name: typing.Optional[str] = None
-    function_description: typing.Optional[str] = None
-    function_parameters: typing.Optional[typing.Dict[str, typing.Any]] = None
-    function_forced: typing.Optional[bool] = None
-    blocks: typing.Optional[typing.List[PromptTemplateBlock]] = None
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
-
-from .prompt_template_block import PromptTemplateBlock  # noqa: E402
-
-PromptTemplateBlockProperties.update_forward_refs()

vellum/types/prompt_template_block_properties_request.py

@@ -1,47 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from __future__ import annotations
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .chat_message_role import ChatMessageRole
-from .vellum_variable_type import VellumVariableType
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class PromptTemplateBlockPropertiesRequest(pydantic.BaseModel):
-    chat_role: typing.Optional[ChatMessageRole] = None
-    chat_message_unterminated: typing.Optional[bool] = None
-    chat_source: typing.Optional[str] = None
-    template: typing.Optional[str] = None
-    template_type: typing.Optional[VellumVariableType] = None
-    function_name: typing.Optional[str] = None
-    function_description: typing.Optional[str] = None
-    function_parameters: typing.Optional[typing.Dict[str, typing.Any]] = None
-    function_forced: typing.Optional[bool] = None
-    blocks: typing.Optional[typing.List[PromptTemplateBlockRequest]] = None
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
-
-from .prompt_template_block_request import PromptTemplateBlockRequest  # noqa: E402
-
-PromptTemplateBlockPropertiesRequest.update_forward_refs()

vellum/types/prompt_template_block_request.py

@@ -1,41 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from __future__ import annotations
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .block_type_enum import BlockTypeEnum
-from .prompt_template_block_state import PromptTemplateBlockState
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class PromptTemplateBlockRequest(pydantic.BaseModel):
-    id: str
-    block_type: BlockTypeEnum
-    properties: PromptTemplateBlockPropertiesRequest
-    state: typing.Optional[PromptTemplateBlockState] = None
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
-
-from .prompt_template_block_properties_request import PromptTemplateBlockPropertiesRequest  # noqa: E402
-
-PromptTemplateBlockRequest.update_forward_refs()

vellum/types/prompt_template_block_state.py

@@ -1,22 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import enum
-import typing
-
-T_Result = typing.TypeVar("T_Result")
-
-
-class PromptTemplateBlockState(str, enum.Enum):
-    """
-    - `ENABLED` - ENABLED
-    - `DISABLED` - DISABLED
-    """
-
-    ENABLED = "ENABLED"
-    DISABLED = "DISABLED"
-
-    def visit(self, enabled: typing.Callable[[], T_Result], disabled: typing.Callable[[], T_Result]) -> T_Result:
-        if self is PromptTemplateBlockState.ENABLED:
-            return enabled()
-        if self is PromptTemplateBlockState.DISABLED:
-            return disabled()

vellum/types/provider_enum.py

@@ -1,81 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import enum
-import typing
-
-T_Result = typing.TypeVar("T_Result")
-
-
-class ProviderEnum(str, enum.Enum):
-    """
-    - `ANTHROPIC` - Anthropic
-    - `AWS_BEDROCK` - AWS Bedrock
-    - `AZURE_OPENAI` - Azure OpenAI
-    - `COHERE` - Cohere
-    - `GOOGLE` - Google
-    - `HOSTED` - Hosted
-    - `MOSAICML` - MosaicML
-    - `OPENAI` - OpenAI
-    - `FIREWORKS_AI` - Fireworks AI
-    - `HUGGINGFACE` - HuggingFace
-    - `MYSTIC` - Mystic
-    - `PYQ` - Pyq
-    - `REPLICATE` - Replicate
-    """
-
-    ANTHROPIC = "ANTHROPIC"
-    AWS_BEDROCK = "AWS_BEDROCK"
-    AZURE_OPENAI = "AZURE_OPENAI"
-    COHERE = "COHERE"
-    GOOGLE = "GOOGLE"
-    HOSTED = "HOSTED"
-    MOSAICML = "MOSAICML"
-    OPENAI = "OPENAI"
-    FIREWORKS_AI = "FIREWORKS_AI"
-    HUGGINGFACE = "HUGGINGFACE"
-    MYSTIC = "MYSTIC"
-    PYQ = "PYQ"
-    REPLICATE = "REPLICATE"
-
-    def visit(
-        self,
-        anthropic: typing.Callable[[], T_Result],
-        aws_bedrock: typing.Callable[[], T_Result],
-        azure_openai: typing.Callable[[], T_Result],
-        cohere: typing.Callable[[], T_Result],
-        google: typing.Callable[[], T_Result],
-        hosted: typing.Callable[[], T_Result],
-        mosaicml: typing.Callable[[], T_Result],
-        openai: typing.Callable[[], T_Result],
-        fireworks_ai: typing.Callable[[], T_Result],
-        huggingface: typing.Callable[[], T_Result],
-        mystic: typing.Callable[[], T_Result],
-        pyq: typing.Callable[[], T_Result],
-        replicate: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is ProviderEnum.ANTHROPIC:
-            return anthropic()
-        if self is ProviderEnum.AWS_BEDROCK:
-            return aws_bedrock()
-        if self is ProviderEnum.AZURE_OPENAI:
-            return azure_openai()
-        if self is ProviderEnum.COHERE:
-            return cohere()
-        if self is ProviderEnum.GOOGLE:
-            return google()
-        if self is ProviderEnum.HOSTED:
-            return hosted()
-        if self is ProviderEnum.MOSAICML:
-            return mosaicml()
-        if self is ProviderEnum.OPENAI:
-            return openai()
-        if self is ProviderEnum.FIREWORKS_AI:
-            return fireworks_ai()
-        if self is ProviderEnum.HUGGINGFACE:
-            return huggingface()
-        if self is ProviderEnum.MYSTIC:
-            return mystic()
-        if self is ProviderEnum.PYQ:
-            return pyq()
-        if self is ProviderEnum.REPLICATE:
-            return replicate()

vellum/types/register_prompt_model_parameters_request.py

@@ -1,37 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class RegisterPromptModelParametersRequest(pydantic.BaseModel):
-    temperature: float
-    max_tokens: int
-    stop: typing.Optional[typing.List[str]] = None
-    top_p: float
-    top_k: typing.Optional[int] = None
-    frequency_penalty: float
-    presence_penalty: float
-    logit_bias: typing.Optional[typing.Dict[str, typing.Optional[float]]] = None
-    custom_parameters: typing.Optional[typing.Dict[str, typing.Any]] = None
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/register_prompt_prompt_info_request.py

@@ -1,35 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .prompt_template_block_data_request import PromptTemplateBlockDataRequest
-from .registered_prompt_input_variable_request import RegisteredPromptInputVariableRequest
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class RegisterPromptPromptInfoRequest(pydantic.BaseModel):
-    prompt_block_data: PromptTemplateBlockDataRequest
-    input_variables: typing.List[RegisteredPromptInputVariableRequest] = pydantic.Field()
-    """
-    The input variables specified in the prompt template.
-    """
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/register_prompt_response.py

@@ -1,62 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .register_prompt_prompt import RegisterPromptPrompt
-from .registered_prompt_deployment import RegisteredPromptDeployment
-from .registered_prompt_model_version import RegisteredPromptModelVersion
-from .registered_prompt_sandbox import RegisteredPromptSandbox
-from .registered_prompt_sandbox_snapshot import RegisteredPromptSandboxSnapshot
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class RegisterPromptResponse(pydantic.BaseModel):
-    prompt: RegisterPromptPrompt = pydantic.Field()
-    """
-    Information about the generated prompt
-    """
-
-    sandbox_snapshot: RegisteredPromptSandboxSnapshot = pydantic.Field()
-    """
-    Information about the generated sandbox snapshot
-    """
-
-    sandbox: RegisteredPromptSandbox = pydantic.Field()
-    """
-    Information about the generated sandbox
-    """
-
-    model_version: RegisteredPromptModelVersion = pydantic.Field()
-    """
-    Information about the generated model version
-    """
-
-    prompt_version_id: str = pydantic.Field()
-    """
-    The ID of the generated prompt version
-    """
-
-    deployment: RegisteredPromptDeployment = pydantic.Field()
-    """
-    Information about the generated deployment
-    """
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/registered_prompt_deployment.py

@@ -1,42 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class RegisteredPromptDeployment(pydantic.BaseModel):
-    id: str = pydantic.Field()
-    """
-    The ID of the generated deployment.
-    """
-
-    name: str = pydantic.Field()
-    """
-    A uniquely-identifying name for generated deployment.
-    """
-
-    label: str = pydantic.Field()
-    """
-    A human-friendly label for the generated deployment.
-    """
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/scenario_input_request.py

@@ -1,34 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .chat_message_request import ChatMessageRequest
-from .scenario_input_type_enum import ScenarioInputTypeEnum
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class ScenarioInputRequest(pydantic.BaseModel):
-    key: str
-    type: typing.Optional[ScenarioInputTypeEnum] = None
-    value: typing.Optional[str] = None
-    chat_history: typing.Optional[typing.List[ChatMessageRequest]] = None
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/scenario_input_type_enum.py

@@ -1,22 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import enum
-import typing
-
-T_Result = typing.TypeVar("T_Result")
-
-
-class ScenarioInputTypeEnum(str, enum.Enum):
-    """
-    - `TEXT` - TEXT
-    - `CHAT_HISTORY` - CHAT_HISTORY
-    """
-
-    TEXT = "TEXT"
-    CHAT_HISTORY = "CHAT_HISTORY"
-
-    def visit(self, text: typing.Callable[[], T_Result], chat_history: typing.Callable[[], T_Result]) -> T_Result:
-        if self is ScenarioInputTypeEnum.TEXT:
-            return text()
-        if self is ScenarioInputTypeEnum.CHAT_HISTORY:
-            return chat_history()