remove underscore_attrs_are_private

Signed-off-by: ssbuild <[email protected]>
ssbuild · Nov 17, 2023 · 0639350 · 0639350
1 parent 6bf1283
commit 0639350
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/serving/openai_api/custom.py b/serving/openai_api/custom.py
@@ -3,6 +3,7 @@
 # @Time    : 2023/8/11 15:31
 from typing import Optional, Union, List, Dict, Any
 from pydantic import BaseModel
+from pydantic.v1 import PrivateAttr
 
 __all__ = [
     'CustomChatParams'
@@ -45,10 +46,11 @@ class CustomChatParams(BaseModel):
     functions: Optional[List[Dict[str, Any]]] = None
     function_call: Union[str, Dict[str, str]] = "auto"
 
-    class Config:
-        underscore_attrs_are_private = True
-#   私有成员
-    _model_type: Optional[str] = None
+#     class Config:
+#         underscore_attrs_are_private = True
+#     _model_type: Optional[str] = None
+
+    _model_type: Optional[str] = PrivateAttr(default=None)
 
     @property
     def model_type(self):