vellum-ai 0.5.1__tar.gz → 0.5.2__tar.gz

{vellum_ai-0.5.1 → vellum_ai-0.5.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.5.1
+Version: 0.5.2
 Summary: 
 Requires-Python: >=3.8,<4.0
 Classifier: Programming Language :: Python :: 3

{vellum_ai-0.5.1 → vellum_ai-0.5.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vellum-ai"
-version = "0.5.1"
+version = "0.5.2"
 description = ""
 readme = "README.md"
 authors = []

{vellum_ai-0.5.1 → vellum_ai-0.5.2}/src/vellum/__init__.py

@@ -71,6 +71,7 @@ from .types import (
     EnvironmentEnum,
     ErrorEnum,
     ErrorVariableValue,
+    ErrorVellumValue,
     ExecutePromptApiErrorResponse,
     ExecutePromptEvent,
     ExecutePromptEvent_Fulfilled,
@@ -121,6 +122,7 @@ from .types import (
     FunctionCallChatMessageContentValueRequest,
     FunctionCallEnum,
     FunctionCallVariableValue,
+    FunctionCallVellumValue,
     FunctionCall_Fulfilled,
     FunctionCall_Rejected,
     GenerateErrorResponse,
@@ -145,6 +147,7 @@ from .types import (
     JsonEnum,
     JsonInputRequest,
     JsonVariableValue,
+    JsonVellumValue,
     LogicalOperator,
     LogprobsEnum,
     MetadataFilterConfigRequest,
@@ -206,7 +209,7 @@ from .types import (
     NodeOutputCompiledArrayValue,
     NodeOutputCompiledChatHistoryValue,
     NodeOutputCompiledErrorValue,
-    NodeOutputCompiledFunctionValue,
+    NodeOutputCompiledFunctionCallValue,
     NodeOutputCompiledJsonValue,
     NodeOutputCompiledNumberValue,
     NodeOutputCompiledSearchResultsValue,
@@ -285,6 +288,7 @@ from .types import (
     StringEnum,
     StringInputRequest,
     StringVariableValue,
+    StringVellumValue,
     SubmitCompletionActualRequest,
     SubmitCompletionActualsErrorResponse,
     SubmitWorkflowExecutionActualRequest,
@@ -575,6 +579,7 @@ __all__ = [
     "EnvironmentEnum",
     "ErrorEnum",
     "ErrorVariableValue",
+    "ErrorVellumValue",
     "ExecutePromptApiErrorResponse",
     "ExecutePromptEvent",
     "ExecutePromptEvent_Fulfilled",
@@ -626,6 +631,7 @@ __all__ = [
     "FunctionCallChatMessageContentValueRequest",
     "FunctionCallEnum",
     "FunctionCallVariableValue",
+    "FunctionCallVellumValue",
     "FunctionCall_Fulfilled",
     "FunctionCall_Rejected",
     "GenerateErrorResponse",
@@ -651,6 +657,7 @@ __all__ = [
     "JsonEnum",
     "JsonInputRequest",
     "JsonVariableValue",
+    "JsonVellumValue",
     "LogicalOperator",
     "LogprobsEnum",
     "MetadataFilterConfigRequest",
@@ -712,7 +719,7 @@ __all__ = [
     "NodeOutputCompiledArrayValue",
     "NodeOutputCompiledChatHistoryValue",
     "NodeOutputCompiledErrorValue",
-    "NodeOutputCompiledFunctionValue",
+    "NodeOutputCompiledFunctionCallValue",
     "NodeOutputCompiledJsonValue",
     "NodeOutputCompiledNumberValue",
     "NodeOutputCompiledSearchResultsValue",
@@ -792,6 +799,7 @@ __all__ = [
     "StringEnum",
     "StringInputRequest",
     "StringVariableValue",
+    "StringVellumValue",
     "SubmitCompletionActualRequest",
     "SubmitCompletionActualsErrorResponse",
     "SubmitWorkflowExecutionActualRequest",

{vellum_ai-0.5.1 → vellum_ai-0.5.2}/src/vellum/core/client_wrapper.py

@@ -18,7 +18,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "0.5.1",
+            "X-Fern-SDK-Version": "0.5.2",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

{vellum_ai-0.5.1 → vellum_ai-0.5.2}/src/vellum/types/__init__.py

@@ -82,6 +82,7 @@ from .entity_status import EntityStatus
 from .environment_enum import EnvironmentEnum
 from .error_enum import ErrorEnum
 from .error_variable_value import ErrorVariableValue
+from .error_vellum_value import ErrorVellumValue
 from .execute_prompt_api_error_response import ExecutePromptApiErrorResponse
 from .execute_prompt_event import (
     ExecutePromptEvent,
@@ -140,6 +141,7 @@ from .function_call_chat_message_content_value import FunctionCallChatMessageCon
 from .function_call_chat_message_content_value_request import FunctionCallChatMessageContentValueRequest
 from .function_call_enum import FunctionCallEnum
 from .function_call_variable_value import FunctionCallVariableValue
+from .function_call_vellum_value import FunctionCallVellumValue
 from .generate_error_response import GenerateErrorResponse
 from .generate_options_request import GenerateOptionsRequest
 from .generate_request import GenerateRequest
@@ -162,6 +164,7 @@ from .initiated_workflow_node_result_event import InitiatedWorkflowNodeResultEve
 from .json_enum import JsonEnum
 from .json_input_request import JsonInputRequest
 from .json_variable_value import JsonVariableValue
+from .json_vellum_value import JsonVellumValue
 from .logical_operator import LogicalOperator
 from .logprobs_enum import LogprobsEnum
 from .metadata_filter_config_request import MetadataFilterConfigRequest
@@ -231,7 +234,7 @@ from .node_input_variable_compiled_value import (
 from .node_output_compiled_array_value import NodeOutputCompiledArrayValue
 from .node_output_compiled_chat_history_value import NodeOutputCompiledChatHistoryValue
 from .node_output_compiled_error_value import NodeOutputCompiledErrorValue
-from .node_output_compiled_function_value import NodeOutputCompiledFunctionValue
+from .node_output_compiled_function_call_value import NodeOutputCompiledFunctionCallValue
 from .node_output_compiled_json_value import NodeOutputCompiledJsonValue
 from .node_output_compiled_number_value import NodeOutputCompiledNumberValue
 from .node_output_compiled_search_results_value import NodeOutputCompiledSearchResultsValue
@@ -314,6 +317,7 @@ from .string_chat_message_content_request import StringChatMessageContentRequest
 from .string_enum import StringEnum
 from .string_input_request import StringInputRequest
 from .string_variable_value import StringVariableValue
+from .string_vellum_value import StringVellumValue
 from .submit_completion_actual_request import SubmitCompletionActualRequest
 from .submit_completion_actuals_error_response import SubmitCompletionActualsErrorResponse
 from .submit_workflow_execution_actual_request import (
@@ -614,6 +618,7 @@ __all__ = [
     "EnvironmentEnum",
     "ErrorEnum",
     "ErrorVariableValue",
+    "ErrorVellumValue",
     "ExecutePromptApiErrorResponse",
     "ExecutePromptEvent",
     "ExecutePromptEvent_Fulfilled",
@@ -664,6 +669,7 @@ __all__ = [
     "FunctionCallChatMessageContentValueRequest",
     "FunctionCallEnum",
     "FunctionCallVariableValue",
+    "FunctionCallVellumValue",
     "FunctionCall_Fulfilled",
     "FunctionCall_Rejected",
     "GenerateErrorResponse",
@@ -688,6 +694,7 @@ __all__ = [
     "JsonEnum",
     "JsonInputRequest",
     "JsonVariableValue",
+    "JsonVellumValue",
     "LogicalOperator",
     "LogprobsEnum",
     "MetadataFilterConfigRequest",
@@ -749,7 +756,7 @@ __all__ = [
     "NodeOutputCompiledArrayValue",
     "NodeOutputCompiledChatHistoryValue",
     "NodeOutputCompiledErrorValue",
-    "NodeOutputCompiledFunctionValue",
+    "NodeOutputCompiledFunctionCallValue",
     "NodeOutputCompiledJsonValue",
     "NodeOutputCompiledNumberValue",
     "NodeOutputCompiledSearchResultsValue",
@@ -828,6 +835,7 @@ __all__ = [
     "StringEnum",
     "StringInputRequest",
     "StringVariableValue",
+    "StringVellumValue",
     "SubmitCompletionActualRequest",
     "SubmitCompletionActualsErrorResponse",
     "SubmitWorkflowExecutionActualRequest",

vellum_ai-0.5.1/src/vellum/types/node_output_compiled_error_value.py → vellum_ai-0.5.2/src/vellum/types/error_vellum_value.py

@@ -8,8 +8,11 @@ from ..core.pydantic_utilities import pydantic_v1
 from .vellum_error import VellumError
 
 
-class NodeOutputCompiledErrorValue(pydantic_v1.BaseModel):
-    node_output_id: str
+class ErrorVellumValue(pydantic_v1.BaseModel):
+    """
+    A value representing an Error.
+    """
+
     value: typing.Optional[VellumError] = None
 
     def json(self, **kwargs: typing.Any) -> str:

{vellum_ai-0.5.1 → vellum_ai-0.5.2}/src/vellum/types/function_call_variable_value.py

@@ -9,7 +9,7 @@ from .function_call import FunctionCall
 
 
 class FunctionCallVariableValue(pydantic_v1.BaseModel):
-    value: FunctionCall
+    value: typing.Optional[FunctionCall] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

vellum_ai-0.5.1/src/vellum/types/node_output_compiled_function_value.py → vellum_ai-0.5.2/src/vellum/types/function_call_vellum_value.py

@@ -8,9 +8,12 @@ from ..core.pydantic_utilities import pydantic_v1
 from .function_call import FunctionCall
 
 
-class NodeOutputCompiledFunctionValue(pydantic_v1.BaseModel):
-    node_output_id: str
-    value: typing.Optional[FunctionCall] = None
+class FunctionCallVellumValue(pydantic_v1.BaseModel):
+    """
+    A value representing a Function Call.
+    """
+
+    value: FunctionCall
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

vellum_ai-0.5.1/src/vellum/types/node_output_compiled_json_value.py → vellum_ai-0.5.2/src/vellum/types/json_vellum_value.py

@@ -7,8 +7,11 @@ from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import pydantic_v1
 
 
-class NodeOutputCompiledJsonValue(pydantic_v1.BaseModel):
-    node_output_id: str
+class JsonVellumValue(pydantic_v1.BaseModel):
+    """
+    A value representing a JSON object.
+    """
+
     value: typing.Optional[typing.Dict[str, typing.Any]] = None
 
     def json(self, **kwargs: typing.Any) -> str:

{vellum_ai-0.5.1 → vellum_ai-0.5.2}/src/vellum/types/node_output_compiled_array_value.py

@@ -6,11 +6,17 @@ 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
 
 
 class NodeOutputCompiledArrayValue(pydantic_v1.BaseModel):
-    node_output_id: str
+    """
+    An output returned by a node that is of type ARRAY.
+    """
+
     value: typing.Optional[typing.List[ArrayVariableValueItem]] = None
+    node_output_id: str
+    state: typing.Optional[WorkflowNodeResultEventState] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

{vellum_ai-0.5.1 → vellum_ai-0.5.2}/src/vellum/types/node_output_compiled_chat_history_value.py

@@ -6,11 +6,17 @@ 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
 
 
 class NodeOutputCompiledChatHistoryValue(pydantic_v1.BaseModel):
-    node_output_id: str
+    """
+    An output returned by a node that is of type CHAT_HISTORY.
+    """
+
     value: typing.Optional[typing.List[ChatMessage]] = None
+    node_output_id: str
+    state: typing.Optional[WorkflowNodeResultEventState] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

vellum_ai-0.5.2/src/vellum/types/node_output_compiled_error_value.py

@@ -0,0 +1,33 @@
+# This file was auto-generated by Fern from our API Definition.
+
+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
+
+
+class NodeOutputCompiledErrorValue(pydantic_v1.BaseModel):
+    """
+    An output returned by a node that is of type ERROR.
+    """
+
+    value: typing.Optional[VellumError] = None
+    node_output_id: str
+    state: typing.Optional[WorkflowNodeResultEventState] = 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: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum_ai-0.5.2/src/vellum/types/node_output_compiled_function_call_value.py

@@ -0,0 +1,33 @@
+# This file was auto-generated by Fern from our API Definition.
+
+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
+
+
+class NodeOutputCompiledFunctionCallValue(pydantic_v1.BaseModel):
+    """
+    An output returned by a node that is of type FUNCTION_CALL.
+    """
+
+    value: typing.Optional[FunctionCall] = None
+    node_output_id: str
+    state: typing.Optional[WorkflowNodeResultEventState] = 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: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum_ai-0.5.2/src/vellum/types/node_output_compiled_json_value.py

@@ -0,0 +1,32 @@
+# This file was auto-generated by Fern from our API Definition.
+
+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
+
+
+class NodeOutputCompiledJsonValue(pydantic_v1.BaseModel):
+    """
+    An output returned by a node that is of type JSON.
+    """
+
+    value: typing.Optional[typing.Dict[str, typing.Any]] = None
+    node_output_id: str
+    state: typing.Optional[WorkflowNodeResultEventState] = 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: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

{vellum_ai-0.5.1 → vellum_ai-0.5.2}/src/vellum/types/node_output_compiled_number_value.py

@@ -5,11 +5,17 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import pydantic_v1
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
 
 class NodeOutputCompiledNumberValue(pydantic_v1.BaseModel):
-    node_output_id: str
+    """
+    An output returned by a node that is of type NUMBER.
+    """
+
     value: typing.Optional[float] = None
+    node_output_id: str
+    state: typing.Optional[WorkflowNodeResultEventState] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

{vellum_ai-0.5.1 → vellum_ai-0.5.2}/src/vellum/types/node_output_compiled_search_results_value.py

@@ -6,11 +6,17 @@ 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
 
 
 class NodeOutputCompiledSearchResultsValue(pydantic_v1.BaseModel):
-    node_output_id: str
+    """
+    An output returned by a node that is of type SEARCH_RESULTS.
+    """
+
     value: typing.Optional[typing.List[SearchResult]] = None
+    node_output_id: str
+    state: typing.Optional[WorkflowNodeResultEventState] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

vellum_ai-0.5.2/src/vellum/types/node_output_compiled_string_value.py

@@ -0,0 +1,32 @@
+# This file was auto-generated by Fern from our API Definition.
+
+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
+
+
+class NodeOutputCompiledStringValue(pydantic_v1.BaseModel):
+    """
+    An output returned by a node that is of type STRING.
+    """
+
+    value: typing.Optional[str] = None
+    node_output_id: str
+    state: typing.Optional[WorkflowNodeResultEventState] = 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: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

{vellum_ai-0.5.1 → vellum_ai-0.5.2}/src/vellum/types/node_output_compiled_value.py

@@ -7,7 +7,7 @@ import typing
 from .node_output_compiled_array_value import NodeOutputCompiledArrayValue
 from .node_output_compiled_chat_history_value import NodeOutputCompiledChatHistoryValue
 from .node_output_compiled_error_value import NodeOutputCompiledErrorValue
-from .node_output_compiled_function_value import NodeOutputCompiledFunctionValue
+from .node_output_compiled_function_call_value import NodeOutputCompiledFunctionCallValue
 from .node_output_compiled_json_value import NodeOutputCompiledJsonValue
 from .node_output_compiled_number_value import NodeOutputCompiledNumberValue
 from .node_output_compiled_search_results_value import NodeOutputCompiledSearchResultsValue
@@ -84,7 +84,7 @@ class NodeOutputCompiledValue_Array(NodeOutputCompiledArrayValue):
         populate_by_name = True
 
 
-class NodeOutputCompiledValue_FunctionCall(NodeOutputCompiledFunctionValue):
+class NodeOutputCompiledValue_FunctionCall(NodeOutputCompiledFunctionCallValue):
     type: typing.Literal["FUNCTION_CALL"] = "FUNCTION_CALL"
 
     class Config:

{vellum_ai-0.5.1 → vellum_ai-0.5.2}/src/vellum/types/prompt_output.py

@@ -4,13 +4,13 @@ from __future__ import annotations
 
 import typing
 
-from .error_variable_value import ErrorVariableValue
-from .function_call_variable_value import FunctionCallVariableValue
-from .json_variable_value import JsonVariableValue
-from .string_variable_value import StringVariableValue
+from .error_vellum_value import ErrorVellumValue
+from .function_call_vellum_value import FunctionCallVellumValue
+from .json_vellum_value import JsonVellumValue
+from .string_vellum_value import StringVellumValue
 
 
-class PromptOutput_String(StringVariableValue):
+class PromptOutput_String(StringVellumValue):
     type: typing.Literal["STRING"] = "STRING"
 
     class Config:
@@ -20,7 +20,7 @@ class PromptOutput_String(StringVariableValue):
         populate_by_name = True
 
 
-class PromptOutput_Json(JsonVariableValue):
+class PromptOutput_Json(JsonVellumValue):
     type: typing.Literal["JSON"] = "JSON"
 
     class Config:
@@ -30,7 +30,7 @@ class PromptOutput_Json(JsonVariableValue):
         populate_by_name = True
 
 
-class PromptOutput_Error(ErrorVariableValue):
+class PromptOutput_Error(ErrorVellumValue):
     type: typing.Literal["ERROR"] = "ERROR"
 
     class Config:
@@ -40,7 +40,7 @@ class PromptOutput_Error(ErrorVariableValue):
         populate_by_name = True
 
 
-class PromptOutput_FunctionCall(FunctionCallVariableValue):
+class PromptOutput_FunctionCall(FunctionCallVellumValue):
     type: typing.Literal["FUNCTION_CALL"] = "FUNCTION_CALL"
 
     class Config:

vellum_ai-0.5.1/src/vellum/types/node_output_compiled_string_value.py → vellum_ai-0.5.2/src/vellum/types/string_vellum_value.py

@@ -7,8 +7,11 @@ from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import pydantic_v1
 
 
-class NodeOutputCompiledStringValue(pydantic_v1.BaseModel):
-    node_output_id: str
+class StringVellumValue(pydantic_v1.BaseModel):
+    """
+    A value representing a string.
+    """
+
     value: typing.Optional[str] = None
 
     def json(self, **kwargs: typing.Any) -> str: