Handle 404 from @databricks/sdk-experimental

Author: janniklasrose

packages/databricks-vscode/src/run/WorkflowRunner.ts

@@ -221,14 +221,18 @@ export class WorkflowRunner implements Disposable {
         try {
             if (e instanceof ApiError) {
                 logger.error("API error while running workflow:", e.message);
+                const response =
+                    typeof e.response === "string"
+                        ? JSON.parse(e.response)
+                        : e.response;
                 panel.showError({
                     message: e.message,
                     stack:
-                        "error_trace" in e.response
-                            ? new Convert().toHtml(e.response.error_trace)
+                        "error_trace" in response
+                            ? new Convert().toHtml(response.error_trace)
                             : undefined,
                 });
-                panel.showStdoutResult(e.response.logs || "");
+                panel.showStdoutResult(response.logs || "");
             } else {
                 logger.error("Unexpected error while running workflow:", e);
                 panel.showError({

packages/databricks-vscode/src/sdk-extensions/wsfs/WorkspaceFsEntity.ts

@@ -122,7 +122,10 @@ export abstract class WorkspaceFsEntity {
             return this;
         } catch (e: unknown) {
             if (e instanceof ApiError) {
-                if (e.errorCode === "RESOURCE_DOES_NOT_EXIST") {
+                if (
+                    e.errorCode === "RESOURCE_DOES_NOT_EXIST" ||
+                    e.statusCode === 404
+                ) {
                     return undefined;
                 }
             }
@@ -154,7 +157,7 @@ export abstract class WorkspaceFsEntity {
         } catch (e) {
             if (
                 e instanceof ApiError &&
-                e.errorCode === "RESOURCE_DOES_NOT_EXIST"
+                (e.errorCode === "RESOURCE_DOES_NOT_EXIST" || e.statusCode === 404)
             ) {
                 return undefined;
             }