add error code in handling

PostHog · Apr 17, 2024 · dc6e3fd · dc6e3fd
1 parent b2773cb
commit dc6e3fd
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/posthog/warehouse/api/external_data_source.py b/posthog/warehouse/api/external_data_source.py
@@ -391,8 +391,7 @@ def reload(self, request: Request, *args: Any, **kwargs: Any):
                 try:
                     trigger_external_data_workflow(schema)
                 except temporalio.service.RPCError as e:
-                    # schedule doesn't exist
-                    if e.message == "sql: no rows in result set":
+                    if e.status == temporalio.service.RPCStatusCode.NOT_FOUND:
                         sync_external_data_job_workflow(schema, create=True)
 
                 except Exception as e: