vellum-ai 0.7.11__py3-none-any.whl → 0.8.5__py3-none-any.whl

vellum/types/workflow_node_result_data.py

@@ -1,286 +1,28 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
-
-import datetime as dt
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-from .api_node_result_data import ApiNodeResultData
-from .code_execution_node_result_data import CodeExecutionNodeResultData
-from .conditional_node_result_data import ConditionalNodeResultData
-from .map_node_result_data import MapNodeResultData
-from .merge_node_result_data import MergeNodeResultData
-from .prompt_node_result_data import PromptNodeResultData
-from .search_node_result_data import SearchNodeResultData
-from .subworkflow_node_result_data import SubworkflowNodeResultData
-from .templating_node_result_data import TemplatingNodeResultData
-from .terminal_node_result_data import TerminalNodeResultData
-
-
-class WorkflowNodeResultData_Prompt(pydantic_v1.BaseModel):
-    data: PromptNodeResultData
-    type: typing.Literal["PROMPT"] = "PROMPT"
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
-
-class WorkflowNodeResultData_Search(pydantic_v1.BaseModel):
-    data: SearchNodeResultData
-    type: typing.Literal["SEARCH"] = "SEARCH"
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
-
-class WorkflowNodeResultData_Templating(pydantic_v1.BaseModel):
-    data: TemplatingNodeResultData
-    type: typing.Literal["TEMPLATING"] = "TEMPLATING"
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
-
-class WorkflowNodeResultData_CodeExecution(pydantic_v1.BaseModel):
-    data: CodeExecutionNodeResultData
-    type: typing.Literal["CODE_EXECUTION"] = "CODE_EXECUTION"
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
-
-class WorkflowNodeResultData_Conditional(pydantic_v1.BaseModel):
-    data: ConditionalNodeResultData
-    type: typing.Literal["CONDITIONAL"] = "CONDITIONAL"
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
-
-class WorkflowNodeResultData_Api(pydantic_v1.BaseModel):
-    data: ApiNodeResultData
-    type: typing.Literal["API"] = "API"
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
-
-class WorkflowNodeResultData_Terminal(pydantic_v1.BaseModel):
-    data: TerminalNodeResultData
-    type: typing.Literal["TERMINAL"] = "TERMINAL"
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
-
-class WorkflowNodeResultData_Merge(pydantic_v1.BaseModel):
-    data: MergeNodeResultData
-    type: typing.Literal["MERGE"] = "MERGE"
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
-
-class WorkflowNodeResultData_Subworkflow(pydantic_v1.BaseModel):
-    data: typing.Optional[SubworkflowNodeResultData] = None
-    type: typing.Literal["SUBWORKFLOW"] = "SUBWORKFLOW"
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
-
-class WorkflowNodeResultData_Metric(pydantic_v1.BaseModel):
-    type: typing.Literal["METRIC"] = "METRIC"
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
-
-class WorkflowNodeResultData_Map(pydantic_v1.BaseModel):
-    data: typing.Optional[MapNodeResultData] = None
-    type: typing.Literal["MAP"] = "MAP"
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
+from .prompt_node_result import PromptNodeResult
+from .search_node_result import SearchNodeResult
+from .templating_node_result import TemplatingNodeResult
+from .code_execution_node_result import CodeExecutionNodeResult
+from .conditional_node_result import ConditionalNodeResult
+from .api_node_result import ApiNodeResult
+from .terminal_node_result import TerminalNodeResult
+from .merge_node_result import MergeNodeResult
+from .subworkflow_node_result import SubworkflowNodeResult
+from .metric_node_result import MetricNodeResult
+from .map_node_result import MapNodeResult
 
 WorkflowNodeResultData = typing.Union[
-    WorkflowNodeResultData_Prompt,
-    WorkflowNodeResultData_Search,
-    WorkflowNodeResultData_Templating,
-    WorkflowNodeResultData_CodeExecution,
-    WorkflowNodeResultData_Conditional,
-    WorkflowNodeResultData_Api,
-    WorkflowNodeResultData_Terminal,
-    WorkflowNodeResultData_Merge,
-    WorkflowNodeResultData_Subworkflow,
-    WorkflowNodeResultData_Metric,
-    WorkflowNodeResultData_Map,
+    PromptNodeResult,
+    SearchNodeResult,
+    TemplatingNodeResult,
+    CodeExecutionNodeResult,
+    ConditionalNodeResult,
+    ApiNodeResult,
+    TerminalNodeResult,
+    MergeNodeResult,
+    SubworkflowNodeResult,
+    MetricNodeResult,
+    MapNodeResult,
 ]

vellum/types/workflow_node_result_event.py

@@ -1,139 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
-
-import datetime as dt
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-from .node_input_variable_compiled_value import NodeInputVariableCompiledValue
-from .node_output_compiled_value import NodeOutputCompiledValue
-from .workflow_event_error import WorkflowEventError
-from .workflow_node_result_data import WorkflowNodeResultData
-
-
-class WorkflowNodeResultEvent_Initiated(pydantic_v1.BaseModel):
-    id: str
-    node_id: str
-    node_result_id: str
-    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
-    state: typing.Literal["INITIATED"] = "INITIATED"
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
-
-class WorkflowNodeResultEvent_Streaming(pydantic_v1.BaseModel):
-    id: str
-    node_id: str
-    node_result_id: str
-    ts: typing.Optional[dt.datetime] = None
-    data: typing.Optional[WorkflowNodeResultData] = None
-    source_execution_id: typing.Optional[str] = None
-    output: typing.Optional[NodeOutputCompiledValue] = None
-    output_index: typing.Optional[int] = None
-    state: typing.Literal["STREAMING"] = "STREAMING"
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
-
-class WorkflowNodeResultEvent_Fulfilled(pydantic_v1.BaseModel):
-    id: str
-    node_id: str
-    node_result_id: str
-    ts: typing.Optional[dt.datetime] = None
-    data: typing.Optional[WorkflowNodeResultData] = None
-    source_execution_id: typing.Optional[str] = None
-    output_values: typing.List[NodeOutputCompiledValue]
-    mocked: typing.Optional[bool] = None
-    state: typing.Literal["FULFILLED"] = "FULFILLED"
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
-
-class WorkflowNodeResultEvent_Rejected(pydantic_v1.BaseModel):
-    id: str
-    node_id: str
-    node_result_id: str
-    ts: typing.Optional[dt.datetime] = None
-    data: typing.Optional[WorkflowNodeResultData] = None
-    source_execution_id: typing.Optional[str] = None
-    error: WorkflowEventError
-    state: typing.Literal["REJECTED"] = "REJECTED"
-
-    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_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
+from .initiated_workflow_node_result_event import InitiatedWorkflowNodeResultEvent
+from .streaming_workflow_node_result_event import StreamingWorkflowNodeResultEvent
+from .fulfilled_workflow_node_result_event import FulfilledWorkflowNodeResultEvent
+from .rejected_workflow_node_result_event import RejectedWorkflowNodeResultEvent
 
 WorkflowNodeResultEvent = typing.Union[
-    WorkflowNodeResultEvent_Initiated,
-    WorkflowNodeResultEvent_Streaming,
-    WorkflowNodeResultEvent_Fulfilled,
-    WorkflowNodeResultEvent_Rejected,
+    InitiatedWorkflowNodeResultEvent,
+    StreamingWorkflowNodeResultEvent,
+    FulfilledWorkflowNodeResultEvent,
+    RejectedWorkflowNodeResultEvent,
 ]