vendor: prefixed icon names are included in the IconName type

Author: jiwon-oai

chatkit/types.py

@@ -3,7 +3,7 @@
 from datetime import datetime
 from typing import Any, Generic, Literal
 
-from pydantic import AnyUrl, BaseModel, Field
+from pydantic import AnyUrl, BaseModel, Field, StringConstraints
 from typing_extensions import Annotated, TypeIs, TypeVar
 
 from chatkit.errors import ErrorCode
@@ -829,70 +829,73 @@ class EntitySource(SourceBase):
 FeedbackKind = Literal["positive", "negative"]
 """Literal type for feedback sentiment."""
 
-
-IconName = Literal[
-    "agent",
-    "analytics",
-    "atom",
-    "batch",
-    "bolt",
-    "book-open",
-    "book-closed",
-    "book-clock",
-    "bug",
-    "calendar",
-    "chart",
-    "check",
-    "check-circle",
-    "check-circle-filled",
-    "chevron-left",
-    "chevron-right",
-    "circle-question",
-    "compass",
-    "confetti",
-    "cube",
-    "desktop",
-    "document",
-    "dot",
-    "dots-horizontal",
-    "dots-vertical",
-    "empty-circle",
-    "external-link",
-    "globe",
-    "keys",
-    "lab",
-    "images",
-    "info",
-    "lifesaver",
-    "lightbulb",
-    "mail",
-    "map-pin",
-    "maps",
-    "mobile",
-    "name",
-    "notebook",
-    "notebook-pencil",
-    "page-blank",
-    "phone",
-    "play",
-    "plus",
-    "profile",
-    "profile-card",
-    "reload",
-    "star",
-    "star-filled",
-    "search",
-    "sparkle",
-    "sparkle-double",
-    "square-code",
-    "square-image",
-    "square-text",
-    "suitcase",
-    "settings-slider",
-    "user",
-    "wreath",
-    "write",
-    "write-alt",
-    "write-alt2",
-]
-"""Literal names of supported progress icons."""
+VendorIconName = Annotated[str, StringConstraints(pattern=r"^vendor:")]
+
+IconName = (
+    Literal[
+        "agent",
+        "analytics",
+        "atom",
+        "batch",
+        "bolt",
+        "book-open",
+        "book-closed",
+        "book-clock",
+        "bug",
+        "calendar",
+        "chart",
+        "check",
+        "check-circle",
+        "check-circle-filled",
+        "chevron-left",
+        "chevron-right",
+        "circle-question",
+        "compass",
+        "confetti",
+        "cube",
+        "desktop",
+        "document",
+        "dot",
+        "dots-horizontal",
+        "dots-vertical",
+        "empty-circle",
+        "external-link",
+        "globe",
+        "keys",
+        "lab",
+        "images",
+        "info",
+        "lifesaver",
+        "lightbulb",
+        "mail",
+        "map-pin",
+        "maps",
+        "mobile",
+        "name",
+        "notebook",
+        "notebook-pencil",
+        "page-blank",
+        "phone",
+        "play",
+        "plus",
+        "profile",
+        "profile-card",
+        "reload",
+        "star",
+        "star-filled",
+        "search",
+        "sparkle",
+        "sparkle-double",
+        "square-code",
+        "square-image",
+        "square-text",
+        "suitcase",
+        "settings-slider",
+        "user",
+        "wreath",
+        "write",
+        "write-alt",
+        "write-alt2",
+    ]
+    | VendorIconName
+)

tests/test_types.py

@@ -0,0 +1,22 @@
+import pytest
+from pydantic import TypeAdapter, ValidationError
+
+from chatkit.types import IconName
+
+
+def test_vendor_icon_names_are_valid():
+    """Icon names prefixed with `vendor:` are valid."""
+    TypeAdapter(IconName).validate_python("vendor:icon-name")
+    TypeAdapter(IconName).validate_python("vendor:another-icon-name")
+
+
+def test_literal_icon_names_are_valid():
+    """Spot check some literal icon names."""
+    TypeAdapter(IconName).validate_python("book-open")
+    TypeAdapter(IconName).validate_python("phone")
+    TypeAdapter(IconName).validate_python("user")
+
+
+def test_invalid_icon_names_are_rejected():
+    with pytest.raises(ValidationError):
+        TypeAdapter(IconName).validate_python("invalid-icon")