diff --git a/package.json b/package.json
index 1297e8c7..77b9df0d 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@reflex-platform/npm-core-sdk",
-  "version": "1.10.0-SNAPSHOT-250331125838",
+  "version": "1.10.0-SNAPSHOT-250331131408",
   "description": "npm libs from core model proto files",
   "homepage": "",
   "main": "index.ts",
diff --git a/workflow.ts b/workflow.ts
index 4a64dcba..e478c107 100644
--- a/workflow.ts
+++ b/workflow.ts
@@ -11,6 +11,7 @@ import { UnknownFieldHandler } from "@protobuf-ts/runtime";
 import type { PartialMessage } from "@protobuf-ts/runtime";
 import { reflectionMergePartial } from "@protobuf-ts/runtime";
 import { MessageType } from "@protobuf-ts/runtime";
+import { EntityType } from "./shared";
 import { RecoverType } from "./shared";
 import { ModelDataSetEnum } from "./shared";
 import { ModuleWorkflowTypeEnum } from "./shared";
@@ -713,9 +714,9 @@ export interface StartWorkflowPruneDataRequest {
      */
     projectID: string;
     /**
-     * @generated from protobuf field: api.ModelDataSetEnum Entity = 2 [json_name = "Entity"];
+     * @generated from protobuf field: api.EntityType Entity = 2 [json_name = "Entity"];
      */
-    entity: ModelDataSetEnum;
+    entity: EntityType;
 }
 /**
  * @generated from protobuf message api.StartWorkflowPruneDataResult
@@ -749,9 +750,9 @@ export interface StartWorkflowAnonymizeDataRequest {
      */
     projectID: string;
     /**
-     * @generated from protobuf field: api.ModelDataSetEnum Entity = 2 [json_name = "Entity"];
+     * @generated from protobuf field: api.EntityType Entity = 2 [json_name = "Entity"];
      */
-    entity: ModelDataSetEnum;
+    entity: EntityType;
 }
 /**
  * @generated from protobuf message api.StartWorkflowAnonymizeDataResult
@@ -3605,7 +3606,7 @@ class StartWorkflowPruneDataRequest$Type extends MessageType<StartWorkflowPruneD
     constructor() {
         super("api.StartWorkflowPruneDataRequest", [
             { no: 1, name: "ProjectID", kind: "scalar", jsonName: "ProjectID", T: 9 /*ScalarType.STRING*/ },
-            { no: 2, name: "Entity", kind: "enum", jsonName: "Entity", T: () => ["api.ModelDataSetEnum", ModelDataSetEnum] }
+            { no: 2, name: "Entity", kind: "enum", jsonName: "Entity", T: () => ["api.EntityType", EntityType, "ENTITY_TYPE_"] }
         ], { "grpc.gateway.protoc_gen_openapiv2.options.openapiv2_schema": { jsonSchema: { required: ["ProjectID", "Entity"] } } });
     }
     create(value?: PartialMessage<StartWorkflowPruneDataRequest>): StartWorkflowPruneDataRequest {
@@ -3624,7 +3625,7 @@ class StartWorkflowPruneDataRequest$Type extends MessageType<StartWorkflowPruneD
                 case /* string ProjectID = 1 [json_name = "ProjectID"];*/ 1:
                     message.projectID = reader.string();
                     break;
-                case /* api.ModelDataSetEnum Entity = 2 [json_name = "Entity"];*/ 2:
+                case /* api.EntityType Entity = 2 [json_name = "Entity"];*/ 2:
                     message.entity = reader.int32();
                     break;
                 default:
@@ -3642,7 +3643,7 @@ class StartWorkflowPruneDataRequest$Type extends MessageType<StartWorkflowPruneD
         /* string ProjectID = 1 [json_name = "ProjectID"]; */
         if (message.projectID !== "")
             writer.tag(1, WireType.LengthDelimited).string(message.projectID);
-        /* api.ModelDataSetEnum Entity = 2 [json_name = "Entity"]; */
+        /* api.EntityType Entity = 2 [json_name = "Entity"]; */
         if (message.entity !== 0)
             writer.tag(2, WireType.Varint).int32(message.entity);
         let u = options.writeUnknownFields;
@@ -3790,7 +3791,7 @@ class StartWorkflowAnonymizeDataRequest$Type extends MessageType<StartWorkflowAn
     constructor() {
         super("api.StartWorkflowAnonymizeDataRequest", [
             { no: 1, name: "ProjectID", kind: "scalar", jsonName: "ProjectID", T: 9 /*ScalarType.STRING*/ },
-            { no: 2, name: "Entity", kind: "enum", jsonName: "Entity", T: () => ["api.ModelDataSetEnum", ModelDataSetEnum] }
+            { no: 2, name: "Entity", kind: "enum", jsonName: "Entity", T: () => ["api.EntityType", EntityType, "ENTITY_TYPE_"] }
         ], { "grpc.gateway.protoc_gen_openapiv2.options.openapiv2_schema": { jsonSchema: { required: ["ProjectID", "Entity"] } } });
     }
     create(value?: PartialMessage<StartWorkflowAnonymizeDataRequest>): StartWorkflowAnonymizeDataRequest {
@@ -3809,7 +3810,7 @@ class StartWorkflowAnonymizeDataRequest$Type extends MessageType<StartWorkflowAn
                 case /* string ProjectID = 1 [json_name = "ProjectID"];*/ 1:
                     message.projectID = reader.string();
                     break;
-                case /* api.ModelDataSetEnum Entity = 2 [json_name = "Entity"];*/ 2:
+                case /* api.EntityType Entity = 2 [json_name = "Entity"];*/ 2:
                     message.entity = reader.int32();
                     break;
                 default:
@@ -3827,7 +3828,7 @@ class StartWorkflowAnonymizeDataRequest$Type extends MessageType<StartWorkflowAn
         /* string ProjectID = 1 [json_name = "ProjectID"]; */
         if (message.projectID !== "")
             writer.tag(1, WireType.LengthDelimited).string(message.projectID);
-        /* api.ModelDataSetEnum Entity = 2 [json_name = "Entity"]; */
+        /* api.EntityType Entity = 2 [json_name = "Entity"]; */
         if (message.entity !== 0)
             writer.tag(2, WireType.Varint).int32(message.entity);
         let u = options.writeUnknownFields;