codemie-sdk-python 0.1.184__tar.gz → 0.1.186__tar.gz

{codemie_sdk_python-0.1.184 → codemie_sdk_python-0.1.186}/PKG-INFO

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

{codemie_sdk_python-0.1.184 → codemie_sdk_python-0.1.186}/pyproject.toml

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

{codemie_sdk_python-0.1.184 → codemie_sdk_python-0.1.186}/src/codemie_sdk/models/workflow_state.py

@@ -2,7 +2,7 @@
 
 from datetime import datetime
 from enum import Enum
-from typing import Optional
+from typing import List, Optional
 
 from pydantic import BaseModel, ConfigDict
 
@@ -24,9 +24,11 @@ class WorkflowExecutionStateThought(BaseModel):
     model_config = ConfigDict(populate_by_name=True)
 
     id: str
-    text: str
-    created_at: datetime
+    execution_state_id: str
     parent_id: Optional[str] = None
+    author_name: str
+    author_type: str
+    date: datetime
 
 
 class WorkflowExecutionState(BaseModel):
@@ -41,6 +43,7 @@ class WorkflowExecutionState(BaseModel):
     status: WorkflowExecutionStatusEnum = WorkflowExecutionStatusEnum.NOT_STARTED
     started_at: Optional[datetime] = None
     completed_at: Optional[datetime] = None
+    thoughts: Optional[List[WorkflowExecutionStateThought]] = None
 
 
 class WorkflowExecutionStateOutput(BaseModel):

codemie_sdk_python-0.1.186/src/codemie_sdk/models/workflow_thoughts.py

@@ -0,0 +1,26 @@
+"""Workflow execution thoughts models."""
+
+from datetime import datetime
+from typing import List, Optional
+
+from pydantic import BaseModel, ConfigDict
+
+
+class WorkflowExecutionThought(BaseModel):
+    """Model for workflow execution thought."""
+
+    model_config = ConfigDict(populate_by_name=True)
+
+    id: str
+    execution_state_id: str
+    parent_id: Optional[str] = None
+    author_name: str
+    author_type: str
+    input_text: str
+    content: str
+    date: datetime
+    children: List["WorkflowExecutionThought"] = []
+
+
+# Update forward references for recursive model
+WorkflowExecutionThought.model_rebuild()

{codemie_sdk_python-0.1.184 → codemie_sdk_python-0.1.186}/src/codemie_sdk/services/workflow_execution.py

@@ -4,6 +4,7 @@ from typing import List, Optional
 
 from ..models.common import PaginationParams
 from ..models.workflow import WorkflowExecution
+from ..models.workflow_thoughts import WorkflowExecutionThought
 from .workflow_execution_state import WorkflowExecutionStateService
 from ..utils import ApiRequestHandler
 
@@ -112,3 +113,21 @@ class WorkflowExecutionService:
         return self._api.put(
             f"/v1/workflows/{self._workflow_id}/executions/{execution_id}/resume", dict
         )
+
+    def get_thoughts(
+        self, execution_id: str, thought_ids: List[str]
+    ) -> List[WorkflowExecutionThought]:
+        """Get detailed thoughts information for specific thought IDs.
+
+        Args:
+            execution_id: ID of the execution to get thoughts for
+            thought_ids: List of thought IDs to retrieve detailed information for
+
+        Returns:
+            List[WorkflowExecutionThought]: List of detailed thought objects
+        """
+        return self._api.post(
+            f"/v1/workflows/{self._workflow_id}/executions/{execution_id}/thoughts",
+            List[WorkflowExecutionThought],
+            json_data=thought_ids,
+        )