vellum-ai 0.9.15__py3-none-any.whl → 0.9.16rc2__py3-none-any.whl

vellum/client/types/workflow_result_event_output_data.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from .workflow_result_event_output_data_string import WorkflowResultEventOutputDataString
+from .workflow_result_event_output_data_number import WorkflowResultEventOutputDataNumber
+from .workflow_result_event_output_data_json import WorkflowResultEventOutputDataJson
+from .workflow_result_event_output_data_chat_history import WorkflowResultEventOutputDataChatHistory
+from .workflow_result_event_output_data_search_results import WorkflowResultEventOutputDataSearchResults
+from .workflow_result_event_output_data_array import WorkflowResultEventOutputDataArray
+from .workflow_result_event_output_data_function_call import WorkflowResultEventOutputDataFunctionCall
+from .workflow_result_event_output_data_error import WorkflowResultEventOutputDataError
+
+WorkflowResultEventOutputData = typing.Union[
+    WorkflowResultEventOutputDataString,
+    WorkflowResultEventOutputDataNumber,
+    WorkflowResultEventOutputDataJson,
+    WorkflowResultEventOutputDataChatHistory,
+    WorkflowResultEventOutputDataSearchResults,
+    WorkflowResultEventOutputDataArray,
+    WorkflowResultEventOutputDataFunctionCall,
+    WorkflowResultEventOutputDataError,
+]

vellum/client/types/workflow_result_event_output_data_array.py

@@ -0,0 +1,41 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+from ..core.pydantic_utilities import UniversalBaseModel
+from .array_variable_value import ArrayVariableValue
+import typing
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
+import pydantic
+from .array_variable_value_item import ArrayVariableValueItem
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import update_forward_refs
+
+
+class WorkflowResultEventOutputDataArray(UniversalBaseModel):
+    """
+    An Array output returned from a Workflow execution.
+    """
+
+    id: typing.Optional[str] = None
+    name: str
+    state: WorkflowNodeResultEventState
+    node_id: str
+    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["ARRAY"] = "ARRAY"
+    value: typing.Optional[typing.List[ArrayVariableValueItem]] = 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.Extra.allow
+
+
+update_forward_refs(ArrayVariableValue, WorkflowResultEventOutputDataArray=WorkflowResultEventOutputDataArray)

vellum/client/types/workflow_result_event_output_data_chat_history.py

@@ -0,0 +1,35 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
+import pydantic
+from .chat_message import ChatMessage
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class WorkflowResultEventOutputDataChatHistory(UniversalBaseModel):
+    """
+    A Chat History output streamed from a Workflow execution.
+    """
+
+    id: typing.Optional[str] = None
+    name: str
+    state: WorkflowNodeResultEventState
+    node_id: str
+    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["CHAT_HISTORY"] = "CHAT_HISTORY"
+    value: typing.Optional[typing.List[ChatMessage]] = 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.Extra.allow

vellum/client/types/workflow_result_event_output_data_error.py

@@ -0,0 +1,35 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
+import pydantic
+from .vellum_error import VellumError
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class WorkflowResultEventOutputDataError(UniversalBaseModel):
+    """
+    An Error output streamed from a Workflow execution.
+    """
+
+    id: typing.Optional[str] = None
+    name: str
+    state: WorkflowNodeResultEventState
+    node_id: str
+    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["ERROR"] = "ERROR"
+    value: typing.Optional[VellumError] = 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.Extra.allow

vellum/client/types/workflow_result_event_output_data_function_call.py

@@ -0,0 +1,35 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
+import pydantic
+from .function_call import FunctionCall
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class WorkflowResultEventOutputDataFunctionCall(UniversalBaseModel):
+    """
+    A Function Call output returned from a Workflow execution.
+    """
+
+    id: typing.Optional[str] = None
+    name: str
+    state: WorkflowNodeResultEventState
+    node_id: str
+    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["FUNCTION_CALL"] = "FUNCTION_CALL"
+    value: typing.Optional[FunctionCall] = 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.Extra.allow

vellum/client/types/workflow_result_event_output_data_json.py

@@ -0,0 +1,34 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class WorkflowResultEventOutputDataJson(UniversalBaseModel):
+    """
+    A JSON output streamed from a Workflow execution.
+    """
+
+    id: typing.Optional[str] = None
+    name: str
+    state: WorkflowNodeResultEventState
+    node_id: str
+    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["JSON"] = "JSON"
+    value: typing.Optional[typing.Any] = 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.Extra.allow

vellum/client/types/workflow_result_event_output_data_number.py

@@ -0,0 +1,34 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class WorkflowResultEventOutputDataNumber(UniversalBaseModel):
+    """
+    A number output streamed from a Workflow execution.
+    """
+
+    id: typing.Optional[str] = None
+    name: str
+    state: WorkflowNodeResultEventState
+    node_id: str
+    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["NUMBER"] = "NUMBER"
+    value: typing.Optional[float] = 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.Extra.allow

vellum/client/types/workflow_result_event_output_data_search_results.py

@@ -0,0 +1,35 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+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(UniversalBaseModel):
+    """
+    A Search Results output streamed from a Workflow execution.
+    """
+
+    id: typing.Optional[str] = None
+    name: str
+    state: WorkflowNodeResultEventState
+    node_id: str
+    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
+
+    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.Extra.allow

vellum/client/types/workflow_result_event_output_data_string.py

@@ -0,0 +1,37 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class WorkflowResultEventOutputDataString(UniversalBaseModel):
+    """
+    A string output streamed from a Workflow execution.
+    """
+
+    id: typing.Optional[str] = None
+    name: str
+    state: WorkflowNodeResultEventState
+    node_id: str
+    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.
+    """
+
+    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.
+    """
+
+    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.Extra.allow

vellum/client/types/workflow_stream_event.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from .workflow_execution_workflow_result_event import WorkflowExecutionWorkflowResultEvent
+from .workflow_execution_node_result_event import WorkflowExecutionNodeResultEvent
+
+WorkflowStreamEvent = typing.Union[WorkflowExecutionWorkflowResultEvent, WorkflowExecutionNodeResultEvent]

vellum/client/types/workspace_secret_read.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import datetime as dt
+from .secret_type_enum import SecretTypeEnum
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class WorkspaceSecretRead(UniversalBaseModel):
+    id: str
+    modified: dt.datetime
+    name: str
+    label: str
+    secret_type: SecretTypeEnum
+
+    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.Extra.allow

vellum/core/__init__.py

@@ -1,47 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from .api_error import ApiError
-from .client_wrapper import AsyncClientWrapper, BaseClientWrapper, SyncClientWrapper
-from .datetime_utils import serialize_datetime
-from .file import File, convert_file_dict_to_httpx_tuples, with_content_type
-from .http_client import AsyncHttpClient, HttpClient
-from .jsonable_encoder import jsonable_encoder
-from .pydantic_utilities import (
-    IS_PYDANTIC_V2,
-    UniversalBaseModel,
-    UniversalRootModel,
-    parse_obj_as,
-    universal_field_validator,
-    universal_root_validator,
-    update_forward_refs,
-)
-from .query_encoder import encode_query
-from .remove_none_from_dict import remove_none_from_dict
-from .request_options import RequestOptions
-from .serialization import FieldMetadata, convert_and_respect_annotation_metadata
-
-__all__ = [
-    "ApiError",
-    "AsyncClientWrapper",
-    "AsyncHttpClient",
-    "BaseClientWrapper",
-    "FieldMetadata",
-    "File",
-    "HttpClient",
-    "IS_PYDANTIC_V2",
-    "RequestOptions",
-    "SyncClientWrapper",
-    "UniversalBaseModel",
-    "UniversalRootModel",
-    "convert_and_respect_annotation_metadata",
-    "convert_file_dict_to_httpx_tuples",
-    "encode_query",
-    "jsonable_encoder",
-    "parse_obj_as",
-    "remove_none_from_dict",
-    "serialize_datetime",
-    "universal_field_validator",
-    "universal_root_validator",
-    "update_forward_refs",
-    "with_content_type",
-]
+from vellum.client.core import *

vellum/core/api_error.py

@@ -1,15 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-import typing
-
-
-class ApiError(Exception):
-    status_code: typing.Optional[int]
-    body: typing.Any
-
-    def __init__(self, *, status_code: typing.Optional[int] = None, body: typing.Any = None):
-        self.status_code = status_code
-        self.body = body
-
-    def __str__(self) -> str:
-        return f"status_code: {self.status_code}, body: {self.body}"
+from vellum.client.core.api_error import *

vellum/core/client_wrapper.py

@@ -1,59 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from ..environment import VellumEnvironment
-import typing
-import httpx
-from .http_client import HttpClient
-from .http_client import AsyncHttpClient
-
-
-class BaseClientWrapper:
-    def __init__(self, *, api_key: str, environment: VellumEnvironment, timeout: typing.Optional[float] = None):
-        self.api_key = api_key
-        self._environment = environment
-        self._timeout = timeout
-
-    def get_headers(self) -> typing.Dict[str, str]:
-        headers: typing.Dict[str, str] = {
-            "X-Fern-Language": "Python",
-            "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "0.9.15",
-        }
-        headers["X_API_KEY"] = self.api_key
-        return headers
-
-    def get_environment(self) -> VellumEnvironment:
-        return self._environment
-
-    def get_timeout(self) -> typing.Optional[float]:
-        return self._timeout
-
-
-class SyncClientWrapper(BaseClientWrapper):
-    def __init__(
-        self,
-        *,
-        api_key: str,
-        environment: VellumEnvironment,
-        timeout: typing.Optional[float] = None,
-        httpx_client: httpx.Client,
-    ):
-        super().__init__(api_key=api_key, environment=environment, timeout=timeout)
-        self.httpx_client = HttpClient(
-            httpx_client=httpx_client, base_headers=self.get_headers, base_timeout=self.get_timeout
-        )
-
-
-class AsyncClientWrapper(BaseClientWrapper):
-    def __init__(
-        self,
-        *,
-        api_key: str,
-        environment: VellumEnvironment,
-        timeout: typing.Optional[float] = None,
-        httpx_client: httpx.AsyncClient,
-    ):
-        super().__init__(api_key=api_key, environment=environment, timeout=timeout)
-        self.httpx_client = AsyncHttpClient(
-            httpx_client=httpx_client, base_headers=self.get_headers, base_timeout=self.get_timeout
-        )
+from vellum.client.core.client_wrapper import *

vellum/core/datetime_utils.py

@@ -1,28 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-import datetime as dt
-
-
-def serialize_datetime(v: dt.datetime) -> str:
-    """
-    Serialize a datetime including timezone info.
-
-    Uses the timezone info provided if present, otherwise uses the current runtime's timezone info.
-
-    UTC datetimes end in "Z" while all other timezones are represented as offset from UTC, e.g. +05:00.
-    """
-
-    def _serialize_zoned_datetime(v: dt.datetime) -> str:
-        if v.tzinfo is not None and v.tzinfo.tzname(None) == dt.timezone.utc.tzname(None):
-            # UTC is a special case where we use "Z" at the end instead of "+00:00"
-            return v.isoformat().replace("+00:00", "Z")
-        else:
-            # Delegate to the typical +/- offset format
-            return v.isoformat()
-
-    if v.tzinfo is not None:
-        return _serialize_zoned_datetime(v)
-    else:
-        local_tz = dt.datetime.now().astimezone().tzinfo
-        localized_dt = v.replace(tzinfo=local_tz)
-        return _serialize_zoned_datetime(localized_dt)
+from vellum.client.core.datetime_utils import *

vellum/core/file.py

@@ -1,62 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from typing import IO, Dict, List, Mapping, Optional, Tuple, Union, cast
-
-# File typing inspired by the flexibility of types within the httpx library
-# https://github.com/encode/httpx/blob/master/httpx/_types.py
-FileContent = Union[IO[bytes], bytes, str]
-File = Union[
-    # file (or bytes)
-    FileContent,
-    # (filename, file (or bytes))
-    Tuple[Optional[str], FileContent],
-    # (filename, file (or bytes), content_type)
-    Tuple[Optional[str], FileContent, Optional[str]],
-    # (filename, file (or bytes), content_type, headers)
-    Tuple[
-        Optional[str],
-        FileContent,
-        Optional[str],
-        Mapping[str, str],
-    ],
-]
-
-
-def convert_file_dict_to_httpx_tuples(
-    d: Dict[str, Union[File, List[File]]],
-) -> List[Tuple[str, File]]:
-    """
-    The format we use is a list of tuples, where the first element is the
-    name of the file and the second is the file object. Typically HTTPX wants
-    a dict, but to be able to send lists of files, you have to use the list
-    approach (which also works for non-lists)
-    https://github.com/encode/httpx/pull/1032
-    """
-
-    httpx_tuples = []
-    for key, file_like in d.items():
-        if isinstance(file_like, list):
-            for file_like_item in file_like:
-                httpx_tuples.append((key, file_like_item))
-        else:
-            httpx_tuples.append((key, file_like))
-    return httpx_tuples
-
-
-def with_content_type(*, file: File, content_type: str) -> File:
-    """ """
-    if isinstance(file, tuple):
-        if len(file) == 2:
-            filename, content = cast(Tuple[Optional[str], FileContent], file)  # type: ignore
-            return (filename, content, content_type)
-        elif len(file) == 3:
-            filename, content, _ = cast(Tuple[Optional[str], FileContent, Optional[str]], file)  # type: ignore
-            return (filename, content, content_type)
-        elif len(file) == 4:
-            filename, content, _, headers = cast(  # type: ignore
-                Tuple[Optional[str], FileContent, Optional[str], Mapping[str, str]], file
-            )
-            return (filename, content, content_type, headers)
-        else:
-            raise ValueError(f"Unexpected tuple length: {len(file)}")
-    return (None, file, content_type)
+from vellum.client.core.file import *