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

vellum/client/types/composio_tool_definition.py

@@ -0,0 +1,31 @@
+# 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 .integration import Integration
+
+
+class ComposioToolDefinition(UniversalBaseModel):
+    """
+    Serializer for Composio tool definition response.
+    """
+
+    provider: typing.Literal["COMPOSIO"] = "COMPOSIO"
+    integration: Integration
+    name: str
+    label: str
+    description: str
+    input_parameters: typing.Dict[str, typing.Optional[typing.Any]]
+    output_parameters: typing.Dict[str, typing.Optional[typing.Any]]
+    toolkit_version: 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/client/types/conditional_node_result.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 .conditional_node_result_data import ConditionalNodeResultData
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .conditional_node_result_data import ConditionalNodeResultData
 
 
 class ConditionalNodeResult(UniversalBaseModel):

vellum/client/types/conditional_node_result_data.py

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

vellum/client/types/container_image_build_config.py

@@ -0,0 +1,22 @@
+# 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 .code_execution_package import CodeExecutionPackage
+
+
+class ContainerImageBuildConfig(UniversalBaseModel):
+    packages: typing.List[CodeExecutionPackage]
+    user_script: typing.Optional[str] = None
+    hotswappable_version: typing.Optional[str] = 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/container_image_container_image_tag.py

@@ -1,15 +1,16 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import datetime as dt
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class ContainerImageContainerImageTag(UniversalBaseModel):
     name: str
     modified: dt.datetime
+    history_item_sha: typing.Optional[str] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/container_image_read.py

@@ -1,12 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .entity_visibility import EntityVisibility
 import datetime as dt
 import typing
-from .container_image_container_image_tag import ContainerImageContainerImageTag
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .build_status_enum import BuildStatusEnum
+from .container_image_build_config import ContainerImageBuildConfig
+from .container_image_container_image_tag import ContainerImageContainerImageTag
+from .entity_visibility import EntityVisibility
 
 
 class ContainerImageRead(UniversalBaseModel):
@@ -18,6 +20,8 @@ class ContainerImageRead(UniversalBaseModel):
     repository: str
     sha: str
     tags: typing.List[ContainerImageContainerImageTag]
+    build_status: typing.Optional[BuildStatusEnum] = None
+    build_config: typing.Optional[ContainerImageBuildConfig] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/create_test_suite_test_case_request.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_request import ArrayVellumValueRequest
+from __future__ import annotations
+
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 from .named_test_case_variable_value_request import NamedTestCaseVariableValueRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class CreateTestSuiteTestCaseRequest(UniversalBaseModel):
@@ -41,3 +42,8 @@ class CreateTestSuiteTestCaseRequest(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value_request import ArrayVellumValueRequest  # noqa: E402, F401, I001
+
+update_forward_refs(CreateTestSuiteTestCaseRequest)

vellum/client/types/create_workflow_event_request.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .workflow_event import WorkflowEvent
+
+CreateWorkflowEventRequest = typing.Union[typing.List[WorkflowEvent], WorkflowEvent]

vellum/client/types/dataset_row_push_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 DatasetRowPushRequest(UniversalBaseModel):
+    id: typing.Optional[str] = None
+    label: str
+    inputs: typing.Dict[str, typing.Optional[typing.Any]]
+    mocks: typing.Optional[typing.List[typing.Dict[str, typing.Optional[typing.Any]]]] = None
+    workflow_trigger_id: typing.Optional[str] = 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/delimiter_chunker_config.py

@@ -0,0 +1,20 @@
+# 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 DelimiterChunkerConfig(UniversalBaseModel):
+    delimiter: typing.Optional[str] = None
+    is_regex: 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

vellum/client/types/delimiter_chunker_config_request.py

@@ -0,0 +1,20 @@
+# 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 DelimiterChunkerConfigRequest(UniversalBaseModel):
+    delimiter: typing.Optional[str] = None
+    is_regex: 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

vellum/client/types/delimiter_chunking.py

@@ -0,0 +1,21 @@
+# 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 .delimiter_chunker_config import DelimiterChunkerConfig
+
+
+class DelimiterChunking(UniversalBaseModel):
+    chunker_name: typing.Literal["delimiter-chunker"] = "delimiter-chunker"
+    chunker_config: typing.Optional[DelimiterChunkerConfig] = 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/delimiter_chunking_request.py

@@ -0,0 +1,21 @@
+# 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 .delimiter_chunker_config_request import DelimiterChunkerConfigRequest
+
+
+class DelimiterChunkingRequest(UniversalBaseModel):
+    chunker_name: typing.Literal["delimiter-chunker"] = "delimiter-chunker"
+    chunker_config: typing.Optional[DelimiterChunkerConfigRequest] = 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/deployment_history_item.py

@@ -1,12 +1,13 @@
 # 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 datetime as dt
-import pydantic
 import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 from .vellum_variable import VellumVariable
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class DeploymentHistoryItem(UniversalBaseModel):
@@ -37,3 +38,8 @@ class DeploymentHistoryItem(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(DeploymentHistoryItem)

vellum/client/types/deployment_provider_payload_response.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .deployment_provider_payload_response_payload import DeploymentProviderPayloadResponsePayload
 import typing
-from .compile_prompt_meta import CompilePromptMeta
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .compile_prompt_meta import CompilePromptMeta
+from .deployment_provider_payload_response_payload import DeploymentProviderPayloadResponsePayload
 
 
 class DeploymentProviderPayloadResponse(UniversalBaseModel):

vellum/client/types/deployment_read.py

@@ -1,17 +1,22 @@
 # 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 datetime as dt
-import pydantic
 import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 from .entity_status import EntityStatus
 from .environment_enum import EnvironmentEnum
 from .vellum_variable import VellumVariable
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class DeploymentRead(UniversalBaseModel):
+    """
+    A Prompt Deployment's full details.
+    """
+
     id: str
     created: dt.datetime
     label: str = pydantic.Field()
@@ -34,11 +39,7 @@ class DeploymentRead(UniversalBaseModel):
 
     environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
     """
-    The environment this deployment is used in
-    
-    * `DEVELOPMENT` - Development
-    * `STAGING` - Staging
-    * `PRODUCTION` - Production
+    Deprecated. The value returned will always be 'PRODUCTION'.
     """
 
     last_deployed_on: dt.datetime
@@ -66,3 +67,8 @@ class DeploymentRead(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(DeploymentRead)

vellum/client/types/deployment_release_tag_deployment_history_item.py

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

vellum/client/types/deployment_release_tag_read.py

@@ -1,12 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+
 import pydantic
-from .release_tag_source import ReleaseTagSource
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .deployment_release_tag_deployment_history_item import DeploymentReleaseTagDeploymentHistoryItem
 from .release_tag_release import ReleaseTagRelease
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import typing
+from .release_tag_source import ReleaseTagSource
 
 
 class DeploymentReleaseTagRead(UniversalBaseModel):

vellum/client/types/deprecated_prompt_request_input.py

@@ -0,0 +1,8 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .prompt_request_input import PromptRequestInput
+from .workflow_input import WorkflowInput
+
+DeprecatedPromptRequestInput = typing.Union[WorkflowInput, PromptRequestInput]

vellum/client/types/docker_service_token.py

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

vellum/client/types/document_chat_message_content.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 .vellum_document import VellumDocument
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_document import VellumDocument
 
 
 class DocumentChatMessageContent(UniversalBaseModel):

vellum/client/types/document_chat_message_content_request.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 .vellum_document_request import VellumDocumentRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_document_request import VellumDocumentRequest
 
 
 class DocumentChatMessageContentRequest(UniversalBaseModel):

vellum/client/types/document_document_to_document_index.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import pydantic
 import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .indexing_state_enum import IndexingStateEnum
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class DocumentDocumentToDocumentIndex(UniversalBaseModel):
@@ -22,7 +22,7 @@ class DocumentDocumentToDocumentIndex(UniversalBaseModel):
     Vellum-generated ID that uniquely identifies the environment index this document is included in.
     """
 
-    document_index_id: str = pydantic.Field()
+    document_index_id: typing.Optional[str] = pydantic.Field(default=None)
     """
     Vellum-generated ID that uniquely identifies the index this document is included in.
     """
@@ -39,6 +39,7 @@ class DocumentDocumentToDocumentIndex(UniversalBaseModel):
     """
 
     extracted_text_file_url: typing.Optional[str] = None
+    processing_state: typing.Optional[str] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/document_index_chunking.py

@@ -1,8 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
+
+from .delimiter_chunking import DelimiterChunking
 from .reducto_chunking import ReductoChunking
 from .sentence_chunking import SentenceChunking
 from .token_overlapping_window_chunking import TokenOverlappingWindowChunking
 
-DocumentIndexChunking = typing.Union[ReductoChunking, SentenceChunking, TokenOverlappingWindowChunking]
+DocumentIndexChunking = typing.Union[
+    ReductoChunking, SentenceChunking, TokenOverlappingWindowChunking, DelimiterChunking
+]

vellum/client/types/document_index_chunking_request.py

@@ -1,10 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
+
+from .delimiter_chunking_request import DelimiterChunkingRequest
 from .reducto_chunking_request import ReductoChunkingRequest
 from .sentence_chunking_request import SentenceChunkingRequest
 from .token_overlapping_window_chunking_request import TokenOverlappingWindowChunkingRequest
 
 DocumentIndexChunkingRequest = typing.Union[
-    ReductoChunkingRequest, SentenceChunkingRequest, TokenOverlappingWindowChunkingRequest
+    ReductoChunkingRequest, SentenceChunkingRequest, TokenOverlappingWindowChunkingRequest, DelimiterChunkingRequest
 ]

vellum/client/types/document_index_indexing_config.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .indexing_config_vectorizer import IndexingConfigVectorizer
 import typing
-from .document_index_chunking import DocumentIndexChunking
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .document_index_chunking import DocumentIndexChunking
+from .indexing_config_vectorizer import IndexingConfigVectorizer
 
 
 class DocumentIndexIndexingConfig(UniversalBaseModel):

vellum/client/types/document_index_indexing_config_request.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .indexing_config_vectorizer_request import IndexingConfigVectorizerRequest
 import typing
-from .document_index_chunking_request import DocumentIndexChunkingRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .document_index_chunking_request import DocumentIndexChunkingRequest
+from .indexing_config_vectorizer_request import IndexingConfigVectorizerRequest
 
 
 class DocumentIndexIndexingConfigRequest(UniversalBaseModel):

vellum/client/types/document_index_read.py

@@ -1,12 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import datetime as dt
-import pydantic
 import typing
-from .entity_status import EntityStatus
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .document_index_indexing_config import DocumentIndexIndexingConfig
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from .entity_status import EntityStatus
 
 
 class DocumentIndexRead(UniversalBaseModel):

vellum/client/types/document_input.py

@@ -0,0 +1,30 @@
+# 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 .vellum_document import VellumDocument
+
+
+class DocumentInput(UniversalBaseModel):
+    """
+    A user input representing a Vellum Document value
+    """
+
+    name: str = pydantic.Field()
+    """
+    The variable's name
+    """
+
+    type: typing.Literal["DOCUMENT"] = "DOCUMENT"
+    value: VellumDocument
+
+    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/document_input_request.py

@@ -0,0 +1,30 @@
+# 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 .vellum_document_request import VellumDocumentRequest
+
+
+class DocumentInputRequest(UniversalBaseModel):
+    """
+    A user input representing a Vellum Document value
+    """
+
+    name: str = pydantic.Field()
+    """
+    The variable's name
+    """
+
+    type: typing.Literal["DOCUMENT"] = "DOCUMENT"
+    value: VellumDocumentRequest
+
+    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/document_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 DocumentPromptBlock(UniversalBaseModel):