vellum-ai 0.3.22__py3-none-any.whl → 0.5.0__py3-none-any.whl

vellum/types/execute_workflow_response.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .execute_workflow_workflow_result_event import ExecuteWorkflowWorkflowResultEvent
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ExecuteWorkflowResponse(pydantic.BaseModel):
+class ExecuteWorkflowResponse(pydantic_v1.BaseModel):
     execution_id: str
     run_id: typing.Optional[str] = None
     external_id: typing.Optional[str] = None
@@ -29,5 +25,5 @@ class ExecuteWorkflowResponse(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/execute_workflow_stream_error_response.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ExecuteWorkflowStreamErrorResponse(pydantic.BaseModel):
-    detail: str = pydantic.Field()
+class ExecuteWorkflowStreamErrorResponse(pydantic_v1.BaseModel):
+    detail: str = pydantic_v1.Field()
     """
     Details about why the request failed.
     """
@@ -28,5 +24,5 @@ class ExecuteWorkflowStreamErrorResponse(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/execution_array_vellum_value.py

@@ -4,16 +4,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .array_variable_value_item import ArrayVariableValueItem
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ExecutionArrayVellumValue(pydantic.BaseModel):
-    id: str = pydantic.Field()
+class ExecutionArrayVellumValue(pydantic_v1.BaseModel):
+    id: str = pydantic_v1.Field()
     """
     The variable's uniquely identifying internal id.
     """
@@ -32,5 +28,5 @@ class ExecutionArrayVellumValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/execution_chat_history_vellum_value.py

@@ -4,16 +4,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .chat_message import ChatMessage
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ExecutionChatHistoryVellumValue(pydantic.BaseModel):
-    id: str = pydantic.Field()
+class ExecutionChatHistoryVellumValue(pydantic_v1.BaseModel):
+    id: str = pydantic_v1.Field()
     """
     The variable's uniquely identifying internal id.
     """
@@ -32,5 +28,5 @@ class ExecutionChatHistoryVellumValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/execution_error_vellum_value.py

@@ -4,16 +4,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .vellum_error import VellumError
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ExecutionErrorVellumValue(pydantic.BaseModel):
-    id: str = pydantic.Field()
+class ExecutionErrorVellumValue(pydantic_v1.BaseModel):
+    id: str = pydantic_v1.Field()
     """
     The variable's uniquely identifying internal id.
     """
@@ -32,5 +28,5 @@ class ExecutionErrorVellumValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/execution_function_call_vellum_value.py

@@ -4,16 +4,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .function_call import FunctionCall
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ExecutionFunctionCallVellumValue(pydantic.BaseModel):
-    id: str = pydantic.Field()
+class ExecutionFunctionCallVellumValue(pydantic_v1.BaseModel):
+    id: str = pydantic_v1.Field()
     """
     The variable's uniquely identifying internal id.
     """
@@ -32,5 +28,5 @@ class ExecutionFunctionCallVellumValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/execution_json_vellum_value.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ExecutionJsonVellumValue(pydantic.BaseModel):
-    id: str = pydantic.Field()
+class ExecutionJsonVellumValue(pydantic_v1.BaseModel):
+    id: str = pydantic_v1.Field()
     """
     The variable's uniquely identifying internal id.
     """
@@ -31,5 +27,5 @@ class ExecutionJsonVellumValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/execution_number_vellum_value.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ExecutionNumberVellumValue(pydantic.BaseModel):
-    id: str = pydantic.Field()
+class ExecutionNumberVellumValue(pydantic_v1.BaseModel):
+    id: str = pydantic_v1.Field()
     """
     The variable's uniquely identifying internal id.
     """
@@ -31,5 +27,5 @@ class ExecutionNumberVellumValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/execution_search_results_vellum_value.py

@@ -4,16 +4,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .search_result import SearchResult
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ExecutionSearchResultsVellumValue(pydantic.BaseModel):
-    id: str = pydantic.Field()
+class ExecutionSearchResultsVellumValue(pydantic_v1.BaseModel):
+    id: str = pydantic_v1.Field()
     """
     The variable's uniquely identifying internal id.
     """
@@ -32,5 +28,5 @@ class ExecutionSearchResultsVellumValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/execution_string_vellum_value.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ExecutionStringVellumValue(pydantic.BaseModel):
-    id: str = pydantic.Field()
+class ExecutionStringVellumValue(pydantic_v1.BaseModel):
+    id: str = pydantic_v1.Field()
     """
     The variable's uniquely identifying internal id.
     """
@@ -31,5 +27,5 @@ class ExecutionStringVellumValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/external_test_case_execution.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
+from .named_test_case_variable_value import NamedTestCaseVariableValue
+
+
+class ExternalTestCaseExecution(pydantic_v1.BaseModel):
+    outputs: typing.List[NamedTestCaseVariableValue] = pydantic_v1.Field()
+    """
+    The output values of a callable that was executed against a Test Case outside of Vellum
+    """
+
+    test_case_id: str
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/external_test_case_execution_request.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
+from .named_test_case_variable_value_request import NamedTestCaseVariableValueRequest
+
+
+class ExternalTestCaseExecutionRequest(pydantic_v1.BaseModel):
+    outputs: typing.List[NamedTestCaseVariableValueRequest] = pydantic_v1.Field()
+    """
+    The output values of a callable that was executed against a Test Case outside of Vellum
+    """
+
+    test_case_id: str
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/finish_reason_enum.py

@@ -1,31 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class FinishReasonEnum(str, enum.Enum):
-    """
-    - `LENGTH` - LENGTH
-    - `STOP` - STOP
-    - `UNKNOWN` - UNKNOWN
-    """
-
-    LENGTH = "LENGTH"
-    STOP = "STOP"
-    UNKNOWN = "UNKNOWN"
-
-    def visit(
-        self,
-        length: typing.Callable[[], T_Result],
-        stop: typing.Callable[[], T_Result],
-        unknown: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is FinishReasonEnum.LENGTH:
-            return length()
-        if self is FinishReasonEnum.STOP:
-            return stop()
-        if self is FinishReasonEnum.UNKNOWN:
-            return unknown()
+FinishReasonEnum = typing.Union[typing.Literal["LENGTH", "STOP", "UNKNOWN"], typing.Any]

vellum/types/fulfilled_execute_prompt_event.py

@@ -4,16 +4,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .fulfilled_prompt_execution_meta import FulfilledPromptExecutionMeta
 from .prompt_output import PromptOutput
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class FulfilledExecutePromptEvent(pydantic.BaseModel):
+class FulfilledExecutePromptEvent(pydantic_v1.BaseModel):
     """
     The final data event returned indicating that the stream has ended and all final resolved values from the model can be found.
     """
@@ -33,5 +29,5 @@ class FulfilledExecutePromptEvent(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/fulfilled_execute_prompt_response.py

@@ -4,27 +4,23 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .prompt_execution_meta import PromptExecutionMeta
 from .prompt_output import PromptOutput
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class FulfilledExecutePromptResponse(pydantic.BaseModel):
+class FulfilledExecutePromptResponse(pydantic_v1.BaseModel):
     """
     The successful response from the model containing all of the resolved values generated by the prompt.
     """
 
     meta: typing.Optional[PromptExecutionMeta] = None
-    raw: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(default=None)
+    raw: typing.Optional[typing.Dict[str, typing.Any]] = pydantic_v1.Field(default=None)
     """
     The subset of the raw response from the model that the request opted into with `expand_raw`.
     """
 
-    execution_id: str = pydantic.Field()
+    execution_id: str = pydantic_v1.Field()
     """
     The ID of the execution.
     """
@@ -42,5 +38,5 @@ class FulfilledExecutePromptResponse(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/fulfilled_execute_workflow_workflow_result_event.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .workflow_output import WorkflowOutput
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class FulfilledExecuteWorkflowWorkflowResultEvent(pydantic.BaseModel):
+class FulfilledExecuteWorkflowWorkflowResultEvent(pydantic_v1.BaseModel):
     """
     The successful response from the Workflow execution containing the produced outputs.
     """
@@ -32,5 +28,5 @@ class FulfilledExecuteWorkflowWorkflowResultEvent(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/fulfilled_function_call.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class FulfilledFunctionCall(pydantic.BaseModel):
+class FulfilledFunctionCall(pydantic_v1.BaseModel):
     """
     The final resolved function call value.
     """
@@ -31,5 +27,5 @@ class FulfilledFunctionCall(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/{registered_prompt_input_variable_request.py → fulfilled_function_call_request.py}

@@ -4,18 +4,19 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .vellum_variable_type import VellumVariableType
+from ..core.pydantic_utilities import pydantic_v1
+from .fulfilled_enum import FulfilledEnum
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
+class FulfilledFunctionCallRequest(pydantic_v1.BaseModel):
+    """
+    The final resolved function call value.
+    """
 
-class RegisteredPromptInputVariableRequest(pydantic.BaseModel):
-    key: str
+    state: FulfilledEnum
+    arguments: typing.Dict[str, typing.Any]
     id: typing.Optional[str] = None
-    type: typing.Optional[VellumVariableType] = None
+    name: str
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -28,5 +29,5 @@ class RegisteredPromptInputVariableRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/fulfilled_prompt_execution_meta.py

@@ -4,16 +4,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .finish_reason_enum import FinishReasonEnum
 from .ml_model_usage import MlModelUsage
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class FulfilledPromptExecutionMeta(pydantic.BaseModel):
+class FulfilledPromptExecutionMeta(pydantic_v1.BaseModel):
     """
     The subset of the metadata tracked by Vellum during prompt execution that the request opted into with `expand_meta`.
     """
@@ -33,5 +29,5 @@ class FulfilledPromptExecutionMeta(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/fulfilled_workflow_node_result_event.py

@@ -4,16 +4,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .node_output_compiled_value import NodeOutputCompiledValue
 from .workflow_node_result_data import WorkflowNodeResultData
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class FulfilledWorkflowNodeResultEvent(pydantic.BaseModel):
+class FulfilledWorkflowNodeResultEvent(pydantic_v1.BaseModel):
     """
     An event that indicates that the node has fulfilled its execution.
     """
@@ -38,5 +34,5 @@ class FulfilledWorkflowNodeResultEvent(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/function_call_chat_message_content.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .function_call_chat_message_content_value import FunctionCallChatMessageContentValue
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class FunctionCallChatMessageContent(pydantic.BaseModel):
+class FunctionCallChatMessageContent(pydantic_v1.BaseModel):
     """
     A function call value that is used in a chat message.
     """
@@ -30,5 +26,5 @@ class FunctionCallChatMessageContent(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/function_call_chat_message_content_request.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .function_call_chat_message_content_value_request import FunctionCallChatMessageContentValueRequest
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class FunctionCallChatMessageContentRequest(pydantic.BaseModel):
+class FunctionCallChatMessageContentRequest(pydantic_v1.BaseModel):
     """
     A function call value that is used in a chat message.
     """
@@ -30,5 +26,5 @@ class FunctionCallChatMessageContentRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/function_call_chat_message_content_value.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class FunctionCallChatMessageContentValue(pydantic.BaseModel):
+class FunctionCallChatMessageContentValue(pydantic_v1.BaseModel):
     """
     The final resolved function call value.
     """
@@ -31,5 +27,5 @@ class FunctionCallChatMessageContentValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/function_call_chat_message_content_value_request.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class FunctionCallChatMessageContentValueRequest(pydantic.BaseModel):
+class FunctionCallChatMessageContentValueRequest(pydantic_v1.BaseModel):
     """
     The final resolved function call value.
     """
@@ -31,5 +27,5 @@ class FunctionCallChatMessageContentValueRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}