fix: make models serializable for telemetry

Author: philipph-askui

src/askui/models/models.py

@@ -195,6 +195,32 @@ def act(
         """  # noqa: E501
         raise NotImplementedError
 
+    def to_telemetry_dict(self) -> dict[str, str]:
+        """Return a JSON-serializable representation for telemetry.
+
+        For ActModels, this includes the model_id and MessagesApi class name.
+
+        Returns:
+            dict[str, str]: A dictionary with telemetry information about this model.
+                Includes: "type", "class", "model_id", "messages_api".
+
+        Example:
+            ```python
+            {
+                "type": "act_model",
+                "class": "AskUIAgent",
+                "model_id": "claude-sonnet-4-20250514",
+                "messages_api": "AnthropicMessagesApi"
+            }
+            ```
+        """
+        return {
+            "type": "act_model",
+            "class": self.__class__.__name__,
+            "model_id": self.__getattribute__("_model_id" or "-"),
+            "messages_api": self._messages_api.__class__.__name__,  # type: ignore[attr-defined]
+        }
+
 
 class GetModel(abc.ABC):
     """Abstract base class for models that can extract information from images and PDFs.
@@ -247,6 +273,28 @@ def get(
         """
         raise NotImplementedError
 
+    def to_telemetry_dict(self) -> dict[str, str]:
+        """Return a JSON-serializable representation for telemetry.
+
+        For GetModels, this includes the class name.
+
+        Returns:
+            dict[str, str]: A dictionary with telemetry information about this model.
+                Includes: "type", "class".
+
+        Example:
+            ```python
+            {
+                "type": "get_model",
+                "class": "AskUiGeminiGetModel"
+            }
+            ```
+        """
+        return {
+            "type": "get_model",
+            "class": self.__class__.__name__,
+        }
+
 
 class LocateModel(abc.ABC):
     """Abstract base class for models that can locate UI elements in images.
@@ -312,6 +360,28 @@ def locate_all_elements(
         """
         raise NotImplementedError
 
+    def to_telemetry_dict(self) -> dict[str, str]:
+        """Return a JSON-serializable representation for telemetry.
+
+        For LocateModels, this includes the class name.
+
+        Returns:
+            dict[str, str]: A dictionary with telemetry information about this model.
+                Includes: "type", "class".
+
+        Example:
+            ```python
+            {
+                "type": "locate_model",
+                "class": "AskUiPtaLocateModel"
+            }
+            ```
+        """
+        return {
+            "type": "locate_model",
+            "class": self.__class__.__name__,
+        }
+
 
 Model = ActModel | GetModel | LocateModel
 """Union type of all abstract model classes.

src/askui/telemetry/telemetry.py

@@ -204,25 +204,24 @@ def wrapper(*args: P.args, **kwargs: P.kwargs) -> R:  # noqa: C901
                 )
                 if exclude_first_arg:
                     processed_args = processed_args[1:] if processed_args else ()
-                processed_args = tuple(
-                    arg.model_dump()
-                    if isinstance(arg, BaseModel)
-                    else str(arg)
-                    if inspect.isclass(arg)
-                    else arg
-                    for arg in processed_args
-                )
+
+                def _serialize_arg(arg: Any) -> Any:
+                    if to_telemetry := getattr(arg, "to_telemetry_dict", None):
+                        if callable(to_telemetry):
+                            return to_telemetry()
+                    if isinstance(arg, BaseModel):
+                        return arg.model_dump()
+                    if inspect.isclass(arg):
+                        return str(arg)
+                    return arg
+
+                processed_args = tuple(_serialize_arg(arg) for arg in processed_args)
                 processed_kwargs = {
                     k: v if k not in _exclude else self._EXCLUDE_MASK
                     for k, v in kwargs.items()
                 }
                 processed_kwargs = {
-                    k: v.model_dump()
-                    if isinstance(v, BaseModel)
-                    else str(v)
-                    if inspect.isclass(v)
-                    else v
-                    for k, v in processed_kwargs.items()
+                    k: _serialize_arg(v) for k, v in processed_kwargs.items()
                 }
                 attributes: dict[str, Any] = {
                     "module": module,