vellum-ai 0.1.13__py3-none-any.whl → 0.2.1__py3-none-any.whl

vellum/types/chat_message_content_request.py

@@ -0,0 +1,56 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+import typing_extensions
+
+from .array_chat_message_content_request import ArrayChatMessageContentRequest
+from .function_call_chat_message_content_request import FunctionCallChatMessageContentRequest
+from .image_chat_message_content_request import ImageChatMessageContentRequest
+from .string_chat_message_content_request import StringChatMessageContentRequest
+
+
+class ChatMessageContentRequest_String(StringChatMessageContentRequest):
+    type: typing_extensions.Literal["STRING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class ChatMessageContentRequest_FunctionCall(FunctionCallChatMessageContentRequest):
+    type: typing_extensions.Literal["FUNCTION_CALL"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class ChatMessageContentRequest_Array(ArrayChatMessageContentRequest):
+    type: typing_extensions.Literal["ARRAY"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class ChatMessageContentRequest_Image(ImageChatMessageContentRequest):
+    type: typing_extensions.Literal["IMAGE"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+ChatMessageContentRequest = typing.Union[
+    ChatMessageContentRequest_String,
+    ChatMessageContentRequest_FunctionCall,
+    ChatMessageContentRequest_Array,
+    ChatMessageContentRequest_Image,
+]

vellum/types/chat_message_request.py

@@ -4,6 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from .chat_message_content_request import ChatMessageContentRequest
 from .chat_message_role import ChatMessageRole
 
 try:
@@ -13,8 +14,9 @@ except ImportError:
 
 
 class ChatMessageRequest(pydantic.BaseModel):
-    text: str
+    text: typing.Optional[str]
     role: ChatMessageRole
+    content: typing.Optional[ChatMessageContentRequest]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

vellum/types/deployment_read.py

@@ -4,7 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .deployment_status import DeploymentStatus
+from .entity_status import EntityStatus
 from .environment_enum import EnvironmentEnum
 from .vellum_variable import VellumVariable
 
@@ -19,14 +19,8 @@ class DeploymentRead(pydantic.BaseModel):
     created: dt.datetime
     label: str = pydantic.Field(description="A human-readable label for the deployment")
     name: str = pydantic.Field(description="A name that uniquely identifies this deployment within its workspace")
-    status: typing.Optional[DeploymentStatus] = pydantic.Field(
-        description=(
-            "The current status of the deployment\n"
-            "\n"
-            "- `ACTIVE` - Active\n"
-            "- `INACTIVE` - Inactive\n"
-            "- `ARCHIVED` - Archived\n"
-        )
+    status: typing.Optional[EntityStatus] = pydantic.Field(
+        description=("The current status of the deployment\n" "\n" "- `ACTIVE` - Active\n" "- `ARCHIVED` - Archived\n")
     )
     environment: typing.Optional[EnvironmentEnum] = pydantic.Field(
         description=(
@@ -37,11 +31,11 @@ class DeploymentRead(pydantic.BaseModel):
             "- `PRODUCTION` - Production\n"
         )
     )
+    last_deployed_on: dt.datetime
+    input_variables: typing.List[VellumVariable]
     active_model_version_ids: typing.List[str] = pydantic.Field(
         description="Deprecated. The Prompt execution endpoints return a `prompt_version_id` that could be used instead."
     )
-    last_deployed_on: dt.datetime
-    input_variables: typing.List[VellumVariable]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

vellum/types/document_index_read.py

@@ -4,7 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .document_index_status import DocumentIndexStatus
+from .entity_status import EntityStatus
 from .environment_enum import EnvironmentEnum
 
 try:
@@ -18,7 +18,7 @@ class DocumentIndexRead(pydantic.BaseModel):
     created: dt.datetime
     label: str = pydantic.Field(description="A human-readable label for the document index")
     name: str = pydantic.Field(description="A name that uniquely identifies this index within its workspace")
-    status: typing.Optional[DocumentIndexStatus] = pydantic.Field(
+    status: typing.Optional[EntityStatus] = pydantic.Field(
         description=(
             "The current status of the document index\n" "\n" "- `ACTIVE` - Active\n" "- `ARCHIVED` - Archived\n"
         )

vellum/types/{document_index_status.py → entity_status.py}

@@ -6,7 +6,7 @@ import typing
 T_Result = typing.TypeVar("T_Result")
 
 
-class DocumentIndexStatus(str, enum.Enum):
+class EntityStatus(str, enum.Enum):
     """
     - `ACTIVE` - Active
     - `ARCHIVED` - Archived
@@ -16,7 +16,7 @@ class DocumentIndexStatus(str, enum.Enum):
     ARCHIVED = "ARCHIVED"
 
     def visit(self, active: typing.Callable[[], T_Result], archived: typing.Callable[[], T_Result]) -> T_Result:
-        if self is DocumentIndexStatus.ACTIVE:
+        if self is EntityStatus.ACTIVE:
             return active()
-        if self is DocumentIndexStatus.ARCHIVED:
+        if self is EntityStatus.ARCHIVED:
             return archived()

vellum/types/error_enum.py

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

vellum/types/execute_workflow_error_response.py

@@ -0,0 +1,28 @@
+# 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 ExecuteWorkflowErrorResponse(pydantic.BaseModel):
+    detail: str = pydantic.Field(description="Details about why the request failed.")
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/execute_workflow_response.py

@@ -0,0 +1,32 @@
+# 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 .execute_workflow_workflow_result_event import ExecuteWorkflowWorkflowResultEvent
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ExecuteWorkflowResponse(pydantic.BaseModel):
+    execution_id: str
+    run_id: typing.Optional[str]
+    external_id: typing.Optional[str]
+    data: ExecuteWorkflowWorkflowResultEvent
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/execute_workflow_workflow_result_event.py

@@ -0,0 +1,33 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+import typing_extensions
+
+from .fulfilled_execute_workflow_workflow_result_event import FulfilledExecuteWorkflowWorkflowResultEvent
+from .rejected_execute_workflow_workflow_result_event import RejectedExecuteWorkflowWorkflowResultEvent
+
+
+class ExecuteWorkflowWorkflowResultEvent_Fulfilled(FulfilledExecuteWorkflowWorkflowResultEvent):
+    state: typing_extensions.Literal["FULFILLED"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class ExecuteWorkflowWorkflowResultEvent_Rejected(RejectedExecuteWorkflowWorkflowResultEvent):
+    state: typing_extensions.Literal["REJECTED"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+ExecuteWorkflowWorkflowResultEvent = typing.Union[
+    ExecuteWorkflowWorkflowResultEvent_Fulfilled, ExecuteWorkflowWorkflowResultEvent_Rejected
+]

vellum/types/fulfilled_execute_workflow_workflow_result_event.py

@@ -0,0 +1,35 @@
+# 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 .workflow_output import WorkflowOutput
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class FulfilledExecuteWorkflowWorkflowResultEvent(pydantic.BaseModel):
+    """
+    The successful response from the Workflow execution containing the produced outputs.
+    """
+
+    id: str
+    ts: dt.datetime
+    outputs: typing.List[WorkflowOutput]
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/function_call_chat_message_content.py

@@ -0,0 +1,33 @@
+# 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 .function_call_chat_message_content_value import FunctionCallChatMessageContentValue
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class FunctionCallChatMessageContent(pydantic.BaseModel):
+    """
+    A function call value that is used in a chat message.
+    """
+
+    value: FunctionCallChatMessageContentValue
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/function_call_chat_message_content_request.py

@@ -0,0 +1,33 @@
+# 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 .function_call_chat_message_content_value_request import FunctionCallChatMessageContentValueRequest
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class FunctionCallChatMessageContentRequest(pydantic.BaseModel):
+    """
+    A function call value that is used in a chat message.
+    """
+
+    value: FunctionCallChatMessageContentValueRequest
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/function_call_chat_message_content_value.py

@@ -0,0 +1,34 @@
+# 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 FunctionCallChatMessageContentValue(pydantic.BaseModel):
+    """
+    The final resolved function call value.
+    """
+
+    name: str
+    arguments: typing.Dict[str, typing.Any]
+    id: typing.Optional[str]
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/function_call_chat_message_content_value_request.py

@@ -0,0 +1,34 @@
+# 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 FunctionCallChatMessageContentValueRequest(pydantic.BaseModel):
+    """
+    The final resolved function call value.
+    """
+
+    name: str
+    arguments: typing.Dict[str, typing.Any]
+    id: typing.Optional[str]
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/function_call_enum.py

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

vellum/types/image_chat_message_content.py

@@ -0,0 +1,33 @@
+# 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 .vellum_image import VellumImage
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ImageChatMessageContent(pydantic.BaseModel):
+    """
+    An image value that is used in a chat message.
+    """
+
+    value: VellumImage
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/image_chat_message_content_request.py

@@ -0,0 +1,33 @@
+# 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 .vellum_image_request import VellumImageRequest
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ImageChatMessageContentRequest(pydantic.BaseModel):
+    """
+    An image value that is used in a chat message.
+    """
+
+    value: VellumImageRequest
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/image_enum.py

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

vellum/types/json_enum.py

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

vellum/types/json_input_request.py

@@ -12,6 +12,10 @@ except ImportError:
 
 
 class JsonInputRequest(pydantic.BaseModel):
+    """
+    A user input representing a JSON object
+    """
+
     name: str = pydantic.Field(description="The variable's name, as defined in the deployment.")
     value: typing.Dict[str, typing.Any]
 

vellum/types/model_version_exec_config_parameters.py

@@ -20,6 +20,7 @@ class ModelVersionExecConfigParameters(pydantic.BaseModel):
     logit_bias: typing.Optional[typing.Dict[str, typing.Optional[float]]]
     stop: typing.Optional[typing.List[str]]
     top_k: typing.Optional[float]
+    custom_parameters: typing.Optional[typing.Dict[str, typing.Any]]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

vellum/types/number_enum.py

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

vellum/types/number_variable_value.py

@@ -0,0 +1,28 @@
+# 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 NumberVariableValue(pydantic.BaseModel):
+    value: typing.Optional[float]
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/paginated_slim_deployment_read_list.py

@@ -0,0 +1,32 @@
+# 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 .slim_deployment_read import SlimDeploymentRead
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class PaginatedSlimDeploymentReadList(pydantic.BaseModel):
+    count: typing.Optional[int]
+    next: typing.Optional[str]
+    previous: typing.Optional[str]
+    results: typing.Optional[typing.List[SlimDeploymentRead]]
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/paginated_slim_workflow_deployment_list.py

@@ -0,0 +1,32 @@
+# 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 .slim_workflow_deployment import SlimWorkflowDeployment
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class PaginatedSlimWorkflowDeploymentList(pydantic.BaseModel):
+    count: typing.Optional[int]
+    next: typing.Optional[str]
+    previous: typing.Optional[str]
+    results: typing.Optional[typing.List[SlimWorkflowDeployment]]
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/register_prompt_model_parameters_request.py

@@ -20,6 +20,7 @@ class RegisterPromptModelParametersRequest(pydantic.BaseModel):
     frequency_penalty: float
     presence_penalty: float
     logit_bias: typing.Optional[typing.Dict[str, typing.Optional[float]]]
+    custom_parameters: typing.Optional[typing.Dict[str, typing.Any]]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}