vellum-ai 0.2.2__py3-none-any.whl → 0.3.2__py3-none-any.whl

vellum/resources/sandboxes/client.py

@@ -7,7 +7,6 @@ from json.decoder import JSONDecodeError
 from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.jsonable_encoder import jsonable_encoder
-from ...types.sandbox_metric_input_params_request import SandboxMetricInputParamsRequest
 from ...types.sandbox_scenario import SandboxScenario
 from ...types.scenario_input_request import ScenarioInputRequest
 
@@ -31,7 +30,6 @@ class SandboxesClient:
         label: typing.Optional[str] = OMIT,
         inputs: typing.List[ScenarioInputRequest],
         scenario_id: typing.Optional[str] = OMIT,
-        metric_input_params: typing.Optional[SandboxMetricInputParamsRequest] = OMIT,
     ) -> SandboxScenario:
         """
         Upserts a new scenario for a sandbox, keying off of the optionally provided scenario id.
@@ -50,17 +48,8 @@ class SandboxesClient:
             - inputs: typing.List[ScenarioInputRequest]. The inputs for the scenario
 
             - scenario_id: typing.Optional[str]. The id of the scenario to update. If none is provided, an id will be generated and a new scenario will be appended.
-
-            - metric_input_params: typing.Optional[SandboxMetricInputParamsRequest].
         ---
-        from vellum import (
-            ChatMessageRequest,
-            ChatMessageRole,
-            EvaluationParamsRequest,
-            SandboxMetricInputParamsRequest,
-            ScenarioInputRequest,
-            ScenarioInputTypeEnum,
-        )
+        from vellum import ScenarioInputRequest, ScenarioInputTypeEnum
         from vellum.client import Vellum
 
         client = Vellum(
@@ -68,20 +57,19 @@ class SandboxesClient:
         )
         client.sandboxes.upsert_sandbox_scenario(
             id="string",
+            label="Scenario 1",
             inputs=[
                 ScenarioInputRequest(
-                    key="string",
+                    key="var_1",
                     type=ScenarioInputTypeEnum.TEXT,
-                    chat_history=[
-                        ChatMessageRequest(
-                            role=ChatMessageRole.SYSTEM,
-                        )
-                    ],
-                )
+                    value="Hello, world!",
+                ),
+                ScenarioInputRequest(
+                    key="var_2",
+                    type=ScenarioInputTypeEnum.TEXT,
+                    value="Why hello, there!",
+                ),
             ],
-            metric_input_params=SandboxMetricInputParamsRequest(
-                params=EvaluationParamsRequest(),
-            ),
         )
         """
         _request: typing.Dict[str, typing.Any] = {"inputs": inputs}
@@ -89,8 +77,6 @@ class SandboxesClient:
             _request["label"] = label
         if scenario_id is not OMIT:
             _request["scenario_id"] = scenario_id
-        if metric_input_params is not OMIT:
-            _request["metric_input_params"] = metric_input_params
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", f"v1/sandboxes/{id}/scenarios"),
@@ -153,7 +139,6 @@ class AsyncSandboxesClient:
         label: typing.Optional[str] = OMIT,
         inputs: typing.List[ScenarioInputRequest],
         scenario_id: typing.Optional[str] = OMIT,
-        metric_input_params: typing.Optional[SandboxMetricInputParamsRequest] = OMIT,
     ) -> SandboxScenario:
         """
         Upserts a new scenario for a sandbox, keying off of the optionally provided scenario id.
@@ -172,17 +157,8 @@ class AsyncSandboxesClient:
             - inputs: typing.List[ScenarioInputRequest]. The inputs for the scenario
 
             - scenario_id: typing.Optional[str]. The id of the scenario to update. If none is provided, an id will be generated and a new scenario will be appended.
-
-            - metric_input_params: typing.Optional[SandboxMetricInputParamsRequest].
         ---
-        from vellum import (
-            ChatMessageRequest,
-            ChatMessageRole,
-            EvaluationParamsRequest,
-            SandboxMetricInputParamsRequest,
-            ScenarioInputRequest,
-            ScenarioInputTypeEnum,
-        )
+        from vellum import ScenarioInputRequest, ScenarioInputTypeEnum
         from vellum.client import AsyncVellum
 
         client = AsyncVellum(
@@ -190,20 +166,19 @@ class AsyncSandboxesClient:
         )
         await client.sandboxes.upsert_sandbox_scenario(
             id="string",
+            label="Scenario 1",
             inputs=[
                 ScenarioInputRequest(
-                    key="string",
+                    key="var_1",
                     type=ScenarioInputTypeEnum.TEXT,
-                    chat_history=[
-                        ChatMessageRequest(
-                            role=ChatMessageRole.SYSTEM,
-                        )
-                    ],
-                )
+                    value="Hello, world!",
+                ),
+                ScenarioInputRequest(
+                    key="var_2",
+                    type=ScenarioInputTypeEnum.TEXT,
+                    value="Why hello, there!",
+                ),
             ],
-            metric_input_params=SandboxMetricInputParamsRequest(
-                params=EvaluationParamsRequest(),
-            ),
         )
         """
         _request: typing.Dict[str, typing.Any] = {"inputs": inputs}
@@ -211,8 +186,6 @@ class AsyncSandboxesClient:
             _request["label"] = label
         if scenario_id is not OMIT:
             _request["scenario_id"] = scenario_id
-        if metric_input_params is not OMIT:
-            _request["metric_input_params"] = metric_input_params
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", f"v1/sandboxes/{id}/scenarios"),

vellum/types/__init__.py

@@ -16,12 +16,9 @@ from .array_chat_message_content_item_request import (
     ArrayChatMessageContentItemRequest_String,
 )
 from .array_chat_message_content_request import ArrayChatMessageContentRequest
-from .array_enum import ArrayEnum
-from .array_variable_value import ArrayVariableValue
 from .block_type_enum import BlockTypeEnum
 from .chat_history_enum import ChatHistoryEnum
 from .chat_history_input_request import ChatHistoryInputRequest
-from .chat_history_variable_value import ChatHistoryVariableValue
 from .chat_message import ChatMessage
 from .chat_message_content import (
     ChatMessageContent,
@@ -69,8 +66,6 @@ from .entity_status import EntityStatus
 from .environment_enum import EnvironmentEnum
 from .error_enum import ErrorEnum
 from .error_variable_value import ErrorVariableValue
-from .evaluation_params import EvaluationParams
-from .evaluation_params_request import EvaluationParamsRequest
 from .execute_prompt_api_error_response import ExecutePromptApiErrorResponse
 from .execute_prompt_event import (
     ExecutePromptEvent,
@@ -99,6 +94,7 @@ from .fulfilled_execute_prompt_response import FulfilledExecutePromptResponse
 from .fulfilled_execute_workflow_workflow_result_event import FulfilledExecuteWorkflowWorkflowResultEvent
 from .fulfilled_function_call import FulfilledFunctionCall
 from .fulfilled_prompt_execution_meta import FulfilledPromptExecutionMeta
+from .fulfilled_workflow_node_result_event import FulfilledWorkflowNodeResultEvent
 from .function_call import FunctionCall, FunctionCall_Fulfilled, FunctionCall_Rejected
 from .function_call_chat_message_content import FunctionCallChatMessageContent
 from .function_call_chat_message_content_request import FunctionCallChatMessageContentRequest
@@ -123,6 +119,7 @@ from .indexing_state_enum import IndexingStateEnum
 from .initiated_enum import InitiatedEnum
 from .initiated_execute_prompt_event import InitiatedExecutePromptEvent
 from .initiated_prompt_execution_meta import InitiatedPromptExecutionMeta
+from .initiated_workflow_node_result_event import InitiatedWorkflowNodeResultEvent
 from .json_enum import JsonEnum
 from .json_input_request import JsonInputRequest
 from .json_variable_value import JsonVariableValue
@@ -167,10 +164,24 @@ from .node_input_variable_compiled_value import (
     NodeInputVariableCompiledValue_SearchResults,
     NodeInputVariableCompiledValue_String,
 )
+from .node_output_compiled_chat_history_value import NodeOutputCompiledChatHistoryValue
+from .node_output_compiled_error_value import NodeOutputCompiledErrorValue
+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
+from .node_output_compiled_string_value import NodeOutputCompiledStringValue
+from .node_output_compiled_value import (
+    NodeOutputCompiledValue,
+    NodeOutputCompiledValue_ChatHistory,
+    NodeOutputCompiledValue_Error,
+    NodeOutputCompiledValue_Json,
+    NodeOutputCompiledValue_Number,
+    NodeOutputCompiledValue_SearchResults,
+    NodeOutputCompiledValue_String,
+)
 from .normalized_log_probs import NormalizedLogProbs
 from .normalized_token_log_probs import NormalizedTokenLogProbs
 from .number_enum import NumberEnum
-from .number_variable_value import NumberVariableValue
 from .paginated_slim_deployment_read_list import PaginatedSlimDeploymentReadList
 from .paginated_slim_document_list import PaginatedSlimDocumentList
 from .paginated_slim_workflow_deployment_list import PaginatedSlimWorkflowDeploymentList
@@ -217,8 +228,7 @@ from .rejected_execute_prompt_response import RejectedExecutePromptResponse
 from .rejected_execute_workflow_workflow_result_event import RejectedExecuteWorkflowWorkflowResultEvent
 from .rejected_function_call import RejectedFunctionCall
 from .rejected_prompt_execution_meta import RejectedPromptExecutionMeta
-from .sandbox_metric_input_params import SandboxMetricInputParams
-from .sandbox_metric_input_params_request import SandboxMetricInputParamsRequest
+from .rejected_workflow_node_result_event import RejectedWorkflowNodeResultEvent
 from .sandbox_scenario import SandboxScenario
 from .scenario_input import ScenarioInput
 from .scenario_input_request import ScenarioInputRequest
@@ -235,7 +245,6 @@ from .search_result_document_request import SearchResultDocumentRequest
 from .search_result_merging_request import SearchResultMergingRequest
 from .search_result_request import SearchResultRequest
 from .search_results_enum import SearchResultsEnum
-from .search_results_variable_value import SearchResultsVariableValue
 from .search_weights_request import SearchWeightsRequest
 from .slim_deployment_read import SlimDeploymentRead
 from .slim_document import SlimDocument
@@ -243,6 +252,7 @@ from .slim_workflow_deployment import SlimWorkflowDeployment
 from .streaming_enum import StreamingEnum
 from .streaming_execute_prompt_event import StreamingExecutePromptEvent
 from .streaming_prompt_execution_meta import StreamingPromptExecutionMeta
+from .streaming_workflow_node_result_event import StreamingWorkflowNodeResultEvent
 from .string_chat_message_content import StringChatMessageContent
 from .string_chat_message_content_request import StringChatMessageContentRequest
 from .string_enum import StringEnum
@@ -308,17 +318,6 @@ from .test_case_variable_value import (
 from .test_suite_test_case import TestSuiteTestCase
 from .upload_document_error_response import UploadDocumentErrorResponse
 from .upload_document_response import UploadDocumentResponse
-from .variable_value import (
-    VariableValue,
-    VariableValue_Array,
-    VariableValue_ChatHistory,
-    VariableValue_Error,
-    VariableValue_FunctionCall,
-    VariableValue_Json,
-    VariableValue_Number,
-    VariableValue_SearchResults,
-    VariableValue_String,
-)
 from .vellum_error import VellumError
 from .vellum_error_code_enum import VellumErrorCodeEnum
 from .vellum_error_request import VellumErrorRequest
@@ -344,11 +343,16 @@ from .workflow_node_result_data import (
     WorkflowNodeResultData_Templating,
     WorkflowNodeResultData_Terminal,
 )
-from .workflow_node_result_event import WorkflowNodeResultEvent
+from .workflow_node_result_event import (
+    WorkflowNodeResultEvent,
+    WorkflowNodeResultEvent_Fulfilled,
+    WorkflowNodeResultEvent_Initiated,
+    WorkflowNodeResultEvent_Rejected,
+    WorkflowNodeResultEvent_Streaming,
+)
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
 from .workflow_output import (
     WorkflowOutput,
-    WorkflowOutput_Array,
     WorkflowOutput_ChatHistory,
     WorkflowOutput_Error,
     WorkflowOutput_FunctionCall,
@@ -358,7 +362,6 @@ from .workflow_output import (
     WorkflowOutput_SearchResults,
     WorkflowOutput_String,
 )
-from .workflow_output_array import WorkflowOutputArray
 from .workflow_output_chat_history import WorkflowOutputChatHistory
 from .workflow_output_error import WorkflowOutputError
 from .workflow_output_function_call import WorkflowOutputFunctionCall
@@ -409,12 +412,9 @@ __all__ = [
     "ArrayChatMessageContentItem_Image",
     "ArrayChatMessageContentItem_String",
     "ArrayChatMessageContentRequest",
-    "ArrayEnum",
-    "ArrayVariableValue",
     "BlockTypeEnum",
     "ChatHistoryEnum",
     "ChatHistoryInputRequest",
-    "ChatHistoryVariableValue",
     "ChatMessage",
     "ChatMessageContent",
     "ChatMessageContentRequest",
@@ -456,8 +456,6 @@ __all__ = [
     "EnvironmentEnum",
     "ErrorEnum",
     "ErrorVariableValue",
-    "EvaluationParams",
-    "EvaluationParamsRequest",
     "ExecutePromptApiErrorResponse",
     "ExecutePromptEvent",
     "ExecutePromptEvent_Fulfilled",
@@ -480,6 +478,7 @@ __all__ = [
     "FulfilledExecuteWorkflowWorkflowResultEvent",
     "FulfilledFunctionCall",
     "FulfilledPromptExecutionMeta",
+    "FulfilledWorkflowNodeResultEvent",
     "FunctionCall",
     "FunctionCallChatMessageContent",
     "FunctionCallChatMessageContentRequest",
@@ -506,6 +505,7 @@ __all__ = [
     "InitiatedEnum",
     "InitiatedExecutePromptEvent",
     "InitiatedPromptExecutionMeta",
+    "InitiatedWorkflowNodeResultEvent",
     "JsonEnum",
     "JsonInputRequest",
     "JsonVariableValue",
@@ -546,10 +546,22 @@ __all__ = [
     "NodeInputVariableCompiledValue_Number",
     "NodeInputVariableCompiledValue_SearchResults",
     "NodeInputVariableCompiledValue_String",
+    "NodeOutputCompiledChatHistoryValue",
+    "NodeOutputCompiledErrorValue",
+    "NodeOutputCompiledJsonValue",
+    "NodeOutputCompiledNumberValue",
+    "NodeOutputCompiledSearchResultsValue",
+    "NodeOutputCompiledStringValue",
+    "NodeOutputCompiledValue",
+    "NodeOutputCompiledValue_ChatHistory",
+    "NodeOutputCompiledValue_Error",
+    "NodeOutputCompiledValue_Json",
+    "NodeOutputCompiledValue_Number",
+    "NodeOutputCompiledValue_SearchResults",
+    "NodeOutputCompiledValue_String",
     "NormalizedLogProbs",
     "NormalizedTokenLogProbs",
     "NumberEnum",
-    "NumberVariableValue",
     "PaginatedSlimDeploymentReadList",
     "PaginatedSlimDocumentList",
     "PaginatedSlimWorkflowDeploymentList",
@@ -592,8 +604,7 @@ __all__ = [
     "RejectedExecuteWorkflowWorkflowResultEvent",
     "RejectedFunctionCall",
     "RejectedPromptExecutionMeta",
-    "SandboxMetricInputParams",
-    "SandboxMetricInputParamsRequest",
+    "RejectedWorkflowNodeResultEvent",
     "SandboxScenario",
     "ScenarioInput",
     "ScenarioInputRequest",
@@ -610,7 +621,6 @@ __all__ = [
     "SearchResultMergingRequest",
     "SearchResultRequest",
     "SearchResultsEnum",
-    "SearchResultsVariableValue",
     "SearchWeightsRequest",
     "SlimDeploymentRead",
     "SlimDocument",
@@ -618,6 +628,7 @@ __all__ = [
     "StreamingEnum",
     "StreamingExecutePromptEvent",
     "StreamingPromptExecutionMeta",
+    "StreamingWorkflowNodeResultEvent",
     "StringChatMessageContent",
     "StringChatMessageContentRequest",
     "StringEnum",
@@ -675,15 +686,6 @@ __all__ = [
     "TestSuiteTestCase",
     "UploadDocumentErrorResponse",
     "UploadDocumentResponse",
-    "VariableValue",
-    "VariableValue_Array",
-    "VariableValue_ChatHistory",
-    "VariableValue_Error",
-    "VariableValue_FunctionCall",
-    "VariableValue_Json",
-    "VariableValue_Number",
-    "VariableValue_SearchResults",
-    "VariableValue_String",
     "VellumError",
     "VellumErrorCodeEnum",
     "VellumErrorRequest",
@@ -709,8 +711,11 @@ __all__ = [
     "WorkflowNodeResultData_Terminal",
     "WorkflowNodeResultEvent",
     "WorkflowNodeResultEventState",
+    "WorkflowNodeResultEvent_Fulfilled",
+    "WorkflowNodeResultEvent_Initiated",
+    "WorkflowNodeResultEvent_Rejected",
+    "WorkflowNodeResultEvent_Streaming",
     "WorkflowOutput",
-    "WorkflowOutputArray",
     "WorkflowOutputChatHistory",
     "WorkflowOutputError",
     "WorkflowOutputFunctionCall",
@@ -719,7 +724,6 @@ __all__ = [
     "WorkflowOutputNumber",
     "WorkflowOutputSearchResults",
     "WorkflowOutputString",
-    "WorkflowOutput_Array",
     "WorkflowOutput_ChatHistory",
     "WorkflowOutput_Error",
     "WorkflowOutput_FunctionCall",

vellum/types/api_node_result.py

@@ -13,6 +13,10 @@ except ImportError:
 
 
 class ApiNodeResult(pydantic.BaseModel):
+    """
+    A Node Result Event emitted from an API Node.
+    """
+
     data: ApiNodeResultData
 
     def json(self, **kwargs: typing.Any) -> str:

vellum/types/code_execution_node_result.py

@@ -13,6 +13,10 @@ except ImportError:
 
 
 class CodeExecutionNodeResult(pydantic.BaseModel):
+    """
+    A Node Result Event emitted from a Code Execution Node.
+    """
+
     data: CodeExecutionNodeResultData
 
     def json(self, **kwargs: typing.Any) -> str:

vellum/types/conditional_node_result.py

@@ -13,6 +13,10 @@ except ImportError:
 
 
 class ConditionalNodeResult(pydantic.BaseModel):
+    """
+    A Node Result Event emitted from a Conditional Node.
+    """
+
     data: ConditionalNodeResultData
 
     def json(self, **kwargs: typing.Any) -> str:

vellum/types/{array_variable_value.py → fulfilled_workflow_node_result_event.py}

@@ -1,11 +1,11 @@
 # 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 .node_output_compiled_value import NodeOutputCompiledValue
+from .workflow_node_result_data import WorkflowNodeResultData
 
 try:
     import pydantic.v1 as pydantic  # type: ignore
@@ -13,8 +13,17 @@ except ImportError:
     import pydantic  # type: ignore
 
 
-class ArrayVariableValue(pydantic.BaseModel):
-    value: typing.List[VariableValue]
+class FulfilledWorkflowNodeResultEvent(pydantic.BaseModel):
+    """
+    An event that indicates that the node has fulfilled its execution.
+    """
+
+    id: str
+    node_id: str
+    node_result_id: str
+    ts: typing.Optional[dt.datetime]
+    data: typing.Optional[WorkflowNodeResultData]
+    output_values: typing.Optional[typing.List[NodeOutputCompiledValue]]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -28,8 +37,3 @@ class ArrayVariableValue(pydantic.BaseModel):
         frozen = True
         smart_union = True
         json_encoders = {dt.datetime: serialize_datetime}
-
-
-from .variable_value import VariableValue  # noqa: E402
-
-ArrayVariableValue.update_forward_refs()

vellum/types/initiated_workflow_node_result_event.py

@@ -0,0 +1,39 @@
+# 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 .node_input_variable_compiled_value import NodeInputVariableCompiledValue
+from .workflow_node_result_data import WorkflowNodeResultData
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class InitiatedWorkflowNodeResultEvent(pydantic.BaseModel):
+    """
+    An event that indicates that the node has initiated its execution.
+    """
+
+    id: str
+    node_id: str
+    node_result_id: str
+    ts: typing.Optional[dt.datetime]
+    data: typing.Optional[WorkflowNodeResultData]
+    input_values: typing.Optional[typing.List[NodeInputVariableCompiledValue]]
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/{chat_history_variable_value.py → node_output_compiled_chat_history_value.py}

@@ -12,7 +12,8 @@ except ImportError:
     import pydantic  # type: ignore
 
 
-class ChatHistoryVariableValue(pydantic.BaseModel):
+class NodeOutputCompiledChatHistoryValue(pydantic.BaseModel):
+    node_output_id: str
     value: typing.Optional[typing.List[ChatMessage]]
 
     def json(self, **kwargs: typing.Any) -> str:

vellum/types/node_output_compiled_error_value.py

@@ -0,0 +1,30 @@
+# 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 .vellum_error import VellumError
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class NodeOutputCompiledErrorValue(pydantic.BaseModel):
+    node_output_id: str
+    value: typing.Optional[VellumError]
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/{sandbox_metric_input_params.py → node_output_compiled_json_value.py}

@@ -4,7 +4,6 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .evaluation_params import EvaluationParams
 
 try:
     import pydantic.v1 as pydantic  # type: ignore
@@ -12,8 +11,9 @@ except ImportError:
     import pydantic  # type: ignore
 
 
-class SandboxMetricInputParams(pydantic.BaseModel):
-    params: typing.Optional[EvaluationParams]
+class NodeOutputCompiledJsonValue(pydantic.BaseModel):
+    node_output_id: str
+    value: typing.Optional[typing.Dict[str, typing.Any]]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

vellum/types/{number_variable_value.py → node_output_compiled_number_value.py}

@@ -11,7 +11,8 @@ except ImportError:
     import pydantic  # type: ignore
 
 
-class NumberVariableValue(pydantic.BaseModel):
+class NodeOutputCompiledNumberValue(pydantic.BaseModel):
+    node_output_id: str
     value: typing.Optional[float]
 
     def json(self, **kwargs: typing.Any) -> str:

vellum/types/{search_results_variable_value.py → node_output_compiled_search_results_value.py}

@@ -12,7 +12,8 @@ except ImportError:
     import pydantic  # type: ignore
 
 
-class SearchResultsVariableValue(pydantic.BaseModel):
+class NodeOutputCompiledSearchResultsValue(pydantic.BaseModel):
+    node_output_id: str
     value: typing.Optional[typing.List[SearchResult]]
 
     def json(self, **kwargs: typing.Any) -> str:

vellum/types/node_output_compiled_string_value.py

@@ -0,0 +1,29 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class NodeOutputCompiledStringValue(pydantic.BaseModel):
+    node_output_id: str
+    value: typing.Optional[str]
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}