fix: 修复 Store 里的插件实例构造问题

BigOrangeQWQ · Aug 27, 2024 · b2db73a · b2db73a
1 parent ad0a6f5
commit b2db73a
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/utils/store_test/models.py b/src/utils/store_test/models.py
@@ -38,8 +38,10 @@ class StorePlugin(BaseModel):
     @field_validator("tags", mode="before")
     @classmethod
     def tags_validator(cls, v: list[dict[str, Any]]):
-        """商店里的标签数据不需要验证，直接构造即可"""
-        return [Tag.model_construct(**tag) for tag in v]
+        return [
+            Tag.model_construct(label=tag["label"], color=Color(tag["color"]))
+            for tag in v
+        ]
 
 
 class Metadata(BaseModel):
@@ -83,8 +85,10 @@ def metadata(self) -> Metadata:
     @field_validator("tags", mode="before")
     @classmethod
     def tags_validator(cls, v: list[dict[str, Any]]):
-        """商店里的标签数据不需要验证，直接构造即可"""
-        return [Tag.model_construct(**tag) for tag in v]
+        return [
+            Tag.model_construct(label=tag["label"], color=Color(tag["color"]))
+            for tag in v
+        ]
 
 
 class TestResult(BaseModel):