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

vellum/types/workflow_result_event_output_data_search_results.py

@@ -1,15 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-from .search_result import SearchResult
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
+import pydantic
+from .search_result import SearchResult
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class WorkflowResultEventOutputDataSearchResults(pydantic_v1.BaseModel):
+class WorkflowResultEventOutputDataSearchResults(UniversalBaseModel):
     """
     A Search Results output streamed from a Workflow execution.
     """
@@ -18,27 +17,19 @@ class WorkflowResultEventOutputDataSearchResults(pydantic_v1.BaseModel):
     name: str
     state: WorkflowNodeResultEventState
     node_id: str
-    delta: typing.Optional[str] = pydantic_v1.Field(default=None)
+    delta: typing.Optional[str] = pydantic.Field(default=None)
     """
     The newly output string value. Only relevant for string outputs with a state of STREAMING.
     """
 
+    type: typing.Literal["SEARCH_RESULTS"] = "SEARCH_RESULTS"
     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}
-        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)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/workflow_result_event_output_data_string.py

@@ -1,14 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class WorkflowResultEventOutputDataString(pydantic_v1.BaseModel):
+class WorkflowResultEventOutputDataString(UniversalBaseModel):
     """
     A string output streamed from a Workflow execution.
     """
@@ -17,30 +16,22 @@ class WorkflowResultEventOutputDataString(pydantic_v1.BaseModel):
     name: str
     state: WorkflowNodeResultEventState
     node_id: str
-    delta: typing.Optional[str] = pydantic_v1.Field(default=None)
+    delta: typing.Optional[str] = pydantic.Field(default=None)
     """
     The newly output string value, meant to be concatenated with all previous. Will be non-null for events of state STREAMING.
     """
 
-    value: typing.Optional[str] = pydantic_v1.Field(default=None)
+    type: typing.Literal["STRING"] = "STRING"
+    value: typing.Optional[str] = pydantic.Field(default=None)
     """
     The entire string value. Will be non-null for events of state 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)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/workflow_stream_event.py

@@ -1,66 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
-
-import datetime as dt
 import typing
+from .workflow_execution_workflow_result_event import WorkflowExecutionWorkflowResultEvent
+from .workflow_execution_node_result_event import WorkflowExecutionNodeResultEvent
 
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-from .workflow_node_result_event import WorkflowNodeResultEvent
-from .workflow_result_event import WorkflowResultEvent
-
-
-class WorkflowStreamEvent_Workflow(pydantic_v1.BaseModel):
-    execution_id: str
-    run_id: typing.Optional[str] = None
-    external_id: typing.Optional[str] = None
-    data: WorkflowResultEvent
-    type: typing.Literal["WORKFLOW"] = "WORKFLOW"
-
-    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 WorkflowStreamEvent_Node(pydantic_v1.BaseModel):
-    execution_id: str
-    run_id: typing.Optional[str] = None
-    external_id: typing.Optional[str] = None
-    data: WorkflowNodeResultEvent
-    type: typing.Literal["NODE"] = "NODE"
-
-    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}
-
-
-WorkflowStreamEvent = typing.Union[WorkflowStreamEvent_Workflow, WorkflowStreamEvent_Node]
+WorkflowStreamEvent = typing.Union[WorkflowExecutionWorkflowResultEvent, WorkflowExecutionNodeResultEvent]

vellum/version.py

@@ -1,4 +1,3 @@
-
 from importlib import metadata
 
 __version__ = metadata.version("vellum-ai")

{vellum_ai-0.7.11.dist-info → vellum_ai-0.8.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.7.11
+Version: 0.8.5
 Summary: 
 License: MIT
 Requires-Python: >=3.8,<4.0
@@ -24,6 +24,7 @@ Requires-Dist: cdktf (>=0.20.5,<0.21.0)
 Requires-Dist: httpx (>=0.21.2)
 Requires-Dist: publication (==0.0.3)
 Requires-Dist: pydantic (>=1.9.2)
+Requires-Dist: pydantic-core (>=2.18.2,<3.0.0)
 Requires-Dist: typing_extensions (>=4.0.0)
 Description-Content-Type: text/markdown
 
@@ -51,7 +52,7 @@ that Vellum supports, check out our [API Reference](https://docs.vellum.ai/api-r
 
 ```python
 from vellum import (
-    PromptDeploymentInputRequest_String,
+    StringInputRequest,
 )
 from vellum.client import Vellum
 
@@ -64,7 +65,7 @@ def execute() -> str:
         prompt_deployment_name="<example-deployment-name>>",
         release_tag="LATEST",
         inputs=[
-            PromptDeploymentInputRequest_String(
+            StringInputRequest(
                 name="input_a",
                 type="STRING",
                 value="Hello, world!",
@@ -103,7 +104,7 @@ async def execute() -> str:
         prompt_deployment_name="<example-deployment-name>>",
         release_tag="LATEST",
         inputs=[
-            vellum.PromptDeploymentInputRequest_String(
+            vellum.StringInputRequest(
                 name="input_a",
                 value="Hello, world!",
             )