vellum-ai 0.14.63__py3-none-any.whl → 1.13.5__py3-none-any.whl

vellum/client/types/open_ai_vectorizer_text_embedding_3_small.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .open_ai_vectorizer_config import OpenAiVectorizerConfig
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .open_ai_vectorizer_config import OpenAiVectorizerConfig
 
 
 class OpenAiVectorizerTextEmbedding3Small(UniversalBaseModel):

vellum/client/types/open_ai_vectorizer_text_embedding_3_small_request.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .open_ai_vectorizer_config_request import OpenAiVectorizerConfigRequest
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .open_ai_vectorizer_config_request import OpenAiVectorizerConfigRequest
 
 
 class OpenAiVectorizerTextEmbedding3SmallRequest(UniversalBaseModel):

vellum/client/types/open_ai_vectorizer_text_embedding_ada_002.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .open_ai_vectorizer_config import OpenAiVectorizerConfig
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .open_ai_vectorizer_config import OpenAiVectorizerConfig
 
 
 class OpenAiVectorizerTextEmbeddingAda002(UniversalBaseModel):

vellum/client/types/open_ai_vectorizer_text_embedding_ada_002_request.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .open_ai_vectorizer_config_request import OpenAiVectorizerConfigRequest
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .open_ai_vectorizer_config_request import OpenAiVectorizerConfigRequest
 
 
 class OpenAiVectorizerTextEmbeddingAda002Request(UniversalBaseModel):

vellum/client/types/organization_read.py

@@ -1,17 +1,20 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+import datetime as dt
 import typing
-from .new_member_join_behavior_enum import NewMemberJoinBehaviorEnum
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .new_member_join_behavior_enum import NewMemberJoinBehaviorEnum
 
 
 class OrganizationRead(UniversalBaseModel):
     id: str
     name: str
+    created: typing.Optional[dt.datetime] = None
     allow_staff_access: typing.Optional[bool] = None
     new_member_join_behavior: NewMemberJoinBehaviorEnum
+    limit_config: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/paginated_container_image_read_list.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .container_image_read import ContainerImageRead
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .container_image_read import ContainerImageRead
 
 
 class PaginatedContainerImageReadList(UniversalBaseModel):

vellum/client/types/paginated_deployment_release_tag_read_list.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .deployment_release_tag_read import DeploymentReleaseTagRead
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .deployment_release_tag_read import DeploymentReleaseTagRead
 
 
 class PaginatedDeploymentReleaseTagReadList(UniversalBaseModel):

vellum/client/types/paginated_document_index_read_list.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .document_index_read import DocumentIndexRead
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .document_index_read import DocumentIndexRead
 
 
 class PaginatedDocumentIndexReadList(UniversalBaseModel):

vellum/client/types/paginated_folder_entity_list.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .folder_entity import FolderEntity
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .folder_entity import FolderEntity
 
 
 class PaginatedFolderEntityList(UniversalBaseModel):

vellum/client/types/paginated_slim_deployment_read_list.py

@@ -1,11 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value import ArrayVellumValue
+from __future__ import annotations
+
 import typing
-from .slim_deployment_read import SlimDeploymentRead
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .slim_deployment_read import SlimDeploymentRead
 
 
 class PaginatedSlimDeploymentReadList(UniversalBaseModel):
@@ -22,3 +23,8 @@ class PaginatedSlimDeploymentReadList(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(PaginatedSlimDeploymentReadList)

vellum/client/types/paginated_slim_document_list.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .slim_document import SlimDocument
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .slim_document import SlimDocument
 
 
 class PaginatedSlimDocumentList(UniversalBaseModel):

vellum/client/types/paginated_slim_integration_auth_config_read_list.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .slim_integration_auth_config_read import SlimIntegrationAuthConfigRead
+
+
+class PaginatedSlimIntegrationAuthConfigReadList(UniversalBaseModel):
+    count: typing.Optional[int] = None
+    next: typing.Optional[str] = None
+    previous: typing.Optional[str] = None
+    results: typing.Optional[typing.List[SlimIntegrationAuthConfigRead]] = 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/client/types/paginated_slim_integration_read_list.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .slim_integration_read import SlimIntegrationRead
+
+
+class PaginatedSlimIntegrationReadList(UniversalBaseModel):
+    count: typing.Optional[int] = None
+    next: typing.Optional[str] = None
+    previous: typing.Optional[str] = None
+    results: typing.Optional[typing.List[SlimIntegrationRead]] = 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/client/types/paginated_slim_tool_definition_list.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .components_schemas_slim_composio_tool_definition import ComponentsSchemasSlimComposioToolDefinition
+
+
+class PaginatedSlimToolDefinitionList(UniversalBaseModel):
+    count: int
+    next: typing.Optional[str] = None
+    previous: typing.Optional[str] = None
+    results: typing.List[ComponentsSchemasSlimComposioToolDefinition]
+
+    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/client/types/paginated_slim_workflow_deployment_list.py

@@ -1,11 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value import ArrayVellumValue
+from __future__ import annotations
+
 import typing
-from .slim_workflow_deployment import SlimWorkflowDeployment
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .slim_workflow_deployment import SlimWorkflowDeployment
 
 
 class PaginatedSlimWorkflowDeploymentList(UniversalBaseModel):
@@ -22,3 +23,8 @@ class PaginatedSlimWorkflowDeploymentList(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(PaginatedSlimWorkflowDeploymentList)

vellum/client/types/paginated_test_suite_run_execution_list.py

@@ -1,11 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value import ArrayVellumValue
+from __future__ import annotations
+
 import typing
-from .test_suite_run_execution import TestSuiteRunExecution
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .test_suite_run_execution import TestSuiteRunExecution
 
 
 class PaginatedTestSuiteRunExecutionList(UniversalBaseModel):
@@ -22,3 +23,8 @@ class PaginatedTestSuiteRunExecutionList(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(PaginatedTestSuiteRunExecutionList)

vellum/client/types/paginated_test_suite_test_case_list.py

@@ -1,11 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value import ArrayVellumValue
+from __future__ import annotations
+
 import typing
-from .test_suite_test_case import TestSuiteTestCase
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .test_suite_test_case import TestSuiteTestCase
 
 
 class PaginatedTestSuiteTestCaseList(UniversalBaseModel):
@@ -22,3 +23,8 @@ class PaginatedTestSuiteTestCaseList(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(PaginatedTestSuiteTestCaseList)

vellum/client/types/paginated_workflow_deployment_release_list.py

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .workflow_deployment_release import WorkflowDeploymentRelease
+
+
+class PaginatedWorkflowDeploymentReleaseList(UniversalBaseModel):
+    count: typing.Optional[int] = None
+    next: typing.Optional[str] = None
+    previous: typing.Optional[str] = None
+    results: typing.Optional[typing.List[WorkflowDeploymentRelease]] = 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 .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(PaginatedWorkflowDeploymentReleaseList)

vellum/client/types/paginated_workflow_release_tag_read_list.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .workflow_release_tag_read import WorkflowReleaseTagRead
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .workflow_release_tag_read import WorkflowReleaseTagRead
 
 
 class PaginatedWorkflowReleaseTagReadList(UniversalBaseModel):

vellum/client/types/paginated_workflow_sandbox_example_list.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .workflow_sandbox_example import WorkflowSandboxExample
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .workflow_sandbox_example import WorkflowSandboxExample
 
 
 class PaginatedWorkflowSandboxExampleList(UniversalBaseModel):

vellum/client/types/parent_context.py

@@ -1,21 +1,31 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from __future__ import annotations
+
 import typing
-import typing
+from pydantic import Field
 
 if typing.TYPE_CHECKING:
-    from .workflow_parent_context import WorkflowParentContext
+    from .api_request_parent_context import ApiRequestParentContext
+    from .external_parent_context import ExternalParentContext
+    from .integration_trigger_context import IntegrationTriggerContext
     from .node_parent_context import NodeParentContext
+    from .prompt_deployment_parent_context import PromptDeploymentParentContext
+    from .scheduled_trigger_context import ScheduledTriggerContext
     from .workflow_deployment_parent_context import WorkflowDeploymentParentContext
+    from .workflow_parent_context import WorkflowParentContext
     from .workflow_sandbox_parent_context import WorkflowSandboxParentContext
-    from .prompt_deployment_parent_context import PromptDeploymentParentContext
-    from .api_request_parent_context import ApiRequestParentContext
-ParentContext = typing.Union[
-    "WorkflowParentContext",
-    "NodeParentContext",
-    "WorkflowDeploymentParentContext",
-    "WorkflowSandboxParentContext",
-    "PromptDeploymentParentContext",
-    "ApiRequestParentContext",
+ParentContext = typing.Annotated[
+    typing.Union[
+        "WorkflowParentContext",
+        "NodeParentContext",
+        "WorkflowDeploymentParentContext",
+        "WorkflowSandboxParentContext",
+        "PromptDeploymentParentContext",
+        "ApiRequestParentContext",
+        "ExternalParentContext",
+        "ScheduledTriggerContext",
+        "IntegrationTriggerContext",
+    ],
+    Field(discriminator="type")
 ]

vellum/client/types/pdf_search_result_meta_source.py

@@ -1,9 +1,9 @@
 # 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
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class PdfSearchResultMetaSource(UniversalBaseModel):

vellum/client/types/pdf_search_result_meta_source_request.py

@@ -1,9 +1,9 @@
 # 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
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class PdfSearchResultMetaSourceRequest(UniversalBaseModel):

vellum/client/types/plain_text_prompt_block.py

@@ -1,11 +1,11 @@
 # 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
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
+from .prompt_block_state import PromptBlockState
 
 
 class PlainTextPromptBlock(UniversalBaseModel):

vellum/client/types/price.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .unit_enum import UnitEnum
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .unit_enum import UnitEnum
 
 
 class Price(UniversalBaseModel):

vellum/client/types/private_vectorizer.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class PrivateVectorizer(UniversalBaseModel):
+    """
+    Serializer for private vectorizer.
+    """
+
+    model_name: typing.Literal["private-vectorizer"] = "private-vectorizer"
+
+    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/client/types/private_vectorizer_request.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class PrivateVectorizerRequest(UniversalBaseModel):
+    """
+    Serializer for private vectorizer.
+    """
+
+    model_name: typing.Literal["private-vectorizer"] = "private-vectorizer"
+
+    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/client/types/processing_failure_reason_enum.py

@@ -2,4 +2,6 @@
 
 import typing
 
-ProcessingFailureReasonEnum = typing.Union[typing.Literal["EXCEEDED_CHARACTER_LIMIT", "INVALID_FILE"], typing.Any]
+ProcessingFailureReasonEnum = typing.Union[
+    typing.Literal["EXCEEDED_CHARACTER_LIMIT", "INVALID_FILE", "INVALID_CREDENTIALS"], typing.Any
+]

vellum/client/types/prompt_block.py

@@ -1,15 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from __future__ import annotations
+
 import typing
-from .jinja_prompt_block import JinjaPromptBlock
-from .variable_prompt_block import VariablePromptBlock
-from .rich_text_prompt_block import RichTextPromptBlock
+
 from .audio_prompt_block import AudioPromptBlock
+from .document_prompt_block import DocumentPromptBlock
 from .function_call_prompt_block import FunctionCallPromptBlock
 from .image_prompt_block import ImagePromptBlock
-from .document_prompt_block import DocumentPromptBlock
-import typing
+from .jinja_prompt_block import JinjaPromptBlock
+from .rich_text_prompt_block import RichTextPromptBlock
+from .variable_prompt_block import VariablePromptBlock
+from .video_prompt_block import VideoPromptBlock
 
 if typing.TYPE_CHECKING:
     from .chat_message_prompt_block import ChatMessagePromptBlock
@@ -18,8 +20,9 @@ PromptBlock = typing.Union[
     "ChatMessagePromptBlock",
     VariablePromptBlock,
     RichTextPromptBlock,
-    AudioPromptBlock,
     FunctionCallPromptBlock,
+    AudioPromptBlock,
+    VideoPromptBlock,
     ImagePromptBlock,
     DocumentPromptBlock,
 ]

vellum/client/types/prompt_deployment_expand_meta_request.py

@@ -1,9 +1,9 @@
 # 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
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class PromptDeploymentExpandMetaRequest(UniversalBaseModel):

vellum/client/types/prompt_deployment_input_request.py

@@ -1,8 +1,21 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .string_input_request import StringInputRequest
-from .json_input_request import JsonInputRequest
+
+from .audio_input_request import AudioInputRequest
 from .chat_history_input_request import ChatHistoryInputRequest
+from .document_input_request import DocumentInputRequest
+from .image_input_request import ImageInputRequest
+from .json_input_request import JsonInputRequest
+from .string_input_request import StringInputRequest
+from .video_input_request import VideoInputRequest
 
-PromptDeploymentInputRequest = typing.Union[StringInputRequest, JsonInputRequest, ChatHistoryInputRequest]
+PromptDeploymentInputRequest = typing.Union[
+    StringInputRequest,
+    JsonInputRequest,
+    ChatHistoryInputRequest,
+    AudioInputRequest,
+    VideoInputRequest,
+    ImageInputRequest,
+    DocumentInputRequest,
+]