vellum-workflow-server 0.14.69__py3-none-any.whl → 0.14.70__py3-none-any.whl

{vellum_workflow_server-0.14.69.dist-info → vellum_workflow_server-0.14.70.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-workflow-server
-Version: 0.14.69
+Version: 0.14.70
 Summary: 
 License: AGPL
 Requires-Python: >=3.9.0,<4
@@ -28,7 +28,7 @@ Requires-Dist: pebble (==5.0.7)
 Requires-Dist: pyjwt (==2.10.0)
 Requires-Dist: python-dotenv (==1.0.1)
 Requires-Dist: sentry-sdk[flask] (==2.20.0)
-Requires-Dist: vellum-ai (==0.14.69)
+Requires-Dist: vellum-ai (==0.14.70)
 Description-Content-Type: text/markdown
 
 # Vellum Workflow Runner Server

{vellum_workflow_server-0.14.69.dist-info → vellum_workflow_server-0.14.70.dist-info}/RECORD

@@ -3,6 +3,7 @@ workflow_server/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 workflow_server/api/auth_middleware.py,sha256=IlZaCiwZ5nwQqk5sYQorvOFj7lt0p1ZSSEqUxfiFaW0,2458
 workflow_server/api/healthz_view.py,sha256=itiRvBDBXncrw8Kbbc73UZLwqMAhgHOR3uSre_dAfgY,404
 workflow_server/api/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+workflow_server/api/tests/test_input_display_mapping.py,sha256=drBZqMudFyB5wgiUOcMgRXz7E7ge-Qgxbstw4E4f0zE,2211
 workflow_server/api/tests/test_workflow_view.py,sha256=wlVFBmKcoI-RdzfGPioeW46k6zaXyUeIerPc6m4aQls,7150
 workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=wWW4sukIUmHnL525MLSAB2E5P6CeASnhsCoKQXEgPcI,21297
 workflow_server/api/workflow_view.py,sha256=duiMnAZ7PRpoPz63s9z37pxUxGR-9yAi3qxG9APXCao,14244
@@ -21,9 +22,9 @@ workflow_server/utils/log_proxy.py,sha256=nugi6fOgAYKX2X9DIc39TG366rsmmDUPoEtG3g
 workflow_server/utils/oom_killer.py,sha256=8WB0nQWjmnjW9QzvNNwfYoBFB3yDHM3_OmnryeC8G3A,3657
 workflow_server/utils/sentry.py,sha256=Pr3xKvHdk0XFSpXgy-55bWI4J3bbf_36gjDyLOs7oVU,855
 workflow_server/utils/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-workflow_server/utils/tests/test_utils.py,sha256=MKGY4F_jnRdwqyQ8Krhlk7MpMEPql5MDATcwBaxpIEA,4786
-workflow_server/utils/utils.py,sha256=dpFeUQW7cmiH2-aLsO65jAa4RubfvptWNkW1ZkvDlmE,3710
-vellum_workflow_server-0.14.69.dist-info/METADATA,sha256=Mw-2DCN-YSs3w4bny4SLReAts1d2VzuuQRJHfx6acyU,2237
-vellum_workflow_server-0.14.69.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-0.14.69.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-0.14.69.dist-info/RECORD,,
+workflow_server/utils/tests/test_utils.py,sha256=qwK5Rmy3RQyjtlUrYAuGuDlBeRzZKsf1yS-y2IpUizQ,6452
+workflow_server/utils/utils.py,sha256=Wqqn-1l2ugkGgy5paWWdt0AVxAyPMQCYcnRSSOMjXlA,4355
+vellum_workflow_server-0.14.70.dist-info/METADATA,sha256=8jVFfOujZb2NwLgEDdy5IgpJKF3kyl7ics_KTn3D1qI,2237
+vellum_workflow_server-0.14.70.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-0.14.70.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-0.14.70.dist-info/RECORD,,

workflow_server/api/tests/test_input_display_mapping.py

@@ -0,0 +1,61 @@
+from uuid import uuid4
+
+from workflow_server.server import create_app
+
+
+def test_input_conversion_with_display_mapping():
+    """
+    Test that validates input conversion behavior for future WorkflowDisplay refactor.
+
+    This test demonstrates the expected behavior when inputs are converted from UI names
+    to SDK attribute names using WorkflowDisplay's inputs_display mapping, rather than
+    the current snake_case conversion approach.
+
+    The test passes on main branch by using the current conversion logic, but establishes
+    the expected input/output patterns for the future refactor.
+    """
+    span_id = uuid4()
+    request_body = {
+        "execution_id": str(span_id),
+        "inputs": [
+            {"name": "User Message", "type": "STRING", "value": "Hello world"},
+            {"name": "123-config", "type": "STRING", "value": "config-value"},
+            {"name": "API-Key", "type": "STRING", "value": "test-key"},
+        ],
+        "environment_api_key": "test",
+        "module": "workflow",
+        "timeout": 360,
+        "files": {
+            "__init__.py": "",
+            "workflow.py": """
+from vellum.workflows import BaseWorkflow
+from vellum.workflows.state import BaseState
+from .inputs import Inputs
+
+class Workflow(BaseWorkflow[Inputs, BaseState]):
+    class Outputs(BaseWorkflow.Outputs):
+        result = "success"
+""",
+            "inputs.py": """
+from vellum.workflows.inputs import BaseInputs
+
+class Inputs(BaseInputs):
+    user_message: str
+    input_123_config: str
+    api_key: str
+""",
+        },
+    }
+
+    flask_app = create_app()
+    with flask_app.test_client() as test_client:
+        response = test_client.post("/workflow/stream", json=request_body)
+        status_code = response.status_code
+
+    assert status_code == 200, f"Request failed with status {status_code}: {response.data}"
+
+    response_lines = [line for line in response.data.decode().split("\n") if line and line not in ["WAITING", "END"]]
+    assert len(response_lines) > 0, "No response events received"
+
+    fulfilled_events = [line for line in response_lines if "workflow.execution.fulfilled" in line]
+    assert len(fulfilled_events) > 0, "No workflow.execution.fulfilled event received"

workflow_server/utils/tests/test_utils.py

@@ -10,7 +10,11 @@ from vellum import (
     VellumError,
     VellumImage,
 )
-from workflow_server.utils.utils import convert_json_inputs_to_vellum
+from workflow_server.utils.utils import (
+    convert_json_inputs_to_vellum,
+    to_python_safe_snake_case,
+    to_valid_python_identifier,
+)
 
 
 @pytest.mark.parametrize(
@@ -134,10 +138,57 @@ def test_input_variables_with_uppercase_gets_sanitized():
     ]
 
     expected = {
-        "foo": "<example-string-value>",
+        "Foo": "<example-string-value>",
         "foo_var": "<another-example-string-value>",
     }
 
     actual = convert_json_inputs_to_vellum(inputs)
 
     assert expected == actual
+
+
+@pytest.mark.parametrize(
+    ["input_string", "safety_prefix", "expected"],
+    [
+        ("Foo", "input", "Foo"),
+        ("test", "input", "test"),
+        ("myVariable", "input", "myVariable"),
+        ("validName123", "input", "validName123"),
+        ("Foo-Var", "input", "foo_var"),
+        ("My Variable", "input", "my_variable"),
+        ("test-case", "input", "test_case"),
+        ("CamelCase", "input", "CamelCase"),
+        ("123", "input", "input_123"),
+        ("_a", "input", "input_a"),
+        ("_123", "input", "input_123"),
+        ("test@#$", "input", "test"),
+        ("@#$test", "input", "test"),
+        ("123", "_", "_123"),
+        ("123", "var", "var_123"),
+        ("123", "var_", "var_123"),
+    ],
+)
+def test_to_valid_python_identifier(input_string, safety_prefix, expected):
+    actual = to_valid_python_identifier(input_string, safety_prefix)
+    assert expected == actual
+
+
+@pytest.mark.parametrize(
+    ["input_string", "safety_prefix", "expected"],
+    [
+        ("Foo", "input", "foo"),
+        ("Foo-Var", "input", "foo_var"),
+        ("CamelCase", "input", "camel_case"),
+        ("test", "input", "test"),
+        ("My Variable", "input", "my_variable"),
+        ("123", "input", "input_123"),
+        ("_a", "input", "input_a"),
+        ("_123", "input", "input_123"),
+        ("123", "_", "_123"),
+        ("123", "var", "var_123"),
+        ("123", "var_", "var_123"),
+    ],
+)
+def test_to_python_safe_snake_case(input_string, safety_prefix, expected):
+    actual = to_python_safe_snake_case(input_string, safety_prefix)
+    assert expected == actual

workflow_server/utils/utils.py

@@ -22,7 +22,7 @@ def convert_json_inputs_to_vellum(inputs: List[dict]) -> dict:
     for input in inputs:
         value = input["value"]
         # sync with vellum-python-sdks/ee/codegen/src/context/input-variable-context.ts
-        name = to_python_safe_snake_case(input["name"], "input")
+        name = to_valid_python_identifier(input["name"], "input")
         type = input["type"]
 
         if type == "CHAT_HISTORY":
@@ -58,7 +58,7 @@ def to_python_safe_snake_case(string: str, safety_prefix: str = "_") -> str:
     # Strip special characters from start of string
     cleaned_str = re.sub(r"^[^a-zA-Z0-9_]+", "", string)
 
-    # Check if cleaned string starts with a number or underscore
+    # Check if cleaned string starts with a number or an underscore
     starts_with_unsafe = bool(re.match(r"^[\d_]", cleaned_str))
 
     # Convert to snake case
@@ -74,6 +74,22 @@ def to_python_safe_snake_case(string: str, safety_prefix: str = "_") -> str:
     return f"{cleaned_safety_prefix}{snake_case}" if starts_with_unsafe else snake_case
 
 
+def to_valid_python_identifier(string: str, safety_prefix: str = "_") -> str:
+    # Strip special characters from start of string
+    cleaned_str = re.sub(r"^[^a-zA-Z0-9_]+", "", string)
+
+    # Check if cleaned string starts with a number or an underscore
+    starts_with_unsafe = bool(re.match(r"^[\d_]", cleaned_str))
+
+    # Check if the string is already a valid Python identifier (preserve case)
+    is_valid_python_identifier = bool(re.match(r"^[a-zA-Z][a-zA-Z0-9]*$", cleaned_str))
+
+    if is_valid_python_identifier and not starts_with_unsafe:
+        return cleaned_str
+
+    return to_python_safe_snake_case(string, safety_prefix)
+
+
 def get_obj_size(obj: Any) -> int:
     marked = {id(obj)}
     obj_q = [obj]