codemie-sdk-python 0.1.519__tar.gz → 0.1.520__tar.gz

{codemie_sdk_python-0.1.519 → codemie_sdk_python-0.1.520}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codemie-sdk-python
-Version: 0.1.519
+Version: 0.1.520
 Summary: CodeMie SDK for Python
 Author: Vadym Vlasenko
 Author-email: vadym_vlasenko@epam.com

{codemie_sdk_python-0.1.519 → codemie_sdk_python-0.1.520}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "codemie-sdk-python"
-version = "0.1.519"
+version = "0.1.520"
 description = "CodeMie SDK for Python"
 authors = [
     "Vadym Vlasenko <vadym_vlasenko@epam.com>",

codemie_sdk_python-0.1.520/src/codemie_sdk/models/workflow_transition.py

@@ -0,0 +1,31 @@
+"""Workflow execution transition models."""
+
+from datetime import datetime
+from typing import Any
+
+from pydantic import BaseModel, ConfigDict
+
+
+class WorkflowContext(BaseModel):
+    """Context snapshot captured at a workflow state transition."""
+
+    model_config = ConfigDict(populate_by_name=True, extra="allow")
+
+    context_store: dict[str, Any] = {}
+    next: list[str] = []
+    messages: list[str] = []
+    user_input: str | None = None
+    final_summary: list[Any] = []
+
+
+class WorkflowTransition(BaseModel):
+    """A single state-to-state transition within a workflow execution."""
+
+    model_config = ConfigDict(populate_by_name=True)
+
+    id: str
+    execution_id: str
+    from_state_id: str | None = None
+    to_state_id: str
+    workflow_context: WorkflowContext
+    date: datetime

{codemie_sdk_python-0.1.519 → codemie_sdk_python-0.1.520}/src/codemie_sdk/services/workflow_execution.py

@@ -6,6 +6,7 @@ from ..models.common import PaginationParams
 from ..models.workflow import WorkflowExecution
 from ..models.workflow_execution_payload import WorkflowExecutionCreateRequest
 from ..models.workflow_thoughts import WorkflowExecutionThought
+from ..models.workflow_transition import WorkflowTransition
 from .workflow_execution_state import WorkflowExecutionStateService
 from ..utils import ApiRequestHandler
 
@@ -152,6 +153,21 @@ class WorkflowExecutionService:
             extra_headers=headers,
         )
 
+    def get_transition_to(self, execution_id: str, state_id: str) -> WorkflowTransition:
+        """Get the transition that led into a specific state.
+
+        Args:
+            execution_id: ID of the execution
+            state_id: ID of the destination state
+
+        Returns:
+            WorkflowTransition including workflow_context.context_store snapshot
+        """
+        return self._api.get(
+            f"/v1/workflows/{self._workflow_id}/executions/{execution_id}/transitions/to/{state_id}",
+            WorkflowTransition,
+        )
+
     def get_thoughts(
         self, execution_id: str, thought_ids: list[str]
     ) -> list[WorkflowExecutionThought]: