codemie-sdk-python 0.1.183__py3-none-any.whl → 0.1.185__py3-none-any.whl

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/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/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,
+        )

{codemie_sdk_python-0.1.183.dist-info → codemie_sdk_python-0.1.185.dist-info}/METADATA

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

{codemie_sdk_python-0.1.183.dist-info → codemie_sdk_python-0.1.185.dist-info}/RECORD

@@ -14,7 +14,8 @@ codemie_sdk/models/llm.py,sha256=ppb9-1dx1UFhRuJpSR3ij7H6Pfhe9nO4C4BEOIbToy4,119
 codemie_sdk/models/task.py,sha256=J4ZFRY3s8qBGrqB5NLQF0rMbInLh4s7OEZ0ZfmnW0Ho,1476
 codemie_sdk/models/user.py,sha256=Q0rjimZh-IbeaPfq6b6fk6ZaCtwLqWHEIlU863suCS4,1777
 codemie_sdk/models/workflow.py,sha256=qfk0rBJnFUMpcEDq_E5GB3hzYKbe_bb2NYJlLZJwUEE,2453
-codemie_sdk/models/workflow_state.py,sha256=CMYFQZ7sy4QxmnWmc83TFfqP7TG_3rW5MdH5fxsS9kY,1251
+codemie_sdk/models/workflow_state.py,sha256=okEMKzkiBU3GHs9VNBoiEMOnOeZRMXGYtpL0NYSg-FY,1374
+codemie_sdk/models/workflow_thoughts.py,sha256=CjHaIPgca9kQAjpoq8IpX4MuDeql1SvaoKS5RmyU2SE,616
 codemie_sdk/services/assistant.py,sha256=BsTR_uoPoZtta6icfw2M0Ea6w2VIhYndDUVqsHJuZJo,7888
 codemie_sdk/services/conversation.py,sha256=Ss0mkGaBi4u8-YKzhRYUJOfoFqJueDp9JurI5DRCGT8,2566
 codemie_sdk/services/datasource.py,sha256=d8LImpireGOnXKvj4h8iafq_ZQMn177_GGLLiE5g7wU,7405
@@ -23,10 +24,10 @@ codemie_sdk/services/llm.py,sha256=0-e4_7RvLHs2giCyoQ5U4KDTh6p5VXgPKNxnDP9ZDFU,1
 codemie_sdk/services/task.py,sha256=3e9t8_LMkR4xfeMBwMCo7ZF87PxPS-ZbzDg85ilda2M,1031
 codemie_sdk/services/user.py,sha256=7B-Qw451qKPD5Io6qLda-kbFDaPRQ3TamJamiGwCQu4,1013
 codemie_sdk/services/workflow.py,sha256=0d5grAsBIH2NZY66UbVhlDqEFYxEYSBByj3GNgcz_DQ,4965
-codemie_sdk/services/workflow_execution.py,sha256=aGoT3rdTmh5-doAsrmBBjLEuOfvL5aqeo3g9th1_aAw,3647
+codemie_sdk/services/workflow_execution.py,sha256=fvDjn0GEm8ObL8N2LUO0kk6zwTl0-DYBr2jZBx9OeLE,4377
 codemie_sdk/services/workflow_execution_state.py,sha256=tXoaa8yT09xgYEUNiHhVULe76TwGwVgZupMIUyyLxdo,2070
 codemie_sdk/utils/__init__.py,sha256=BXAJJfAzO89-kMYvWWo9wSNhSbGgF3vB1In9sePFhMM,109
 codemie_sdk/utils/http.py,sha256=ZtXb-hlwqXmjSTKRncgO-9MYoXnmHt2Dljno0kPy7mA,9506
-codemie_sdk_python-0.1.183.dist-info/METADATA,sha256=eQL8TNW2bWYh5XWMY5SGyw-dRuquv_oZyWBkftKhd7E,24494
-codemie_sdk_python-0.1.183.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-codemie_sdk_python-0.1.183.dist-info/RECORD,,
+codemie_sdk_python-0.1.185.dist-info/METADATA,sha256=I98ehWvhd2mh19qK2asNNS6TNkXFTvOtjO1qBHpFluQ,24494
+codemie_sdk_python-0.1.185.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+codemie_sdk_python-0.1.185.dist-info/RECORD,,