vellum-ai 0.14.41__py3-none-any.whl → 1.11.2__py3-none-any.whl

vellum/client/types/chat_history_vellum_value_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 .chat_message_request import ChatMessageRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .chat_message_request import ChatMessageRequest
 
 
 class ChatHistoryVellumValueRequest(UniversalBaseModel):

vellum/client/types/chat_message.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 .chat_message_role import ChatMessageRole
-from .chat_message_content import ChatMessageContent
+
 import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .chat_message_content import ChatMessageContent
+from .chat_message_role import ChatMessageRole
 
 
 class ChatMessage(UniversalBaseModel):

vellum/client/types/chat_message_content.py

@@ -1,18 +1,21 @@
 # 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 .array_chat_message_content import ArrayChatMessageContent
-from .image_chat_message_content import ImageChatMessageContent
 from .audio_chat_message_content import AudioChatMessageContent
 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
 
 ChatMessageContent = typing.Union[
     StringChatMessageContent,
     FunctionCallChatMessageContent,
     ArrayChatMessageContent,
-    ImageChatMessageContent,
     AudioChatMessageContent,
+    VideoChatMessageContent,
+    ImageChatMessageContent,
     DocumentChatMessageContent,
 ]

vellum/client/types/chat_message_content_request.py

@@ -1,18 +1,21 @@
 # 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 .array_chat_message_content_request import ArrayChatMessageContentRequest
-from .image_chat_message_content_request import ImageChatMessageContentRequest
 from .audio_chat_message_content_request import AudioChatMessageContentRequest
 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
 
 ChatMessageContentRequest = typing.Union[
     StringChatMessageContentRequest,
     FunctionCallChatMessageContentRequest,
     ArrayChatMessageContentRequest,
-    ImageChatMessageContentRequest,
     AudioChatMessageContentRequest,
+    VideoChatMessageContentRequest,
+    ImageChatMessageContentRequest,
     DocumentChatMessageContentRequest,
 ]

vellum/client/types/chat_message_prompt_block.py

@@ -1,14 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from __future__ import annotations
-from ..core.pydantic_utilities import UniversalBaseModel
+
 import typing
-from .prompt_block_state import PromptBlockState
-from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
-from .chat_message_role import ChatMessageRole
-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
+from .chat_message_role import ChatMessageRole
+from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
+from .prompt_block_state import PromptBlockState
 
 
 class ChatMessagePromptBlock(UniversalBaseModel):
@@ -34,6 +34,6 @@ class ChatMessagePromptBlock(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-from .prompt_block import PromptBlock  # noqa: E402
+from .prompt_block import PromptBlock  # noqa: E402, F401, I001
 
 update_forward_refs(ChatMessagePromptBlock)

vellum/client/types/chat_message_request.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 .chat_message_role import ChatMessageRole
-from .chat_message_content_request import ChatMessageContentRequest
+
 import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .chat_message_content_request import ChatMessageContentRequest
+from .chat_message_role import ChatMessageRole
 
 
 class ChatMessageRequest(UniversalBaseModel):

vellum/client/types/code_execution_node_array_result.py

@@ -1,19 +1,17 @@
 # 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 import ArrayVellumValue
+
 import typing
-from .vellum_value import VellumValue
-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 CodeExecutionNodeArrayResult(UniversalBaseModel):
     id: str
     type: typing.Literal["ARRAY"] = "ARRAY"
-    value: typing.Optional[typing.List[VellumValue]] = None
+    value: typing.Optional[typing.List["VellumValue"]] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
@@ -25,4 +23,7 @@ class CodeExecutionNodeArrayResult(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-update_forward_refs(ArrayVellumValue, CodeExecutionNodeArrayResult=CodeExecutionNodeArrayResult)
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+from .vellum_value import VellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(CodeExecutionNodeArrayResult)

vellum/client/types/code_execution_node_chat_history_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 .chat_message import ChatMessage
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .chat_message import ChatMessage
 
 
 class CodeExecutionNodeChatHistoryResult(UniversalBaseModel):

vellum/client/types/code_execution_node_error_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 .vellum_error import VellumError
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_error import VellumError
 
 
 class CodeExecutionNodeErrorResult(UniversalBaseModel):

vellum/client/types/code_execution_node_function_call_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 .function_call import FunctionCall
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .function_call import FunctionCall
 
 
 class CodeExecutionNodeFunctionCallResult(UniversalBaseModel):

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

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

vellum/client/types/code_execution_node_result.py

@@ -1,13 +1,12 @@
 # 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 import ArrayVellumValue
+
 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
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .code_execution_node_result_data import CodeExecutionNodeResultData
 
 
 class CodeExecutionNodeResult(UniversalBaseModel):
@@ -28,4 +27,6 @@ class CodeExecutionNodeResult(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-update_forward_refs(ArrayVellumValue, CodeExecutionNodeResult=CodeExecutionNodeResult)
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(CodeExecutionNodeResult)

vellum/client/types/code_execution_node_result_data.py

@@ -1,13 +1,12 @@
 # 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 import ArrayVellumValue
-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
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .code_execution_node_result_output import CodeExecutionNodeResultOutput
 
 
 class CodeExecutionNodeResultData(UniversalBaseModel):
@@ -24,4 +23,6 @@ class CodeExecutionNodeResultData(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-update_forward_refs(ArrayVellumValue, CodeExecutionNodeResultData=CodeExecutionNodeResultData)
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(CodeExecutionNodeResultData)

vellum/client/types/code_execution_node_result_output.py

@@ -1,14 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .code_execution_node_string_result import CodeExecutionNodeStringResult
-from .code_execution_node_number_result import CodeExecutionNodeNumberResult
-from .code_execution_node_json_result import CodeExecutionNodeJsonResult
+
+from .code_execution_node_array_result import CodeExecutionNodeArrayResult
 from .code_execution_node_chat_history_result import CodeExecutionNodeChatHistoryResult
-from .code_execution_node_search_results_result import CodeExecutionNodeSearchResultsResult
 from .code_execution_node_error_result import CodeExecutionNodeErrorResult
-from .code_execution_node_array_result import CodeExecutionNodeArrayResult
 from .code_execution_node_function_call_result import CodeExecutionNodeFunctionCallResult
+from .code_execution_node_json_result import CodeExecutionNodeJsonResult
+from .code_execution_node_number_result import CodeExecutionNodeNumberResult
+from .code_execution_node_search_results_result import CodeExecutionNodeSearchResultsResult
+from .code_execution_node_string_result import CodeExecutionNodeStringResult
 
 CodeExecutionNodeResultOutput = typing.Union[
     CodeExecutionNodeStringResult,

vellum/client/types/code_execution_node_search_results_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 .search_result import SearchResult
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .search_result import SearchResult
 
 
 class CodeExecutionNodeSearchResultsResult(UniversalBaseModel):

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

vellum/client/types/code_execution_package.py

@@ -1,14 +1,15 @@
 # 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 CodeExecutionPackage(UniversalBaseModel):
     version: str
     name: str
+    repository: 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/code_executor_input.py

@@ -1,15 +1,20 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .string_input import StringInput
-from .json_input import JsonInput
+
+from .array_input import ArrayInput
+from .audio_input import AudioInput
 from .chat_history_input import ChatHistoryInput
-from .number_input import NumberInput
-from .search_results_input import SearchResultsInput
+from .code_executor_secret_input import CodeExecutorSecretInput
+from .document_input import DocumentInput
 from .error_input import ErrorInput
-from .array_input import ArrayInput
 from .function_call_input import FunctionCallInput
-from .code_executor_secret_input import CodeExecutorSecretInput
+from .image_input import ImageInput
+from .json_input import JsonInput
+from .number_input import NumberInput
+from .search_results_input import SearchResultsInput
+from .string_input import StringInput
+from .video_input import VideoInput
 
 CodeExecutorInput = typing.Union[
     StringInput,
@@ -20,5 +25,9 @@ CodeExecutorInput = typing.Union[
     ErrorInput,
     ArrayInput,
     FunctionCallInput,
+    AudioInput,
+    VideoInput,
+    ImageInput,
+    DocumentInput,
     CodeExecutorSecretInput,
 ]

vellum/client/types/code_executor_response.py

@@ -1,18 +1,16 @@
 # 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 import ArrayVellumValue
-from .vellum_value import VellumValue
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import typing
+
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 
 
 class CodeExecutorResponse(UniversalBaseModel):
     log: str
-    output: VellumValue
+    output: "VellumValue"
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
@@ -24,4 +22,7 @@ class CodeExecutorResponse(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-update_forward_refs(ArrayVellumValue, CodeExecutorResponse=CodeExecutorResponse)
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+from .vellum_value import VellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(CodeExecutorResponse)

vellum/client/types/code_executor_secret_input.py

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

vellum/client/types/code_resource_definition.py

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

vellum/client/types/compile_prompt_deployment_expand_meta_request.py

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

vellum/client/types/compile_prompt_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 CompilePromptMeta(UniversalBaseModel):

vellum/client/types/components_schemas_composio_execute_tool_request.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .composio_execute_tool_request import ComposioExecuteToolRequest
+
+ComponentsSchemasComposioExecuteToolRequest = ComposioExecuteToolRequest

vellum/client/types/components_schemas_composio_execute_tool_response.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .composio_execute_tool_response import ComposioExecuteToolResponse
+
+ComponentsSchemasComposioExecuteToolResponse = ComposioExecuteToolResponse

vellum/client/types/components_schemas_composio_integration_exec_config.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .composio_integration_exec_config import ComposioIntegrationExecConfig
+
+ComponentsSchemasComposioIntegrationExecConfig = ComposioIntegrationExecConfig

vellum/client/types/components_schemas_composio_tool_definition.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .composio_tool_definition import ComposioToolDefinition
+
+ComponentsSchemasComposioToolDefinition = ComposioToolDefinition

vellum/client/types/components_schemas_prompt_version_build_config_sandbox.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .prompt_version_build_config_sandbox import PromptVersionBuildConfigSandbox
+
+ComponentsSchemasPromptVersionBuildConfigSandbox = PromptVersionBuildConfigSandbox

vellum/client/types/components_schemas_slim_composio_tool_definition.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .slim_composio_tool_definition import SlimComposioToolDefinition
+
+ComponentsSchemasSlimComposioToolDefinition = SlimComposioToolDefinition

vellum/client/types/composio_execute_tool_request.py

@@ -0,0 +1,24 @@
+# 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 ComposioExecuteToolRequest(UniversalBaseModel):
+    """
+    Payload for executing a Composio tool with provider id and tool arguments.
+    """
+
+    provider: typing.Literal["COMPOSIO"] = "COMPOSIO"
+    arguments: typing.Dict[str, typing.Optional[typing.Any]]
+
+    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/composio_execute_tool_response.py

@@ -0,0 +1,24 @@
+# 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 ComposioExecuteToolResponse(UniversalBaseModel):
+    """
+    Response payload with provider id and execution output from a Composio tool.
+    """
+
+    provider: typing.Literal["COMPOSIO"] = "COMPOSIO"
+    data: typing.Dict[str, typing.Optional[typing.Any]]
+
+    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/composio_integration_exec_config.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
+
+
+class ComposioIntegrationExecConfig(UniversalBaseModel):
+    type: typing.Literal["COMPOSIO"] = "COMPOSIO"
+    slug: str
+    supports_webhook_triggers: 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/composio_tool_definition.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 .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]]
+
+    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):