vellum-ai 0.14.63__tar.gz → 0.14.64__tar.gz

{vellum_ai-0.14.63 → vellum_ai-0.14.64}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.14.63
+Version: 0.14.64
 Summary: 
 License: MIT
 Requires-Python: >=3.9,<4.0
@@ -20,22 +20,21 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Typing :: Typed
-Requires-Dist: Jinja2 (>=3.1.2,<4.0.0)
-Requires-Dist: cdktf (>=0.20.5,<0.21.0)
-Requires-Dist: click (==8.1.7)
-Requires-Dist: docker (==7.1.0)
+Requires-Dist: Jinja2 (>=3.1.0,<4.0.0)
+Requires-Dist: click (>=8.1.0,<9.0.0)
+Requires-Dist: docker (>=7.1.0,<8.0.0)
 Requires-Dist: httpx (>=0.21.2)
-Requires-Dist: openai (>=1.0.0)
+Requires-Dist: openai (>=1.0.0,<2.0.0)
 Requires-Dist: orderly-set (>=5.2.2,<6.0.0)
 Requires-Dist: publication (==0.0.3)
 Requires-Dist: pydantic (>=1.9.2)
 Requires-Dist: pydantic-core (>=2.18.2,<3.0.0)
-Requires-Dist: pydash (==7.0.6)
-Requires-Dist: python-dotenv (==1.0.1)
-Requires-Dist: pytz (==2025.1)
-Requires-Dist: pyyaml (==6.0.2)
-Requires-Dist: requests (==2.32.3)
-Requires-Dist: tomli (==2.0.2)
+Requires-Dist: pydash (>=7.0.0,<8.0.0)
+Requires-Dist: python-dotenv (>=1.0.0,<2.0.0)
+Requires-Dist: pytz (>=2022.0,<2026.0)
+Requires-Dist: pyyaml (>=6.0.0,<7.0.0)
+Requires-Dist: requests (>=2.31.0,<3.0.0)
+Requires-Dist: tomli (>=2.0.0,<3.0.0)
 Requires-Dist: typing_extensions (>=4.0.0)
 Project-URL: Repository, https://github.com/vellum-ai/vellum-python-sdks
 Description-Content-Type: text/markdown

{vellum_ai-0.14.63 → vellum_ai-0.14.64}/ee/vellum_ee/workflows/display/nodes/vellum/map_node.py

@@ -3,6 +3,7 @@ from typing import Generic, Optional, TypeVar, cast
 
 from vellum.workflows.nodes import MapNode
 from vellum.workflows.types.core import JsonObject
+from vellum.workflows.workflows.base import BaseWorkflow
 from vellum_ee.workflows.display.nodes.utils import raise_if_descriptor
 from vellum_ee.workflows.display.nodes.vellum.base_adornment_node import BaseAdornmentNodeDisplay
 from vellum_ee.workflows.display.nodes.vellum.utils import create_node_input
@@ -21,7 +22,7 @@ class BaseMapNodeDisplay(BaseAdornmentNodeDisplay[_MapNodeType], Generic[_MapNod
         node = self._node
         node_id = self.node_id
 
-        subworkflow = raise_if_descriptor(node.subworkflow)
+        subworkflow = cast(type[BaseWorkflow], raise_if_descriptor(node.subworkflow))
 
         items_node_input = create_node_input(
             node_id=node_id,

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

@@ -125,7 +125,7 @@ def test_serialize_node__prompt_inputs__mixed_values():
     }
 
 
-def test_serialize_node__function_packages():
+def test_serialize_node__function_configs():
     # GIVEN a tool calling node with function packages
     def foo():
         pass
@@ -135,14 +135,18 @@ def test_serialize_node__function_packages():
 
     class MyToolCallingNode(ToolCallingNode):
         functions = [foo, bar]
-        function_packages = {
-            foo: [
-                CodeExecutionPackage(name="first_package", version="1.0.0"),
-                CodeExecutionPackage(name="second_package", version="2.0.0"),
-            ],
-            bar: [
-                CodeExecutionPackage(name="third_package", version="3.0.0"),
-            ],
+        function_configs = {
+            "foo": {
+                "runtime": "PYTHON_3_11_6",
+                "packages": [
+                    CodeExecutionPackage(name="first_package", version="1.0.0"),
+                    CodeExecutionPackage(name="second_package", version="2.0.0"),
+                ],
+            },
+            "bar": {
+                "runtime": "PYTHON_3_11_6",
+                "packages": [CodeExecutionPackage(name="third_package", version="3.0.0")],
+            },
         }
 
     # AND a workflow with the tool calling node
@@ -160,31 +164,34 @@ def test_serialize_node__function_packages():
         if node["id"] == str(MyToolCallingNode.__id__)
     )
 
-    function_packages_attribute = next(
-        attribute for attribute in my_tool_calling_node["attributes"] if attribute["name"] == "function_packages"
+    function_configs_attribute = next(
+        attribute for attribute in my_tool_calling_node["attributes"] if attribute["name"] == "function_configs"
     )
 
-    assert function_packages_attribute == {
-        "id": "c315228f-870a-4288-abf7-a217b5b8c253",
-        "name": "function_packages",
+    assert function_configs_attribute == {
+        "id": "90cc5fc7-9fb3-450f-be5a-e90e7412a601",
+        "name": "function_configs",
         "value": {
             "type": "CONSTANT_VALUE",
             "value": {
                 "type": "JSON",
                 "value": {
-                    "foo": [
-                        {"version": "1.0.0", "name": "first_package"},
-                        {"version": "2.0.0", "name": "second_package"},
-                    ],
-                    "bar": [{"version": "3.0.0", "name": "third_package"}],
+                    "foo": {
+                        "runtime": "PYTHON_3_11_6",
+                        "packages": [
+                            {"version": "1.0.0", "name": "first_package"},
+                            {"version": "2.0.0", "name": "second_package"},
+                        ],
+                    },
+                    "bar": {"runtime": "PYTHON_3_11_6", "packages": [{"version": "3.0.0", "name": "third_package"}]},
                 },
             },
         },
     }
 
 
-def test_serialize_node__none_function_packages():
-    # GIVEN a tool calling node with no function packages
+def test_serialize_node__function_configs__none():
+    # GIVEN a tool calling node with no function configs
     def foo():
         pass
 
@@ -206,12 +213,12 @@ def test_serialize_node__none_function_packages():
         if node["id"] == str(MyToolCallingNode.__id__)
     )
 
-    function_packages_attribute = next(
-        attribute for attribute in my_tool_calling_node["attributes"] if attribute["name"] == "function_packages"
+    function_configs_attribute = next(
+        attribute for attribute in my_tool_calling_node["attributes"] if attribute["name"] == "function_configs"
     )
 
-    assert function_packages_attribute == {
-        "id": "05c86dea-ea4b-46c4-883b-7912a6680491",
-        "name": "function_packages",
+    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.63 → vellum_ai-0.14.64}/ee/vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_serialization.py

@@ -150,6 +150,22 @@ 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"}],
+                            }
+                        },
+                    },
+                },
+            },
             {
                 "id": "0f6dc102-3460-4963-91fa-7ba85d65ef7a",
                 "name": "prompt_inputs",
@@ -167,16 +183,6 @@ def test_serialize_workflow():
                     ],
                 },
             },
-            {
-                "id": "b31575f0-633b-4bd0-ba6a-960532f3887a",
-                "name": "function_packages",
-                "value": {"type": "CONSTANT_VALUE", "value": {"type": "JSON", "value": None}},
-            },
-            {
-                "id": "d316c45c-97ea-4f39-9a2d-5c43e2d355de",
-                "name": "runtime",
-                "value": {"type": "CONSTANT_VALUE", "value": {"type": "STRING", "value": "PYTHON_3_11_6"}},
-            },
         ],
         "outputs": [
             {"id": "e62bc785-a914-4066-b79e-8c89a5d0ec6c", "name": "text", "type": "STRING", "value": None},

{vellum_ai-0.14.63 → vellum_ai-0.14.64}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-ai"
 
 [tool.poetry]
 name = "vellum-ai"
-version = "0.14.63"
+version = "0.14.64"
 description = ""
 readme = "README.md"
 authors = []
@@ -36,22 +36,21 @@ packages = [
 
 [tool.poetry.dependencies]
 python = "^3.9"
-Jinja2 = "^3.1.2"
-cdktf = "^0.20.5"
-click = "8.1.7"
-docker = "7.1.0"
+Jinja2 = ">=3.1.0,<4.0.0"
+click = ">=8.1.0,<9.0.0"
+docker = ">=7.1.0,<8.0.0"
 httpx = ">=0.21.2"
-openai = ">= 1.0.0"
+openai = ">=1.0.0,<2.0.0"
 orderly-set = "^5.2.2"
 publication = "0.0.3"
 pydantic = ">= 1.9.2"
 pydantic-core = "^2.18.2"
-pydash = "7.0.6"
-python-dotenv = "1.0.1"
-pytz = "2025.1"
-pyyaml = "6.0.2"
-requests = "2.32.3"
-tomli = "2.0.2"
+pydash = ">=7.0.0,<8.0.0"
+python-dotenv = ">=1.0.0,<2.0.0"
+pytz = ">=2022.0,<2026.0"
+pyyaml = ">=6.0.0,<7.0.0"
+requests = ">=2.31.0,<3.0.0"
+tomli = ">=2.0.0,<3.0.0"
 typing_extensions = ">= 4.0.0"
 
 [tool.poetry.dev-dependencies]

{vellum_ai-0.14.63 → vellum_ai-0.14.64}/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.63",
+            "X-Fern-SDK-Version": "0.14.64",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

{vellum_ai-0.14.63 → vellum_ai-0.14.64}/src/vellum/workflows/nodes/bases/base.py

@@ -1,3 +1,4 @@
+from abc import ABC, ABCMeta
 from dataclasses import field
 from functools import cached_property, reduce
 import inspect
@@ -50,7 +51,7 @@ def _is_annotated(cls: Type, name: str) -> bool:
     return False
 
 
-class BaseNodeMeta(type):
+class BaseNodeMeta(ABCMeta):
     def __new__(mcs, name: str, bases: Tuple[Type, ...], dct: Dict[str, Any]) -> Any:
         if "Outputs" in dct:
             outputs_class = dct["Outputs"]
@@ -258,7 +259,7 @@ class _BaseNodeExecutionMeta(type):
 NodeRunResponse = Union[BaseOutputs, Iterator[BaseOutput]]
 
 
-class BaseNode(Generic[StateType], metaclass=BaseNodeMeta):
+class BaseNode(Generic[StateType], ABC, metaclass=BaseNodeMeta):
     __id__: UUID = uuid4_from_hash(__qualname__)
     __output_ids__: Dict[str, UUID] = {}
     state: StateType
@@ -490,7 +491,6 @@ class BaseNode(Generic[StateType], metaclass=BaseNodeMeta):
                         node_output_descriptor.name,
                         node_output_descriptor.instance.resolve(self.state),
                     )
-            delattr(self.Outputs, "_outputs_post_init")
 
         setattr(self.Outputs, "_outputs_post_init", _outputs_post_init)
 

{vellum_ai-0.14.63 → vellum_ai-0.14.64}/src/vellum/workflows/nodes/bases/base_adornment_node.py

@@ -1,3 +1,4 @@
+from abc import ABC
 from typing import TYPE_CHECKING, Any, Dict, Generic, Optional, Tuple, Type
 
 from vellum.workflows.inputs.base import BaseInputs
@@ -67,6 +68,7 @@ class _BaseAdornmentNodeMeta(BaseNodeMeta):
 class BaseAdornmentNode(
     BaseNode[StateType],
     Generic[StateType],
+    ABC,
     metaclass=_BaseAdornmentNodeMeta,
 ):
     """

{vellum_ai-0.14.63 → vellum_ai-0.14.64}/src/vellum/workflows/nodes/displayable/subworkflow_deployment_node/node.py

@@ -63,6 +63,10 @@ class SubworkflowDeploymentNode(BaseNode[StateType], Generic[StateType]):
         compiled_inputs: List[WorkflowRequestInputRequest] = []
 
         for input_name, input_value in self.subworkflow_inputs.items():
+            # Exclude inputs that resolved to be null. This ensure that we don't pass input values
+            # to optional subworkflow inputs whose values were unresolved.
+            if input_value is None:
+                continue
             if isinstance(input_value, str):
                 compiled_inputs.append(
                     WorkflowRequestStringInputRequest(

{vellum_ai-0.14.63 → vellum_ai-0.14.64}/src/vellum/workflows/nodes/experimental/tool_calling_node/node.py

@@ -1,10 +1,11 @@
-from collections.abc import Callable
+from collections.abc import Callable, Sequence
 from typing import Any, ClassVar, Dict, List, Optional, cast
 
 from pydash import snake_case
 
 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
@@ -29,16 +30,14 @@ class ToolCallingNode(BaseNode):
         functions: List[FunctionDefinition] - The functions that can be called
         function_callables: List[Callable] - The callables that can be called
         prompt_inputs: Optional[EntityInputsInterface] - Mapping of input variable names to values
-        function_packages: Optional[Dict[Callable, List[CodeExecutionPackage]]] - Mapping of functions to their packages
-        runtime: str - The runtime to use for code execution (default: "PYTHON_3_11_6")
+        function_configs: Optional[Dict[str, Dict[str, Any]]] - Mapping of function names to their configuration
     """
 
     ml_model: ClassVar[str] = "gpt-4o-mini"
     blocks: ClassVar[List[PromptBlock]] = []
     functions: ClassVar[List[Callable[..., Any]]] = []
     prompt_inputs: ClassVar[Optional[EntityInputsInterface]] = None
-    function_packages: ClassVar[Optional[Dict[Callable[..., Any], List[CodeExecutionPackage]]]] = None
-    runtime: ClassVar[str] = "PYTHON_3_11_6"
+    function_configs: ClassVar[Optional[Dict[str, Dict[str, Any]]]] = None
 
     class Outputs(BaseOutputs):
         """
@@ -108,17 +107,24 @@ class ToolCallingNode(BaseNode):
         self._function_nodes = {}
         for function in self.functions:
             function_name = snake_case(function.__name__)
-            # Get packages for this function if specified
-            packages = None
-            if self.function_packages and function in self.function_packages:
-                # Cast to tell mypy this is a plain list, not a descriptor
-                packages = cast(List[CodeExecutionPackage], self.function_packages[function])
+
+            # Get configuration for this function
+            config = {}
+            if 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=self.runtime,
+                runtime=runtime,
             )
 
         graph_set = set()

{vellum_ai-0.14.63 → vellum_ai-0.14.64}/src/vellum/workflows/nodes/experimental/tool_calling_node/utils.py

@@ -1,4 +1,4 @@
-from collections.abc import Callable
+from collections.abc import Callable, Sequence
 import inspect
 import json
 import types
@@ -8,6 +8,7 @@ from pydash import snake_case
 
 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.client.types.function_call_chat_message_content import FunctionCallChatMessageContent
 from vellum.client.types.function_call_chat_message_content_value import FunctionCallChatMessageContentValue
 from vellum.client.types.string_chat_message_content import StringChatMessageContent
@@ -128,8 +129,8 @@ def create_tool_router_node(
 def create_function_node(
     function: Callable[..., Any],
     tool_router_node: Type[ToolRouterNode],
-    packages: Optional[List[CodeExecutionPackage]] = None,
-    runtime: str = "PYTHON_3_11_6",
+    packages: Optional[Sequence[CodeExecutionPackage]] = None,
+    runtime: CodeExecutionRuntime = "PYTHON_3_11_6",
 ) -> Type[FunctionNode]:
     """
     Create a FunctionNode class for a given function.
@@ -198,20 +199,17 @@ def create_function_node(
         function_source = inspect.getsource(function)
         function_name = function.__name__
 
-        _code = f'''
+        code = f'''
 {function_source}
 
 def main(arguments):
     """Main function that calls the original function with the provided arguments."""
     return {function_name}(**arguments)
 '''
-        _packages = packages
-        _tool_router_node = tool_router_node
-        _runtime = runtime
 
         def execute_code_execution_function(self) -> BaseNode.Outputs:
             # Get the function call from the tool router output
-            function_call_output = self.state.meta.node_outputs.get(_tool_router_node.Outputs.results)
+            function_call_output = self.state.meta.node_outputs.get(tool_router_node.Outputs.results)
             if function_call_output and len(function_call_output) > 0:
                 function_call = function_call_output[0]
                 arguments = function_call.value.arguments
@@ -246,11 +244,11 @@ def main(arguments):
             {},
             lambda ns: ns.update(
                 {
-                    "code": _code,
+                    "code": code,
                     "code_inputs": {},  # No inputs needed since we handle function call extraction in run()
                     "run": execute_code_execution_function,
-                    "runtime": _runtime,
-                    "packages": _packages,
+                    "runtime": runtime,
+                    "packages": packages,
                     "__module__": __name__,
                 }
             ),