Only return None for the correct error

Signed-off-by: Elena Kolevska <[email protected]>

Author: elena-kolevska

examples/workflow/monitor.py

@@ -63,7 +63,12 @@ def send_alert(ctx, message: str):
 
     wf_client = wf.DaprWorkflowClient()
     job_id = 'job1'
-    status = wf_client.get_workflow_state(job_id)
+    status = None
+    try:
+        status = wf_client.get_workflow_state(job_id)
+    except Exception:
+        pass
+
     if not status or status.runtime_status.name != 'RUNNING':
         # TODO update to use reuse_id_policy
         instance_id = wf_client.schedule_new_workflow(

ext/dapr-ext-workflow/dapr/ext/workflow/dapr_workflow_client.py

@@ -24,6 +24,7 @@
 from dapr.ext.workflow.workflow_state import WorkflowState
 from dapr.ext.workflow.workflow_context import Workflow
 from dapr.ext.workflow.util import getAddress
+from grpc import RpcError
 
 from dapr.clients import DaprInternalError
 from dapr.clients.http.client import DAPR_API_TOKEN_HEADER
@@ -133,9 +134,12 @@ def get_workflow_state(
         try:
             state = self.__obj.get_orchestration_state(instance_id, fetch_payloads=fetch_payloads)
             return WorkflowState(state) if state else None
-        except Exception as error:
-            self._logger.error(f'Error fetching workflow state: {error}')
-            return None
+        except RpcError as error:
+            if 'no such instance exists' in error.details():
+                self._logger.warning(f"Workflow instance not found: {instance_id}")
+                return None
+            self._logger.error(f"Unhandled RPC error while fetching workflow state: {error.code()} - {error.details()}")
+            raise error
 
     def wait_for_workflow_start(
         self, instance_id: str, *, fetch_payloads: bool = False, timeout_in_seconds: int = 60