vellum-ai 0.0.41__tar.gz → 0.0.43__tar.gz

{vellum_ai-0.0.41 → vellum_ai-0.0.43}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.0.41
+Version: 0.0.43
 Summary: 
 Requires-Python: >=3.7,<4.0
 Classifier: Programming Language :: Python :: 3
@@ -10,7 +10,7 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Requires-Dist: httpx (>=0.21.2)
-Requires-Dist: pydantic (>=1.9.2,<2.0.0)
+Requires-Dist: pydantic (>=1.9.2,<3)
 Description-Content-Type: text/markdown
 
 # Vellum Python Library

{vellum_ai-0.0.41 → vellum_ai-0.0.43}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vellum-ai"
-version = "v0.0.41"
+version = "v0.0.43"
 description = ""
 readme = "README.md"
 authors = []
@@ -11,7 +11,7 @@ packages = [
 [tool.poetry.dependencies]
 python = "^3.7"
 httpx = ">=0.21.2"
-pydantic = "^1.9.2"
+pydantic = ">= 1.9.2, < 3"
 
 [tool.poetry.dev-dependencies]
 mypy = "0.971"

{vellum_ai-0.0.41 → vellum_ai-0.0.43}/src/vellum/__init__.py

@@ -1,13 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .types import (
+    ApiNodeResult,
+    ApiNodeResultData,
     BlockTypeEnum,
     ChatMessage,
     ChatMessageRequest,
     ChatMessageRole,
     ConditionalNodeResult,
     ConditionalNodeResultData,
-    ContentType,
     DeploymentRead,
     DeploymentStatus,
     DocumentDocumentToDocumentIndex,
@@ -48,11 +49,13 @@ from .types import (
     ModelVersionSandboxSnapshot,
     NodeInputCompiledChatHistoryValue,
     NodeInputCompiledJsonValue,
+    NodeInputCompiledNumberValue,
     NodeInputCompiledSearchResultsValue,
     NodeInputCompiledStringValue,
     NodeInputVariableCompiledValue,
     NodeInputVariableCompiledValue_ChatHistory,
     NodeInputVariableCompiledValue_Json,
+    NodeInputVariableCompiledValue_Number,
     NodeInputVariableCompiledValue_SearchResults,
     NodeInputVariableCompiledValue_String,
     NormalizedLogProbs,
@@ -104,22 +107,26 @@ from .types import (
     SubmitWorkflowExecutionActualRequest_String,
     TemplatingNodeChatHistoryResult,
     TemplatingNodeJsonResult,
+    TemplatingNodeNumberResult,
     TemplatingNodeResult,
     TemplatingNodeResultData,
     TemplatingNodeResultOutput,
     TemplatingNodeResultOutput_ChatHistory,
     TemplatingNodeResultOutput_Json,
+    TemplatingNodeResultOutput_Number,
     TemplatingNodeResultOutput_SearchResults,
     TemplatingNodeResultOutput_String,
     TemplatingNodeSearchResultsResult,
     TemplatingNodeStringResult,
     TerminalNodeChatHistoryResult,
     TerminalNodeJsonResult,
+    TerminalNodeNumberResult,
     TerminalNodeResult,
     TerminalNodeResultData,
     TerminalNodeResultOutput,
     TerminalNodeResultOutput_ChatHistory,
     TerminalNodeResultOutput_Json,
+    TerminalNodeResultOutput_Number,
     TerminalNodeResultOutput_SearchResults,
     TerminalNodeResultOutput_String,
     TerminalNodeSearchResultsResult,
@@ -138,6 +145,7 @@ from .types import (
     WorkflowExecutionNodeResultEvent,
     WorkflowExecutionWorkflowResultEvent,
     WorkflowNodeResultData,
+    WorkflowNodeResultData_Api,
     WorkflowNodeResultData_Conditional,
     WorkflowNodeResultData_Prompt,
     WorkflowNodeResultData_Search,
@@ -156,10 +164,12 @@ from .types import (
     WorkflowResultEventOutputData,
     WorkflowResultEventOutputDataChatHistory,
     WorkflowResultEventOutputDataJson,
+    WorkflowResultEventOutputDataNumber,
     WorkflowResultEventOutputDataSearchResults,
     WorkflowResultEventOutputDataString,
     WorkflowResultEventOutputData_ChatHistory,
     WorkflowResultEventOutputData_Json,
+    WorkflowResultEventOutputData_Number,
     WorkflowResultEventOutputData_SearchResults,
     WorkflowResultEventOutputData_String,
     WorkflowStreamEvent,
@@ -179,6 +189,8 @@ from .resources import (
 from .environment import VellumEnvironment
 
 __all__ = [
+    "ApiNodeResult",
+    "ApiNodeResultData",
     "BadRequestError",
     "BlockTypeEnum",
     "ChatMessage",
@@ -187,7 +199,6 @@ __all__ = [
     "ConditionalNodeResult",
     "ConditionalNodeResultData",
     "ConflictError",
-    "ContentType",
     "DeploymentRead",
     "DeploymentStatus",
     "DocumentDocumentToDocumentIndex",
@@ -230,11 +241,13 @@ __all__ = [
     "ModelVersionSandboxSnapshot",
     "NodeInputCompiledChatHistoryValue",
     "NodeInputCompiledJsonValue",
+    "NodeInputCompiledNumberValue",
     "NodeInputCompiledSearchResultsValue",
     "NodeInputCompiledStringValue",
     "NodeInputVariableCompiledValue",
     "NodeInputVariableCompiledValue_ChatHistory",
     "NodeInputVariableCompiledValue_Json",
+    "NodeInputVariableCompiledValue_Number",
     "NodeInputVariableCompiledValue_SearchResults",
     "NodeInputVariableCompiledValue_String",
     "NormalizedLogProbs",
@@ -287,22 +300,26 @@ __all__ = [
     "SubmitWorkflowExecutionActualRequest_String",
     "TemplatingNodeChatHistoryResult",
     "TemplatingNodeJsonResult",
+    "TemplatingNodeNumberResult",
     "TemplatingNodeResult",
     "TemplatingNodeResultData",
     "TemplatingNodeResultOutput",
     "TemplatingNodeResultOutput_ChatHistory",
     "TemplatingNodeResultOutput_Json",
+    "TemplatingNodeResultOutput_Number",
     "TemplatingNodeResultOutput_SearchResults",
     "TemplatingNodeResultOutput_String",
     "TemplatingNodeSearchResultsResult",
     "TemplatingNodeStringResult",
     "TerminalNodeChatHistoryResult",
     "TerminalNodeJsonResult",
+    "TerminalNodeNumberResult",
     "TerminalNodeResult",
     "TerminalNodeResultData",
     "TerminalNodeResultOutput",
     "TerminalNodeResultOutput_ChatHistory",
     "TerminalNodeResultOutput_Json",
+    "TerminalNodeResultOutput_Number",
     "TerminalNodeResultOutput_SearchResults",
     "TerminalNodeResultOutput_String",
     "TerminalNodeSearchResultsResult",
@@ -322,6 +339,7 @@ __all__ = [
     "WorkflowExecutionNodeResultEvent",
     "WorkflowExecutionWorkflowResultEvent",
     "WorkflowNodeResultData",
+    "WorkflowNodeResultData_Api",
     "WorkflowNodeResultData_Conditional",
     "WorkflowNodeResultData_Prompt",
     "WorkflowNodeResultData_Search",
@@ -340,10 +358,12 @@ __all__ = [
     "WorkflowResultEventOutputData",
     "WorkflowResultEventOutputDataChatHistory",
     "WorkflowResultEventOutputDataJson",
+    "WorkflowResultEventOutputDataNumber",
     "WorkflowResultEventOutputDataSearchResults",
     "WorkflowResultEventOutputDataString",
     "WorkflowResultEventOutputData_ChatHistory",
     "WorkflowResultEventOutputData_Json",
+    "WorkflowResultEventOutputData_Number",
     "WorkflowResultEventOutputData_SearchResults",
     "WorkflowResultEventOutputData_String",
     "WorkflowStreamEvent",

{vellum_ai-0.0.41 → vellum_ai-0.0.43}/src/vellum/client.py

@@ -6,7 +6,6 @@ import urllib.parse
 from json.decoder import JSONDecodeError
 
 import httpx
-import pydantic
 
 from .core.api_error import ApiError
 from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
@@ -36,6 +35,11 @@ from .types.workflow_execution_event_type import WorkflowExecutionEventType
 from .types.workflow_request_input_request import WorkflowRequestInputRequest
 from .types.workflow_stream_event import WorkflowStreamEvent
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
 
@@ -140,7 +144,7 @@ class Vellum:
 
             - deployment_name: typing.Optional[str]. The name of the deployment. Must provide either this or deployment_id.
 
-            - requests: typing.List[GenerateRequest]. The generation requests to make. Supplying multiple will perform a bulk request to the LLM provided when possible.
+            - requests: typing.List[GenerateRequest]. The generation request to make. Bulk requests are no longer supported, this field must be an array of length 1.
 
             - options: typing.Optional[GenerateOptionsRequest]. Additional configuration that can be used to control what's included in the response.
         """
@@ -192,7 +196,7 @@ class Vellum:
 
             - deployment_name: typing.Optional[str]. The name of the deployment. Must provide either this or deployment_id.
 
-            - requests: typing.List[GenerateRequest]. The generation requests to make. Supplying multiple will perform a bulk request to the LLM provided when possible.
+            - requests: typing.List[GenerateRequest]. The generation request to make. Bulk requests are no longer supported, this field must be an array of length 1.
 
             - options: typing.Optional[GenerateOptionsRequest]. Additional configuration that can be used to control what's included in the response.
         """
@@ -468,7 +472,7 @@ class AsyncVellum:
 
             - deployment_name: typing.Optional[str]. The name of the deployment. Must provide either this or deployment_id.
 
-            - requests: typing.List[GenerateRequest]. The generation requests to make. Supplying multiple will perform a bulk request to the LLM provided when possible.
+            - requests: typing.List[GenerateRequest]. The generation request to make. Bulk requests are no longer supported, this field must be an array of length 1.
 
             - options: typing.Optional[GenerateOptionsRequest]. Additional configuration that can be used to control what's included in the response.
         """
@@ -520,7 +524,7 @@ class AsyncVellum:
 
             - deployment_name: typing.Optional[str]. The name of the deployment. Must provide either this or deployment_id.
 
-            - requests: typing.List[GenerateRequest]. The generation requests to make. Supplying multiple will perform a bulk request to the LLM provided when possible.
+            - requests: typing.List[GenerateRequest]. The generation request to make. Bulk requests are no longer supported, this field must be an array of length 1.
 
             - options: typing.Optional[GenerateOptionsRequest]. Additional configuration that can be used to control what's included in the response.
         """

{vellum_ai-0.0.41 → vellum_ai-0.0.43}/src/vellum/core/client_wrapper.py

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

{vellum_ai-0.0.41 → vellum_ai-0.0.43}/src/vellum/core/jsonable_encoder.py

@@ -16,8 +16,10 @@ from pathlib import PurePath
 from types import GeneratorType
 from typing import Any, Callable, Dict, List, Optional, Set, Tuple, Union
 
-from pydantic import BaseModel
-from pydantic.json import ENCODERS_BY_TYPE
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
 
 from .datetime_utils import serialize_datetime
 
@@ -34,7 +36,7 @@ def generate_encoders_by_class_tuples(
     return encoders_by_class_tuples
 
 
-encoders_by_class_tuples = generate_encoders_by_class_tuples(ENCODERS_BY_TYPE)
+encoders_by_class_tuples = generate_encoders_by_class_tuples(pydantic.json.ENCODERS_BY_TYPE)
 
 
 def jsonable_encoder(obj: Any, custom_encoder: Optional[Dict[Any, Callable[[Any], Any]]] = None) -> Any:
@@ -46,7 +48,7 @@ def jsonable_encoder(obj: Any, custom_encoder: Optional[Dict[Any, Callable[[Any]
             for encoder_type, encoder_instance in custom_encoder.items():
                 if isinstance(obj, encoder_type):
                     return encoder_instance(obj)
-    if isinstance(obj, BaseModel):
+    if isinstance(obj, pydantic.BaseModel):
         encoder = getattr(obj.__config__, "json_encoders", {})
         if custom_encoder:
             encoder.update(custom_encoder)
@@ -82,8 +84,8 @@ def jsonable_encoder(obj: Any, custom_encoder: Optional[Dict[Any, Callable[[Any]
             encoded_list.append(jsonable_encoder(item, custom_encoder=custom_encoder))
         return encoded_list
 
-    if type(obj) in ENCODERS_BY_TYPE:
-        return ENCODERS_BY_TYPE[type(obj)](obj)
+    if type(obj) in pydantic.json.ENCODERS_BY_TYPE:
+        return pydantic.json.ENCODERS_BY_TYPE[type(obj)](obj)
     for encoder, classes_tuple in encoders_by_class_tuples.items():
         if isinstance(obj, classes_tuple):
             return encoder(obj)

{vellum_ai-0.0.41 → vellum_ai-0.0.43}/src/vellum/resources/deployments/client.py

@@ -3,12 +3,15 @@
 import urllib.parse
 from json.decoder import JSONDecodeError
 
-import pydantic
-
 from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...types.deployment_read import DeploymentRead
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class DeploymentsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):

{vellum_ai-0.0.41 → vellum_ai-0.0.43}/src/vellum/resources/document_indexes/client.py

@@ -4,8 +4,6 @@ import typing
 import urllib.parse
 from json.decoder import JSONDecodeError
 
-import pydantic
-
 from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.jsonable_encoder import jsonable_encoder
@@ -13,6 +11,11 @@ from ...types.document_index_read import DocumentIndexRead
 from ...types.document_index_status import DocumentIndexStatus
 from ...types.environment_enum import EnvironmentEnum
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
 
@@ -74,6 +77,27 @@ class DocumentIndexesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def retrieve(self, id: str) -> DocumentIndexRead:
+        """
+        Used to retrieve a Document Index given its ID or name.
+
+        Parameters:
+            - id: str. Either the Document Index's ID or its unique name
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", f"v1/document-indexes/{id}"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=None,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(DocumentIndexRead, _response.json())  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
 
 class AsyncDocumentIndexesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -131,3 +155,24 @@ class AsyncDocumentIndexesClient:
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def retrieve(self, id: str) -> DocumentIndexRead:
+        """
+        Used to retrieve a Document Index given its ID or name.
+
+        Parameters:
+            - id: str. Either the Document Index's ID or its unique name
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", f"v1/document-indexes/{id}"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=None,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(DocumentIndexRead, _response.json())  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)

{vellum_ai-0.0.41 → vellum_ai-0.0.43}/src/vellum/resources/documents/client.py

@@ -4,8 +4,6 @@ import typing
 import urllib.parse
 from json.decoder import JSONDecodeError
 
-import pydantic
-
 from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.jsonable_encoder import jsonable_encoder
@@ -18,6 +16,11 @@ from ...types.document_status import DocumentStatus
 from ...types.paginated_slim_document_list import PaginatedSlimDocumentList
 from ...types.upload_document_response import UploadDocumentResponse
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
 

{vellum_ai-0.0.41 → vellum_ai-0.0.43}/src/vellum/resources/model_versions/client.py

@@ -4,14 +4,17 @@ import typing
 import urllib.parse
 from json.decoder import JSONDecodeError
 
-import pydantic
-
 from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.jsonable_encoder import jsonable_encoder
 from ...types.model_version_compile_prompt_response import ModelVersionCompilePromptResponse
 from ...types.model_version_read import ModelVersionRead
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
 

{vellum_ai-0.0.41 → vellum_ai-0.0.43}/src/vellum/resources/registered_prompts/client.py

@@ -4,8 +4,6 @@ import typing
 import urllib.parse
 from json.decoder import JSONDecodeError
 
-import pydantic
-
 from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.jsonable_encoder import jsonable_encoder
@@ -16,6 +14,11 @@ from ...types.register_prompt_model_parameters_request import RegisterPromptMode
 from ...types.register_prompt_prompt_info_request import RegisterPromptPromptInfoRequest
 from ...types.register_prompt_response import RegisterPromptResponse
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
 

{vellum_ai-0.0.41 → vellum_ai-0.0.43}/src/vellum/resources/sandboxes/client.py

@@ -4,8 +4,6 @@ import typing
 import urllib.parse
 from json.decoder import JSONDecodeError
 
-import pydantic
-
 from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.jsonable_encoder import jsonable_encoder
@@ -13,6 +11,11 @@ from ...types.sandbox_metric_input_params_request import SandboxMetricInputParam
 from ...types.sandbox_scenario import SandboxScenario
 from ...types.scenario_input_request import ScenarioInputRequest
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
 

{vellum_ai-0.0.41 → vellum_ai-0.0.43}/src/vellum/resources/test_suites/client.py

@@ -4,14 +4,17 @@ import typing
 import urllib.parse
 from json.decoder import JSONDecodeError
 
-import pydantic
-
 from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.jsonable_encoder import jsonable_encoder
 from ...types.evaluation_params_request import EvaluationParamsRequest
 from ...types.test_suite_test_case import TestSuiteTestCase
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
 

{vellum_ai-0.0.41 → vellum_ai-0.0.43}/src/vellum/types/__init__.py

@@ -1,12 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from .api_node_result import ApiNodeResult
+from .api_node_result_data import ApiNodeResultData
 from .block_type_enum import BlockTypeEnum
 from .chat_message import ChatMessage
 from .chat_message_request import ChatMessageRequest
 from .chat_message_role import ChatMessageRole
 from .conditional_node_result import ConditionalNodeResult
 from .conditional_node_result_data import ConditionalNodeResultData
-from .content_type import ContentType
 from .deployment_read import DeploymentRead
 from .deployment_status import DeploymentStatus
 from .document_document_to_document_index import DocumentDocumentToDocumentIndex
@@ -47,12 +48,14 @@ from .model_version_read_status_enum import ModelVersionReadStatusEnum
 from .model_version_sandbox_snapshot import ModelVersionSandboxSnapshot
 from .node_input_compiled_chat_history_value import NodeInputCompiledChatHistoryValue
 from .node_input_compiled_json_value import NodeInputCompiledJsonValue
+from .node_input_compiled_number_value import NodeInputCompiledNumberValue
 from .node_input_compiled_search_results_value import NodeInputCompiledSearchResultsValue
 from .node_input_compiled_string_value import NodeInputCompiledStringValue
 from .node_input_variable_compiled_value import (
     NodeInputVariableCompiledValue,
     NodeInputVariableCompiledValue_ChatHistory,
     NodeInputVariableCompiledValue_Json,
+    NodeInputVariableCompiledValue_Number,
     NodeInputVariableCompiledValue_SearchResults,
     NodeInputVariableCompiledValue_String,
 )
@@ -107,12 +110,14 @@ from .submit_workflow_execution_actual_request import (
 )
 from .templating_node_chat_history_result import TemplatingNodeChatHistoryResult
 from .templating_node_json_result import TemplatingNodeJsonResult
+from .templating_node_number_result import TemplatingNodeNumberResult
 from .templating_node_result import TemplatingNodeResult
 from .templating_node_result_data import TemplatingNodeResultData
 from .templating_node_result_output import (
     TemplatingNodeResultOutput,
     TemplatingNodeResultOutput_ChatHistory,
     TemplatingNodeResultOutput_Json,
+    TemplatingNodeResultOutput_Number,
     TemplatingNodeResultOutput_SearchResults,
     TemplatingNodeResultOutput_String,
 )
@@ -120,12 +125,14 @@ from .templating_node_search_results_result import TemplatingNodeSearchResultsRe
 from .templating_node_string_result import TemplatingNodeStringResult
 from .terminal_node_chat_history_result import TerminalNodeChatHistoryResult
 from .terminal_node_json_result import TerminalNodeJsonResult
+from .terminal_node_number_result import TerminalNodeNumberResult
 from .terminal_node_result import TerminalNodeResult
 from .terminal_node_result_data import TerminalNodeResultData
 from .terminal_node_result_output import (
     TerminalNodeResultOutput,
     TerminalNodeResultOutput_ChatHistory,
     TerminalNodeResultOutput_Json,
+    TerminalNodeResultOutput_Number,
     TerminalNodeResultOutput_SearchResults,
     TerminalNodeResultOutput_String,
 )
@@ -146,6 +153,7 @@ from .workflow_execution_node_result_event import WorkflowExecutionNodeResultEve
 from .workflow_execution_workflow_result_event import WorkflowExecutionWorkflowResultEvent
 from .workflow_node_result_data import (
     WorkflowNodeResultData,
+    WorkflowNodeResultData_Api,
     WorkflowNodeResultData_Conditional,
     WorkflowNodeResultData_Prompt,
     WorkflowNodeResultData_Search,
@@ -168,23 +176,26 @@ from .workflow_result_event_output_data import (
     WorkflowResultEventOutputData,
     WorkflowResultEventOutputData_ChatHistory,
     WorkflowResultEventOutputData_Json,
+    WorkflowResultEventOutputData_Number,
     WorkflowResultEventOutputData_SearchResults,
     WorkflowResultEventOutputData_String,
 )
 from .workflow_result_event_output_data_chat_history import WorkflowResultEventOutputDataChatHistory
 from .workflow_result_event_output_data_json import WorkflowResultEventOutputDataJson
+from .workflow_result_event_output_data_number import WorkflowResultEventOutputDataNumber
 from .workflow_result_event_output_data_search_results import WorkflowResultEventOutputDataSearchResults
 from .workflow_result_event_output_data_string import WorkflowResultEventOutputDataString
 from .workflow_stream_event import WorkflowStreamEvent, WorkflowStreamEvent_Node, WorkflowStreamEvent_Workflow
 
 __all__ = [
+    "ApiNodeResult",
+    "ApiNodeResultData",
     "BlockTypeEnum",
     "ChatMessage",
     "ChatMessageRequest",
     "ChatMessageRole",
     "ConditionalNodeResult",
     "ConditionalNodeResultData",
-    "ContentType",
     "DeploymentRead",
     "DeploymentStatus",
     "DocumentDocumentToDocumentIndex",
@@ -225,11 +236,13 @@ __all__ = [
     "ModelVersionSandboxSnapshot",
     "NodeInputCompiledChatHistoryValue",
     "NodeInputCompiledJsonValue",
+    "NodeInputCompiledNumberValue",
     "NodeInputCompiledSearchResultsValue",
     "NodeInputCompiledStringValue",
     "NodeInputVariableCompiledValue",
     "NodeInputVariableCompiledValue_ChatHistory",
     "NodeInputVariableCompiledValue_Json",
+    "NodeInputVariableCompiledValue_Number",
     "NodeInputVariableCompiledValue_SearchResults",
     "NodeInputVariableCompiledValue_String",
     "NormalizedLogProbs",
@@ -281,22 +294,26 @@ __all__ = [
     "SubmitWorkflowExecutionActualRequest_String",
     "TemplatingNodeChatHistoryResult",
     "TemplatingNodeJsonResult",
+    "TemplatingNodeNumberResult",
     "TemplatingNodeResult",
     "TemplatingNodeResultData",
     "TemplatingNodeResultOutput",
     "TemplatingNodeResultOutput_ChatHistory",
     "TemplatingNodeResultOutput_Json",
+    "TemplatingNodeResultOutput_Number",
     "TemplatingNodeResultOutput_SearchResults",
     "TemplatingNodeResultOutput_String",
     "TemplatingNodeSearchResultsResult",
     "TemplatingNodeStringResult",
     "TerminalNodeChatHistoryResult",
     "TerminalNodeJsonResult",
+    "TerminalNodeNumberResult",
     "TerminalNodeResult",
     "TerminalNodeResultData",
     "TerminalNodeResultOutput",
     "TerminalNodeResultOutput_ChatHistory",
     "TerminalNodeResultOutput_Json",
+    "TerminalNodeResultOutput_Number",
     "TerminalNodeResultOutput_SearchResults",
     "TerminalNodeResultOutput_String",
     "TerminalNodeSearchResultsResult",
@@ -315,6 +332,7 @@ __all__ = [
     "WorkflowExecutionNodeResultEvent",
     "WorkflowExecutionWorkflowResultEvent",
     "WorkflowNodeResultData",
+    "WorkflowNodeResultData_Api",
     "WorkflowNodeResultData_Conditional",
     "WorkflowNodeResultData_Prompt",
     "WorkflowNodeResultData_Search",
@@ -333,10 +351,12 @@ __all__ = [
     "WorkflowResultEventOutputData",
     "WorkflowResultEventOutputDataChatHistory",
     "WorkflowResultEventOutputDataJson",
+    "WorkflowResultEventOutputDataNumber",
     "WorkflowResultEventOutputDataSearchResults",
     "WorkflowResultEventOutputDataString",
     "WorkflowResultEventOutputData_ChatHistory",
     "WorkflowResultEventOutputData_Json",
+    "WorkflowResultEventOutputData_Number",
     "WorkflowResultEventOutputData_SearchResults",
     "WorkflowResultEventOutputData_String",
     "WorkflowStreamEvent",