vellum-ai 0.3.23__py3-none-any.whl → 0.5.0__py3-none-any.whl

vellum/types/workflow_output_json.py

@@ -4,20 +4,16 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowOutputJson(pydantic.BaseModel):
+class WorkflowOutputJson(pydantic_v1.BaseModel):
     """
     A JSON output from a Workflow execution.
     """
 
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The output's name, as defined in the workflow
     """
@@ -35,5 +31,5 @@ class WorkflowOutputJson(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_output_number.py

@@ -4,20 +4,16 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowOutputNumber(pydantic.BaseModel):
+class WorkflowOutputNumber(pydantic_v1.BaseModel):
     """
     A number output from a Workflow execution.
     """
 
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The output's name, as defined in the workflow
     """
@@ -35,5 +31,5 @@ class WorkflowOutputNumber(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_output_search_results.py

@@ -4,21 +4,17 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .search_result import SearchResult
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowOutputSearchResults(pydantic.BaseModel):
+class WorkflowOutputSearchResults(pydantic_v1.BaseModel):
     """
     A search results output from a Workflow execution.
     """
 
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The output's name, as defined in the workflow
     """
@@ -36,5 +32,5 @@ class WorkflowOutputSearchResults(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_output_string.py

@@ -4,20 +4,16 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowOutputString(pydantic.BaseModel):
+class WorkflowOutputString(pydantic_v1.BaseModel):
     """
     A string output from a Workflow execution.
     """
 
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The output's name, as defined in the workflow
     """
@@ -35,5 +31,5 @@ class WorkflowOutputString(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_request_chat_history_input_request.py

@@ -4,20 +4,16 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .chat_message_request import ChatMessageRequest
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowRequestChatHistoryInputRequest(pydantic.BaseModel):
+class WorkflowRequestChatHistoryInputRequest(pydantic_v1.BaseModel):
     """
     The input for a chat history variable in a Workflow.
     """
 
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The variable's name, as defined in the Workflow.
     """
@@ -35,5 +31,5 @@ class WorkflowRequestChatHistoryInputRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_request_json_input_request.py

@@ -4,19 +4,15 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowRequestJsonInputRequest(pydantic.BaseModel):
+class WorkflowRequestJsonInputRequest(pydantic_v1.BaseModel):
     """
     The input for a JSON variable in a Workflow.
     """
 
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The variable's name, as defined in the Workflow.
     """
@@ -34,5 +30,5 @@ class WorkflowRequestJsonInputRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_request_number_input_request.py

@@ -4,19 +4,15 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowRequestNumberInputRequest(pydantic.BaseModel):
+class WorkflowRequestNumberInputRequest(pydantic_v1.BaseModel):
     """
     The input for a number variable in a Workflow.
     """
 
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The variable's name, as defined in the Workflow.
     """
@@ -34,5 +30,5 @@ class WorkflowRequestNumberInputRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_request_string_input_request.py

@@ -4,19 +4,15 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowRequestStringInputRequest(pydantic.BaseModel):
+class WorkflowRequestStringInputRequest(pydantic_v1.BaseModel):
     """
     The input for a string variable in a Workflow.
     """
 
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The variable's name, as defined in the Workflow.
     """
@@ -34,5 +30,5 @@ class WorkflowRequestStringInputRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_result_event.py

@@ -4,19 +4,15 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .execution_vellum_value import ExecutionVellumValue
 from .workflow_event_error import WorkflowEventError
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
 from .workflow_output import WorkflowOutput
 from .workflow_result_event_output_data import WorkflowResultEventOutputData
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowResultEvent(pydantic.BaseModel):
+class WorkflowResultEvent(pydantic_v1.BaseModel):
     id: str
     state: WorkflowNodeResultEventState
     ts: dt.datetime
@@ -36,5 +32,5 @@ class WorkflowResultEvent(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_result_event_output_data_array.py

@@ -4,16 +4,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .array_variable_value_item import ArrayVariableValueItem
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowResultEventOutputDataArray(pydantic.BaseModel):
+class WorkflowResultEventOutputDataArray(pydantic_v1.BaseModel):
     """
     An Array output returned from a Workflow execution.
     """
@@ -22,7 +18,7 @@ class WorkflowResultEventOutputDataArray(pydantic.BaseModel):
     name: str
     state: WorkflowNodeResultEventState
     node_id: str
-    delta: typing.Optional[str] = pydantic.Field(default=None)
+    delta: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The newly output string value. Only relevant for string outputs with a state of STREAMING.
     """
@@ -40,5 +36,5 @@ class WorkflowResultEventOutputDataArray(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_result_event_output_data_chat_history.py

@@ -4,16 +4,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .chat_message import ChatMessage
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowResultEventOutputDataChatHistory(pydantic.BaseModel):
+class WorkflowResultEventOutputDataChatHistory(pydantic_v1.BaseModel):
     """
     A Chat History output streamed from a Workflow execution.
     """
@@ -22,7 +18,7 @@ class WorkflowResultEventOutputDataChatHistory(pydantic.BaseModel):
     name: str
     state: WorkflowNodeResultEventState
     node_id: str
-    delta: typing.Optional[str] = pydantic.Field(default=None)
+    delta: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The newly output string value. Only relevant for string outputs with a state of STREAMING.
     """
@@ -40,5 +36,5 @@ class WorkflowResultEventOutputDataChatHistory(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_result_event_output_data_error.py

@@ -4,16 +4,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .vellum_error import VellumError
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowResultEventOutputDataError(pydantic.BaseModel):
+class WorkflowResultEventOutputDataError(pydantic_v1.BaseModel):
     """
     An Error output streamed from a Workflow execution.
     """
@@ -22,7 +18,7 @@ class WorkflowResultEventOutputDataError(pydantic.BaseModel):
     name: str
     state: WorkflowNodeResultEventState
     node_id: str
-    delta: typing.Optional[str] = pydantic.Field(default=None)
+    delta: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The newly output string value. Only relevant for string outputs with a state of STREAMING.
     """
@@ -40,5 +36,5 @@ class WorkflowResultEventOutputDataError(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_result_event_output_data_function_call.py

@@ -4,16 +4,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .function_call import FunctionCall
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowResultEventOutputDataFunctionCall(pydantic.BaseModel):
+class WorkflowResultEventOutputDataFunctionCall(pydantic_v1.BaseModel):
     """
     A Function Call output returned from a Workflow execution.
     """
@@ -22,7 +18,7 @@ class WorkflowResultEventOutputDataFunctionCall(pydantic.BaseModel):
     name: str
     state: WorkflowNodeResultEventState
     node_id: str
-    delta: typing.Optional[str] = pydantic.Field(default=None)
+    delta: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The newly output string value. Only relevant for string outputs with a state of STREAMING.
     """
@@ -40,5 +36,5 @@ class WorkflowResultEventOutputDataFunctionCall(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_result_event_output_data_json.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowResultEventOutputDataJson(pydantic.BaseModel):
+class WorkflowResultEventOutputDataJson(pydantic_v1.BaseModel):
     """
     A JSON output streamed from a Workflow execution.
     """
@@ -21,7 +17,7 @@ class WorkflowResultEventOutputDataJson(pydantic.BaseModel):
     name: str
     state: WorkflowNodeResultEventState
     node_id: str
-    delta: typing.Optional[str] = pydantic.Field(default=None)
+    delta: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The newly output string value. Only relevant for string outputs with a state of STREAMING.
     """
@@ -39,5 +35,5 @@ class WorkflowResultEventOutputDataJson(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_result_event_output_data_number.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowResultEventOutputDataNumber(pydantic.BaseModel):
+class WorkflowResultEventOutputDataNumber(pydantic_v1.BaseModel):
     """
     A number output streamed from a Workflow execution.
     """
@@ -21,7 +17,7 @@ class WorkflowResultEventOutputDataNumber(pydantic.BaseModel):
     name: str
     state: WorkflowNodeResultEventState
     node_id: str
-    delta: typing.Optional[str] = pydantic.Field(default=None)
+    delta: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The newly output string value. Only relevant for string outputs with a state of STREAMING.
     """
@@ -39,5 +35,5 @@ class WorkflowResultEventOutputDataNumber(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_result_event_output_data_search_results.py

@@ -4,16 +4,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .search_result import SearchResult
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowResultEventOutputDataSearchResults(pydantic.BaseModel):
+class WorkflowResultEventOutputDataSearchResults(pydantic_v1.BaseModel):
     """
     A Search Results output streamed from a Workflow execution.
     """
@@ -22,7 +18,7 @@ class WorkflowResultEventOutputDataSearchResults(pydantic.BaseModel):
     name: str
     state: WorkflowNodeResultEventState
     node_id: str
-    delta: typing.Optional[str] = pydantic.Field(default=None)
+    delta: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The newly output string value. Only relevant for string outputs with a state of STREAMING.
     """
@@ -40,5 +36,5 @@ class WorkflowResultEventOutputDataSearchResults(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/workflow_result_event_output_data_string.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowResultEventOutputDataString(pydantic.BaseModel):
+class WorkflowResultEventOutputDataString(pydantic_v1.BaseModel):
     """
     A string output streamed from a Workflow execution.
     """
@@ -21,12 +17,12 @@ class WorkflowResultEventOutputDataString(pydantic.BaseModel):
     name: str
     state: WorkflowNodeResultEventState
     node_id: str
-    delta: typing.Optional[str] = pydantic.Field(default=None)
+    delta: typing.Optional[str] = pydantic_v1.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.Field(default=None)
+    value: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The entire string value. Will be non-null for events of state FULFILLED.
     """
@@ -42,5 +38,5 @@ class WorkflowResultEventOutputDataString(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

{vellum_ai-0.3.23.dist-info → vellum_ai-0.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.3.23
+Version: 0.5.0
 Summary: 
 Requires-Python: >=3.8,<4.0
 Classifier: Programming Language :: Python :: 3
@@ -12,7 +12,6 @@ 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: requests_mock (>=1.11.0,<2.0.0)
 Requires-Dist: typing_extensions (>=4.0.0)
 Description-Content-Type: text/markdown