vellum-ai 0.14.58__tar.gz → 0.14.60__tar.gz

{vellum_ai-0.14.58 → vellum_ai-0.14.60}/PKG-INFO

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

{vellum_ai-0.14.58 → vellum_ai-0.14.60}/pyproject.toml

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

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

{vellum_ai-0.14.58 → vellum_ai-0.14.60}/src/vellum/client/reference.md

@@ -1900,7 +1900,7 @@ client.deployments.retrieve(
 <dl>
 <dd>
 
-**id:** `str` — Either the Deployment's ID or its unique name
+**id:** `str` — Either the Prompt Deployment's ID or its unique name
     
 </dd>
 </dl>
@@ -1980,7 +1980,7 @@ client.deployments.deployment_history_item_retrieve(
 <dl>
 <dd>
 
-**id:** `str` — A UUID string identifying this deployment.
+**id:** `str` — Either the Prompt Deployment's ID or its unique name
     
 </dd>
 </dl>
@@ -2153,7 +2153,7 @@ client.deployments.retrieve_deployment_release_tag(
 <dl>
 <dd>
 
-**id:** `str` — A UUID string identifying this deployment.
+**id:** `str` — Either the Prompt Deployment's ID or its unique name
     
 </dd>
 </dl>
@@ -2232,7 +2232,7 @@ client.deployments.update_deployment_release_tag(
 <dl>
 <dd>
 
-**id:** `str` — A UUID string identifying this deployment.
+**id:** `str` — Either the Prompt Deployment's ID or its unique name
     
 </dd>
 </dl>
@@ -2319,7 +2319,7 @@ client.deployments.retrieve_prompt_deployment_release(
 <dl>
 <dd>
 
-**id:** `str` — A UUID string identifying this deployment.
+**id:** `str` — Either the Prompt Deployment's ID or its unique name
     
 </dd>
 </dl>
@@ -5633,7 +5633,7 @@ client.workflow_deployments.workflow_deployment_history_item_retrieve(
 <dl>
 <dd>
 
-**id:** `str` — A UUID string identifying this workflow deployment.
+**id:** `str` — Either the Workflow Deployment's ID or its unique name
     
 </dd>
 </dl>
@@ -5806,7 +5806,7 @@ client.workflow_deployments.retrieve_workflow_release_tag(
 <dl>
 <dd>
 
-**id:** `str` — A UUID string identifying this workflow deployment.
+**id:** `str` — Either the Workflow Deployment's ID or its unique name
     
 </dd>
 </dl>
@@ -5885,7 +5885,7 @@ client.workflow_deployments.update_workflow_release_tag(
 <dl>
 <dd>
 
-**id:** `str` — A UUID string identifying this workflow deployment.
+**id:** `str` — Either the Workflow Deployment's ID or its unique name
     
 </dd>
 </dl>
@@ -5973,7 +5973,7 @@ client.release_reviews.retrieve_workflow_deployment_release(
 <dl>
 <dd>
 
-**id:** `str` — A UUID string identifying this workflow deployment.
+**id:** `str` — Either the Workflow Deployment's ID or its unique name
     
 </dd>
 </dl>

{vellum_ai-0.14.58 → vellum_ai-0.14.60}/src/vellum/client/resources/deployments/client.py

@@ -109,7 +109,7 @@ class DeploymentsClient:
         Parameters
         ----------
         id : str
-            Either the Deployment's ID or its unique name
+            Either the Prompt Deployment's ID or its unique name
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -163,7 +163,7 @@ class DeploymentsClient:
             Either the UUID of Deployment History Item you'd like to retrieve, or the name of a Release Tag that's pointing to the Deployment History Item you'd like to retrieve.
 
         id : str
-            A UUID string identifying this deployment.
+            Either the Prompt Deployment's ID or its unique name
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -288,7 +288,7 @@ class DeploymentsClient:
         Parameters
         ----------
         id : str
-            A UUID string identifying this deployment.
+            Either the Prompt Deployment's ID or its unique name
 
         name : str
             The name of the Release Tag associated with this Deployment that you'd like to retrieve.
@@ -347,7 +347,7 @@ class DeploymentsClient:
         Parameters
         ----------
         id : str
-            A UUID string identifying this deployment.
+            Either the Prompt Deployment's ID or its unique name
 
         name : str
             The name of the Release Tag associated with this Deployment that you'd like to update.
@@ -411,7 +411,7 @@ class DeploymentsClient:
         Parameters
         ----------
         id : str
-            A UUID string identifying this deployment.
+            Either the Prompt Deployment's ID or its unique name
 
         release_id_or_release_tag : str
             Either the UUID of Prompt Deployment Release you'd like to retrieve, or the name of a Release Tag that's pointing to the Prompt Deployment Release you'd like to retrieve.
@@ -682,7 +682,7 @@ class AsyncDeploymentsClient:
         Parameters
         ----------
         id : str
-            Either the Deployment's ID or its unique name
+            Either the Prompt Deployment's ID or its unique name
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -744,7 +744,7 @@ class AsyncDeploymentsClient:
             Either the UUID of Deployment History Item you'd like to retrieve, or the name of a Release Tag that's pointing to the Deployment History Item you'd like to retrieve.
 
         id : str
-            A UUID string identifying this deployment.
+            Either the Prompt Deployment's ID or its unique name
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -885,7 +885,7 @@ class AsyncDeploymentsClient:
         Parameters
         ----------
         id : str
-            A UUID string identifying this deployment.
+            Either the Prompt Deployment's ID or its unique name
 
         name : str
             The name of the Release Tag associated with this Deployment that you'd like to retrieve.
@@ -952,7 +952,7 @@ class AsyncDeploymentsClient:
         Parameters
         ----------
         id : str
-            A UUID string identifying this deployment.
+            Either the Prompt Deployment's ID or its unique name
 
         name : str
             The name of the Release Tag associated with this Deployment that you'd like to update.
@@ -1024,7 +1024,7 @@ class AsyncDeploymentsClient:
         Parameters
         ----------
         id : str
-            A UUID string identifying this deployment.
+            Either the Prompt Deployment's ID or its unique name
 
         release_id_or_release_tag : str
             Either the UUID of Prompt Deployment Release you'd like to retrieve, or the name of a Release Tag that's pointing to the Prompt Deployment Release you'd like to retrieve.

{vellum_ai-0.14.58 → vellum_ai-0.14.60}/src/vellum/client/resources/release_reviews/client.py

@@ -24,7 +24,7 @@ class ReleaseReviewsClient:
         Parameters
         ----------
         id : str
-            A UUID string identifying this workflow deployment.
+            Either the Workflow Deployment's ID or its unique name
 
         release_id_or_release_tag : str
             Either the UUID of Workflow Deployment Release you'd like to retrieve, or the name of a Release Tag that's pointing to the Workflow Deployment Release you'd like to retrieve.
@@ -83,7 +83,7 @@ class AsyncReleaseReviewsClient:
         Parameters
         ----------
         id : str
-            A UUID string identifying this workflow deployment.
+            Either the Workflow Deployment's ID or its unique name
 
         release_id_or_release_tag : str
             Either the UUID of Workflow Deployment Release you'd like to retrieve, or the name of a Release Tag that's pointing to the Workflow Deployment Release you'd like to retrieve.

{vellum_ai-0.14.58 → vellum_ai-0.14.60}/src/vellum/client/resources/workflow_deployments/client.py

@@ -272,7 +272,7 @@ class WorkflowDeploymentsClient:
             Either the UUID of Workflow Deployment History Item you'd like to retrieve, or the name of a Release Tag that's pointing to the Workflow Deployment History Item you'd like to retrieve.
 
         id : str
-            A UUID string identifying this workflow deployment.
+            Either the Workflow Deployment's ID or its unique name
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -397,7 +397,7 @@ class WorkflowDeploymentsClient:
         Parameters
         ----------
         id : str
-            A UUID string identifying this workflow deployment.
+            Either the Workflow Deployment's ID or its unique name
 
         name : str
             The name of the Release Tag associated with this Workflow Deployment that you'd like to retrieve.
@@ -456,7 +456,7 @@ class WorkflowDeploymentsClient:
         Parameters
         ----------
         id : str
-            A UUID string identifying this workflow deployment.
+            Either the Workflow Deployment's ID or its unique name
 
         name : str
             The name of the Release Tag associated with this Workflow Deployment that you'd like to update.
@@ -796,7 +796,7 @@ class AsyncWorkflowDeploymentsClient:
             Either the UUID of Workflow Deployment History Item you'd like to retrieve, or the name of a Release Tag that's pointing to the Workflow Deployment History Item you'd like to retrieve.
 
         id : str
-            A UUID string identifying this workflow deployment.
+            Either the Workflow Deployment's ID or its unique name
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -937,7 +937,7 @@ class AsyncWorkflowDeploymentsClient:
         Parameters
         ----------
         id : str
-            A UUID string identifying this workflow deployment.
+            Either the Workflow Deployment's ID or its unique name
 
         name : str
             The name of the Release Tag associated with this Workflow Deployment that you'd like to retrieve.
@@ -1004,7 +1004,7 @@ class AsyncWorkflowDeploymentsClient:
         Parameters
         ----------
         id : str
-            A UUID string identifying this workflow deployment.
+            Either the Workflow Deployment's ID or its unique name
 
         name : str
             The name of the Release Tag associated with this Workflow Deployment that you'd like to update.

{vellum_ai-0.14.58 → vellum_ai-0.14.60}/src/vellum/client/types/document_document_to_document_index.py

@@ -17,6 +17,11 @@ class DocumentDocumentToDocumentIndex(UniversalBaseModel):
     Vellum-generated ID that uniquely identifies this link.
     """
 
+    environment_document_index_id: str = pydantic.Field()
+    """
+    Vellum-generated ID that uniquely identifies the environment index this document is included in.
+    """
+
     document_index_id: str = pydantic.Field()
     """
     Vellum-generated ID that uniquely identifies the index this document is included in.

{vellum_ai-0.14.58 → vellum_ai-0.14.60}/src/vellum/client/types/slim_document_document_to_document_index.py

@@ -17,6 +17,11 @@ class SlimDocumentDocumentToDocumentIndex(UniversalBaseModel):
     Vellum-generated ID that uniquely identifies this link.
     """
 
+    environment_document_index_id: str = pydantic.Field()
+    """
+    Vellum-generated ID that uniquely identifies the environment index this document is included in.
+    """
+
     document_index_id: str = pydantic.Field()
     """
     Vellum-generated ID that uniquely identifies the index this document is included in.

{vellum_ai-0.14.58 → vellum_ai-0.14.60}/src/vellum/workflows/nodes/bases/base.py

@@ -356,8 +356,78 @@ class BaseNode(Generic[StateType], metaclass=BaseNodeMeta):
                 state.meta.node_execution_cache._dependencies_invoked[execution_id].add(invoked_by)
                 return execution_id
 
+            # For AWAIT_ANY in workflows, we have two cases:
+            # 1. The node is being re-executed because of a fork
+            # 2. The node is being re-executed because of a loop
+            # For case 1, we need to remove the fork id from the node_to_fork_id mapping
+            # For case 2, we need to check if the node is in a loop
+            in_loop = False
+            # Default to true, will be set to false if the merged node has already been triggered
+            should_retrigger = True
+            if cls.merge_behavior == MergeBehavior.AWAIT_ANY:
+                # Get the node that triggered the current execution
+                invoked_by_node = state.meta.node_execution_cache.__node_execution_lookup__.get(invoked_by)
+
+                # Check if invoked by node is a forked node
+                if invoked_by_node is not None:
+                    fork_id = state.meta.node_execution_cache.__node_to_fork_id__.get(invoked_by_node, None)
+                    if fork_id:
+                        # If the invoked by node has a fork id and that fork id is in __all_fork_ids__
+                        #   We will
+                        #   1. remove the fork id from __all_fork_ids__
+                        #   2. remove the fork id from the __node_to_fork_id__ mapping
+                        # else (this mean the fork has already been triggered)
+                        #   remove the id from the node_to_fork_id mapping and not retrigger again
+                        all_fork_ids = state.meta.node_execution_cache.__all_fork_ids__
+                        if fork_id in all_fork_ids:
+                            # When the next forked node merge, it will not trigger the node again
+                            # We should consider adding a lock here to prevent race condition
+                            all_fork_ids.remove(fork_id)
+                            state.meta.node_execution_cache.__node_to_fork_id__.pop(invoked_by_node, None)
+                        else:
+                            should_retrigger = False
+                            state.meta.node_execution_cache.__node_to_fork_id__.pop(invoked_by_node, None)
+
+                # If should_retrigger is false, then we will not trigger the node already
+                # So we don't need to check loop behavior
+                if should_retrigger:
+                    # Trace back through the dependency chain to detect if this node triggers itself
+                    visited = set()
+                    current_execution_id = invoked_by
+
+                    # Walk backwards through the dependency chain
+                    while current_execution_id and current_execution_id not in visited:
+                        visited.add(current_execution_id)
+
+                        # Get the dependencies that triggered this execution
+                        dependencies_for_current = state.meta.node_execution_cache._dependencies_invoked.get(
+                            current_execution_id, set()
+                        )
+
+                        # If we've reached the end of the chain, it means the node is not in a loop
+                        # we can break out of the loop
+                        if not dependencies_for_current:
+                            break
+
+                        # Move to the previous node in the dependency chain
+                        current_execution_id = next(iter(dependencies_for_current))
+
+                        current_node_class = state.meta.node_execution_cache.__node_execution_lookup__.get(
+                            current_execution_id
+                        )
+
+                        # If we've found our target node class in the chain, we're in a loop
+                        if current_node_class == cls.node_class:
+                            in_loop = True
+                            break
+
             for queued_node_execution_id in state.meta.node_execution_cache._node_executions_queued[cls.node_class]:
-                if invoked_by not in state.meta.node_execution_cache._dependencies_invoked[queued_node_execution_id]:
+                # When should_retrigger is false, it means the merged node has already been triggered
+                # So we don't need to trigger the node again
+                if not should_retrigger or (
+                    invoked_by not in state.meta.node_execution_cache._dependencies_invoked[queued_node_execution_id]
+                    and not in_loop
+                ):
                     state.meta.node_execution_cache._invoke_dependency(
                         queued_node_execution_id, cls.node_class, invoked_by, dependencies
                     )

{vellum_ai-0.14.58 → vellum_ai-0.14.60}/src/vellum/workflows/runner/runner.py

@@ -398,7 +398,9 @@ class WorkflowRunner(Generic[StateType]):
             return
 
         for port in ports:
+            nodes_to_fork = []
             for edge in port.edges:
+                nodes_to_fork.append(edge.to_node)
                 if port.fork_state:
                     next_state = deepcopy(state)
                     self._state_forks.add(next_state)
@@ -409,6 +411,21 @@ class WorkflowRunner(Generic[StateType]):
                     self._concurrency_queue.put((next_state, edge.to_node, invoked_by))
                 else:
                     self._run_node_if_ready(next_state, edge.to_node, invoked_by)
+                if invoked_by:
+                    previous_node = state.meta.node_execution_cache.__node_execution_lookup__.get(invoked_by)
+                    if previous_node:
+                        previous_node_fork_id = state.meta.node_execution_cache.__node_to_fork_id__.get(previous_node)
+                        if previous_node_fork_id:
+                            state.meta.node_execution_cache.__node_to_fork_id__[edge.to_node] = previous_node_fork_id
+                            # Remove previous node fork in __node_to_fork_id__
+                            state.meta.node_execution_cache.__node_to_fork_id__.pop(previous_node, None)
+
+            # If there are multiple edges, we need to create a fork ID
+            if len(nodes_to_fork) > 1:
+                fork_id = uuid4()
+                state.meta.node_execution_cache.__all_fork_ids__.add(fork_id)
+                for node in nodes_to_fork:
+                    state.meta.node_execution_cache.__node_to_fork_id__[node] = fork_id
 
         if self._max_concurrency:
             num_nodes_to_run = self._max_concurrency - len(self._active_nodes_by_execution_id)

{vellum_ai-0.14.58 → vellum_ai-0.14.60}/src/vellum/workflows/state/base.py

@@ -104,7 +104,9 @@ class NodeExecutionCache:
     _dependencies_invoked: DependenciesInvoked
 
     # Derived fields - no need to serialize
-    __node_execution_lookup__: NodeExecutionLookup
+    __node_execution_lookup__: NodeExecutionLookup  # execution_id -> node_class
+    __node_to_fork_id__: Dict[Type["BaseNode"], UUID]  # node_class -> fork_id
+    __all_fork_ids__: Set[UUID]
 
     def __init__(self) -> None:
         self._dependencies_invoked = defaultdict(set)
@@ -112,6 +114,8 @@ class NodeExecutionCache:
         self._node_executions_initiated = defaultdict(set)
         self._node_executions_queued = defaultdict(list)
         self.__node_execution_lookup__ = {}
+        self.__node_to_fork_id__ = {}
+        self.__all_fork_ids__ = set()
 
     @classmethod
     def deserialize(cls, raw_data: dict, nodes: Dict[Union[str, UUID], Type["BaseNode"]]):