vellum-ai 1.2.4__py3-none-any.whl → 1.2.5__py3-none-any.whl

vellum/client/types/video_input.py

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_video import VellumVideo
+
+
+class VideoInput(UniversalBaseModel):
+    """
+    A user input representing a Vellum Video value
+    """
+
+    name: str = pydantic.Field()
+    """
+    The variable's name
+    """
+
+    type: typing.Literal["VIDEO"] = "VIDEO"
+    value: VellumVideo
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/workflow_push_deployment_config_request.py

@@ -11,6 +11,7 @@ class WorkflowPushDeploymentConfigRequest(UniversalBaseModel):
     name: typing.Optional[str] = None
     description: typing.Optional[str] = None
     release_tags: typing.Optional[typing.List[str]] = None
+    release_description: typing.Optional[str] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/types/audio_input.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.audio_input import *

vellum/types/document_input.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.document_input import *

vellum/types/image_input.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.image_input import *

vellum/types/named_scenario_input_audio_variable_value_request.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.named_scenario_input_audio_variable_value_request import *

vellum/types/named_scenario_input_document_variable_value_request.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.named_scenario_input_document_variable_value_request import *

vellum/types/named_scenario_input_image_variable_value_request.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.named_scenario_input_image_variable_value_request import *

vellum/types/named_scenario_input_video_variable_value_request.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.named_scenario_input_video_variable_value_request import *

vellum/types/named_test_case_audio_variable_value.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.named_test_case_audio_variable_value import *

vellum/types/named_test_case_audio_variable_value_request.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.named_test_case_audio_variable_value_request import *

vellum/types/named_test_case_document_variable_value.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.named_test_case_document_variable_value import *

vellum/types/named_test_case_document_variable_value_request.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.named_test_case_document_variable_value_request import *

vellum/types/named_test_case_image_variable_value.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.named_test_case_image_variable_value import *

vellum/types/named_test_case_image_variable_value_request.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.named_test_case_image_variable_value_request import *

vellum/types/named_test_case_video_variable_value.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.named_test_case_video_variable_value import *

vellum/types/named_test_case_video_variable_value_request.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.named_test_case_video_variable_value_request import *

vellum/types/scenario_input_audio_variable_value.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.scenario_input_audio_variable_value import *

vellum/types/scenario_input_document_variable_value.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.scenario_input_document_variable_value import *

vellum/types/scenario_input_image_variable_value.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.scenario_input_image_variable_value import *

vellum/types/scenario_input_video_variable_value.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.scenario_input_video_variable_value import *

vellum/types/test_case_audio_variable_value.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.test_case_audio_variable_value import *

vellum/types/test_case_document_variable_value.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.test_case_document_variable_value import *

vellum/types/test_case_image_variable_value.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.test_case_image_variable_value import *

vellum/types/test_case_video_variable_value.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.test_case_video_variable_value import *

vellum/types/video_input.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.video_input import *

vellum/workflows/resolvers/base.py

@@ -1,11 +1,24 @@
 from abc import ABC, abstractmethod
-from typing import Iterator
+from uuid import UUID
+from typing import TYPE_CHECKING, Iterator, Optional, Type
 
 from vellum.workflows.events.workflow import WorkflowEvent
 from vellum.workflows.state.base import BaseState
 
+if TYPE_CHECKING:
+    from vellum.workflows.state.context import WorkflowContext
+    from vellum.workflows.workflows.base import BaseWorkflow
+
 
 class BaseWorkflowResolver(ABC):
+    def __init__(self):
+        self._context: Optional["WorkflowContext"] = None
+        self._workflow_class: Optional[Type["BaseWorkflow"]] = None
+
+    def register_workflow_instance(self, workflow_instance: "BaseWorkflow") -> None:
+        self._workflow_class = type(workflow_instance)
+        self._context = workflow_instance.context
+
     @abstractmethod
     def get_latest_execution_events(self) -> Iterator[WorkflowEvent]:
         pass
@@ -13,3 +26,7 @@ class BaseWorkflowResolver(ABC):
     @abstractmethod
     def get_state_snapshot_history(self) -> Iterator[BaseState]:
         pass
+
+    @abstractmethod
+    def load_state(self, previous_execution_id: Optional[UUID] = None) -> Optional[BaseState]:
+        pass

vellum/workflows/resolvers/resolver.py

@@ -0,0 +1,42 @@
+import logging
+from uuid import UUID
+from typing import Iterator, Optional
+
+from vellum.workflows.events.workflow import WorkflowEvent
+from vellum.workflows.resolvers.base import BaseWorkflowResolver
+from vellum.workflows.state.base import BaseState, StateMeta
+
+logger = logging.getLogger(__name__)
+
+
+class VellumResolver(BaseWorkflowResolver):
+    def get_latest_execution_events(self) -> Iterator[WorkflowEvent]:
+        return iter([])
+
+    def get_state_snapshot_history(self) -> Iterator[BaseState]:
+        return iter([])
+
+    def load_state(self, previous_execution_id: Optional[UUID] = None) -> Optional[BaseState]:
+        if previous_execution_id is None:
+            return None
+
+        if not self._context:
+            logger.warning("Cannot load state: No workflow context registered")
+            return None
+
+        client = self._context.vellum_client
+        response = client.workflow_executions.retrieve_workflow_execution_detail(
+            execution_id=str(previous_execution_id),
+        )
+
+        if response.state is None:
+            return None
+
+        meta = StateMeta.model_validate(response.state.pop("meta"))
+
+        if self._workflow_class:
+            state_class = self._workflow_class.get_state_class()
+            return state_class(**response.state, meta=meta)
+        else:
+            logger.warning("No workflow class registered, falling back to BaseState")
+            return BaseState(**response.state, meta=meta)

vellum/workflows/resolvers/tests/test_resolver.py

@@ -0,0 +1,59 @@
+from datetime import datetime
+from unittest.mock import Mock
+from uuid import uuid4
+
+from vellum.client.types.workflow_execution_detail import WorkflowExecutionDetail
+from vellum.workflows import BaseWorkflow
+from vellum.workflows.inputs.base import BaseInputs
+from vellum.workflows.resolvers.resolver import VellumResolver
+from vellum.workflows.state.base import BaseState, NodeExecutionCache
+from vellum.workflows.state.context import WorkflowContext
+
+
+def test_load_state_with_context_success():
+    """Test load_state successfully loads state when context and client are available."""
+    resolver = VellumResolver()
+    execution_id = uuid4()
+
+    class TestState(BaseState):
+        test_key: str = "test_value"
+
+    class TestWorkflow(BaseWorkflow[BaseInputs, TestState]):
+        pass
+
+    # GIVEN a state dictionary that matches what the resolver expects
+    state_dict = {
+        "test_key": "test_value",
+        "meta": {
+            "workflow_definition": "MockWorkflow",
+            "id": str(uuid4()),
+            "span_id": str(uuid4()),
+            "updated_ts": datetime.now().isoformat(),
+            "workflow_inputs": BaseInputs(),
+            "external_inputs": {},
+            "node_outputs": {},
+            "node_execution_cache": NodeExecutionCache(),
+            "parent": None,
+        },
+    }
+
+    mock_response = WorkflowExecutionDetail(
+        span_id="test-span-id", start=datetime.now(), inputs=[], outputs=[], spans=[], state=state_dict
+    )
+
+    mock_client = Mock()
+    mock_client.workflow_executions.retrieve_workflow_execution_detail.return_value = mock_response
+
+    # AND context with the test workflow class
+    context = WorkflowContext(vellum_client=mock_client)
+    TestWorkflow(context=context, resolvers=[resolver])
+
+    result = resolver.load_state(previous_execution_id=execution_id)
+
+    # THEN should return an instance of TestWorkflow.State, not BaseState
+    assert isinstance(result, TestState)
+    assert result.test_key == "test_value"
+
+    mock_client.workflow_executions.retrieve_workflow_execution_detail.assert_called_once_with(
+        execution_id=str(execution_id)
+    )

vellum/workflows/workflows/base.py

@@ -251,6 +251,9 @@ class BaseWorkflow(Generic[InputsType, StateType], metaclass=_BaseWorkflowMeta):
         for emitter in self.emitters:
             emitter.register_context(self._context)
 
+        for resolver in self.resolvers:
+            resolver.register_workflow_instance(self)
+
         self.validate()
 
     @property

{vellum_ai-1.2.4.dist-info → vellum_ai-1.2.5.dist-info}/METADATA

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