vellum-ai 0.14.55__py3-none-any.whl → 0.14.56__py3-none-any.whl

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.55",
+            "X-Fern-SDK-Version": "0.14.56",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

vellum/workflows/nodes/experimental/tool_calling_node/utils.py

@@ -57,26 +57,32 @@ def create_tool_router_node(
     functions: List[Callable[..., Any]],
     prompt_inputs: Optional[EntityInputsInterface],
 ) -> Type[ToolRouterNode]:
-    Ports = type("Ports", (), {})
-    for function in functions:
-        function_name = function.__name__
-
-        # Avoid using lambda to capture function_name
-        # lambda will capture the function_name by reference,
-        # and if the function_name is changed, the port_condition will also change.
-        def create_port_condition(fn_name):
-            return LazyReference(
-                lambda: (
-                    node.Outputs.results[0]["type"].equals("FUNCTION_CALL")
-                    & node.Outputs.results[0]["value"]["name"].equals(fn_name)
+    if functions and len(functions) > 0:
+        # If we have functions, create dynamic ports for each function
+        Ports = type("Ports", (), {})
+        for function in functions:
+            function_name = function.__name__
+
+            # Avoid using lambda to capture function_name
+            # lambda will capture the function_name by reference,
+            # and if the function_name is changed, the port_condition will also change.
+            def create_port_condition(fn_name):
+                return LazyReference(
+                    lambda: (
+                        node.Outputs.results[0]["type"].equals("FUNCTION_CALL")
+                        & node.Outputs.results[0]["value"]["name"].equals(fn_name)
+                    )
                 )
-            )
 
-        port_condition = create_port_condition(function_name)
-        port = Port.on_if(port_condition)
-        setattr(Ports, function_name, port)
+            port_condition = create_port_condition(function_name)
+            port = Port.on_if(port_condition)
+            setattr(Ports, function_name, port)
 
-    setattr(Ports, "default", Port.on_else())
+        # Add the else port for when no function conditions match
+        setattr(Ports, "default", Port.on_else())
+    else:
+        # If no functions exist, create a simple Ports class with just a default port
+        Ports = type("Ports", (), {"default": Port(default=True)})
 
     # Add a chat history block to blocks
     blocks.append(

vellum/workflows/state/encoder.py

@@ -1,6 +1,7 @@
 from dataclasses import asdict, is_dataclass
 from datetime import datetime
 import enum
+import inspect
 from json import JSONEncoder
 from queue import Queue
 from uuid import UUID
@@ -59,7 +60,13 @@ class DefaultStateEncoder(JSONEncoder):
             return str(obj)
 
         if callable(obj):
-            return compile_function_definition(obj)
+            function_definition = compile_function_definition(obj)
+            try:
+                source_code = inspect.getsource(obj)
+            except Exception:
+                source_code = f"<source code not available for {obj.__name__}>"
+
+            return {"definition": function_definition, "src": source_code}
 
         if obj.__class__ in self.encoders:
             return self.encoders[obj.__class__](obj)

{vellum_ai-0.14.55.dist-info → vellum_ai-0.14.56.dist-info}/METADATA

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

{vellum_ai-0.14.55.dist-info → vellum_ai-0.14.56.dist-info}/RECORD

@@ -89,7 +89,7 @@ vellum_ee/workflows/display/tests/workflow_serialization/test_basic_search_node_
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_subworkflow_deployment_serialization.py,sha256=KkYZc_bZuq1lmDcvUz3QxIqJLpQPCZioD1FHUNsMJY8,11211
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_templating_node_serialization.py,sha256=aZaqRDrkO3ytcmdM2eKJqHSt60MF070NMj6M2vgzOKc,7711
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_terminal_node_serialization.py,sha256=r748dpS13HtwY7t_KQFExFssxRy0xI2d-wxmhiUHRe0,3850
-vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_serialization.py,sha256=Lws9-tWRsbtiS_WqRrO03BLMMRe5_6reXX_XfkkHkJs,7478
+vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_serialization.py,sha256=kEhaRJmwg5rewuGYi4FjJB9D7xEiaO9gvQi5zQfuBic,7888
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_try_node_serialization.py,sha256=EL5kfakuoEcwD85dGjhMta-J-PpCHRSDoc80SdbBrQk,2769
 vellum_ee/workflows/display/tests/workflow_serialization/test_complex_terminal_node_serialization.py,sha256=RmFUDx8dYdfsOE2CGLvdXqNNRtLLpVzXDN8dqZyMcZ8,5822
 vellum_ee/workflows/display/types.py,sha256=i4T7ElU5b5h-nA1i3scmEhO1BqmNDc4eJDHavATD88w,2821
@@ -133,7 +133,7 @@ vellum/client/README.md,sha256=qmaVIP42MnxAu8jV7u-CsgVFfs3-pHQODrXdZdFxtaw,4749
 vellum/client/__init__.py,sha256=AYopGv2ZRVn3zsU8_km6KOvEHDbXiTPCVuYVI7bWvdA,120166
 vellum/client/core/__init__.py,sha256=SQ85PF84B9MuKnBwHNHWemSGuy-g_515gFYNFhvEE0I,1438
 vellum/client/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-vellum/client/core/client_wrapper.py,sha256=GMn5XtiuHqXf7vTEZMBtzd-oj9clWCurZASqSMpcAcI,1869
+vellum/client/core/client_wrapper.py,sha256=M88FTiEEacH6N8wlBA4RkxOXM_ha2OYXYDPzIWnOOXk,1869
 vellum/client/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 vellum/client/core/file.py,sha256=d4NNbX8XvXP32z8KpK2Xovv33nFfruIrpz0QWxlgpZk,2663
 vellum/client/core/http_client.py,sha256=Z77OIxIbL4OAB2IDqjRq_sYa5yNYAWfmdhdCSSvh6Y4,19552
@@ -1645,7 +1645,7 @@ vellum/workflows/nodes/experimental/openai_chat_completion_node/node.py,sha256=c
 vellum/workflows/nodes/experimental/tool_calling_node/__init__.py,sha256=S7OzT3I4cyOU5Beoz87nPwCejCMP2FsHBFL8OcVmxJ4,118
 vellum/workflows/nodes/experimental/tool_calling_node/node.py,sha256=Vxa0hs_tK1zdU-ux5j1XtwQkTph8crdd56wrMfAqWUY,4849
 vellum/workflows/nodes/experimental/tool_calling_node/tests/test_tool_calling_node.py,sha256=sxG26mOwt4N36RLoPJ-ngginPqC5qFzD_kGj9izdCFI,1833
-vellum/workflows/nodes/experimental/tool_calling_node/utils.py,sha256=mIQpAxLT2IF4SNQvUpRsX-t0jODxQ3Xvb_LEF7bRMF8,5214
+vellum/workflows/nodes/experimental/tool_calling_node/utils.py,sha256=JAxPP0DNNSxQlaC0pFHVs3B2WlY5Xf8MFzKLPmhlZYQ,5623
 vellum/workflows/nodes/mocks.py,sha256=a1FjWEIocseMfjzM-i8DNozpUsaW0IONRpZmXBoWlyc,10455
 vellum/workflows/nodes/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/nodes/tests/test_mocks.py,sha256=mfPvrs75PKcsNsbJLQAN6PDFoVqs9TmQxpdyFKDdO60,7837
@@ -1678,7 +1678,7 @@ vellum/workflows/sandbox.py,sha256=GVJzVjMuYzOBnSrboB0_6MMRZWBluAyQ2o7syeaeBd0,2
 vellum/workflows/state/__init__.py,sha256=yUUdR-_Vl7UiixNDYQZ-GEM_kJI9dnOia75TtuNEsnE,60
 vellum/workflows/state/base.py,sha256=-0b-nNBEXvGVau4c1BUwmCsXfo5wZD5VjLb8-eqi0Y8,21502
 vellum/workflows/state/context.py,sha256=KOAI1wEGn8dGmhmAemJaf4SZbitP3jpIBcwKfznQaRE,3076
-vellum/workflows/state/encoder.py,sha256=z7Mk6jQC-92wCj6XTK7VEnJ8px_lU8qy0BINqwGDN4I,2063
+vellum/workflows/state/encoder.py,sha256=_z9V34I9-_ie5TSxJNjJYc_DMmE6f1Q4VtGE08Uo7Yo,2349
 vellum/workflows/state/store.py,sha256=uVe-oN73KwGV6M6YLhwZMMUQhzTQomsVfVnb8V91gVo,1147
 vellum/workflows/state/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/state/tests/test_state.py,sha256=YOiC9qZAzkdiqb7nRarNWeDwxo7xHv3y3czlHl81ezg,6741
@@ -1711,8 +1711,8 @@ vellum/workflows/workflows/event_filters.py,sha256=GSxIgwrX26a1Smfd-6yss2abGCnad
 vellum/workflows/workflows/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/workflows/tests/test_base_workflow.py,sha256=8P5YIsNMO78_CR1NNK6wkEdkMB4b3Q_Ni1qxh78OnHo,20481
 vellum/workflows/workflows/tests/test_context.py,sha256=VJBUcyWVtMa_lE5KxdhgMu0WYNYnUQUDvTF7qm89hJ0,2333
-vellum_ai-0.14.55.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
-vellum_ai-0.14.55.dist-info/METADATA,sha256=ApsBqWl_U1jFrSqdnv2Rksz_onftgT0T2yu6KxfOpPc,5484
-vellum_ai-0.14.55.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_ai-0.14.55.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
-vellum_ai-0.14.55.dist-info/RECORD,,
+vellum_ai-0.14.56.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
+vellum_ai-0.14.56.dist-info/METADATA,sha256=NyxQYI0fSWAKolY31nX0fymQDu35425SCSXs1GJVZRI,5484
+vellum_ai-0.14.56.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_ai-0.14.56.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
+vellum_ai-0.14.56.dist-info/RECORD,,

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_serialization.py

@@ -130,17 +130,20 @@ def test_serialize_workflow():
                         "type": "JSON",
                         "value": [
                             {
-                                "state": None,
-                                "cache_config": None,
-                                "name": "get_current_weather",
-                                "description": None,
-                                "parameters": {
-                                    "type": "object",
-                                    "properties": {"location": {"type": "string"}, "unit": {"type": "string"}},
-                                    "required": ["location", "unit"],
+                                "definition": {
+                                    "state": None,
+                                    "cache_config": None,
+                                    "name": "get_current_weather",
+                                    "description": None,
+                                    "parameters": {
+                                        "type": "object",
+                                        "properties": {"location": {"type": "string"}, "unit": {"type": "string"}},
+                                        "required": ["location", "unit"],
+                                    },
+                                    "forced": None,
+                                    "strict": None,
                                 },
-                                "forced": None,
-                                "strict": None,
+                                "src": 'def get_current_weather(location: str, unit: str) -> str:\n    """\n    Get the current weather in a given location.\n    """\n    return f"The current weather in {location} is sunny with a temperature of 70 degrees {unit}."\n',  # noqa: E501
                             }
                         ],
                     },