vellum-workflow-server 0.14.72.post4__py3-none-any.whl → 0.14.72.post6__py3-none-any.whl

{vellum_workflow_server-0.14.72.post4.dist-info → vellum_workflow_server-0.14.72.post6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-workflow-server
-Version: 0.14.72.post4
+Version: 0.14.72.post6
 Summary: 
 License: AGPL
 Requires-Python: >=3.9.0,<4

{vellum_workflow_server-0.14.72.post4.dist-info → vellum_workflow_server-0.14.72.post6.dist-info}/RECORD

@@ -6,14 +6,14 @@ workflow_server/api/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 workflow_server/api/tests/test_input_display_mapping.py,sha256=drBZqMudFyB5wgiUOcMgRXz7E7ge-Qgxbstw4E4f0zE,2211
 workflow_server/api/tests/test_workflow_view.py,sha256=2nscM_QsYPHkkTG8_JhNbE2LmGL5FQKXEtaLLjXouw0,14591
 workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=wmeH8oGo0lKx3YzEJQ9nGjw7adqQD0EL7-xGLPgYWqc,24872
-workflow_server/api/workflow_view.py,sha256=9ZU2cziP-Fe-wFezanesjau3eTDaHNu2sO9iMuKR4Xw,18247
+workflow_server/api/workflow_view.py,sha256=x9SSsWCzLuwXBhLJMc-vDMtmP626O6tw1AGh6KII_lc,17697
 workflow_server/code_exec_runner.py,sha256=p6lGaEGAfMFCxIpOJtH9KTvkffCDsJ3bbpmHdCGWquM,3265
 workflow_server/config.py,sha256=K5Tavm7wiqCZt0RWWue7zzb8N6e8aWnFOTNlBqEJPcI,1330
 workflow_server/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 workflow_server/core/cancel_workflow.py,sha256=Ffkc3mzmrdMEUcD-sHfEhX4IwVrka-E--SxKA1dUfIU,2185
 workflow_server/core/events.py,sha256=24MA66DVQuaLJJcZrS8IL1Zq4Ohi9CoouKZ5VgoH3Cs,1402
 workflow_server/core/executor.py,sha256=-FR91th7vp_v-R_Mo3NVS5GcsmD7iPo3D18Xi-bUrHo,18017
-workflow_server/core/workflow_executor_context.py,sha256=JHGlrEaf7IkCJjrAPMo64M7X0aiwWXe6exwXwxhwow4,1298
+workflow_server/core/workflow_executor_context.py,sha256=a-v48GJbOWUh4JIf_bNwDX-BvfKkg4xwRSPEyRVQmp4,1373
 workflow_server/server.py,sha256=QBU12AaAfAgLqfCDBd24qIJl_mbheiq0-hfcWV7rZM4,1234
 workflow_server/start.py,sha256=DgtQhuCLc07BIWyJPLPZKZsQ8jwEFsvvfIo7MdwVrpw,1998
 workflow_server/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -26,7 +26,7 @@ workflow_server/utils/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMp
 workflow_server/utils/tests/test_system_utils.py,sha256=MdBxI9gxUOpR_JBAHpEz6dGFY6JjxhMSM2oExpqFvNA,4314
 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.72.post4.dist-info/METADATA,sha256=9Isp-DKUVQduS6Sadl5OqHFPoiy4Ix9tJG8JqWQGfAY,2243
-vellum_workflow_server-0.14.72.post4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-0.14.72.post4.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-0.14.72.post4.dist-info/RECORD,,
+vellum_workflow_server-0.14.72.post6.dist-info/METADATA,sha256=pmr1HyQdJhIDgb7E0Qwu5YtfYqjXq9U1RAhfNqE_JyI,2243
+vellum_workflow_server-0.14.72.post6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-0.14.72.post6.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-0.14.72.post6.dist-info/RECORD,,

workflow_server/api/workflow_view.py

@@ -398,7 +398,7 @@ def get_version_route() -> tuple[dict, int]:
                             node_display_class = get_node_display_class(obj)
                             exec_config_raw = node_display_class().serialize(WorkflowDisplayContext())
                             exec_config = cast(Dict[str, Any], exec_config_raw)
-                            module = exec_config["definition"]["module"]
+                            config_module = exec_config["definition"]["module"]
                             label = exec_config["label"]
                             description = (
                                 exec_config["display_data"]["comment"]["value"]
@@ -408,7 +408,7 @@ def get_version_route() -> tuple[dict, int]:
                             nodes.append(
                                 {
                                     "id": str(uuid4()),
-                                    "module": module,
+                                    "module": config_module,
                                     "name": obj.__name__,
                                     "label": pascal_to_title_case(label),
                                     "description": description,
@@ -428,16 +428,9 @@ def get_version_route() -> tuple[dict, int]:
 
 
 def get_workflow_request_context(data: dict) -> WorkflowExecutorContext:
-    # not sure if this is the filter we want to pass forward?
-    trace_id = uuid4()
-    if "execution_context" in data and isinstance(data["execution_context"], dict):
-        if "trace_id" in data["execution_context"]:
-            trace_id = data["execution_context"]["trace_id"]
-
     context_data = {
         **data,
         "inputs": convert_json_inputs_to_vellum(data.get("inputs") or []),
-        "trace_id": trace_id,
         "request_start_time": time.time_ns(),
     }
 
@@ -445,15 +438,9 @@ def get_workflow_request_context(data: dict) -> WorkflowExecutorContext:
 
 
 def get_node_request_context(data: dict) -> NodeExecutorContext:
-    trace_id = uuid4()
-    if "execution_context" in data and isinstance(data["execution_context"], dict):
-        if "trace_id" in data["execution_context"]:
-            trace_id = data["execution_context"]["trace_id"]
-
     context_data = {
         **data,
         "inputs": convert_json_inputs_to_vellum(data["inputs"]),
-        "trace_id": trace_id,
         "request_start_time": time.time_ns(),
     }
 

workflow_server/core/workflow_executor_context.py

@@ -15,7 +15,6 @@ class BaseExecutorContext(UniversalBaseModel):
     files: dict[str, str]
     environment_api_key: str
     execution_id: UUID
-    trace_id: UUID
     module: str
     execution_context: ExecutionContext = field(default_factory=ExecutionContext)
     request_start_time: int
@@ -28,6 +27,10 @@ class BaseExecutorContext(UniversalBaseModel):
     def container_overhead_latency(self) -> int:
         return self.stream_start_time - self.request_start_time if self.stream_start_time else -1
 
+    @property
+    def trace_id(self) -> UUID:
+        return self.execution_context.trace_id
+
     def __hash__(self) -> int:
         # do we think we need anything else for a unique hash for caching?
         return hash(str(self.execution_id))