From 0f85e3557ba12f1bf0175ebe491452c34ec2a6f7 Mon Sep 17 00:00:00 2001
From: Hash Brown <hi@xzd.me>
Date: Tue, 26 Nov 2024 10:23:03 +0800
Subject: [PATCH] fix: site icon not showing (#11094)

---
 api/libs/helper.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/api/libs/helper.py b/api/libs/helper.py
index 023240a9a4..b98a4829e8 100644
--- a/api/libs/helper.py
+++ b/api/libs/helper.py
@@ -31,12 +31,12 @@ class AppIconUrlField(fields.Raw):
         if obj is None:
             return None
 
-        from models.model import App, IconType
+        from models.model import App, IconType, Site
 
         if isinstance(obj, dict) and "app" in obj:
             obj = obj["app"]
 
-        if isinstance(obj, App) and obj.icon_type == IconType.IMAGE.value:
+        if isinstance(obj, App | Site) and obj.icon_type == IconType.IMAGE.value:
             return file_helpers.get_signed_file_url(obj.icon)
         return None