vellum-ai 0.8.14__py3-none-any.whl → 0.8.16__py3-none-any.whl

vellum/__init__.py

@@ -14,9 +14,12 @@ from .types import (
     ArrayChatMessageContentItem,
     ArrayChatMessageContentItemRequest,
     ArrayChatMessageContentRequest,
+    ArrayVariableValue,
     ArrayVariableValueItem,
+    ArrayVellumValue,
     ArrayVellumValueItem,
     ArrayVellumValueItemRequest,
+    ArrayVellumValueRequest,
     BasicVectorizerIntfloatMultilingualE5Large,
     BasicVectorizerIntfloatMultilingualE5LargeRequest,
     BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1,
@@ -24,6 +27,9 @@ from .types import (
     BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1,
     BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request,
     ChatHistoryInputRequest,
+    ChatHistoryVariableValue,
+    ChatHistoryVellumValue,
+    ChatHistoryVellumValueRequest,
     ChatMessage,
     ChatMessageContent,
     ChatMessageContentRequest,
@@ -258,6 +264,9 @@ from .types import (
     SearchResultMeta,
     SearchResultMetaRequest,
     SearchResultRequest,
+    SearchResultsVariableValue,
+    SearchResultsVellumValue,
+    SearchResultsVellumValueRequest,
     SearchWeightsRequest,
     SentenceChunkerConfig,
     SentenceChunkerConfigRequest,
@@ -452,9 +461,12 @@ __all__ = [
     "ArrayChatMessageContentItem",
     "ArrayChatMessageContentItemRequest",
     "ArrayChatMessageContentRequest",
+    "ArrayVariableValue",
     "ArrayVariableValueItem",
+    "ArrayVellumValue",
     "ArrayVellumValueItem",
     "ArrayVellumValueItemRequest",
+    "ArrayVellumValueRequest",
     "AsyncVellum",
     "BadRequestError",
     "BasicVectorizerIntfloatMultilingualE5Large",
@@ -464,6 +476,9 @@ __all__ = [
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1",
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request",
     "ChatHistoryInputRequest",
+    "ChatHistoryVariableValue",
+    "ChatHistoryVellumValue",
+    "ChatHistoryVellumValueRequest",
     "ChatMessage",
     "ChatMessageContent",
     "ChatMessageContentRequest",
@@ -703,6 +718,9 @@ __all__ = [
     "SearchResultMeta",
     "SearchResultMetaRequest",
     "SearchResultRequest",
+    "SearchResultsVariableValue",
+    "SearchResultsVellumValue",
+    "SearchResultsVellumValueRequest",
     "SearchWeightsRequest",
     "SentenceChunkerConfig",
     "SentenceChunkerConfigRequest",

vellum/core/client_wrapper.py

@@ -17,7 +17,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "0.8.14",
+            "X-Fern-SDK-Version": "0.8.16",
         }
         headers["X_API_KEY"] = self.api_key
         return headers
@@ -40,7 +40,7 @@ class SyncClientWrapper(BaseClientWrapper):
     ):
         super().__init__(api_key=api_key, environment=environment, timeout=timeout)
         self.httpx_client = HttpClient(
-            httpx_client=httpx_client, base_headers=self.get_headers(), base_timeout=self.get_timeout()
+            httpx_client=httpx_client, base_headers=self.get_headers, base_timeout=self.get_timeout
         )
 
 
@@ -55,5 +55,5 @@ class AsyncClientWrapper(BaseClientWrapper):
     ):
         super().__init__(api_key=api_key, environment=environment, timeout=timeout)
         self.httpx_client = AsyncHttpClient(
-            httpx_client=httpx_client, base_headers=self.get_headers(), base_timeout=self.get_timeout()
+            httpx_client=httpx_client, base_headers=self.get_headers, base_timeout=self.get_timeout
         )

vellum/core/http_client.py

@@ -152,9 +152,9 @@ class HttpClient:
         self,
         *,
         httpx_client: httpx.Client,
-        base_timeout: typing.Optional[float],
-        base_headers: typing.Dict[str, str],
-        base_url: typing.Optional[str] = None,
+        base_timeout: typing.Callable[[], typing.Optional[float]],
+        base_headers: typing.Callable[[], typing.Dict[str, str]],
+        base_url: typing.Optional[typing.Callable[[], str]] = None,
     ):
         self.base_url = base_url
         self.base_timeout = base_timeout
@@ -162,7 +162,10 @@ class HttpClient:
         self.httpx_client = httpx_client
 
     def get_base_url(self, maybe_base_url: typing.Optional[str]) -> str:
-        base_url = self.base_url if maybe_base_url is None else maybe_base_url
+        base_url = maybe_base_url
+        if self.base_url is not None and base_url is None:
+            base_url = self.base_url()
+
         if base_url is None:
             raise ValueError("A base_url is required to make this request, please provide one and try again.")
         return base_url
@@ -187,7 +190,7 @@ class HttpClient:
         timeout = (
             request_options.get("timeout_in_seconds")
             if request_options is not None and request_options.get("timeout_in_seconds") is not None
-            else self.base_timeout
+            else self.base_timeout()
         )
 
         json_body, data_body = get_request_body(json=json, data=data, request_options=request_options, omit=omit)
@@ -198,7 +201,7 @@ class HttpClient:
             headers=jsonable_encoder(
                 remove_none_from_dict(
                     {
-                        **self.base_headers,
+                        **self.base_headers(),
                         **(headers if headers is not None else {}),
                         **(request_options.get("additional_headers", {}) or {} if request_options is not None else {}),
                     }
@@ -271,7 +274,7 @@ class HttpClient:
         timeout = (
             request_options.get("timeout_in_seconds")
             if request_options is not None and request_options.get("timeout_in_seconds") is not None
-            else self.base_timeout
+            else self.base_timeout()
         )
 
         json_body, data_body = get_request_body(json=json, data=data, request_options=request_options, omit=omit)
@@ -282,7 +285,7 @@ class HttpClient:
             headers=jsonable_encoder(
                 remove_none_from_dict(
                     {
-                        **self.base_headers,
+                        **self.base_headers(),
                         **(headers if headers is not None else {}),
                         **(request_options.get("additional_headers", {}) if request_options is not None else {}),
                     }
@@ -321,9 +324,9 @@ class AsyncHttpClient:
         self,
         *,
         httpx_client: httpx.AsyncClient,
-        base_timeout: typing.Optional[float],
-        base_headers: typing.Dict[str, str],
-        base_url: typing.Optional[str] = None,
+        base_timeout: typing.Callable[[], typing.Optional[float]],
+        base_headers: typing.Callable[[], typing.Dict[str, str]],
+        base_url: typing.Optional[typing.Callable[[], str]] = None,
     ):
         self.base_url = base_url
         self.base_timeout = base_timeout
@@ -331,7 +334,10 @@ class AsyncHttpClient:
         self.httpx_client = httpx_client
 
     def get_base_url(self, maybe_base_url: typing.Optional[str]) -> str:
-        base_url = self.base_url if maybe_base_url is None else maybe_base_url
+        base_url = maybe_base_url
+        if self.base_url is not None and base_url is None:
+            base_url = self.base_url()
+
         if base_url is None:
             raise ValueError("A base_url is required to make this request, please provide one and try again.")
         return base_url
@@ -356,7 +362,7 @@ class AsyncHttpClient:
         timeout = (
             request_options.get("timeout_in_seconds")
             if request_options is not None and request_options.get("timeout_in_seconds") is not None
-            else self.base_timeout
+            else self.base_timeout()
         )
 
         json_body, data_body = get_request_body(json=json, data=data, request_options=request_options, omit=omit)
@@ -368,7 +374,7 @@ class AsyncHttpClient:
             headers=jsonable_encoder(
                 remove_none_from_dict(
                     {
-                        **self.base_headers,
+                        **self.base_headers(),
                         **(headers if headers is not None else {}),
                         **(request_options.get("additional_headers", {}) or {} if request_options is not None else {}),
                     }
@@ -438,7 +444,7 @@ class AsyncHttpClient:
         timeout = (
             request_options.get("timeout_in_seconds")
             if request_options is not None and request_options.get("timeout_in_seconds") is not None
-            else self.base_timeout
+            else self.base_timeout()
         )
 
         json_body, data_body = get_request_body(json=json, data=data, request_options=request_options, omit=omit)
@@ -449,7 +455,7 @@ class AsyncHttpClient:
             headers=jsonable_encoder(
                 remove_none_from_dict(
                     {
-                        **self.base_headers,
+                        **self.base_headers(),
                         **(headers if headers is not None else {}),
                         **(request_options.get("additional_headers", {}) if request_options is not None else {}),
                     }

vellum/core/pydantic_utilities.py

@@ -152,7 +152,7 @@ class UniversalBaseModel(pydantic.BaseModel):
             )
 
         else:
-            _fields_set = self.__fields_set__
+            _fields_set = self.__fields_set__.copy()
 
             fields = _get_model_fields(self.__class__)
             for name, field in fields.items():
@@ -162,9 +162,12 @@ class UniversalBaseModel(pydantic.BaseModel):
                     # If the default values are non-null act like they've been set
                     # This effectively allows exclude_unset to work like exclude_none where
                     # the latter passes through intentionally set none values.
-                    if default != None:
+                    if default is not None or ("exclude_unset" in kwargs and not kwargs["exclude_unset"]):
                         _fields_set.add(name)
 
+                        if default is not None:
+                            self.__fields_set__.add(name)
+
             kwargs_with_defaults_exclude_unset_include_fields: typing.Any = {
                 "by_alias": True,
                 "exclude_unset": True,
@@ -177,13 +180,33 @@ class UniversalBaseModel(pydantic.BaseModel):
         return convert_and_respect_annotation_metadata(object_=dict_dump, annotation=self.__class__, direction="write")
 
 
+def _union_list_of_pydantic_dicts(
+    source: typing.List[typing.Any], destination: typing.List[typing.Any]
+) -> typing.List[typing.Any]:
+    converted_list: typing.List[typing.Any] = []
+    for i, item in enumerate(source):
+        destination_value = destination[i]  # type: ignore
+        if isinstance(item, dict):
+            converted_list.append(deep_union_pydantic_dicts(item, destination_value))
+        elif isinstance(item, list):
+            converted_list.append(_union_list_of_pydantic_dicts(item, destination_value))
+        else:
+            converted_list.append(item)
+    return converted_list
+
+
 def deep_union_pydantic_dicts(
     source: typing.Dict[str, typing.Any], destination: typing.Dict[str, typing.Any]
 ) -> typing.Dict[str, typing.Any]:
     for key, value in source.items():
+        node = destination.setdefault(key, {})
         if isinstance(value, dict):
-            node = destination.setdefault(key, {})
             deep_union_pydantic_dicts(value, node)
+        # Note: we do not do this same processing for sets given we do not have sets of models
+        # and given the sets are unordered, the processing of the set and matching objects would
+        # be non-trivial.
+        elif isinstance(value, list):
+            destination[key] = _union_list_of_pydantic_dicts(value, node)
         else:
             destination[key] = value
 

vellum/types/__init__.py

@@ -13,9 +13,12 @@ from .array_chat_message_content import ArrayChatMessageContent
 from .array_chat_message_content_item import ArrayChatMessageContentItem
 from .array_chat_message_content_item_request import ArrayChatMessageContentItemRequest
 from .array_chat_message_content_request import ArrayChatMessageContentRequest
+from .array_variable_value import ArrayVariableValue
 from .array_variable_value_item import ArrayVariableValueItem
+from .array_vellum_value import ArrayVellumValue
 from .array_vellum_value_item import ArrayVellumValueItem
 from .array_vellum_value_item_request import ArrayVellumValueItemRequest
+from .array_vellum_value_request import ArrayVellumValueRequest
 from .basic_vectorizer_intfloat_multilingual_e_5_large import BasicVectorizerIntfloatMultilingualE5Large
 from .basic_vectorizer_intfloat_multilingual_e_5_large_request import BasicVectorizerIntfloatMultilingualE5LargeRequest
 from .basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_cos_v_1 import (
@@ -31,6 +34,9 @@ from .basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1_request
     BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request,
 )
 from .chat_history_input_request import ChatHistoryInputRequest
+from .chat_history_variable_value import ChatHistoryVariableValue
+from .chat_history_vellum_value import ChatHistoryVellumValue
+from .chat_history_vellum_value_request import ChatHistoryVellumValueRequest
 from .chat_message import ChatMessage
 from .chat_message_content import ChatMessageContent
 from .chat_message_content_request import ChatMessageContentRequest
@@ -265,6 +271,9 @@ from .search_result_merging_request import SearchResultMergingRequest
 from .search_result_meta import SearchResultMeta
 from .search_result_meta_request import SearchResultMetaRequest
 from .search_result_request import SearchResultRequest
+from .search_results_variable_value import SearchResultsVariableValue
+from .search_results_vellum_value import SearchResultsVellumValue
+from .search_results_vellum_value_request import SearchResultsVellumValueRequest
 from .search_weights_request import SearchWeightsRequest
 from .sentence_chunker_config import SentenceChunkerConfig
 from .sentence_chunker_config_request import SentenceChunkerConfigRequest
@@ -443,9 +452,12 @@ __all__ = [
     "ArrayChatMessageContentItem",
     "ArrayChatMessageContentItemRequest",
     "ArrayChatMessageContentRequest",
+    "ArrayVariableValue",
     "ArrayVariableValueItem",
+    "ArrayVellumValue",
     "ArrayVellumValueItem",
     "ArrayVellumValueItemRequest",
+    "ArrayVellumValueRequest",
     "BasicVectorizerIntfloatMultilingualE5Large",
     "BasicVectorizerIntfloatMultilingualE5LargeRequest",
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1",
@@ -453,6 +465,9 @@ __all__ = [
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1",
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request",
     "ChatHistoryInputRequest",
+    "ChatHistoryVariableValue",
+    "ChatHistoryVellumValue",
+    "ChatHistoryVellumValueRequest",
     "ChatMessage",
     "ChatMessageContent",
     "ChatMessageContentRequest",
@@ -687,6 +702,9 @@ __all__ = [
     "SearchResultMeta",
     "SearchResultMetaRequest",
     "SearchResultRequest",
+    "SearchResultsVariableValue",
+    "SearchResultsVellumValue",
+    "SearchResultsVellumValueRequest",
     "SearchWeightsRequest",
     "SentenceChunkerConfig",
     "SentenceChunkerConfigRequest",

vellum/types/array_variable_value.py

@@ -0,0 +1,27 @@
+# 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
+
+
+class ArrayVariableValue(UniversalBaseModel):
+    type: typing.Literal["ARRAY"] = "ARRAY"
+    value: typing.Optional[typing.List["ArrayVariableValueItem"]] = 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_variable_value_item import ArrayVariableValueItem  # noqa: E402
+
+update_forward_refs(ArrayVariableValue)

vellum/types/array_variable_value_item.py

@@ -1,5 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
 import typing
 from .string_variable_value import StringVariableValue
 from .number_variable_value import NumberVariableValue
@@ -7,7 +8,12 @@ from .json_variable_value import JsonVariableValue
 from .error_variable_value import ErrorVariableValue
 from .function_call_variable_value import FunctionCallVariableValue
 from .image_variable_value import ImageVariableValue
+from .chat_history_variable_value import ChatHistoryVariableValue
+from .search_results_variable_value import SearchResultsVariableValue
+import typing
 
+if typing.TYPE_CHECKING:
+    from .array_variable_value import ArrayVariableValue
 ArrayVariableValueItem = typing.Union[
     StringVariableValue,
     NumberVariableValue,
@@ -15,4 +21,7 @@ ArrayVariableValueItem = typing.Union[
     ErrorVariableValue,
     FunctionCallVariableValue,
     ImageVariableValue,
+    ChatHistoryVariableValue,
+    SearchResultsVariableValue,
+    "ArrayVariableValue",
 ]

vellum/types/array_vellum_value.py

@@ -0,0 +1,31 @@
+# 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
+
+
+class ArrayVellumValue(UniversalBaseModel):
+    """
+    A value representing an array of Vellum variable values.
+    """
+
+    type: typing.Literal["ARRAY"] = "ARRAY"
+    value: typing.Optional[typing.List["ArrayVellumValueItem"]] = 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_item import ArrayVellumValueItem  # noqa: E402
+
+update_forward_refs(ArrayVellumValue)

vellum/types/array_vellum_value_item.py

@@ -1,5 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
 import typing
 from .string_vellum_value import StringVellumValue
 from .number_vellum_value import NumberVellumValue
@@ -7,7 +8,20 @@ from .json_vellum_value import JsonVellumValue
 from .image_vellum_value import ImageVellumValue
 from .function_call_vellum_value import FunctionCallVellumValue
 from .error_vellum_value import ErrorVellumValue
+from .chat_history_vellum_value import ChatHistoryVellumValue
+from .search_results_vellum_value import SearchResultsVellumValue
+import typing
 
+if typing.TYPE_CHECKING:
+    from .array_vellum_value import ArrayVellumValue
 ArrayVellumValueItem = typing.Union[
-    StringVellumValue, NumberVellumValue, JsonVellumValue, ImageVellumValue, FunctionCallVellumValue, ErrorVellumValue
+    StringVellumValue,
+    NumberVellumValue,
+    JsonVellumValue,
+    ImageVellumValue,
+    FunctionCallVellumValue,
+    ErrorVellumValue,
+    ChatHistoryVellumValue,
+    SearchResultsVellumValue,
+    "ArrayVellumValue",
 ]

vellum/types/array_vellum_value_item_request.py

@@ -1,5 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
 import typing
 from .string_vellum_value_request import StringVellumValueRequest
 from .number_vellum_value_request import NumberVellumValueRequest
@@ -7,7 +8,12 @@ from .json_vellum_value_request import JsonVellumValueRequest
 from .image_vellum_value_request import ImageVellumValueRequest
 from .function_call_vellum_value_request import FunctionCallVellumValueRequest
 from .error_vellum_value_request import ErrorVellumValueRequest
+from .chat_history_vellum_value_request import ChatHistoryVellumValueRequest
+from .search_results_vellum_value_request import SearchResultsVellumValueRequest
+import typing
 
+if typing.TYPE_CHECKING:
+    from .array_vellum_value_request import ArrayVellumValueRequest
 ArrayVellumValueItemRequest = typing.Union[
     StringVellumValueRequest,
     NumberVellumValueRequest,
@@ -15,4 +21,7 @@ ArrayVellumValueItemRequest = typing.Union[
     ImageVellumValueRequest,
     FunctionCallVellumValueRequest,
     ErrorVellumValueRequest,
+    ChatHistoryVellumValueRequest,
+    SearchResultsVellumValueRequest,
+    "ArrayVellumValueRequest",
 ]

vellum/types/array_vellum_value_request.py

@@ -0,0 +1,31 @@
+# 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
+
+
+class ArrayVellumValueRequest(UniversalBaseModel):
+    """
+    A value representing an array of Vellum variable values.
+    """
+
+    type: typing.Literal["ARRAY"] = "ARRAY"
+    value: typing.Optional[typing.List["ArrayVellumValueItemRequest"]] = 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_item_request import ArrayVellumValueItemRequest  # noqa: E402
+
+update_forward_refs(ArrayVellumValueRequest)

vellum/types/chat_history_variable_value.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .chat_message import ChatMessage
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class ChatHistoryVariableValue(UniversalBaseModel):
+    type: typing.Literal["CHAT_HISTORY"] = "CHAT_HISTORY"
+    value: typing.Optional[typing.List[ChatMessage]] = 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/types/chat_history_vellum_value.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .chat_message import ChatMessage
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class ChatHistoryVellumValue(UniversalBaseModel):
+    """
+    A value representing Chat History.
+    """
+
+    type: typing.Literal["CHAT_HISTORY"] = "CHAT_HISTORY"
+    value: typing.Optional[typing.List[ChatMessage]] = 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/types/chat_history_vellum_value_request.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .chat_message_request import ChatMessageRequest
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class ChatHistoryVellumValueRequest(UniversalBaseModel):
+    """
+    A value representing Chat History.
+    """
+
+    type: typing.Literal["CHAT_HISTORY"] = "CHAT_HISTORY"
+    value: typing.Optional[typing.List[ChatMessageRequest]] = 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/types/code_execution_node_array_result.py

@@ -1,10 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
+from .array_variable_value import ArrayVariableValue
 import typing
 from .array_variable_value_item import ArrayVariableValueItem
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
+from ..core.pydantic_utilities import update_forward_refs
 
 
 class CodeExecutionNodeArrayResult(UniversalBaseModel):
@@ -20,3 +23,6 @@ class CodeExecutionNodeArrayResult(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+update_forward_refs(ArrayVariableValue, CodeExecutionNodeArrayResult=CodeExecutionNodeArrayResult)

vellum/types/code_execution_node_result.py

@@ -1,10 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
+from .array_variable_value import ArrayVariableValue
 import typing
 from .code_execution_node_result_data import CodeExecutionNodeResultData
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
+from ..core.pydantic_utilities import update_forward_refs
 
 
 class CodeExecutionNodeResult(UniversalBaseModel):
@@ -23,3 +26,6 @@ class CodeExecutionNodeResult(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+update_forward_refs(ArrayVariableValue, CodeExecutionNodeResult=CodeExecutionNodeResult)

vellum/types/code_execution_node_result_data.py

@@ -1,10 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
+from .array_variable_value import ArrayVariableValue
 from .code_execution_node_result_output import CodeExecutionNodeResultOutput
 import typing
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
+from ..core.pydantic_utilities import update_forward_refs
 
 
 class CodeExecutionNodeResultData(UniversalBaseModel):
@@ -19,3 +22,6 @@ class CodeExecutionNodeResultData(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+update_forward_refs(ArrayVariableValue, CodeExecutionNodeResultData=CodeExecutionNodeResultData)

vellum/types/create_test_suite_test_case_request.py

@@ -1,10 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
+from .array_vellum_value_request import ArrayVellumValueRequest
 import typing
 import pydantic
 from .named_test_case_variable_value_request import NamedTestCaseVariableValueRequest
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import update_forward_refs
 
 
 class CreateTestSuiteTestCaseRequest(UniversalBaseModel):
@@ -40,3 +43,6 @@ class CreateTestSuiteTestCaseRequest(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+update_forward_refs(ArrayVellumValueRequest, CreateTestSuiteTestCaseRequest=CreateTestSuiteTestCaseRequest)