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

vellum/types/workflow_output.py

@@ -4,8 +4,6 @@ from __future__ import annotations
 
 import typing
 
-import typing_extensions
-
 from .workflow_output_array import WorkflowOutputArray
 from .workflow_output_chat_history import WorkflowOutputChatHistory
 from .workflow_output_error import WorkflowOutputError
@@ -18,84 +16,93 @@ from .workflow_output_string import WorkflowOutputString
 
 
 class WorkflowOutput_String(WorkflowOutputString):
-    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 WorkflowOutput_Number(WorkflowOutputNumber):
-    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 WorkflowOutput_Json(WorkflowOutputJson):
-    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 WorkflowOutput_ChatHistory(WorkflowOutputChatHistory):
-    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 WorkflowOutput_SearchResults(WorkflowOutputSearchResults):
-    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 WorkflowOutput_Array(WorkflowOutputArray):
-    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 WorkflowOutput_Error(WorkflowOutputError):
-    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 WorkflowOutput_FunctionCall(WorkflowOutputFunctionCall):
-    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 WorkflowOutput_Image(WorkflowOutputImage):
-    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
 
 
 WorkflowOutput = typing.Union[

vellum/types/workflow_output_array.py

@@ -18,8 +18,12 @@ class WorkflowOutputArray(pydantic.BaseModel):
     """
 
     id: str
-    name: str = pydantic.Field(description="The output's name, as defined in the workflow")
-    value: typing.Optional[typing.List[ArrayVariableValueItem]]
+    name: str = pydantic.Field()
+    """
+    The output's name, as defined in the workflow
+    """
+
+    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}
@@ -32,4 +36,5 @@ class WorkflowOutputArray(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_output_chat_history.py

@@ -18,8 +18,12 @@ class WorkflowOutputChatHistory(pydantic.BaseModel):
     """
 
     id: str
-    name: str = pydantic.Field(description="The output's name, as defined in the workflow")
-    value: typing.Optional[typing.List[ChatMessage]]
+    name: str = pydantic.Field()
+    """
+    The output's name, as defined in the workflow
+    """
+
+    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}
@@ -32,4 +36,5 @@ class WorkflowOutputChatHistory(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_output_error.py

@@ -18,8 +18,12 @@ class WorkflowOutputError(pydantic.BaseModel):
     """
 
     id: str
-    name: str = pydantic.Field(description="The output's name, as defined in the workflow")
-    value: typing.Optional[VellumError]
+    name: str = pydantic.Field()
+    """
+    The output's name, as defined in the workflow
+    """
+
+    value: typing.Optional[VellumError] = 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 WorkflowOutputError(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_output_function_call.py

@@ -18,8 +18,12 @@ class WorkflowOutputFunctionCall(pydantic.BaseModel):
     """
 
     id: str
-    name: str = pydantic.Field(description="The output's name, as defined in the workflow")
-    value: typing.Optional[FunctionCall]
+    name: str = pydantic.Field()
+    """
+    The output's name, as defined in the workflow
+    """
+
+    value: typing.Optional[FunctionCall] = 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 WorkflowOutputFunctionCall(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_output_image.py

@@ -18,8 +18,12 @@ class WorkflowOutputImage(pydantic.BaseModel):
     """
 
     id: str
-    name: str = pydantic.Field(description="The output's name, as defined in the workflow")
-    value: typing.Optional[VellumImage]
+    name: str = pydantic.Field()
+    """
+    The output's name, as defined in the workflow
+    """
+
+    value: typing.Optional[VellumImage] = 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 WorkflowOutputImage(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_output_json.py

@@ -17,8 +17,12 @@ class WorkflowOutputJson(pydantic.BaseModel):
     """
 
     id: str
-    name: str = pydantic.Field(description="The output's name, as defined in the workflow")
-    value: typing.Optional[typing.Dict[str, typing.Any]]
+    name: str = pydantic.Field()
+    """
+    The output's name, as defined in the workflow
+    """
+
+    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}
@@ -31,4 +35,5 @@ class WorkflowOutputJson(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_output_number.py

@@ -17,8 +17,12 @@ class WorkflowOutputNumber(pydantic.BaseModel):
     """
 
     id: str
-    name: str = pydantic.Field(description="The output's name, as defined in the workflow")
-    value: typing.Optional[float]
+    name: str = pydantic.Field()
+    """
+    The output's name, as defined in the workflow
+    """
+
+    value: typing.Optional[float] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -31,4 +35,5 @@ class WorkflowOutputNumber(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_output_search_results.py

@@ -18,8 +18,12 @@ class WorkflowOutputSearchResults(pydantic.BaseModel):
     """
 
     id: str
-    name: str = pydantic.Field(description="The output's name, as defined in the workflow")
-    value: typing.Optional[typing.List[SearchResult]]
+    name: str = pydantic.Field()
+    """
+    The output's name, as defined in the workflow
+    """
+
+    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}
@@ -32,4 +36,5 @@ class WorkflowOutputSearchResults(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_output_string.py

@@ -17,8 +17,12 @@ class WorkflowOutputString(pydantic.BaseModel):
     """
 
     id: str
-    name: str = pydantic.Field(description="The output's name, as defined in the workflow")
-    value: typing.Optional[str]
+    name: str = pydantic.Field()
+    """
+    The output's name, as defined in the workflow
+    """
+
+    value: 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 +35,5 @@ class WorkflowOutputString(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_request_chat_history_input_request.py

@@ -17,7 +17,11 @@ class WorkflowRequestChatHistoryInputRequest(pydantic.BaseModel):
     The input for a chat history variable in a Workflow.
     """
 
-    name: str = pydantic.Field(description="The variable's name, as defined in the Workflow.")
+    name: str = pydantic.Field()
+    """
+    The variable's name, as defined in the Workflow.
+    """
+
     value: typing.List[ChatMessageRequest]
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -31,4 +35,5 @@ class WorkflowRequestChatHistoryInputRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_request_input_request.py

@@ -4,8 +4,6 @@ from __future__ import annotations
 
 import typing
 
-import typing_extensions
-
 from .workflow_request_chat_history_input_request import WorkflowRequestChatHistoryInputRequest
 from .workflow_request_json_input_request import WorkflowRequestJsonInputRequest
 from .workflow_request_number_input_request import WorkflowRequestNumberInputRequest
@@ -13,39 +11,43 @@ from .workflow_request_string_input_request import WorkflowRequestStringInputReq
 
 
 class WorkflowRequestInputRequest_String(WorkflowRequestStringInputRequest):
-    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 WorkflowRequestInputRequest_Json(WorkflowRequestJsonInputRequest):
-    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 WorkflowRequestInputRequest_ChatHistory(WorkflowRequestChatHistoryInputRequest):
-    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 WorkflowRequestInputRequest_Number(WorkflowRequestNumberInputRequest):
-    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
 
 
 WorkflowRequestInputRequest = typing.Union[

vellum/types/workflow_request_json_input_request.py

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

vellum/types/workflow_request_number_input_request.py

@@ -16,7 +16,11 @@ class WorkflowRequestNumberInputRequest(pydantic.BaseModel):
     The input for a number variable in a Workflow.
     """
 
-    name: str = pydantic.Field(description="The variable's name, as defined in the Workflow.")
+    name: str = pydantic.Field()
+    """
+    The variable's name, as defined in the Workflow.
+    """
+
     value: float
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -30,4 +34,5 @@ class WorkflowRequestNumberInputRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_request_string_input_request.py

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

vellum/types/workflow_result_event.py

@@ -20,10 +20,10 @@ class WorkflowResultEvent(pydantic.BaseModel):
     id: str
     state: WorkflowNodeResultEventState
     ts: dt.datetime
-    output: typing.Optional[WorkflowResultEventOutputData]
-    error: typing.Optional[WorkflowEventError]
-    outputs: typing.Optional[typing.List[WorkflowOutput]]
-    inputs: typing.Optional[typing.List[ExecutionVellumValue]]
+    output: typing.Optional[WorkflowResultEventOutputData] = None
+    error: typing.Optional[WorkflowEventError] = None
+    outputs: typing.Optional[typing.List[WorkflowOutput]] = None
+    inputs: typing.Optional[typing.List[ExecutionVellumValue]] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -36,4 +36,5 @@ class WorkflowResultEvent(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_result_event_output_data.py

@@ -4,8 +4,6 @@ from __future__ import annotations
 
 import typing
 
-import typing_extensions
-
 from .workflow_result_event_output_data_array import WorkflowResultEventOutputDataArray
 from .workflow_result_event_output_data_chat_history import WorkflowResultEventOutputDataChatHistory
 from .workflow_result_event_output_data_error import WorkflowResultEventOutputDataError
@@ -17,75 +15,83 @@ from .workflow_result_event_output_data_string import WorkflowResultEventOutputD
 
 
 class WorkflowResultEventOutputData_String(WorkflowResultEventOutputDataString):
-    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 WorkflowResultEventOutputData_Number(WorkflowResultEventOutputDataNumber):
-    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 WorkflowResultEventOutputData_Json(WorkflowResultEventOutputDataJson):
-    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 WorkflowResultEventOutputData_ChatHistory(WorkflowResultEventOutputDataChatHistory):
-    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 WorkflowResultEventOutputData_SearchResults(WorkflowResultEventOutputDataSearchResults):
-    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 WorkflowResultEventOutputData_Array(WorkflowResultEventOutputDataArray):
-    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 WorkflowResultEventOutputData_FunctionCall(WorkflowResultEventOutputDataFunctionCall):
-    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 WorkflowResultEventOutputData_Error(WorkflowResultEventOutputDataError):
-    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
 
 
 WorkflowResultEventOutputData = typing.Union[

vellum/types/workflow_result_event_output_data_array.py

@@ -18,14 +18,16 @@ class WorkflowResultEventOutputDataArray(pydantic.BaseModel):
     An Array output returned from a Workflow execution.
     """
 
-    id: typing.Optional[str]
+    id: typing.Optional[str] = None
     name: str
     state: WorkflowNodeResultEventState
     node_id: str
-    delta: typing.Optional[str] = pydantic.Field(
-        description="The newly output string value. Only relevant for string outputs with a state of STREAMING."
-    )
-    value: typing.Optional[typing.List[ArrayVariableValueItem]]
+    delta: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The newly output string value. Only relevant for string outputs with a state of STREAMING.
+    """
+
+    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}
@@ -38,4 +40,5 @@ class WorkflowResultEventOutputDataArray(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_result_event_output_data_chat_history.py

@@ -18,14 +18,16 @@ class WorkflowResultEventOutputDataChatHistory(pydantic.BaseModel):
     A Chat History output streamed from a Workflow execution.
     """
 
-    id: typing.Optional[str]
+    id: typing.Optional[str] = None
     name: str
     state: WorkflowNodeResultEventState
     node_id: str
-    delta: typing.Optional[str] = pydantic.Field(
-        description="The newly output string value. Only relevant for string outputs with a state of STREAMING."
-    )
-    value: typing.Optional[typing.List[ChatMessage]]
+    delta: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The newly output string value. Only relevant for string outputs with a state of STREAMING.
+    """
+
+    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}
@@ -38,4 +40,5 @@ class WorkflowResultEventOutputDataChatHistory(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_result_event_output_data_error.py

@@ -18,14 +18,16 @@ class WorkflowResultEventOutputDataError(pydantic.BaseModel):
     An Error output streamed from a Workflow execution.
     """
 
-    id: typing.Optional[str]
+    id: typing.Optional[str] = None
     name: str
     state: WorkflowNodeResultEventState
     node_id: str
-    delta: typing.Optional[str] = pydantic.Field(
-        description="The newly output string value. Only relevant for string outputs with a state of STREAMING."
-    )
-    value: typing.Optional[VellumError]
+    delta: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The newly output string value. Only relevant for string outputs with a state of STREAMING.
+    """
+
+    value: typing.Optional[VellumError] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -38,4 +40,5 @@ class WorkflowResultEventOutputDataError(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
+        extra = pydantic.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}