feat: make interrupt models serializable

Author: radu-mocanu

src/uipath/models/__init__.py

@@ -4,7 +4,13 @@
 from .connections import Connection, ConnectionToken
 from .context_grounding import ContextGroundingQueryResponse
 from .exceptions import IngestionInProgressException
-from .interrupt_models import CreateAction, InvokeProcess, WaitAction, WaitJob
+from .interrupt_models import (
+    CreateAction,
+    InvokeProcess,
+    Serializable,
+    WaitAction,
+    WaitJob,
+)
 from .job import Job
 from .processes import Process
 from .queues import (
@@ -34,4 +40,5 @@
     "WaitAction",
     "CreateAction",
     "IngestionInProgressException",
+    "Serializable",
 ]

src/uipath/models/interrupt_models.py

@@ -1,3 +1,4 @@
+from abc import ABC, abstractmethod
 from typing import Any, Dict, Optional
 
 from pydantic import BaseModel
@@ -6,23 +7,53 @@
 from .job import Job
 
 
-class InvokeProcess(BaseModel):
+class Serializable(ABC):
+    @abstractmethod
+    def serialize(self) -> str:
+        pass
+
+
+class InvokeProcess(BaseModel, Serializable):
     name: str
     input_arguments: Optional[Dict[str, Any]]
 
+    def serialize(self) -> str:
+        try:
+            return self.model_dump_json()
+        except ValueError:
+            return str(self)
+
 
-class WaitJob(BaseModel):
+class WaitJob(BaseModel, Serializable):
     job: Job
 
+    def serialize(self) -> str:
+        try:
+            return self.model_dump_json()
+        except ValueError:
+            return str(self)
 
-class CreateAction(BaseModel):
+
+class CreateAction(BaseModel, Serializable):
     name: Optional[str] = None
     key: Optional[str] = None
     title: str
     data: Optional[Dict[str, Any]] = None
     app_version: Optional[int] = 1
     assignee: Optional[str] = ""
 
+    def serialize(self) -> str:
+        try:
+            return self.model_dump_json()
+        except ValueError:
+            return str(self)
+
 
-class WaitAction(BaseModel):
+class WaitAction(BaseModel, Serializable):
     action: Action
+
+    def serialize(self) -> str:
+        try:
+            return self.model_dump_json()
+        except ValueError:
+            return str(self)