fix after review

Author: andy-fisher

src/yandex_cloud_ml_sdk/_runs/domain.py

@@ -27,7 +27,7 @@
 
 class BaseRuns(BaseDomain, Generic[RunTypeT]):
     """
-    Abstract class for Runs operations with configurable RunTypeT.
+    Class for Runs operations.
     Provides core functionality for managing assistant execution in streams.
 
     For usage examples see `runs example <https://github.com/yandex-cloud/yandex-cloud-ml-sdk/blob/master/examples/{link}/assistants/runs.py>`_.

src/yandex_cloud_ml_sdk/_runs/result.py

@@ -13,7 +13,9 @@
 from yandex_cloud_ml_sdk._models.completions.result import Usage
 from yandex_cloud_ml_sdk._tools.tool_call import HaveToolCalls, ToolCallTypeT
 from yandex_cloud_ml_sdk._tools.tool_call_list import ProtoAssistantToolCallList, ToolCallList
+from yandex_cloud_ml_sdk._types.operation import BaseOperationStatus
 from yandex_cloud_ml_sdk._types.result import BaseProtoResult, ProtoMessageTypeT_contra
+from yandex_cloud_ml_sdk._utils.doc import doc_from
 
 from .status import BaseRunStatus, RunStatus, StreamEvent
 
@@ -45,24 +47,18 @@ def _from_proto(cls, *, proto: ProtoMessageTypeT_contra, sdk: BaseSDK) -> BaseRu
         pass
 
     @property
+    @doc_from(BaseOperationStatus.is_running)
     def is_running(self) -> bool:
-        """
-        Check if run is in progress.
-        """
         return self.status.is_running
 
     @property
+    @doc_from(BaseOperationStatus.is_succeeded)
     def is_succeeded(self) -> bool:
-        """
-        Check if run completed successfully.
-        """
         return self.status.is_succeeded
 
     @property
+    @doc_from(BaseOperationStatus.is_failed)
     def is_failed(self) -> bool:
-        """
-        Check if run failed.
-        """
         return self.status.is_failed
 
     @property

src/yandex_cloud_ml_sdk/_runs/status.py

@@ -7,14 +7,15 @@
 from yandex.cloud.ai.assistants.v1.runs.run_service_pb2 import StreamEvent as ProtoStreamEvent
 
 from yandex_cloud_ml_sdk._types.operation import BaseOperationStatus
+from yandex_cloud_ml_sdk._utils.doc import doc_from
 from yandex_cloud_ml_sdk._utils.proto import ProtoEnumBase
 
 
 # pylint: disable=abstract-method
 class BaseRunStatus(BaseOperationStatus):
     pass
     """
-    Base class for run status enumerations.
+    Class for run status enumerations.
     """
 
 
@@ -38,24 +39,18 @@ class RunStatus(BaseRunStatus, ProtoEnumBase, IntEnum):
     TOOL_CALLS = ProtoRunState.TOOL_CALLS
 
     @property
+    @doc_from(BaseOperationStatus.is_running)
     def is_running(self) -> bool:
-        """
-        Check if execution is still in progress.
-        """
         return self in (self.IN_PROGRESS, self.PENDING)
 
     @property
+    @doc_from(BaseOperationStatus.is_succeeded)
     def is_succeeded(self) -> bool:
-        """
-        Check if execution completed successfully.
-        """
         return self in (self.COMPLETED, self.TOOL_CALLS)
 
     @property
+    @doc_from(BaseOperationStatus.is_failed)
     def is_failed(self) -> bool:
-        """
-        Check if execution failed.
-        """
         return self is self.FAILED
 
 
@@ -77,22 +72,16 @@ class StreamEvent(BaseRunStatus, ProtoEnumBase, IntEnum):
     TOOL_CALLS = ProtoStreamEvent.TOOL_CALLS
 
     @property
+    @doc_from(BaseOperationStatus.is_running)
     def is_running(self) -> bool:
-        """
-        Check if execution is still in progress.
-        """
         return self is self.PARTIAL_MESSAGE
 
     @property
+    @doc_from(BaseOperationStatus.is_succeeded)
     def is_succeeded(self) -> bool:
-        """
-        Check if execution completed successfully.
-        """
         return self in (self.DONE, self.TOOL_CALLS)
 
     @property
+    @doc_from(BaseOperationStatus.is_failed)
     def is_failed(self) -> bool:
-        """
-        Check if execution failed.
-        """
         return self is self.ERROR

src/yandex_cloud_ml_sdk/_types/operation.py

@@ -33,22 +33,37 @@
 class BaseOperationStatus:
     @property
     def is_running(self) -> bool:
+        """
+        Check if operation execution is still in progress.
+        """
         raise NotImplementedError()
 
     @property
     def is_succeeded(self) -> bool:
+        """
+        Check if operation execution completed successfully.
+        """
         raise NotImplementedError()
 
     @property
     def is_failed(self) -> bool:
+        """
+        Check if operation execution failed.
+        """
         raise NotImplementedError()
 
     @property
     def is_finished(self) -> bool:
+        """
+        Check if operation execution finished.
+        """
         return self.is_succeeded or self.is_failed
 
     @property
     def status_name(self) -> str:
+        """
+        Get operation execution status name.
+        """
         if self.is_succeeded:
             return 'success'
         if self.is_failed: