vellum-ai 0.3.14__py3-none-any.whl → 0.3.16__py3-none-any.whl

vellum/types/chat_message_content.py

@@ -4,8 +4,6 @@ from __future__ import annotations
 
 import typing
 
-import typing_extensions
-
 from .array_chat_message_content import ArrayChatMessageContent
 from .function_call_chat_message_content import FunctionCallChatMessageContent
 from .image_chat_message_content import ImageChatMessageContent
@@ -13,39 +11,43 @@ from .string_chat_message_content import StringChatMessageContent
 
 
 class ChatMessageContent_String(StringChatMessageContent):
-    type: typing_extensions.Literal["STRING"]
+    type: typing.Literal["STRING"] = "STRING"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class ChatMessageContent_FunctionCall(FunctionCallChatMessageContent):
-    type: typing_extensions.Literal["FUNCTION_CALL"]
+    type: typing.Literal["FUNCTION_CALL"] = "FUNCTION_CALL"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class ChatMessageContent_Array(ArrayChatMessageContent):
-    type: typing_extensions.Literal["ARRAY"]
+    type: typing.Literal["ARRAY"] = "ARRAY"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class ChatMessageContent_Image(ImageChatMessageContent):
-    type: typing_extensions.Literal["IMAGE"]
+    type: typing.Literal["IMAGE"] = "IMAGE"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 ChatMessageContent = typing.Union[

vellum/types/chat_message_content_request.py

@@ -4,8 +4,6 @@ from __future__ import annotations
 
 import typing
 
-import typing_extensions
-
 from .array_chat_message_content_request import ArrayChatMessageContentRequest
 from .function_call_chat_message_content_request import FunctionCallChatMessageContentRequest
 from .image_chat_message_content_request import ImageChatMessageContentRequest
@@ -13,39 +11,43 @@ from .string_chat_message_content_request import StringChatMessageContentRequest
 
 
 class ChatMessageContentRequest_String(StringChatMessageContentRequest):
-    type: typing_extensions.Literal["STRING"]
+    type: typing.Literal["STRING"] = "STRING"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class ChatMessageContentRequest_FunctionCall(FunctionCallChatMessageContentRequest):
-    type: typing_extensions.Literal["FUNCTION_CALL"]
+    type: typing.Literal["FUNCTION_CALL"] = "FUNCTION_CALL"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class ChatMessageContentRequest_Array(ArrayChatMessageContentRequest):
-    type: typing_extensions.Literal["ARRAY"]
+    type: typing.Literal["ARRAY"] = "ARRAY"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class ChatMessageContentRequest_Image(ImageChatMessageContentRequest):
-    type: typing_extensions.Literal["IMAGE"]
+    type: typing.Literal["IMAGE"] = "IMAGE"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 ChatMessageContentRequest = typing.Union[

vellum/types/chat_message_request.py

@@ -14,12 +14,13 @@ except ImportError:
 
 
 class ChatMessageRequest(pydantic.BaseModel):
-    text: typing.Optional[str]
+    text: typing.Optional[str] = None
     role: ChatMessageRole
-    content: typing.Optional[ChatMessageContentRequest]
-    source: typing.Optional[str] = pydantic.Field(
-        description="An optional identifier representing who or what generated this message."
-    )
+    content: typing.Optional[ChatMessageContentRequest] = None
+    source: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    An optional identifier representing who or what generated this message.
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -32,4 +33,5 @@ class ChatMessageRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/code_execution_node_array_result.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 .array_variable_value_item import ArrayVariableValueItem
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class CodeExecutionNodeArrayResult(pydantic.BaseModel):
+    id: str
+    value: typing.Optional[typing.List[ArrayVariableValueItem]] = None
+
+    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.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/code_execution_node_chat_history_result.py

@@ -14,7 +14,7 @@ except ImportError:
 
 class CodeExecutionNodeChatHistoryResult(pydantic.BaseModel):
     id: str
-    value: typing.Optional[typing.List[ChatMessage]]
+    value: typing.Optional[typing.List[ChatMessage]] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -27,4 +27,5 @@ class CodeExecutionNodeChatHistoryResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/code_execution_node_error_result.py

@@ -14,7 +14,7 @@ except ImportError:
 
 class CodeExecutionNodeErrorResult(pydantic.BaseModel):
     id: str
-    value: typing.Optional[VellumError]
+    value: typing.Optional[VellumError] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -27,4 +27,5 @@ class CodeExecutionNodeErrorResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/code_execution_node_function_call_result.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 .function_call import FunctionCall
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class CodeExecutionNodeFunctionCallResult(pydantic.BaseModel):
+    id: str
+    value: typing.Optional[FunctionCall] = None
+
+    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.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/code_execution_node_json_result.py

@@ -13,7 +13,7 @@ except ImportError:
 
 class CodeExecutionNodeJsonResult(pydantic.BaseModel):
     id: str
-    value: typing.Optional[typing.Dict[str, typing.Any]]
+    value: typing.Optional[typing.Dict[str, typing.Any]] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -26,4 +26,5 @@ class CodeExecutionNodeJsonResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/code_execution_node_number_result.py

@@ -13,7 +13,7 @@ except ImportError:
 
 class CodeExecutionNodeNumberResult(pydantic.BaseModel):
     id: str
-    value: typing.Optional[float]
+    value: typing.Optional[float] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -26,4 +26,5 @@ class CodeExecutionNodeNumberResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/code_execution_node_result.py

@@ -30,4 +30,5 @@ class CodeExecutionNodeResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/code_execution_node_result_data.py

@@ -14,7 +14,7 @@ except ImportError:
 
 class CodeExecutionNodeResultData(pydantic.BaseModel):
     output: CodeExecutionNodeResultOutput
-    log_output_id: typing.Optional[str]
+    log_output_id: typing.Optional[str] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -27,4 +27,5 @@ class CodeExecutionNodeResultData(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/code_execution_node_result_output.py

@@ -4,10 +4,10 @@ from __future__ import annotations
 
 import typing
 
-import typing_extensions
-
+from .code_execution_node_array_result import CodeExecutionNodeArrayResult
 from .code_execution_node_chat_history_result import CodeExecutionNodeChatHistoryResult
 from .code_execution_node_error_result import CodeExecutionNodeErrorResult
+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
@@ -15,57 +15,83 @@ from .code_execution_node_string_result import CodeExecutionNodeStringResult
 
 
 class CodeExecutionNodeResultOutput_String(CodeExecutionNodeStringResult):
-    type: typing_extensions.Literal["STRING"]
+    type: typing.Literal["STRING"] = "STRING"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class CodeExecutionNodeResultOutput_Number(CodeExecutionNodeNumberResult):
-    type: typing_extensions.Literal["NUMBER"]
+    type: typing.Literal["NUMBER"] = "NUMBER"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class CodeExecutionNodeResultOutput_Json(CodeExecutionNodeJsonResult):
-    type: typing_extensions.Literal["JSON"]
+    type: typing.Literal["JSON"] = "JSON"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class CodeExecutionNodeResultOutput_ChatHistory(CodeExecutionNodeChatHistoryResult):
-    type: typing_extensions.Literal["CHAT_HISTORY"]
+    type: typing.Literal["CHAT_HISTORY"] = "CHAT_HISTORY"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class CodeExecutionNodeResultOutput_SearchResults(CodeExecutionNodeSearchResultsResult):
-    type: typing_extensions.Literal["SEARCH_RESULTS"]
+    type: typing.Literal["SEARCH_RESULTS"] = "SEARCH_RESULTS"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class CodeExecutionNodeResultOutput_Error(CodeExecutionNodeErrorResult):
-    type: typing_extensions.Literal["ERROR"]
+    type: typing.Literal["ERROR"] = "ERROR"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
+class CodeExecutionNodeResultOutput_Array(CodeExecutionNodeArrayResult):
+    type: typing.Literal["ARRAY"] = "ARRAY"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
+class CodeExecutionNodeResultOutput_FunctionCall(CodeExecutionNodeFunctionCallResult):
+    type: typing.Literal["FUNCTION_CALL"] = "FUNCTION_CALL"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 CodeExecutionNodeResultOutput = typing.Union[
@@ -75,4 +101,6 @@ CodeExecutionNodeResultOutput = typing.Union[
     CodeExecutionNodeResultOutput_ChatHistory,
     CodeExecutionNodeResultOutput_SearchResults,
     CodeExecutionNodeResultOutput_Error,
+    CodeExecutionNodeResultOutput_Array,
+    CodeExecutionNodeResultOutput_FunctionCall,
 ]

vellum/types/code_execution_node_search_results_result.py

@@ -14,7 +14,7 @@ except ImportError:
 
 class CodeExecutionNodeSearchResultsResult(pydantic.BaseModel):
     id: str
-    value: typing.Optional[typing.List[SearchResult]]
+    value: typing.Optional[typing.List[SearchResult]] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -27,4 +27,5 @@ class CodeExecutionNodeSearchResultsResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/code_execution_node_string_result.py

@@ -13,7 +13,7 @@ except ImportError:
 
 class CodeExecutionNodeStringResult(pydantic.BaseModel):
     id: str
-    value: typing.Optional[str]
+    value: typing.Optional[str] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -26,4 +26,5 @@ class CodeExecutionNodeStringResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/conditional_node_result.py

@@ -30,4 +30,5 @@ class ConditionalNodeResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/conditional_node_result_data.py

@@ -12,7 +12,7 @@ except ImportError:
 
 
 class ConditionalNodeResultData(pydantic.BaseModel):
-    source_handle_id: typing.Optional[str]
+    source_handle_id: typing.Optional[str] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -25,4 +25,5 @@ class ConditionalNodeResultData(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/deployment_provider_payload_response.py

@@ -25,4 +25,5 @@ class DeploymentProviderPayloadResponse(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/deployment_read.py

@@ -17,25 +17,39 @@ except ImportError:
 class DeploymentRead(pydantic.BaseModel):
     id: str
     created: dt.datetime
-    label: str = pydantic.Field(description="A human-readable label for the deployment")
-    name: str = pydantic.Field(description="A name that uniquely identifies this deployment within its workspace")
-    status: typing.Optional[EntityStatus] = pydantic.Field(
-        description=("The current status of the deployment\n" "\n" "- `ACTIVE` - Active\n" "- `ARCHIVED` - Archived\n")
-    )
-    environment: typing.Optional[EnvironmentEnum] = pydantic.Field(
-        description=(
-            "The environment this deployment is used in\n"
-            "\n"
-            "- `DEVELOPMENT` - Development\n"
-            "- `STAGING` - Staging\n"
-            "- `PRODUCTION` - Production\n"
-        )
-    )
+    label: str = pydantic.Field()
+    """
+    A human-readable label for the deployment
+    """
+
+    name: str = pydantic.Field()
+    """
+    A name that uniquely identifies this deployment within its workspace
+    """
+
+    status: typing.Optional[EntityStatus] = pydantic.Field(default=None)
+    """
+    The current status of the deployment
+    
+    - `ACTIVE` - Active
+    - `ARCHIVED` - Archived
+    """
+
+    environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
+    """
+    The environment this deployment is used in
+    
+    - `DEVELOPMENT` - Development
+    - `STAGING` - Staging
+    - `PRODUCTION` - Production
+    """
+
     last_deployed_on: dt.datetime
     input_variables: typing.List[VellumVariable]
-    active_model_version_ids: typing.List[str] = pydantic.Field(
-        description="Deprecated. The Prompt execution endpoints return a `prompt_version_id` that could be used instead."
-    )
+    active_model_version_ids: typing.List[str] = pydantic.Field()
+    """
+    Deprecated. The Prompt execution endpoints return a `prompt_version_id` that could be used instead.
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -48,4 +62,5 @@ class DeploymentRead(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/document_document_to_document_index.py

@@ -13,21 +13,26 @@ except ImportError:
 
 
 class DocumentDocumentToDocumentIndex(pydantic.BaseModel):
-    id: str = pydantic.Field(description="Vellum-generated ID that uniquely identifies this link.")
-    document_index_id: str = pydantic.Field(
-        description="Vellum-generated ID that uniquely identifies the index this document is included in."
-    )
-    indexing_state: typing.Optional[IndexingStateEnum] = pydantic.Field(
-        description=(
-            "An enum value representing where this document is along its indexing lifecycle for this index.\n"
-            "\n"
-            "- `AWAITING_PROCESSING` - Awaiting Processing\n"
-            "- `QUEUED` - Queued\n"
-            "- `INDEXING` - Indexing\n"
-            "- `INDEXED` - Indexed\n"
-            "- `FAILED` - Failed\n"
-        )
-    )
+    id: str = pydantic.Field()
+    """
+    Vellum-generated ID that uniquely identifies this link.
+    """
+
+    document_index_id: str = pydantic.Field()
+    """
+    Vellum-generated ID that uniquely identifies the index this document is included in.
+    """
+
+    indexing_state: typing.Optional[IndexingStateEnum] = pydantic.Field(default=None)
+    """
+    An enum value representing where this document is along its indexing lifecycle for this index.
+    
+    - `AWAITING_PROCESSING` - Awaiting Processing
+    - `QUEUED` - Queued
+    - `INDEXING` - Indexing
+    - `INDEXED` - Indexed
+    - `FAILED` - Failed
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -40,4 +45,5 @@ class DocumentDocumentToDocumentIndex(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/document_index_read.py

@@ -16,25 +16,37 @@ except ImportError:
 class DocumentIndexRead(pydantic.BaseModel):
     id: str
     created: dt.datetime
-    label: str = pydantic.Field(description="A human-readable label for the document index")
-    name: str = pydantic.Field(description="A name that uniquely identifies this index within its workspace")
-    status: typing.Optional[EntityStatus] = pydantic.Field(
-        description=(
-            "The current status of the document index\n" "\n" "- `ACTIVE` - Active\n" "- `ARCHIVED` - Archived\n"
-        )
-    )
-    environment: typing.Optional[EnvironmentEnum] = pydantic.Field(
-        description=(
-            "The environment this document index is used in\n"
-            "\n"
-            "- `DEVELOPMENT` - Development\n"
-            "- `STAGING` - Staging\n"
-            "- `PRODUCTION` - Production\n"
-        )
-    )
-    indexing_config: typing.Dict[str, typing.Any] = pydantic.Field(
-        description="Configuration representing how documents should be indexed"
-    )
+    label: str = pydantic.Field()
+    """
+    A human-readable label for the document index
+    """
+
+    name: str = pydantic.Field()
+    """
+    A name that uniquely identifies this index within its workspace
+    """
+
+    status: typing.Optional[EntityStatus] = pydantic.Field(default=None)
+    """
+    The current status of the document index
+    
+    - `ACTIVE` - Active
+    - `ARCHIVED` - Archived
+    """
+
+    environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
+    """
+    The environment this document index is used in
+    
+    - `DEVELOPMENT` - Development
+    - `STAGING` - Staging
+    - `PRODUCTION` - Production
+    """
+
+    indexing_config: typing.Dict[str, typing.Any] = pydantic.Field()
+    """
+    Configuration representing how documents should be indexed
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -47,4 +59,5 @@ class DocumentIndexRead(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}