refactor: code cleanup

providers/shared/model_response.py

@@ -0,0 +1,26 @@
+"""Dataclass used to normalise provider SDK responses."""
+
+from dataclasses import dataclass, field
+from typing import Any
+
+from .provider_type import ProviderType
+
+__all__ = ["ModelResponse"]
+
+
+@dataclass
+class ModelResponse:
+    """Portable representation of a provider completion."""
+
+    content: str
+    usage: dict[str, int] = field(default_factory=dict)
+    model_name: str = ""
+    friendly_name: str = ""
+    provider: ProviderType = ProviderType.GOOGLE
+    metadata: dict[str, Any] = field(default_factory=dict)
+
+    @property
+    def total_tokens(self) -> int:
+        """Return the total token count if the provider reported usage data."""
+
+        return self.usage.get("total_tokens", 0)