Update ai_event.py

ai_event.py

@@ -1,10 +1,13 @@
 from datetime import datetime
-from typing import Optional, List, Dict, Any
+from typing import Optional, List, Literal
 from pydantic import Field
 from agentic_reliability_framework.core.models.event import ReliabilityEvent
 
 class AIEvent(ReliabilityEvent):
-    """Extended event for AI system monitoring."""
+    """Extended event for AI system monitoring across multiple modalities (text, image, audio, etc.)."""
+    action_category: Literal["chat", "code", "summary", "image", "audio", "iot"] = Field(
+        ..., description="Type of AI task or domain"
+    )
     model_name: str = Field(..., description="Identifier of the AI model used")
     model_version: str = Field(..., description="Version/tag of the model")
     prompt: str = Field(..., description="Input prompt/query")