Fix pyright strict errors: use Enum isinstance checks, add override decorators in tests

Author: prassanna-ravishankar

src/agentex/lib/sdk/state_machine/state_machine.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 from abc import ABC, abstractmethod
+from enum import Enum
 from typing import Any, Generic, TypeVar
 
 from agentex.lib import adk
@@ -140,13 +141,11 @@ def get_lifecycle(self) -> dict[str, Any]:
                 "waits_for_input": workflow.waits_for_input,
                 "accepts": list(workflow.accepts),
                 "transitions": [
-                    t.value if hasattr(t, "value") else str(t)
+                    t.value if isinstance(t, Enum) else str(t)
                     for t in workflow.transitions
                 ],
             })
-        initial = self._initial_state
-        if hasattr(initial, "value"):
-            initial = initial.value
+        initial: str = self._initial_state.value if isinstance(self._initial_state, Enum) else self._initial_state
 
         return {
             "states": states,

tests/lib/test_agent_card.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from enum import Enum
-from typing import Literal
+from typing import Literal, override
 from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
@@ -25,6 +25,7 @@ class WaitingWorkflow(StateWorkflow):
     accepts = ["text", "doc_upload"]
     transitions = [SampleState.PROCESSING]
 
+    @override
     async def execute(self, state_machine, state_machine_data=None):
         return SampleState.PROCESSING
 
@@ -34,6 +35,7 @@ class ProcessingWorkflow(StateWorkflow):
     accepts = ["text"]
     transitions = [SampleState.DONE, SampleState.WAITING]
 
+    @override
     async def execute(self, state_machine, state_machine_data=None):
         return SampleState.DONE
 
@@ -42,6 +44,7 @@ class DoneWorkflow(StateWorkflow):
     description = "Terminal state"
     transitions = []
 
+    @override
     async def execute(self, state_machine, state_machine_data=None):
         return SampleState.DONE
 
@@ -51,6 +54,7 @@ class SampleData(AgentexBaseModel):
 
 
 class SampleStateMachine(StateMachine[SampleData]):
+    @override
     async def terminal_condition(self):
         return self.get_current_state() == SampleState.DONE