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

vellum/types/fulfilled_execute_prompt_event.py

@@ -20,7 +20,7 @@ class FulfilledExecutePromptEvent(pydantic.BaseModel):
 
     outputs: typing.List[PromptOutput]
     execution_id: str
-    meta: typing.Optional[FulfilledPromptExecutionMeta]
+    meta: typing.Optional[FulfilledPromptExecutionMeta] = 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 FulfilledExecutePromptEvent(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/fulfilled_execute_prompt_response.py

@@ -18,11 +18,17 @@ class FulfilledExecutePromptResponse(pydantic.BaseModel):
     The successful response from the model containing all of the resolved values generated by the prompt.
     """
 
-    meta: typing.Optional[PromptExecutionMeta]
-    raw: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(
-        description="The subset of the raw response from the model that the request opted into with `expand_raw`."
-    )
-    execution_id: str = pydantic.Field(description="The ID of the execution.")
+    meta: typing.Optional[PromptExecutionMeta] = None
+    raw: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.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()
+    """
+    The ID of the execution.
+    """
+
     outputs: typing.List[PromptOutput]
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -36,4 +42,5 @@ class FulfilledExecutePromptResponse(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/fulfilled_execute_workflow_workflow_result_event.py

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

vellum/types/fulfilled_function_call.py

@@ -17,7 +17,7 @@ class FulfilledFunctionCall(pydantic.BaseModel):
     """
 
     arguments: typing.Dict[str, typing.Any]
-    id: typing.Optional[str]
+    id: typing.Optional[str] = None
     name: str
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -31,4 +31,5 @@ class FulfilledFunctionCall(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/fulfilled_prompt_execution_meta.py

@@ -17,8 +17,8 @@ class FulfilledPromptExecutionMeta(pydantic.BaseModel):
     The subset of the metadata tracked by Vellum during prompt execution that the request opted into with `expand_meta`.
     """
 
-    latency: typing.Optional[int]
-    finish_reason: typing.Optional[FinishReasonEnum]
+    latency: typing.Optional[int] = None
+    finish_reason: typing.Optional[FinishReasonEnum] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -31,4 +31,5 @@ class FulfilledPromptExecutionMeta(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/fulfilled_workflow_node_result_event.py

@@ -21,11 +21,11 @@ class FulfilledWorkflowNodeResultEvent(pydantic.BaseModel):
     id: str
     node_id: str
     node_result_id: str
-    ts: typing.Optional[dt.datetime]
-    data: typing.Optional[WorkflowNodeResultData]
-    source_execution_id: typing.Optional[str]
-    output_values: typing.Optional[typing.List[NodeOutputCompiledValue]]
-    mocked: typing.Optional[bool]
+    ts: typing.Optional[dt.datetime] = None
+    data: typing.Optional[WorkflowNodeResultData] = None
+    source_execution_id: typing.Optional[str] = None
+    output_values: typing.Optional[typing.List[NodeOutputCompiledValue]] = None
+    mocked: typing.Optional[bool] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -38,4 +38,5 @@ class FulfilledWorkflowNodeResultEvent(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/function_call.py

@@ -4,28 +4,28 @@ from __future__ import annotations
 
 import typing
 
-import typing_extensions
-
 from .fulfilled_function_call import FulfilledFunctionCall
 from .rejected_function_call import RejectedFunctionCall
 
 
 class FunctionCall_Fulfilled(FulfilledFunctionCall):
-    state: typing_extensions.Literal["FULFILLED"]
+    state: typing.Literal["FULFILLED"] = "FULFILLED"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 class FunctionCall_Rejected(RejectedFunctionCall):
-    state: typing_extensions.Literal["REJECTED"]
+    state: typing.Literal["REJECTED"] = "REJECTED"
 
     class Config:
         frozen = True
         smart_union = True
         allow_population_by_field_name = True
+        populate_by_name = True
 
 
 FunctionCall = typing.Union[FunctionCall_Fulfilled, FunctionCall_Rejected]

vellum/types/function_call_chat_message_content.py

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

vellum/types/function_call_chat_message_content_request.py

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

vellum/types/function_call_chat_message_content_value.py

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

vellum/types/function_call_chat_message_content_value_request.py

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

vellum/types/function_call_enum.py

@@ -1,5 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import typing_extensions
+import typing
 
-FunctionCallEnum = typing_extensions.Literal["FUNCTION_CALL"]
+FunctionCallEnum = typing.Literal["FUNCTION_CALL"]

vellum/types/function_call_variable_value.py

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

vellum/types/generate_error_response.py

@@ -12,7 +12,10 @@ except ImportError:
 
 
 class GenerateErrorResponse(pydantic.BaseModel):
-    detail: str = pydantic.Field(description="Details about why the request failed.")
+    detail: str = pydantic.Field()
+    """
+    Details about why the request failed.
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -25,4 +28,5 @@ class GenerateErrorResponse(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/generate_options_request.py

@@ -13,11 +13,13 @@ except ImportError:
 
 
 class GenerateOptionsRequest(pydantic.BaseModel):
-    logprobs: typing.Optional[LogprobsEnum] = pydantic.Field(
-        description=(
-            "Which logprobs to include, if any. Defaults to NONE.\n" "\n" "- `ALL` - ALL\n" "- `NONE` - NONE\n"
-        )
-    )
+    logprobs: typing.Optional[LogprobsEnum] = pydantic.Field(default=None)
+    """
+    Which logprobs to include, if any. Defaults to NONE.
+    
+    - `ALL` - ALL
+    - `NONE` - NONE
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -30,4 +32,5 @@ class GenerateOptionsRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/generate_request.py

@@ -13,15 +13,20 @@ except ImportError:
 
 
 class GenerateRequest(pydantic.BaseModel):
-    input_values: typing.Dict[str, typing.Any] = pydantic.Field(
-        description="Key/value pairs for each template variable defined in the deployment's prompt."
-    )
-    chat_history: typing.Optional[typing.List[ChatMessageRequest]] = pydantic.Field(
-        description="Optionally provide a list of chat messages that'll be used in place of the special chat_history variable, if included in the prompt."
-    )
-    external_ids: typing.Optional[typing.List[str]] = pydantic.Field(
-        description="Optionally include a unique identifier for each generation, as represented outside of Vellum. Note that this should generally be a list of length one."
-    )
+    input_values: typing.Dict[str, typing.Any] = pydantic.Field()
+    """
+    Key/value pairs for each template variable defined in the deployment's prompt.
+    """
+
+    chat_history: typing.Optional[typing.List[ChatMessageRequest]] = pydantic.Field(default=None)
+    """
+    Optionally provide a list of chat messages that'll be used in place of the special chat_history variable, if included in the prompt.
+    """
+
+    external_ids: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    """
+    Optionally include a unique identifier for each generation, as represented outside of Vellum. Note that this should generally be a list of length one.
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -34,4 +39,5 @@ class GenerateRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/generate_result.py

@@ -14,12 +14,15 @@ except ImportError:
 
 
 class GenerateResult(pydantic.BaseModel):
-    data: typing.Optional[GenerateResultData] = pydantic.Field(
-        description="An object containing the resulting generation. This key will be absent if the LLM provider experienced an error."
-    )
-    error: typing.Optional[GenerateResultError] = pydantic.Field(
-        description="An object containing details about the error that occurred. This key will be absent if the LLM provider did not experience an error."
-    )
+    data: typing.Optional[GenerateResultData] = pydantic.Field(default=None)
+    """
+    An object containing the resulting generation. This key will be absent if the LLM provider experienced an error.
+    """
+
+    error: typing.Optional[GenerateResultError] = pydantic.Field(default=None)
+    """
+    An object containing details about the error that occurred. This key will be absent if the LLM provider did not experience an error.
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -32,4 +35,5 @@ class GenerateResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/generate_result_data.py

@@ -13,9 +13,10 @@ except ImportError:
 
 
 class GenerateResultData(pydantic.BaseModel):
-    completions: typing.List[EnrichedNormalizedCompletion] = pydantic.Field(
-        description="The generated completions. This will generally be a list of length one."
-    )
+    completions: typing.List[EnrichedNormalizedCompletion] = pydantic.Field()
+    """
+    The generated completions. This will generally be a list of length one.
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -28,4 +29,5 @@ class GenerateResultData(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/generate_result_error.py

@@ -12,7 +12,10 @@ except ImportError:
 
 
 class GenerateResultError(pydantic.BaseModel):
-    message: str = pydantic.Field(description="The error message returned by the LLM provider.")
+    message: str = pydantic.Field()
+    """
+    The error message returned by the LLM provider.
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -25,4 +28,5 @@ class GenerateResultError(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/generate_stream_response.py

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

vellum/types/generate_stream_result.py

@@ -15,8 +15,8 @@ except ImportError:
 
 class GenerateStreamResult(pydantic.BaseModel):
     request_index: int
-    data: typing.Optional[GenerateStreamResultData]
-    error: typing.Optional[GenerateResultError]
+    data: typing.Optional[GenerateStreamResultData] = None
+    error: typing.Optional[GenerateResultError] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -29,4 +29,5 @@ class GenerateStreamResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/generate_stream_result_data.py

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

vellum/types/image_chat_message_content.py

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

vellum/types/image_chat_message_content_request.py

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

vellum/types/image_enum.py

@@ -1,5 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import typing_extensions
+import typing
 
-ImageEnum = typing_extensions.Literal["IMAGE"]
+ImageEnum = typing.Literal["IMAGE"]

vellum/types/image_variable_value.py

@@ -17,7 +17,7 @@ class ImageVariableValue(pydantic.BaseModel):
     A base Vellum primitive value representing an image.
     """
 
-    value: typing.Optional[VellumImage]
+    value: typing.Optional[VellumImage] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -30,4 +30,5 @@ class ImageVariableValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/initiated_enum.py

@@ -1,5 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import typing_extensions
+import typing
 
-InitiatedEnum = typing_extensions.Literal["INITIATED"]
+InitiatedEnum = typing.Literal["INITIATED"]

vellum/types/initiated_execute_prompt_event.py

@@ -17,7 +17,7 @@ class InitiatedExecutePromptEvent(pydantic.BaseModel):
     The initial data returned indicating that the response from the model has returned and begun streaming.
     """
 
-    meta: typing.Optional[InitiatedPromptExecutionMeta]
+    meta: typing.Optional[InitiatedPromptExecutionMeta] = None
     execution_id: str
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -31,4 +31,5 @@ class InitiatedExecutePromptEvent(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/initiated_prompt_execution_meta.py

@@ -16,10 +16,10 @@ class InitiatedPromptExecutionMeta(pydantic.BaseModel):
     The subset of the metadata tracked by Vellum during prompt execution that the request opted into with `expand_meta`.
     """
 
-    model_name: typing.Optional[str]
-    latency: typing.Optional[int]
-    deployment_release_tag: typing.Optional[str]
-    prompt_version_id: typing.Optional[str]
+    model_name: typing.Optional[str] = None
+    latency: typing.Optional[int] = None
+    deployment_release_tag: typing.Optional[str] = None
+    prompt_version_id: typing.Optional[str] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -32,4 +32,5 @@ class InitiatedPromptExecutionMeta(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/initiated_workflow_node_result_event.py

@@ -21,10 +21,10 @@ class InitiatedWorkflowNodeResultEvent(pydantic.BaseModel):
     id: str
     node_id: str
     node_result_id: str
-    ts: typing.Optional[dt.datetime]
-    data: typing.Optional[WorkflowNodeResultData]
-    source_execution_id: typing.Optional[str]
-    input_values: typing.Optional[typing.List[NodeInputVariableCompiledValue]]
+    ts: typing.Optional[dt.datetime] = None
+    data: typing.Optional[WorkflowNodeResultData] = None
+    source_execution_id: typing.Optional[str] = None
+    input_values: typing.Optional[typing.List[NodeInputVariableCompiledValue]] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -37,4 +37,5 @@ class InitiatedWorkflowNodeResultEvent(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/json_enum.py

@@ -1,5 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import typing_extensions
+import typing
 
-JsonEnum = typing_extensions.Literal["JSON"]
+JsonEnum = typing.Literal["JSON"]

vellum/types/json_input_request.py

@@ -16,7 +16,11 @@ class JsonInputRequest(pydantic.BaseModel):
     A user input representing a JSON object
     """
 
-    name: str = pydantic.Field(description="The variable's name, as defined in the deployment.")
+    name: str = pydantic.Field()
+    """
+    The variable's name, as defined in the deployment.
+    """
+
     value: typing.Dict[str, typing.Any]
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -30,4 +34,5 @@ class JsonInputRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/json_variable_value.py

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

vellum/types/logprobs_enum.py

@@ -15,8 +15,8 @@ class LogprobsEnum(str, enum.Enum):
     ALL = "ALL"
     NONE = "NONE"
 
-    def visit(self, all: typing.Callable[[], T_Result], none: typing.Callable[[], T_Result]) -> T_Result:
+    def visit(self, all_: typing.Callable[[], T_Result], none: typing.Callable[[], T_Result]) -> T_Result:
         if self is LogprobsEnum.ALL:
-            return all()
+            return all_()
         if self is LogprobsEnum.NONE:
             return none()

vellum/types/metadata_filter_config_request.py

@@ -15,12 +15,12 @@ except ImportError:
 
 
 class MetadataFilterConfigRequest(pydantic.BaseModel):
-    combinator: typing.Optional[MetadataFilterRuleCombinator]
-    negated: typing.Optional[bool]
-    rules: typing.Optional[typing.List[MetadataFilterRuleRequest]]
-    field: typing.Optional[str]
-    operator: typing.Optional[LogicalOperator]
-    value: typing.Optional[str]
+    combinator: typing.Optional[MetadataFilterRuleCombinator] = None
+    negated: typing.Optional[bool] = None
+    rules: typing.Optional[typing.List[MetadataFilterRuleRequest]] = None
+    field: typing.Optional[str] = None
+    operator: typing.Optional[LogicalOperator] = None
+    value: typing.Optional[str] = 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 MetadataFilterConfigRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/metadata_filter_rule_request.py

@@ -16,12 +16,12 @@ except ImportError:
 
 
 class MetadataFilterRuleRequest(pydantic.BaseModel):
-    combinator: typing.Optional[MetadataFilterRuleCombinator]
-    negated: typing.Optional[bool]
-    rules: typing.Optional[typing.List[MetadataFilterRuleRequest]]
-    field: typing.Optional[str]
-    operator: typing.Optional[LogicalOperator]
-    value: typing.Optional[str]
+    combinator: typing.Optional[MetadataFilterRuleCombinator] = None
+    negated: typing.Optional[bool] = None
+    rules: typing.Optional[typing.List[MetadataFilterRuleRequest]] = None
+    field: typing.Optional[str] = None
+    operator: typing.Optional[LogicalOperator] = None
+    value: typing.Optional[str] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -34,6 +34,7 @@ class MetadataFilterRuleRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}
 
 

vellum/types/model_version_build_config.py

@@ -13,13 +13,17 @@ except ImportError:
 
 
 class ModelVersionBuildConfig(pydantic.BaseModel):
-    base_model: str = pydantic.Field(
-        description="The name of the base model used to create this model version, as identified by the LLM provider."
-    )
-    sandbox_snapshot: typing.Optional[ModelVersionSandboxSnapshot] = pydantic.Field(
-        description="Information about the sandbox snapshot that was used to create this model version, if applicable."
-    )
-    prompt_version_id: typing.Optional[str]
+    base_model: str = pydantic.Field()
+    """
+    The name of the base model used to create this model version, as identified by the LLM provider.
+    """
+
+    sandbox_snapshot: typing.Optional[ModelVersionSandboxSnapshot] = pydantic.Field(default=None)
+    """
+    Information about the sandbox snapshot that was used to create this model version, if applicable.
+    """
+
+    prompt_version_id: typing.Optional[str] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -32,4 +36,5 @@ class ModelVersionBuildConfig(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}