WIP: unexpected error

Author: mnicstruwig

common/common/models.py

@@ -1,6 +1,6 @@
 from typing import Any, Literal
 from uuid import UUID
-from pydantic import BaseModel, Field, JsonValue, field_validator
+from pydantic import BaseModel, Field, JsonValue, field_validator, model_validator
 from enum import Enum
 import json
 
@@ -153,3 +153,30 @@ class FunctionCallSSEData(BaseModel):
 class FunctionCallSSE(BaseSSE):
     event: Literal["copilotFunctionCall"] = "copilotFunctionCall"
     data: FunctionCallSSEData
+
+
+
+class StatusUpdateSSEData(BaseModel):
+    eventType: Literal["INFO", "WARNING", "ERROR"]
+    message: str
+    group: Literal["reasoning"] = "reasoning"
+    details: list[dict[str, str | int | float | None]] | None = None
+    #artifacts: list[ClientArtifact] | None = None
+
+    @model_validator(mode="before")
+    @classmethod
+    def exclude_fields(cls, values):
+        # Exclude these fields from being in the "details" field.  (since this
+        # pollutes the JSON output)
+        _exclude_fields = []
+
+        if details := values.get("details"):
+            for detail in details:
+                for key in list(detail.keys()):
+                    if key.lower() in _exclude_fields:
+                        detail.pop(key, None)
+        return values
+
+class StatusUpdateSSE(BaseSSE):
+    event: Literal["copilotStatusUpdate"] = "copilotStatusUpdate"
+    data: StatusUpdateSSEData