vellum-ai 0.3.13__py3-none-any.whl → 0.3.15__py3-none-any.whl

vellum/types/model_version_exec_config.py

@@ -15,17 +15,23 @@ except ImportError:
 
 
 class ModelVersionExecConfig(pydantic.BaseModel):
-    parameters: ModelVersionExecConfigParameters = pydantic.Field(
-        description="The generation parameters that are passed to the LLM provider at runtime."
-    )
-    input_variables: typing.List[VellumVariable] = pydantic.Field(
-        description="Input variables specified in the prompt template."
-    )
-    prompt_template: typing.Optional[str] = pydantic.Field(
-        description="The template used to generate prompts for this model version."
-    )
-    prompt_block_data: typing.Optional[PromptTemplateBlockData]
-    prompt_syntax_version: typing.Optional[int]
+    parameters: ModelVersionExecConfigParameters = pydantic.Field()
+    """
+    The generation parameters that are passed to the LLM provider at runtime.
+    """
+
+    input_variables: typing.List[VellumVariable] = pydantic.Field()
+    """
+    Input variables specified in the prompt template.
+    """
+
+    prompt_template: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The template used to generate prompts for this model version.
+    """
+
+    prompt_block_data: typing.Optional[PromptTemplateBlockData] = None
+    prompt_syntax_version: typing.Optional[int] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -38,4 +44,5 @@ class ModelVersionExecConfig(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/model_version_exec_config_parameters.py

@@ -12,15 +12,15 @@ except ImportError:
 
 
 class ModelVersionExecConfigParameters(pydantic.BaseModel):
-    temperature: typing.Optional[float]
-    max_tokens: typing.Optional[int]
+    temperature: typing.Optional[float] = None
+    max_tokens: typing.Optional[int] = None
     top_p: float
     frequency_penalty: float
     presence_penalty: float
-    logit_bias: typing.Optional[typing.Dict[str, typing.Optional[float]]]
-    stop: typing.Optional[typing.List[str]]
-    top_k: typing.Optional[float]
-    custom_parameters: typing.Optional[typing.Dict[str, typing.Any]]
+    logit_bias: typing.Optional[typing.Dict[str, typing.Optional[float]]] = None
+    stop: typing.Optional[typing.List[str]] = None
+    top_k: typing.Optional[float] = None
+    custom_parameters: 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}
@@ -33,4 +33,5 @@ class ModelVersionExecConfigParameters(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/model_version_read.py

@@ -16,38 +16,56 @@ except ImportError:
 
 
 class ModelVersionRead(pydantic.BaseModel):
-    id: str = pydantic.Field(description="Vellum-generated ID that uniquely identifies this model version.")
-    created: dt.datetime = pydantic.Field(description="Timestamp of when this model version was created.")
-    label: str = pydantic.Field(description="Human-friendly name for this model version.")
-    provider: ProviderEnum = pydantic.Field(
-        description=(
-            "Which LLM provider this model version is associated with.\n"
-            "\n"
-            "- `ANTHROPIC` - Anthropic\n"
-            "- `AWS_BEDROCK` - AWS Bedrock\n"
-            "- `AZURE_OPENAI` - Azure OpenAI\n"
-            "- `COHERE` - Cohere\n"
-            "- `GOOGLE` - Google\n"
-            "- `HOSTED` - Hosted\n"
-            "- `MOSAICML` - MosaicML\n"
-            "- `OPENAI` - OpenAI\n"
-            "- `FIREWORKS_AI` - Fireworks AI\n"
-            "- `HUGGINGFACE` - HuggingFace\n"
-            "- `MYSTIC` - Mystic\n"
-            "- `PYQ` - Pyq\n"
-            "- `REPLICATE` - Replicate\n"
-        )
-    )
-    external_id: str = pydantic.Field(
-        description="The unique id of this model version as it exists in the above provider's system."
-    )
-    build_config: ModelVersionBuildConfig = pydantic.Field(
-        description="Configuration used to build this model version."
-    )
-    exec_config: ModelVersionExecConfig = pydantic.Field(
-        description="Configuration used to execute this model version."
-    )
-    status: typing.Optional[ModelVersionReadStatusEnum]
+    id: str = pydantic.Field()
+    """
+    Vellum-generated ID that uniquely identifies this model version.
+    """
+
+    created: dt.datetime = pydantic.Field()
+    """
+    Timestamp of when this model version was created.
+    """
+
+    label: str = pydantic.Field()
+    """
+    Human-friendly name for this model version.
+    """
+
+    provider: ProviderEnum = pydantic.Field()
+    """
+    Which LLM provider this model version is associated with.
+    
+    - `ANTHROPIC` - Anthropic
+    - `AWS_BEDROCK` - AWS Bedrock
+    - `AZURE_OPENAI` - Azure OpenAI
+    - `COHERE` - Cohere
+    - `GOOGLE` - Google
+    - `HOSTED` - Hosted
+    - `MOSAICML` - MosaicML
+    - `OPENAI` - OpenAI
+    - `FIREWORKS_AI` - Fireworks AI
+    - `HUGGINGFACE` - HuggingFace
+    - `MYSTIC` - Mystic
+    - `PYQ` - Pyq
+    - `REPLICATE` - Replicate
+    """
+
+    external_id: str = pydantic.Field()
+    """
+    The unique id of this model version as it exists in the above provider's system.
+    """
+
+    build_config: ModelVersionBuildConfig = pydantic.Field()
+    """
+    Configuration used to build this model version.
+    """
+
+    exec_config: ModelVersionExecConfig = pydantic.Field()
+    """
+    Configuration used to execute this model version.
+    """
+
+    status: typing.Optional[ModelVersionReadStatusEnum] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -60,4 +78,5 @@ class ModelVersionRead(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/model_version_sandbox_snapshot.py

@@ -12,10 +12,22 @@ except ImportError:
 
 
 class ModelVersionSandboxSnapshot(pydantic.BaseModel):
-    id: str = pydantic.Field(description="The ID of the sandbox snapshot.")
-    prompt_index: typing.Optional[int] = pydantic.Field(description="The index of the prompt in the sandbox snapshot.")
-    prompt_id: typing.Optional[str] = pydantic.Field(description="The id of the prompt in the sandbox snapshot.")
-    sandbox_id: typing.Optional[str]
+    id: str = pydantic.Field()
+    """
+    The ID of the sandbox snapshot.
+    """
+
+    prompt_index: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    The index of the prompt in the sandbox snapshot.
+    """
+
+    prompt_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The id of the prompt in the sandbox snapshot.
+    """
+
+    sandbox_id: typing.Optional[str] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -28,4 +40,5 @@ class ModelVersionSandboxSnapshot(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_test_case_chat_history_variable_value_request.py

@@ -14,7 +14,7 @@ except ImportError:
 
 class NamedTestCaseChatHistoryVariableValueRequest(pydantic.BaseModel):
     name: str
-    value: typing.Optional[typing.List[ChatMessageRequest]]
+    value: typing.Optional[typing.List[ChatMessageRequest]] = 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 NamedTestCaseChatHistoryVariableValueRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_test_case_error_variable_value_request.py

@@ -14,7 +14,7 @@ except ImportError:
 
 class NamedTestCaseErrorVariableValueRequest(pydantic.BaseModel):
     name: str
-    value: typing.Optional[VellumErrorRequest]
+    value: typing.Optional[VellumErrorRequest] = 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 NamedTestCaseErrorVariableValueRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_test_case_json_variable_value_request.py

@@ -13,7 +13,7 @@ except ImportError:
 
 class NamedTestCaseJsonVariableValueRequest(pydantic.BaseModel):
     name: 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 NamedTestCaseJsonVariableValueRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_test_case_number_variable_value_request.py

@@ -13,7 +13,7 @@ except ImportError:
 
 class NamedTestCaseNumberVariableValueRequest(pydantic.BaseModel):
     name: 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 NamedTestCaseNumberVariableValueRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_test_case_search_results_variable_value_request.py

@@ -14,7 +14,7 @@ except ImportError:
 
 class NamedTestCaseSearchResultsVariableValueRequest(pydantic.BaseModel):
     name: str
-    value: typing.Optional[typing.List[SearchResultRequest]]
+    value: typing.Optional[typing.List[SearchResultRequest]] = 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 NamedTestCaseSearchResultsVariableValueRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_test_case_string_variable_value_request.py

@@ -13,7 +13,7 @@ except ImportError:
 
 class NamedTestCaseStringVariableValueRequest(pydantic.BaseModel):
     name: 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 NamedTestCaseStringVariableValueRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_test_case_variable_value_request.py

@@ -4,8 +4,6 @@ from __future__ import annotations
 
 import typing
 
-import typing_extensions
-
 from .named_test_case_chat_history_variable_value_request import NamedTestCaseChatHistoryVariableValueRequest
 from .named_test_case_error_variable_value_request import NamedTestCaseErrorVariableValueRequest
 from .named_test_case_json_variable_value_request import NamedTestCaseJsonVariableValueRequest
@@ -15,57 +13,63 @@ from .named_test_case_string_variable_value_request import NamedTestCaseStringVa
 
 
 class NamedTestCaseVariableValueRequest_String(NamedTestCaseStringVariableValueRequest):
-    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 NamedTestCaseVariableValueRequest_Number(NamedTestCaseNumberVariableValueRequest):
-    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 NamedTestCaseVariableValueRequest_Json(NamedTestCaseJsonVariableValueRequest):
-    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 NamedTestCaseVariableValueRequest_ChatHistory(NamedTestCaseChatHistoryVariableValueRequest):
-    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 NamedTestCaseVariableValueRequest_SearchResults(NamedTestCaseSearchResultsVariableValueRequest):
-    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 NamedTestCaseVariableValueRequest_Error(NamedTestCaseErrorVariableValueRequest):
-    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
 
 
 NamedTestCaseVariableValueRequest = typing.Union[

vellum/types/node_input_compiled_array_value.py

@@ -15,7 +15,7 @@ except ImportError:
 class NodeInputCompiledArrayValue(pydantic.BaseModel):
     node_input_id: str
     key: str
-    value: typing.Optional[typing.List[ArrayVariableValueItem]]
+    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}
@@ -28,4 +28,5 @@ class NodeInputCompiledArrayValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_input_compiled_chat_history_value.py

@@ -15,7 +15,7 @@ except ImportError:
 class NodeInputCompiledChatHistoryValue(pydantic.BaseModel):
     node_input_id: str
     key: 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}
@@ -28,4 +28,5 @@ class NodeInputCompiledChatHistoryValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_input_compiled_error_value.py

@@ -15,7 +15,7 @@ except ImportError:
 class NodeInputCompiledErrorValue(pydantic.BaseModel):
     node_input_id: str
     key: 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}
@@ -28,4 +28,5 @@ class NodeInputCompiledErrorValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_input_compiled_function_call.py

@@ -0,0 +1,32 @@
+# 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 NodeInputCompiledFunctionCall(pydantic.BaseModel):
+    node_input_id: str
+    key: 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/node_input_compiled_json_value.py

@@ -14,7 +14,7 @@ except ImportError:
 class NodeInputCompiledJsonValue(pydantic.BaseModel):
     node_input_id: str
     key: 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}
@@ -27,4 +27,5 @@ class NodeInputCompiledJsonValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_input_compiled_number_value.py

@@ -14,7 +14,7 @@ except ImportError:
 class NodeInputCompiledNumberValue(pydantic.BaseModel):
     node_input_id: str
     key: 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}
@@ -27,4 +27,5 @@ class NodeInputCompiledNumberValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_input_compiled_search_results_value.py

@@ -15,7 +15,7 @@ except ImportError:
 class NodeInputCompiledSearchResultsValue(pydantic.BaseModel):
     node_input_id: str
     key: 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}
@@ -28,4 +28,5 @@ class NodeInputCompiledSearchResultsValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_input_compiled_string_value.py

@@ -14,7 +14,7 @@ except ImportError:
 class NodeInputCompiledStringValue(pydantic.BaseModel):
     node_input_id: str
     key: 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}
@@ -27,4 +27,5 @@ class NodeInputCompiledStringValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_input_variable_compiled_value.py

@@ -4,11 +4,10 @@ from __future__ import annotations
 
 import typing
 
-import typing_extensions
-
 from .node_input_compiled_array_value import NodeInputCompiledArrayValue
 from .node_input_compiled_chat_history_value import NodeInputCompiledChatHistoryValue
 from .node_input_compiled_error_value import NodeInputCompiledErrorValue
+from .node_input_compiled_function_call import NodeInputCompiledFunctionCall
 from .node_input_compiled_json_value import NodeInputCompiledJsonValue
 from .node_input_compiled_number_value import NodeInputCompiledNumberValue
 from .node_input_compiled_search_results_value import NodeInputCompiledSearchResultsValue
@@ -16,66 +15,83 @@ from .node_input_compiled_string_value import NodeInputCompiledStringValue
 
 
 class NodeInputVariableCompiledValue_String(NodeInputCompiledStringValue):
-    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 NodeInputVariableCompiledValue_Number(NodeInputCompiledNumberValue):
-    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 NodeInputVariableCompiledValue_Json(NodeInputCompiledJsonValue):
-    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 NodeInputVariableCompiledValue_ChatHistory(NodeInputCompiledChatHistoryValue):
-    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 NodeInputVariableCompiledValue_SearchResults(NodeInputCompiledSearchResultsValue):
-    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 NodeInputVariableCompiledValue_Error(NodeInputCompiledErrorValue):
-    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 NodeInputVariableCompiledValue_Array(NodeInputCompiledArrayValue):
-    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 NodeInputVariableCompiledValue_FunctionCall(NodeInputCompiledFunctionCall):
+    type: typing.Literal["FUNCTION_CALL"] = "FUNCTION_CALL"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 NodeInputVariableCompiledValue = typing.Union[
@@ -86,4 +102,5 @@ NodeInputVariableCompiledValue = typing.Union[
     NodeInputVariableCompiledValue_SearchResults,
     NodeInputVariableCompiledValue_Error,
     NodeInputVariableCompiledValue_Array,
+    NodeInputVariableCompiledValue_FunctionCall,
 ]

vellum/types/node_output_compiled_array_value.py

@@ -14,7 +14,7 @@ except ImportError:
 
 class NodeOutputCompiledArrayValue(pydantic.BaseModel):
     node_output_id: str
-    value: typing.Optional[typing.List[ArrayVariableValueItem]]
+    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}
@@ -27,4 +27,5 @@ class NodeOutputCompiledArrayValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/node_output_compiled_chat_history_value.py

@@ -14,7 +14,7 @@ except ImportError:
 
 class NodeOutputCompiledChatHistoryValue(pydantic.BaseModel):
     node_output_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 NodeOutputCompiledChatHistoryValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}