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

vellum/types/prompt_template_block_properties_request.py

@@ -16,16 +16,16 @@ except ImportError:
 
 
 class PromptTemplateBlockPropertiesRequest(pydantic.BaseModel):
-    chat_role: typing.Optional[ChatMessageRole]
-    chat_message_unterminated: typing.Optional[bool]
-    chat_source: typing.Optional[str]
-    template: typing.Optional[str]
-    template_type: typing.Optional[VellumVariableType]
-    function_name: typing.Optional[str]
-    function_description: typing.Optional[str]
-    function_parameters: typing.Optional[typing.Dict[str, typing.Any]]
-    function_forced: typing.Optional[bool]
-    blocks: typing.Optional[typing.List[PromptTemplateBlockRequest]]
+    chat_role: typing.Optional[ChatMessageRole] = None
+    chat_message_unterminated: typing.Optional[bool] = None
+    chat_source: typing.Optional[str] = None
+    template: typing.Optional[str] = None
+    template_type: typing.Optional[VellumVariableType] = None
+    function_name: typing.Optional[str] = None
+    function_description: typing.Optional[str] = None
+    function_parameters: typing.Optional[typing.Dict[str, typing.Any]] = None
+    function_forced: typing.Optional[bool] = None
+    blocks: typing.Optional[typing.List[PromptTemplateBlockRequest]] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -38,6 +38,7 @@ class PromptTemplateBlockPropertiesRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}
 
 

vellum/types/prompt_template_block_request.py

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

vellum/types/raw_prompt_execution_overrides_request.py

@@ -12,11 +12,16 @@ except ImportError:
 
 
 class RawPromptExecutionOverridesRequest(pydantic.BaseModel):
-    body: typing.Optional[typing.Dict[str, typing.Any]]
-    headers: typing.Optional[typing.Dict[str, typing.Optional[str]]] = pydantic.Field(
-        description="The raw headers to send to the model host."
-    )
-    url: typing.Optional[str] = pydantic.Field(description="The raw URL to send to the model host.")
+    body: typing.Optional[typing.Dict[str, typing.Any]] = None
+    headers: typing.Optional[typing.Dict[str, typing.Optional[str]]] = pydantic.Field(default=None)
+    """
+    The raw headers to send to the model host.
+    """
+
+    url: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The raw URL to send to the model host.
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -29,4 +34,5 @@ class RawPromptExecutionOverridesRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/register_prompt_error_response.py

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

vellum/types/register_prompt_model_parameters_request.py

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

vellum/types/register_prompt_prompt.py

@@ -12,8 +12,15 @@ except ImportError:
 
 
 class RegisterPromptPrompt(pydantic.BaseModel):
-    id: str = pydantic.Field(description="The ID of the generated prompt.")
-    label: str = pydantic.Field(description="A human-friendly label for the generated prompt.")
+    id: str = pydantic.Field()
+    """
+    The ID of the generated prompt.
+    """
+
+    label: str = pydantic.Field()
+    """
+    A human-friendly label for the generated prompt.
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -26,4 +33,5 @@ class RegisterPromptPrompt(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/register_prompt_prompt_info_request.py

@@ -15,9 +15,10 @@ except ImportError:
 
 class RegisterPromptPromptInfoRequest(pydantic.BaseModel):
     prompt_block_data: PromptTemplateBlockDataRequest
-    input_variables: typing.List[RegisteredPromptInputVariableRequest] = pydantic.Field(
-        description="The input variables specified in the prompt template."
-    )
+    input_variables: typing.List[RegisteredPromptInputVariableRequest] = pydantic.Field()
+    """
+    The input variables specified in the prompt template.
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -30,4 +31,5 @@ class RegisterPromptPromptInfoRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/register_prompt_response.py

@@ -17,16 +17,35 @@ except ImportError:
 
 
 class RegisterPromptResponse(pydantic.BaseModel):
-    prompt: RegisterPromptPrompt = pydantic.Field(description="Information about the generated prompt")
-    sandbox_snapshot: RegisteredPromptSandboxSnapshot = pydantic.Field(
-        description="Information about the generated sandbox snapshot"
-    )
-    sandbox: RegisteredPromptSandbox = pydantic.Field(description="Information about the generated sandbox")
-    model_version: RegisteredPromptModelVersion = pydantic.Field(
-        description="Information about the generated model version"
-    )
-    prompt_version_id: str = pydantic.Field(description="The ID of the generated prompt version")
-    deployment: RegisteredPromptDeployment = pydantic.Field(description="Information about the generated deployment")
+    prompt: RegisterPromptPrompt = pydantic.Field()
+    """
+    Information about the generated prompt
+    """
+
+    sandbox_snapshot: RegisteredPromptSandboxSnapshot = pydantic.Field()
+    """
+    Information about the generated sandbox snapshot
+    """
+
+    sandbox: RegisteredPromptSandbox = pydantic.Field()
+    """
+    Information about the generated sandbox
+    """
+
+    model_version: RegisteredPromptModelVersion = pydantic.Field()
+    """
+    Information about the generated model version
+    """
+
+    prompt_version_id: str = pydantic.Field()
+    """
+    The ID of the generated prompt version
+    """
+
+    deployment: RegisteredPromptDeployment = pydantic.Field()
+    """
+    Information about the generated deployment
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -39,4 +58,5 @@ class RegisterPromptResponse(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/registered_prompt_deployment.py

@@ -12,9 +12,20 @@ except ImportError:
 
 
 class RegisteredPromptDeployment(pydantic.BaseModel):
-    id: str = pydantic.Field(description="The ID of the generated deployment.")
-    name: str = pydantic.Field(description="A uniquely-identifying name for generated deployment.")
-    label: str = pydantic.Field(description="A human-friendly label for the generated deployment.")
+    id: str = pydantic.Field()
+    """
+    The ID of the generated deployment.
+    """
+
+    name: str = pydantic.Field()
+    """
+    A uniquely-identifying name for generated deployment.
+    """
+
+    label: str = pydantic.Field()
+    """
+    A human-friendly label for the generated deployment.
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -27,4 +38,5 @@ class RegisteredPromptDeployment(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/registered_prompt_input_variable_request.py

@@ -14,8 +14,8 @@ except ImportError:
 
 class RegisteredPromptInputVariableRequest(pydantic.BaseModel):
     key: str
-    id: typing.Optional[str]
-    type: typing.Optional[VellumVariableType]
+    id: typing.Optional[str] = None
+    type: typing.Optional[VellumVariableType] = 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 RegisteredPromptInputVariableRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/registered_prompt_model_version.py

@@ -12,8 +12,15 @@ except ImportError:
 
 
 class RegisteredPromptModelVersion(pydantic.BaseModel):
-    id: str = pydantic.Field(description="The ID of the generated model version.")
-    label: str = pydantic.Field(description="A human-friendly label for the generated model version.")
+    id: str = pydantic.Field()
+    """
+    The ID of the generated model version.
+    """
+
+    label: str = pydantic.Field()
+    """
+    A human-friendly label for the generated model version.
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -26,4 +33,5 @@ class RegisteredPromptModelVersion(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/registered_prompt_sandbox.py

@@ -12,8 +12,15 @@ except ImportError:
 
 
 class RegisteredPromptSandbox(pydantic.BaseModel):
-    id: str = pydantic.Field(description="The ID of the generated sandbox.")
-    label: str = pydantic.Field(description="A human-friendly label for the generated sandbox.")
+    id: str = pydantic.Field()
+    """
+    The ID of the generated sandbox.
+    """
+
+    label: str = pydantic.Field()
+    """
+    A human-friendly label for the generated sandbox.
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -26,4 +33,5 @@ class RegisteredPromptSandbox(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/registered_prompt_sandbox_snapshot.py

@@ -12,7 +12,10 @@ except ImportError:
 
 
 class RegisteredPromptSandboxSnapshot(pydantic.BaseModel):
-    id: str = pydantic.Field(description="The ID of the generated sandbox snapshot.")
+    id: str = pydantic.Field()
+    """
+    The ID of the generated sandbox snapshot.
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -25,4 +28,5 @@ class RegisteredPromptSandboxSnapshot(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/rejected_enum.py

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

vellum/types/rejected_execute_prompt_event.py

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

vellum/types/rejected_execute_prompt_response.py

@@ -18,11 +18,17 @@ class RejectedExecutePromptResponse(pydantic.BaseModel):
     The unsuccessful response from the model containing an error of what went wrong.
     """
 
-    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.
+    """
+
     error: VellumError
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -36,4 +42,5 @@ class RejectedExecutePromptResponse(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/rejected_execute_workflow_workflow_result_event.py

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

vellum/types/rejected_function_call.py

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

vellum/types/rejected_prompt_execution_meta.py

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

vellum/types/rejected_workflow_node_result_event.py

@@ -21,9 +21,9 @@ class RejectedWorkflowNodeResultEvent(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]
+    ts: typing.Optional[dt.datetime] = None
+    data: typing.Optional[WorkflowNodeResultData] = None
+    source_execution_id: typing.Optional[str] = None
     error: WorkflowEventError
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -37,4 +37,5 @@ class RejectedWorkflowNodeResultEvent(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/sandbox_scenario.py

@@ -13,9 +13,16 @@ except ImportError:
 
 
 class SandboxScenario(pydantic.BaseModel):
-    label: typing.Optional[str]
-    inputs: typing.List[ScenarioInput] = pydantic.Field(description="The inputs for the scenario")
-    id: str = pydantic.Field(description="The id of the scenario")
+    label: typing.Optional[str] = None
+    inputs: typing.List[ScenarioInput] = pydantic.Field()
+    """
+    The inputs for the scenario
+    """
+
+    id: str = pydantic.Field()
+    """
+    The id of the scenario
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -28,4 +35,5 @@ class SandboxScenario(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/scenario_input.py

@@ -15,9 +15,9 @@ except ImportError:
 
 class ScenarioInput(pydantic.BaseModel):
     key: str
-    type: typing.Optional[ScenarioInputTypeEnum]
-    value: typing.Optional[str]
-    chat_history: typing.Optional[typing.List[ChatMessage]]
+    type: typing.Optional[ScenarioInputTypeEnum] = None
+    value: typing.Optional[str] = None
+    chat_history: 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}
@@ -30,4 +30,5 @@ class ScenarioInput(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/scenario_input_request.py

@@ -15,9 +15,9 @@ except ImportError:
 
 class ScenarioInputRequest(pydantic.BaseModel):
     key: str
-    type: typing.Optional[ScenarioInputTypeEnum]
-    value: typing.Optional[str]
-    chat_history: typing.Optional[typing.List[ChatMessageRequest]]
+    type: typing.Optional[ScenarioInputTypeEnum] = None
+    value: typing.Optional[str] = None
+    chat_history: 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}
@@ -30,4 +30,5 @@ class ScenarioInputRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/search_error_response.py

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

vellum/types/search_filters_request.py

@@ -13,12 +13,15 @@ except ImportError:
 
 
 class SearchFiltersRequest(pydantic.BaseModel):
-    external_ids: typing.Optional[typing.List[str]] = pydantic.Field(
-        description="The document external IDs to filter by"
-    )
-    metadata: typing.Optional[MetadataFilterConfigRequest] = pydantic.Field(
-        description="The metadata filters to apply to the search"
-    )
+    external_ids: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    """
+    The document external IDs to filter by
+    """
+
+    metadata: typing.Optional[MetadataFilterConfigRequest] = pydantic.Field(default=None)
+    """
+    The metadata filters to apply to the search
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -31,4 +34,5 @@ class SearchFiltersRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/search_node_result.py

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

vellum/types/search_node_result_data.py

@@ -14,11 +14,13 @@ except ImportError:
 
 class SearchNodeResultData(pydantic.BaseModel):
     results_output_id: str
-    results: typing.List[SearchResult] = pydantic.Field(
-        description="The results of the search. Each result represents a chunk that matches the search query."
-    )
+    results: typing.List[SearchResult] = pydantic.Field()
+    """
+    The results of the search. Each result represents a chunk that matches the search query.
+    """
+
     text_output_id: str
-    text: typing.Optional[str]
+    text: 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 +33,5 @@ class SearchNodeResultData(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/search_request_options_request.py

@@ -15,14 +15,25 @@ except ImportError:
 
 
 class SearchRequestOptionsRequest(pydantic.BaseModel):
-    limit: typing.Optional[int] = pydantic.Field(description="The maximum number of results to return.")
-    weights: typing.Optional[SearchWeightsRequest] = pydantic.Field(
-        description="The weights to use for the search. Must add up to 1.0."
-    )
-    result_merging: typing.Optional[SearchResultMergingRequest] = pydantic.Field(
-        description="The configuration for merging results."
-    )
-    filters: typing.Optional[SearchFiltersRequest] = pydantic.Field(description="The filters to apply to the search.")
+    limit: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    The maximum number of results to return.
+    """
+
+    weights: typing.Optional[SearchWeightsRequest] = pydantic.Field(default=None)
+    """
+    The weights to use for the search. Must add up to 1.0.
+    """
+
+    result_merging: typing.Optional[SearchResultMergingRequest] = pydantic.Field(default=None)
+    """
+    The configuration for merging results.
+    """
+
+    filters: typing.Optional[SearchFiltersRequest] = pydantic.Field(default=None)
+    """
+    The filters to apply to the search.
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -35,4 +46,5 @@ class SearchRequestOptionsRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/search_response.py

@@ -13,9 +13,10 @@ except ImportError:
 
 
 class SearchResponse(pydantic.BaseModel):
-    results: typing.List[SearchResult] = pydantic.Field(
-        description="The results of the search. Each result represents a chunk that matches the search query."
-    )
+    results: typing.List[SearchResult] = pydantic.Field()
+    """
+    The results of the search. Each result represents a chunk that matches the search query.
+    """
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -28,4 +29,5 @@ class SearchResponse(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}