Improve errors, propagating the error code

packages/restate-sdk-cloudflare-workers/patches/vm/sdk_shared_core_wasm_bindings_bg.js

@@ -1309,6 +1309,26 @@ export function __wbindgen_error_new(arg0, arg1) {
     return addHeapObject(ret);
 };
 
+export function __wbindgen_jsval_loose_eq(arg0, arg1) {
+    const ret = getObject(arg0) == getObject(arg1);
+    return ret;
+};
+
+export function __wbindgen_boolean_get(arg0) {
+    const v = getObject(arg0);
+    const ret = typeof(v) === 'boolean' ? (v ? 1 : 0) : 2;
+    return ret;
+};
+
+export function __wbindgen_string_new(arg0, arg1) {
+    const ret = getStringFromWasm0(arg0, arg1);
+    return addHeapObject(ret);
+};
+
+export function __wbg_set_f975102236d3c502(arg0, arg1, arg2) {
+    getObject(arg0)[takeObject(arg1)] = takeObject(arg2);
+};
+
 export function __wbg_new_abda76e883ba8a5f() {
     const ret = new Error();
     return addHeapObject(ret);
@@ -1334,17 +1354,6 @@ export function __wbg_error_f851667af71bcfc6(arg0, arg1) {
     }
 };
 
-export function __wbindgen_jsval_loose_eq(arg0, arg1) {
-    const ret = getObject(arg0) == getObject(arg1);
-    return ret;
-};
-
-export function __wbindgen_boolean_get(arg0) {
-    const v = getObject(arg0);
-    const ret = typeof(v) === 'boolean' ? (v ? 1 : 0) : 2;
-    return ret;
-};
-
 export function __wbg_String_88810dfeb4021902(arg0, arg1) {
     const ret = String(getObject(arg1));
     const ptr1 = passStringToWasm0(ret, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
@@ -1353,11 +1362,6 @@ export function __wbg_String_88810dfeb4021902(arg0, arg1) {
     getDataViewMemory0().setInt32(arg0 + 4 * 0, ptr1, true);
 };
 
-export function __wbindgen_string_new(arg0, arg1) {
-    const ret = getStringFromWasm0(arg0, arg1);
-    return addHeapObject(ret);
-};
-
 export function __wbg_getwithrefkey_5e6d9547403deab8(arg0, arg1) {
     const ret = getObject(arg0)[getObject(arg1)];
     return addHeapObject(ret);

packages/restate-sdk/src/context_impl.ts

@@ -29,6 +29,8 @@ import type * as vm from "./endpoint/handlers/vm/sdk_shared_core_wasm_bindings.j
 import {
   ensureError,
   INTERNAL_ERROR_CODE,
+  RestateError,
+  SUSPENDED_ERROR_CODE,
   TerminalError,
   TimeoutError,
   UNKNOWN_ERROR_CODE,
@@ -803,8 +805,14 @@ export class ContextImpl implements ObjectContext, WorkflowContext {
   }
 
   handleInvocationEndError(e: unknown) {
-    const err = ensureError(e);
-    this.coreVm.notify_error(err.message, err.stack);
+    const error = ensureError(e);
+    if (
+      !(error instanceof RestateError) ||
+      error.code != SUSPENDED_ERROR_CODE
+    ) {
+      this.console.warn("Function completed with an error.\n", error);
+    }
+    this.coreVm.notify_error(error.message, error.stack);
 
     // From now on, no progress will be made.
     this.invocationEndPromise.resolve();

packages/restate-sdk/src/endpoint/handlers/generic.ts

@@ -9,7 +9,12 @@
  * https://github.com/restatedev/sdk-typescript/blob/main/LICENSE
  */
 
-import { ensureError, TerminalError } from "../../types/errors.js";
+import {
+  ensureError,
+  RestateError,
+  SUSPENDED_ERROR_CODE,
+  TerminalError,
+} from "../../types/errors.js";
 import type { ProtocolMode } from "../../types/discovery.js";
 import type { ComponentHandler } from "../../types/components.js";
 import { parseUrlComponents } from "../../types/components.js";
@@ -311,7 +316,12 @@ export class GenericHandler implements RestateHandler {
       })
       .catch((e) => {
         const error = ensureError(e);
-        console.warn("Function completed with an error.\n", error);
+        if (
+          !(error instanceof RestateError) ||
+          error.code != SUSPENDED_ERROR_CODE
+        ) {
+          console.warn("Function completed with an error.\n", error);
+        }
 
         if (error instanceof TerminalError) {
           coreVm.sys_write_output_failure({

packages/restate-sdk/src/types/errors.ts

@@ -15,10 +15,19 @@ export const INTERNAL_ERROR_CODE = 500;
 export const TIMEOUT_ERROR_CODE = 408;
 export const UNKNOWN_ERROR_CODE = 500;
 
+// From shared core!
+export const SUSPENDED_ERROR_CODE = 599;
+
 export function ensureError(e: unknown): Error {
   if (e instanceof Error) {
     return e;
   }
+  if (typeof e == "object" && e != null && "code" in e && "message" in e) {
+    // This is an error from the VM
+    return new RestateError(e.message as string, {
+      errorCode: e.code as number,
+    });
+  }
 
   let msg;
   try {