Add ArtifactEvent model and update workflows to use it

Author: leehuwuj

python/llama-index-server/examples/artifact/code_workflow.py

@@ -18,6 +18,7 @@
 )
 from llama_index.server.api.models import (
     Artifact,
+    ArtifactEvent,
     ArtifactType,
     ChatRequest,
     CodeArtifactData,
@@ -306,8 +307,7 @@ async def generate_artifact(
         )
         # To show the Canvas panel for the artifact
         ctx.write_event_to_stream(
-            UIEvent(
-                type="artifact",
+            ArtifactEvent(
                 data=Artifact(
                     type=ArtifactType.CODE,
                     created_at=int(time.time()),

python/llama-index-server/examples/artifact/document_workflow.py

@@ -18,6 +18,7 @@
 )
 from llama_index.server.api.models import (
     Artifact,
+    ArtifactEvent,
     ArtifactType,
     ChatRequest,
     DocumentArtifactData,
@@ -274,8 +275,7 @@ async def generate_artifact(
         )
         # To show the Canvas panel for the artifact
         ctx.write_event_to_stream(
-            UIEvent(
-                type="artifact",
+            ArtifactEvent(
                 data=Artifact(
                     type=ArtifactType.DOCUMENT,
                     created_at=int(time.time()),

python/llama-index-server/llama_index/server/api/models.py

@@ -193,3 +193,14 @@ def from_message(cls, message: ChatAPIMessage) -> Optional["Artifact"]:
                     )
 
         return None
+
+
+class ArtifactEvent(Event):
+    type: str = "artifact"
+    data: Artifact
+
+    def to_response(self) -> dict:
+        return {
+            "type": self.type,
+            "data": self.data.model_dump(),
+        }