vellum-ai 0.14.81__tar.gz → 0.14.82__tar.gz

{vellum_ai-0.14.81 → vellum_ai-0.14.82}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.14.81
+Version: 0.14.82
 Summary: 
 License: MIT
 Requires-Python: >=3.9,<4.0

{vellum_ai-0.14.81 → vellum_ai-0.14.82}/ee/vellum_ee/workflows/display/nodes/vellum/tests/test_tool_calling_node.py

@@ -1,8 +1,7 @@
-from vellum.client.types.code_execution_package import CodeExecutionPackage
 from vellum.workflows import BaseWorkflow
 from vellum.workflows.inputs import BaseInputs
 from vellum.workflows.nodes.displayable.inline_prompt_node.node import InlinePromptNode
-from vellum.workflows.nodes.experimental.tool_calling_node.node import ToolCallingNode
+from vellum.workflows.nodes.displayable.tool_calling_node.node import ToolCallingNode
 from vellum.workflows.state.base import BaseState
 from vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import get_workflow_display
 
@@ -123,105 +122,3 @@ def test_serialize_node__prompt_inputs__mixed_values():
             ],
         },
     }
-
-
-def test_serialize_node__function_configs():
-    # GIVEN a tool calling node with function packages
-    def foo():
-        pass
-
-    def bar():
-        pass
-
-    class MyToolCallingNode(ToolCallingNode):
-        functions = [foo, bar]
-        function_configs = {
-            "foo": {
-                "runtime": "PYTHON_3_11_6",
-                "packages": [
-                    CodeExecutionPackage(name="first_package", version="1.0.0", repository="test-repo"),
-                    CodeExecutionPackage(name="second_package", version="2.0.0", repository="test-repo"),
-                ],
-            },
-            "bar": {
-                "runtime": "PYTHON_3_11_6",
-                "packages": [CodeExecutionPackage(name="third_package", version="3.0.0")],
-            },
-        }
-
-    # AND a workflow with the tool calling node
-    class Workflow(BaseWorkflow):
-        graph = MyToolCallingNode
-
-    # WHEN the workflow is serialized
-    workflow_display = get_workflow_display(workflow_class=Workflow)
-    serialized_workflow: dict = workflow_display.serialize()
-
-    # THEN the node should properly serialize the function packages
-    my_tool_calling_node = next(
-        node
-        for node in serialized_workflow["workflow_raw_data"]["nodes"]
-        if node["id"] == str(MyToolCallingNode.__id__)
-    )
-
-    function_configs_attribute = next(
-        attribute for attribute in my_tool_calling_node["attributes"] if attribute["name"] == "function_configs"
-    )
-
-    assert function_configs_attribute == {
-        "id": "90cc5fc7-9fb3-450f-be5a-e90e7412a601",
-        "name": "function_configs",
-        "value": {
-            "type": "CONSTANT_VALUE",
-            "value": {
-                "type": "JSON",
-                "value": {
-                    "foo": {
-                        "runtime": "PYTHON_3_11_6",
-                        "packages": [
-                            {"version": "1.0.0", "name": "first_package", "repository": "test-repo"},
-                            {"version": "2.0.0", "name": "second_package", "repository": "test-repo"},
-                        ],
-                    },
-                    "bar": {
-                        "runtime": "PYTHON_3_11_6",
-                        "packages": [{"version": "3.0.0", "name": "third_package", "repository": None}],
-                    },
-                },
-            },
-        },
-    }
-
-
-def test_serialize_node__function_configs__none():
-    # GIVEN a tool calling node with no function configs
-    def foo():
-        pass
-
-    class MyToolCallingNode(ToolCallingNode):
-        functions = [foo]
-
-    # AND a workflow with the tool calling node
-    class Workflow(BaseWorkflow):
-        graph = MyToolCallingNode
-
-    # WHEN the workflow is serialized
-    workflow_display = get_workflow_display(workflow_class=Workflow)
-    serialized_workflow: dict = workflow_display.serialize()
-
-    # THEN the node should properly serialize the functions
-    my_tool_calling_node = next(
-        node
-        for node in serialized_workflow["workflow_raw_data"]["nodes"]
-        if node["id"] == str(MyToolCallingNode.__id__)
-    )
-
-    function_configs_attribute = next(
-        attribute for attribute in my_tool_calling_node["attributes"] if attribute["name"] == "function_configs"
-    )
-
-    assert function_configs_attribute == {
-        "id": "5f63f2aa-72d7-47ad-a552-630753418b7d",
-        "name": "function_configs",
-        "value": {"type": "CONSTANT_VALUE", "value": {"type": "JSON", "value": None}},
-    }

{vellum_ai-0.14.81 → vellum_ai-0.14.82}/ee/vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_inline_workflow_serialization.py

@@ -48,7 +48,7 @@ def test_serialize_workflow():
         },
         "base": {
             "name": "ToolCallingNode",
-            "module": ["vellum", "workflows", "nodes", "experimental", "tool_calling_node", "node"],
+            "module": ["vellum", "workflows", "nodes", "displayable", "tool_calling_node", "node"],
         },
         "definition": {
             "name": "GetCurrentWeatherNode",
@@ -407,11 +407,6 @@ def test_serialize_workflow():
                     ],
                 },
             },
-            {
-                "id": "a4e3bc9f-7112-4d2f-94fb-7362a85db27a",
-                "name": "function_configs",
-                "value": {"type": "CONSTANT_VALUE", "value": {"type": "JSON", "value": None}},
-            },
             {
                 "id": "1668419e-a193-43a5-8a97-3394e89bf278",
                 "name": "max_prompt_iterations",

{vellum_ai-0.14.81 → vellum_ai-0.14.82}/ee/vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_serialization.py

@@ -48,7 +48,7 @@ def test_serialize_workflow():
         },
         "base": {
             "name": "ToolCallingNode",
-            "module": ["vellum", "workflows", "nodes", "experimental", "tool_calling_node", "node"],
+            "module": ["vellum", "workflows", "nodes", "displayable", "tool_calling_node", "node"],
         },
         "definition": {
             "name": "GetCurrentWeatherNode",
@@ -152,22 +152,6 @@ def test_serialize_workflow():
                     },
                 },
             },
-            {
-                "id": "a4e3bc9f-7112-4d2f-94fb-7362a85db27a",
-                "name": "function_configs",
-                "value": {
-                    "type": "CONSTANT_VALUE",
-                    "value": {
-                        "type": "JSON",
-                        "value": {
-                            "get_current_weather": {
-                                "runtime": "PYTHON_3_11_6",
-                                "packages": [{"version": "2.26.0", "name": "requests", "repository": None}],
-                            }
-                        },
-                    },
-                },
-            },
             {
                 "id": "0f6dc102-3460-4963-91fa-7ba85d65ef7a",
                 "name": "prompt_inputs",

{vellum_ai-0.14.81 → vellum_ai-0.14.82}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-ai"
 
 [tool.poetry]
 name = "vellum-ai"
-version = "0.14.81"
+version = "0.14.82"
 description = ""
 readme = "README.md"
 authors = []

{vellum_ai-0.14.81 → vellum_ai-0.14.82}/src/vellum/__init__.py

@@ -121,6 +121,7 @@ from .types import (
     ExecutionNumberVellumValue,
     ExecutionSearchResultsVellumValue,
     ExecutionStringVellumValue,
+    ExecutionThinkingVellumValue,
     ExecutionVellumValue,
     ExternalInputDescriptor,
     ExternalTestCaseExecution,
@@ -487,6 +488,8 @@ from .types import (
     TestSuiteTestCaseReplacedBulkResult,
     TestSuiteTestCaseReplacedBulkResultData,
     TestSuiteTestCaseUpsertBulkOperationRequest,
+    ThinkingVellumValue,
+    ThinkingVellumValueRequest,
     TokenOverlappingWindowChunkerConfig,
     TokenOverlappingWindowChunkerConfigRequest,
     TokenOverlappingWindowChunking,
@@ -754,6 +757,7 @@ __all__ = [
     "ExecutionNumberVellumValue",
     "ExecutionSearchResultsVellumValue",
     "ExecutionStringVellumValue",
+    "ExecutionThinkingVellumValue",
     "ExecutionVellumValue",
     "ExternalInputDescriptor",
     "ExternalTestCaseExecution",
@@ -1127,6 +1131,8 @@ __all__ = [
     "TestSuiteTestCaseReplacedBulkResult",
     "TestSuiteTestCaseReplacedBulkResultData",
     "TestSuiteTestCaseUpsertBulkOperationRequest",
+    "ThinkingVellumValue",
+    "ThinkingVellumValueRequest",
     "TokenOverlappingWindowChunkerConfig",
     "TokenOverlappingWindowChunkerConfigRequest",
     "TokenOverlappingWindowChunking",

{vellum_ai-0.14.81 → vellum_ai-0.14.82}/src/vellum/client/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.14.81",
+            "X-Fern-SDK-Version": "0.14.82",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

{vellum_ai-0.14.81 → vellum_ai-0.14.82}/src/vellum/client/types/__init__.py

@@ -125,6 +125,7 @@ from .execution_json_vellum_value import ExecutionJsonVellumValue
 from .execution_number_vellum_value import ExecutionNumberVellumValue
 from .execution_search_results_vellum_value import ExecutionSearchResultsVellumValue
 from .execution_string_vellum_value import ExecutionStringVellumValue
+from .execution_thinking_vellum_value import ExecutionThinkingVellumValue
 from .execution_vellum_value import ExecutionVellumValue
 from .external_input_descriptor import ExternalInputDescriptor
 from .external_test_case_execution import ExternalTestCaseExecution
@@ -511,6 +512,8 @@ from .test_suite_test_case_replace_bulk_operation_request import TestSuiteTestCa
 from .test_suite_test_case_replaced_bulk_result import TestSuiteTestCaseReplacedBulkResult
 from .test_suite_test_case_replaced_bulk_result_data import TestSuiteTestCaseReplacedBulkResultData
 from .test_suite_test_case_upsert_bulk_operation_request import TestSuiteTestCaseUpsertBulkOperationRequest
+from .thinking_vellum_value import ThinkingVellumValue
+from .thinking_vellum_value_request import ThinkingVellumValueRequest
 from .token_overlapping_window_chunker_config import TokenOverlappingWindowChunkerConfig
 from .token_overlapping_window_chunker_config_request import TokenOverlappingWindowChunkerConfigRequest
 from .token_overlapping_window_chunking import TokenOverlappingWindowChunking
@@ -741,6 +744,7 @@ __all__ = [
     "ExecutionNumberVellumValue",
     "ExecutionSearchResultsVellumValue",
     "ExecutionStringVellumValue",
+    "ExecutionThinkingVellumValue",
     "ExecutionVellumValue",
     "ExternalInputDescriptor",
     "ExternalTestCaseExecution",
@@ -1107,6 +1111,8 @@ __all__ = [
     "TestSuiteTestCaseReplacedBulkResult",
     "TestSuiteTestCaseReplacedBulkResultData",
     "TestSuiteTestCaseUpsertBulkOperationRequest",
+    "ThinkingVellumValue",
+    "ThinkingVellumValueRequest",
     "TokenOverlappingWindowChunkerConfig",
     "TokenOverlappingWindowChunkerConfigRequest",
     "TokenOverlappingWindowChunking",

vellum_ai-0.14.82/src/vellum/client/types/execution_thinking_vellum_value.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import pydantic
+import typing
+from .string_vellum_value import StringVellumValue
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class ExecutionThinkingVellumValue(UniversalBaseModel):
+    """
+    A value representing Thinking mode output.
+    """
+
+    id: str = pydantic.Field()
+    """
+    The variable's uniquely identifying internal id.
+    """
+
+    name: str
+    type: typing.Literal["THINKING"] = "THINKING"
+    value: typing.List[StringVellumValue]
+
+    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_ai-0.14.81 → vellum_ai-0.14.82}/src/vellum/client/types/execution_vellum_value.py

@@ -9,6 +9,7 @@ from .execution_search_results_vellum_value import ExecutionSearchResultsVellumV
 from .execution_error_vellum_value import ExecutionErrorVellumValue
 from .execution_array_vellum_value import ExecutionArrayVellumValue
 from .execution_function_call_vellum_value import ExecutionFunctionCallVellumValue
+from .execution_thinking_vellum_value import ExecutionThinkingVellumValue
 
 ExecutionVellumValue = typing.Union[
     ExecutionStringVellumValue,
@@ -19,4 +20,5 @@ ExecutionVellumValue = typing.Union[
     ExecutionErrorVellumValue,
     ExecutionArrayVellumValue,
     ExecutionFunctionCallVellumValue,
+    ExecutionThinkingVellumValue,
 ]

{vellum_ai-0.14.81 → vellum_ai-0.14.82}/src/vellum/client/types/prompt_output.py

@@ -5,5 +5,8 @@ from .string_vellum_value import StringVellumValue
 from .json_vellum_value import JsonVellumValue
 from .error_vellum_value import ErrorVellumValue
 from .function_call_vellum_value import FunctionCallVellumValue
+from .thinking_vellum_value import ThinkingVellumValue
 
-PromptOutput = typing.Union[StringVellumValue, JsonVellumValue, ErrorVellumValue, FunctionCallVellumValue]
+PromptOutput = typing.Union[
+    StringVellumValue, JsonVellumValue, ErrorVellumValue, FunctionCallVellumValue, ThinkingVellumValue
+]

vellum_ai-0.14.82/src/vellum/client/types/thinking_vellum_value.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .string_vellum_value import StringVellumValue
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class ThinkingVellumValue(UniversalBaseModel):
+    """
+    A value representing Thinking mode output.
+    """
+
+    type: typing.Literal["THINKING"] = "THINKING"
+    value: typing.List[StringVellumValue]
+
+    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_ai-0.14.82/src/vellum/client/types/thinking_vellum_value_request.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .string_vellum_value_request import StringVellumValueRequest
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class ThinkingVellumValueRequest(UniversalBaseModel):
+    """
+    A value representing Thinking mode output.
+    """
+
+    type: typing.Literal["THINKING"] = "THINKING"
+    value: typing.List[StringVellumValueRequest]
+
+    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_ai-0.14.81 → vellum_ai-0.14.82}/src/vellum/client/types/vellum_value.py

@@ -12,6 +12,7 @@ from .function_call_vellum_value import FunctionCallVellumValue
 from .error_vellum_value import ErrorVellumValue
 from .chat_history_vellum_value import ChatHistoryVellumValue
 from .search_results_vellum_value import SearchResultsVellumValue
+from .thinking_vellum_value import ThinkingVellumValue
 import typing
 
 if typing.TYPE_CHECKING:
@@ -28,4 +29,5 @@ VellumValue = typing.Union[
     "ArrayVellumValue",
     ChatHistoryVellumValue,
     SearchResultsVellumValue,
+    ThinkingVellumValue,
 ]

{vellum_ai-0.14.81 → vellum_ai-0.14.82}/src/vellum/client/types/vellum_value_request.py

@@ -12,6 +12,7 @@ from .function_call_vellum_value_request import FunctionCallVellumValueRequest
 from .error_vellum_value_request import ErrorVellumValueRequest
 from .chat_history_vellum_value_request import ChatHistoryVellumValueRequest
 from .search_results_vellum_value_request import SearchResultsVellumValueRequest
+from .thinking_vellum_value_request import ThinkingVellumValueRequest
 import typing
 
 if typing.TYPE_CHECKING:
@@ -28,4 +29,5 @@ VellumValueRequest = typing.Union[
     "ArrayVellumValueRequest",
     ChatHistoryVellumValueRequest,
     SearchResultsVellumValueRequest,
+    ThinkingVellumValueRequest,
 ]

{vellum_ai-0.14.81 → vellum_ai-0.14.82}/src/vellum/client/types/vellum_variable_type.py

@@ -16,6 +16,7 @@ VellumVariableType = typing.Union[
         "AUDIO",
         "DOCUMENT",
         "NULL",
+        "THINKING",
     ],
     typing.Any,
 ]

vellum_ai-0.14.82/src/vellum/types/execution_thinking_vellum_value.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.execution_thinking_vellum_value import *

vellum_ai-0.14.82/src/vellum/types/thinking_vellum_value.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.thinking_vellum_value import *

vellum_ai-0.14.82/src/vellum/types/thinking_vellum_value_request.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.thinking_vellum_value_request import *

{vellum_ai-0.14.81 → vellum_ai-0.14.82}/src/vellum/workflows/nodes/displayable/__init__.py

@@ -13,6 +13,7 @@ from .note_node import NoteNode
 from .prompt_deployment_node import PromptDeploymentNode
 from .search_node import SearchNode
 from .subworkflow_deployment_node import SubworkflowDeploymentNode
+from .tool_calling_node import ToolCallingNode
 
 __all__ = [
     "APINode",
@@ -29,5 +30,6 @@ __all__ = [
     "PromptDeploymentNode",
     "SearchNode",
     "TemplatingNode",
+    "ToolCallingNode",
     "FinalOutputNode",
 ]

{vellum_ai-0.14.81 → vellum_ai-0.14.82}/src/vellum/workflows/nodes/displayable/inline_prompt_node/node.py

@@ -63,6 +63,9 @@ class InlinePromptNode(BaseInlinePromptNode[StateType]):
                 json_output = output.value
             elif output.type == "FUNCTION_CALL":
                 string_outputs.append(output.value.model_dump_json(indent=4))
+            elif output.type == "THINKING":
+                thinking_strings = [item.value for item in output.value if item.value is not None]
+                string_outputs.append("\n".join(thinking_strings))
             else:
                 string_outputs.append(output.value.message)
 

{vellum_ai-0.14.81 → vellum_ai-0.14.82}/src/vellum/workflows/nodes/displayable/prompt_deployment_node/node.py

@@ -65,6 +65,9 @@ class PromptDeploymentNode(BasePromptDeploymentNode[StateType]):
                 string_outputs.append(json.dumps(output.value, indent=4))
             elif output.type == "FUNCTION_CALL":
                 string_outputs.append(output.value.model_dump_json(indent=4))
+            elif output.type == "THINKING":
+                thinking_strings = [item.value for item in output.value if item.value is not None]
+                string_outputs.append("\n".join(thinking_strings))
             else:
                 string_outputs.append(output.value.message)
 

vellum_ai-0.14.82/src/vellum/workflows/nodes/displayable/tool_calling_node/__init__.py

@@ -0,0 +1,3 @@
+from vellum.workflows.nodes.displayable.tool_calling_node.node import ToolCallingNode
+
+__all__ = ["ToolCallingNode"]

{vellum_ai-0.14.81/src/vellum/workflows/nodes/experimental → vellum_ai-0.14.82/src/vellum/workflows/nodes/displayable}/tool_calling_node/node.py

@@ -1,16 +1,13 @@
-from collections.abc import Sequence
-from typing import Any, ClassVar, Dict, List, Optional, cast
+from typing import ClassVar, List, Optional
 
 from vellum import ChatMessage, PromptBlock
-from vellum.client.types.code_execution_package import CodeExecutionPackage
-from vellum.client.types.code_execution_runtime import CodeExecutionRuntime
 from vellum.workflows.context import execution_context, get_parent_context
 from vellum.workflows.errors.types import WorkflowErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.graph.graph import Graph
 from vellum.workflows.inputs.base import BaseInputs
 from vellum.workflows.nodes.bases import BaseNode
-from vellum.workflows.nodes.experimental.tool_calling_node.utils import (
+from vellum.workflows.nodes.displayable.tool_calling_node.utils import (
     create_function_node,
     create_tool_router_node,
     get_function_name,
@@ -19,7 +16,6 @@ from vellum.workflows.outputs.base import BaseOutputs
 from vellum.workflows.state.base import BaseState
 from vellum.workflows.state.context import WorkflowContext
 from vellum.workflows.types.core import EntityInputsInterface, Tool
-from vellum.workflows.workflows.base import BaseWorkflow
 
 
 class ToolCallingNode(BaseNode):
@@ -31,7 +27,6 @@ class ToolCallingNode(BaseNode):
         blocks: List[PromptBlock] - The prompt blocks to use (same format as InlinePromptNode)
         functions: List[Tool] - The functions that can be called
         prompt_inputs: Optional[EntityInputsInterface] - Mapping of input variable names to values
-        function_configs: Optional[Dict[str, Dict[str, Any]]] - Mapping of function names to their configuration
         max_prompt_iterations: Optional[int] - Maximum number of prompt iterations before stopping
     """
 
@@ -39,7 +34,6 @@ class ToolCallingNode(BaseNode):
     blocks: ClassVar[List[PromptBlock]] = []
     functions: ClassVar[List[Tool]] = []
     prompt_inputs: ClassVar[Optional[EntityInputsInterface]] = None
-    function_configs: ClassVar[Optional[Dict[str, Dict[str, Any]]]] = None
     max_prompt_iterations: ClassVar[Optional[int]] = 5
 
     class Outputs(BaseOutputs):
@@ -69,6 +63,8 @@ class ToolCallingNode(BaseNode):
                 chat_history: List[ChatMessage] = []
                 prompt_iterations: int = 0
 
+            from vellum.workflows.workflows.base import BaseWorkflow
+
             class ToolCallingWorkflow(BaseWorkflow[BaseInputs, ToolCallingState]):
                 graph = self._graph
 
@@ -112,23 +108,10 @@ class ToolCallingNode(BaseNode):
         self._function_nodes = {}
         for function in self.functions:
             function_name = get_function_name(function)
-            # Get configuration for this function
-            config = {}
-            if callable(function) and self.function_configs and function.__name__ in self.function_configs:
-                config = self.function_configs[function.__name__]
-
-            packages = config.get("packages", None)
-            if packages is not None:
-                packages = cast(Sequence[CodeExecutionPackage], packages)
-
-            runtime_raw = config.get("runtime", "PYTHON_3_11_6")
-            runtime = cast(CodeExecutionRuntime, runtime_raw)
 
             self._function_nodes[function_name] = create_function_node(
                 function=function,
                 tool_router_node=self.tool_router_node,
-                packages=packages,
-                runtime=runtime,
             )
 
         graph_set = set()

{vellum_ai-0.14.81/src/vellum/workflows/nodes/experimental → vellum_ai-0.14.82/src/vellum/workflows/nodes/displayable}/tool_calling_node/tests/test_node.py

@@ -8,7 +8,7 @@ from vellum.client.types.string_chat_message_content import StringChatMessageCon
 from vellum.workflows import BaseWorkflow
 from vellum.workflows.inputs.base import BaseInputs
 from vellum.workflows.nodes.bases import BaseNode
-from vellum.workflows.nodes.experimental.tool_calling_node.utils import create_function_node, create_tool_router_node
+from vellum.workflows.nodes.displayable.tool_calling_node.utils import create_function_node, create_tool_router_node
 from vellum.workflows.outputs.base import BaseOutputs
 from vellum.workflows.state.base import BaseState, StateMeta
 from vellum.workflows.state.context import WorkflowContext

{vellum_ai-0.14.81/src/vellum/workflows/nodes/experimental → vellum_ai-0.14.82/src/vellum/workflows/nodes/displayable}/tool_calling_node/tests/test_utils.py

@@ -1,7 +1,7 @@
 from vellum.workflows import BaseWorkflow
 from vellum.workflows.inputs.base import BaseInputs
 from vellum.workflows.nodes.bases import BaseNode
-from vellum.workflows.nodes.experimental.tool_calling_node.utils import get_function_name
+from vellum.workflows.nodes.displayable.tool_calling_node.utils import get_function_name
 from vellum.workflows.outputs.base import BaseOutputs
 from vellum.workflows.state.base import BaseState
 from vellum.workflows.types.definition import DeploymentDefinition