vellum-ai 1.1.3__py3-none-any.whl → 1.1.4__py3-none-any.whl

vellum/client/resources/workflows/raw_client.py

@@ -1,21 +1,20 @@
 # This file was auto-generated by Fern from our API Definition.
 
+import contextlib
 import typing
-from ...core.client_wrapper import SyncClientWrapper
-from ...core.request_options import RequestOptions
-from ...core.http_response import HttpResponse
-from ...core.jsonable_encoder import jsonable_encoder
-from ...errors.bad_request_error import BadRequestError
-from ...core.pydantic_utilities import parse_obj_as
 from json.decoder import JSONDecodeError
+
+from ... import core
 from ...core.api_error import ApiError
-import contextlib
-from ...types.workflow_push_exec_config import WorkflowPushExecConfig
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.http_response import AsyncHttpResponse, HttpResponse
+from ...core.jsonable_encoder import jsonable_encoder
+from ...core.pydantic_utilities import parse_obj_as
+from ...core.request_options import RequestOptions
+from ...errors.bad_request_error import BadRequestError
 from ...types.workflow_push_deployment_config_request import WorkflowPushDeploymentConfigRequest
-from ... import core
+from ...types.workflow_push_exec_config import WorkflowPushExecConfig
 from ...types.workflow_push_response import WorkflowPushResponse
-from ...core.client_wrapper import AsyncClientWrapper
-from ...core.http_response import AsyncHttpResponse
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -75,7 +74,7 @@ class RawWorkflowsClient:
             request_options=request_options,
         ) as _response:
 
-            def stream() -> HttpResponse[typing.Iterator[bytes]]:
+            def _stream() -> HttpResponse[typing.Iterator[bytes]]:
                 try:
                     if 200 <= _response.status_code < 300:
                         _chunk_size = request_options.get("chunk_size", None) if request_options is not None else None
@@ -85,20 +84,23 @@ class RawWorkflowsClient:
                     _response.read()
                     if _response.status_code == 400:
                         raise BadRequestError(
-                            typing.cast(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
                                 typing.Optional[typing.Any],
                                 parse_obj_as(
                                     type_=typing.Optional[typing.Any],  # type: ignore
                                     object_=_response.json(),
                                 ),
-                            )
+                            ),
                         )
                     _response_json = _response.json()
                 except JSONDecodeError:
-                    raise ApiError(status_code=_response.status_code, body=_response.text)
-                raise ApiError(status_code=_response.status_code, body=_response_json)
+                    raise ApiError(
+                        status_code=_response.status_code, headers=dict(_response.headers), body=_response.text
+                    )
+                raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
-            yield stream()
+            yield _stream()
 
     def push(
         self,
@@ -152,6 +154,7 @@ class RawWorkflowsClient:
             },
             request_options=request_options,
             omit=OMIT,
+            force_multipart=True,
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -165,8 +168,8 @@ class RawWorkflowsClient:
                 return HttpResponse(response=_response, data=_data)
             _response_json = _response.json()
         except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
 
 class AsyncRawWorkflowsClient:
@@ -223,7 +226,7 @@ class AsyncRawWorkflowsClient:
             request_options=request_options,
         ) as _response:
 
-            async def stream() -> AsyncHttpResponse[typing.AsyncIterator[bytes]]:
+            async def _stream() -> AsyncHttpResponse[typing.AsyncIterator[bytes]]:
                 try:
                     if 200 <= _response.status_code < 300:
                         _chunk_size = request_options.get("chunk_size", None) if request_options is not None else None
@@ -234,20 +237,23 @@ class AsyncRawWorkflowsClient:
                     await _response.aread()
                     if _response.status_code == 400:
                         raise BadRequestError(
-                            typing.cast(
+                            headers=dict(_response.headers),
+                            body=typing.cast(
                                 typing.Optional[typing.Any],
                                 parse_obj_as(
                                     type_=typing.Optional[typing.Any],  # type: ignore
                                     object_=_response.json(),
                                 ),
-                            )
+                            ),
                         )
                     _response_json = _response.json()
                 except JSONDecodeError:
-                    raise ApiError(status_code=_response.status_code, body=_response.text)
-                raise ApiError(status_code=_response.status_code, body=_response_json)
+                    raise ApiError(
+                        status_code=_response.status_code, headers=dict(_response.headers), body=_response.text
+                    )
+                raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
-            yield await stream()
+            yield await _stream()
 
     async def push(
         self,
@@ -301,6 +307,7 @@ class AsyncRawWorkflowsClient:
             },
             request_options=request_options,
             omit=OMIT,
+            force_multipart=True,
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -314,5 +321,5 @@ class AsyncRawWorkflowsClient:
                 return AsyncHttpResponse(response=_response, data=_data)
             _response_json = _response.json()
         except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

vellum/client/resources/workspace_secrets/__init__.py

@@ -1,2 +1,4 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+

vellum/client/resources/workspace_secrets/client.py

@@ -1,12 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ...core.client_wrapper import SyncClientWrapper
-from .raw_client import RawWorkspaceSecretsClient
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
 from ...types.workspace_secret_read import WorkspaceSecretRead
-from ...core.client_wrapper import AsyncClientWrapper
-from .raw_client import AsyncRawWorkspaceSecretsClient
+from .raw_client import AsyncRawWorkspaceSecretsClient, RawWorkspaceSecretsClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -60,8 +59,8 @@ class WorkspaceSecretsClient:
             id="id",
         )
         """
-        response = self._raw_client.retrieve(id, request_options=request_options)
-        return response.data
+        _response = self._raw_client.retrieve(id, request_options=request_options)
+        return _response.data
 
     def partial_update(
         self,
@@ -103,8 +102,8 @@ class WorkspaceSecretsClient:
             id="id",
         )
         """
-        response = self._raw_client.partial_update(id, label=label, value=value, request_options=request_options)
-        return response.data
+        _response = self._raw_client.partial_update(id, label=label, value=value, request_options=request_options)
+        return _response.data
 
 
 class AsyncWorkspaceSecretsClient:
@@ -161,8 +160,8 @@ class AsyncWorkspaceSecretsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.retrieve(id, request_options=request_options)
-        return response.data
+        _response = await self._raw_client.retrieve(id, request_options=request_options)
+        return _response.data
 
     async def partial_update(
         self,
@@ -212,5 +211,5 @@ class AsyncWorkspaceSecretsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.partial_update(id, label=label, value=value, request_options=request_options)
-        return response.data
+        _response = await self._raw_client.partial_update(id, label=label, value=value, request_options=request_options)
+        return _response.data

vellum/client/resources/workspace_secrets/raw_client.py

@@ -1,16 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ...core.client_wrapper import SyncClientWrapper
-from ...core.request_options import RequestOptions
-from ...core.http_response import HttpResponse
-from ...types.workspace_secret_read import WorkspaceSecretRead
-from ...core.jsonable_encoder import jsonable_encoder
-from ...core.pydantic_utilities import parse_obj_as
 from json.decoder import JSONDecodeError
+
 from ...core.api_error import ApiError
-from ...core.client_wrapper import AsyncClientWrapper
-from ...core.http_response import AsyncHttpResponse
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.http_response import AsyncHttpResponse, HttpResponse
+from ...core.jsonable_encoder import jsonable_encoder
+from ...core.pydantic_utilities import parse_obj_as
+from ...core.request_options import RequestOptions
+from ...types.workspace_secret_read import WorkspaceSecretRead
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -57,8 +56,8 @@ class RawWorkspaceSecretsClient:
                 return HttpResponse(response=_response, data=_data)
             _response_json = _response.json()
         except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
     def partial_update(
         self,
@@ -114,8 +113,8 @@ class RawWorkspaceSecretsClient:
                 return HttpResponse(response=_response, data=_data)
             _response_json = _response.json()
         except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
 
 class AsyncRawWorkspaceSecretsClient:
@@ -159,8 +158,8 @@ class AsyncRawWorkspaceSecretsClient:
                 return AsyncHttpResponse(response=_response, data=_data)
             _response_json = _response.json()
         except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
     async def partial_update(
         self,
@@ -216,5 +215,5 @@ class AsyncRawWorkspaceSecretsClient:
                 return AsyncHttpResponse(response=_response, data=_data)
             _response_json = _response.json()
         except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

vellum/client/resources/workspaces/__init__.py

@@ -1,2 +1,4 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+

vellum/client/resources/workspaces/client.py

@@ -1,12 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ...core.client_wrapper import SyncClientWrapper
-from .raw_client import RawWorkspacesClient
 import typing
+
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
 from ...types.workspace_read import WorkspaceRead
-from ...core.client_wrapper import AsyncClientWrapper
-from .raw_client import AsyncRawWorkspacesClient
+from .raw_client import AsyncRawWorkspacesClient, RawWorkspacesClient
 
 
 class WorkspacesClient:
@@ -52,8 +51,8 @@ class WorkspacesClient:
         )
         client.workspaces.workspace_identity()
         """
-        response = self._raw_client.workspace_identity(request_options=request_options)
-        return response.data
+        _response = self._raw_client.workspace_identity(request_options=request_options)
+        return _response.data
 
 
 class AsyncWorkspacesClient:
@@ -103,5 +102,5 @@ class AsyncWorkspacesClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.workspace_identity(request_options=request_options)
-        return response.data
+        _response = await self._raw_client.workspace_identity(request_options=request_options)
+        return _response.data

vellum/client/resources/workspaces/raw_client.py

@@ -1,15 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ...core.client_wrapper import SyncClientWrapper
 import typing
-from ...core.request_options import RequestOptions
-from ...core.http_response import HttpResponse
-from ...types.workspace_read import WorkspaceRead
-from ...core.pydantic_utilities import parse_obj_as
 from json.decoder import JSONDecodeError
+
 from ...core.api_error import ApiError
-from ...core.client_wrapper import AsyncClientWrapper
-from ...core.http_response import AsyncHttpResponse
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.http_response import AsyncHttpResponse, HttpResponse
+from ...core.pydantic_utilities import parse_obj_as
+from ...core.request_options import RequestOptions
+from ...types.workspace_read import WorkspaceRead
 
 
 class RawWorkspacesClient:
@@ -50,8 +49,8 @@ class RawWorkspacesClient:
                 return HttpResponse(response=_response, data=_data)
             _response_json = _response.json()
         except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
 
 class AsyncRawWorkspacesClient:
@@ -92,5 +91,5 @@ class AsyncRawWorkspacesClient:
                 return AsyncHttpResponse(response=_response, data=_data)
             _response_json = _response.json()
         except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

vellum/client/types/__init__.py

@@ -1,5 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+
 from .ad_hoc_execute_prompt_event import AdHocExecutePromptEvent
 from .ad_hoc_expand_meta import AdHocExpandMeta
 from .ad_hoc_fulfilled_prompt_execution_meta import AdHocFulfilledPromptExecutionMeta

vellum/client/types/ad_hoc_execute_prompt_event.py

@@ -1,10 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .initiated_ad_hoc_execute_prompt_event import InitiatedAdHocExecutePromptEvent
-from .streaming_ad_hoc_execute_prompt_event import StreamingAdHocExecutePromptEvent
+
 from .fulfilled_ad_hoc_execute_prompt_event import FulfilledAdHocExecutePromptEvent
+from .initiated_ad_hoc_execute_prompt_event import InitiatedAdHocExecutePromptEvent
 from .rejected_ad_hoc_execute_prompt_event import RejectedAdHocExecutePromptEvent
+from .streaming_ad_hoc_execute_prompt_event import StreamingAdHocExecutePromptEvent
 
 AdHocExecutePromptEvent = typing.Union[
     InitiatedAdHocExecutePromptEvent,

vellum/client/types/ad_hoc_expand_meta.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 AdHocExpandMeta(UniversalBaseModel):

vellum/client/types/ad_hoc_fulfilled_prompt_execution_meta.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 ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .finish_reason_enum import FinishReasonEnum
 from .ml_model_usage import MlModelUsage
 from .price import Price
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
 
 
 class AdHocFulfilledPromptExecutionMeta(UniversalBaseModel):

vellum/client/types/ad_hoc_initiated_prompt_execution_meta.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 AdHocInitiatedPromptExecutionMeta(UniversalBaseModel):

vellum/client/types/ad_hoc_rejected_prompt_execution_meta.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 .finish_reason_enum import FinishReasonEnum
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .finish_reason_enum import FinishReasonEnum
 
 
 class AdHocRejectedPromptExecutionMeta(UniversalBaseModel):

vellum/client/types/ad_hoc_streaming_prompt_execution_meta.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 AdHocStreamingPromptExecutionMeta(UniversalBaseModel):

vellum/client/types/api_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 .api_node_result_data import ApiNodeResultData
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .api_node_result_data import ApiNodeResultData
 
 
 class ApiNodeResult(UniversalBaseModel):

vellum/client/types/api_node_result_data.py

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

vellum/client/types/api_request_parent_context.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from __future__ import annotations
-from ..core.pydantic_utilities import UniversalBaseModel
+
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 
 
 class ApiRequestParentContext(UniversalBaseModel):
@@ -24,13 +24,13 @@ class ApiRequestParentContext(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-from .external_parent_context import ExternalParentContext  # noqa: E402
-from .node_parent_context import NodeParentContext  # noqa: E402
-from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402
-from .span_link import SpanLink  # noqa: E402
-from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402
-from .workflow_parent_context import WorkflowParentContext  # noqa: E402
-from .workflow_sandbox_parent_context import WorkflowSandboxParentContext  # noqa: E402
-from .parent_context import ParentContext  # noqa: E402
+from .external_parent_context import ExternalParentContext  # noqa: E402, F401, I001
+from .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
+from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402, F401, I001
+from .span_link import SpanLink  # noqa: E402, F401, I001
+from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402, F401, I001
+from .workflow_parent_context import WorkflowParentContext  # noqa: E402, F401, I001
+from .workflow_sandbox_parent_context import WorkflowSandboxParentContext  # noqa: E402, F401, I001
+from .parent_context import ParentContext  # noqa: E402, F401, I001
 
 update_forward_refs(ApiRequestParentContext)

vellum/client/types/array_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 .array_chat_message_content_item import ArrayChatMessageContentItem
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .array_chat_message_content_item import ArrayChatMessageContentItem
 
 
 class ArrayChatMessageContent(UniversalBaseModel):

vellum/client/types/array_chat_message_content_item.py

@@ -1,12 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .string_chat_message_content import StringChatMessageContent
-from .function_call_chat_message_content import FunctionCallChatMessageContent
+
 from .audio_chat_message_content import AudioChatMessageContent
-from .video_chat_message_content import VideoChatMessageContent
-from .image_chat_message_content import ImageChatMessageContent
 from .document_chat_message_content import DocumentChatMessageContent
+from .function_call_chat_message_content import FunctionCallChatMessageContent
+from .image_chat_message_content import ImageChatMessageContent
+from .string_chat_message_content import StringChatMessageContent
+from .video_chat_message_content import VideoChatMessageContent
 
 ArrayChatMessageContentItem = typing.Union[
     StringChatMessageContent,

vellum/client/types/array_chat_message_content_item_request.py

@@ -1,12 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .string_chat_message_content_request import StringChatMessageContentRequest
-from .function_call_chat_message_content_request import FunctionCallChatMessageContentRequest
+
 from .audio_chat_message_content_request import AudioChatMessageContentRequest
-from .video_chat_message_content_request import VideoChatMessageContentRequest
-from .image_chat_message_content_request import ImageChatMessageContentRequest
 from .document_chat_message_content_request import DocumentChatMessageContentRequest
+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
+from .video_chat_message_content_request import VideoChatMessageContentRequest
 
 ArrayChatMessageContentItemRequest = typing.Union[
     StringChatMessageContentRequest,

vellum/client/types/array_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 .array_chat_message_content_item_request import ArrayChatMessageContentItemRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .array_chat_message_content_item_request import ArrayChatMessageContentItemRequest
 
 
 class ArrayChatMessageContentRequest(UniversalBaseModel):

vellum/client/types/array_input.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value import ArrayVellumValue
-import pydantic
+from __future__ import annotations
+
 import typing
-from .vellum_value import VellumValue
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 
 
 class ArrayInput(UniversalBaseModel):
@@ -19,7 +19,7 @@ class ArrayInput(UniversalBaseModel):
     """
 
     type: typing.Literal["ARRAY"] = "ARRAY"
-    value: typing.List[VellumValue]
+    value: typing.List["VellumValue"]
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
@@ -29,3 +29,9 @@ class ArrayInput(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+from .vellum_value import VellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(ArrayInput)

vellum/client/types/array_vellum_value.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from __future__ import annotations
-from ..core.pydantic_utilities import UniversalBaseModel
+
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 
 
 class ArrayVellumValue(UniversalBaseModel):
@@ -26,6 +26,6 @@ class ArrayVellumValue(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-from .vellum_value import VellumValue  # noqa: E402
+from .vellum_value import VellumValue  # noqa: E402, F401, I001
 
 update_forward_refs(ArrayVellumValue)