Pydantic 2 upgrade for root model

Author: jianshen92

src/clip_api_service/service.py

@@ -31,7 +31,7 @@ class Item(BaseItem):
 
 
 class ItemList(ListModel):
-    __root__: List[Item]
+    root: List[Item]
 
 
 class RankInput(BaseItem):
@@ -43,7 +43,6 @@ class RankOutput(BaseItem):
     probabilities: List[List[float]]
     cosine_similarities: List[List[float]]
 
-
 bento_model = init_model()
 logit_scale = np.exp(bento_model.info.metadata.get("logit_scale", 4.60517))
 

src/clip_api_service/utils.py

@@ -8,7 +8,7 @@
 import numpy as np
 from bentoml.exceptions import BadInput
 from PIL import Image
-from pydantic import BaseModel
+from pydantic import BaseModel, RootModel
 
 
 def base64_to_image(base64_string: str) -> Image:
@@ -56,14 +56,14 @@ def dict(self, *args, **kwargs) -> dict[str, Any]:
         return super().dict(*args, exclude_none=True, **kwargs)
 
 
-class ListModel(BaseItem):
-    __root__: List[Any]
+class ListModel(RootModel):
+    root: List[Any]
 
     def __iter__(self):
-        return iter(self.__root__)
+        return iter(self.root)
 
     def __getitem__(self, item):
-        return self.__root__[item]
+        return self.root[item]
 
     def append(self, item):
-        return self.__root__.append(item)
+        return self.root.append(item)