fix /docs default api params (Pydantic v2)

2023-11-07 22:53:11 +08:00
parent 26d4c9f0ed
commit cfa3669f6f
4 changed files with 27 additions and 18 deletions
--- a/backend-python/utils/rwkv.py
+++ b/backend-python/utils/rwkv.py
@@ -551,8 +551,8 @@ class ModelConfigBody(BaseModel):
    presence_penalty: float = Field(default=None, ge=-2, le=2)
    frequency_penalty: float = Field(default=None, ge=-2, le=2)

-    class Config:
-        json_schema_extra = {
+    model_config = {
+        "json_schema_extra": {
            "example": {
                "max_tokens": 1000,
                "temperature": 1.2,
@@ -561,6 +561,7 @@ class ModelConfigBody(BaseModel):
                "frequency_penalty": 0.4,
            }
        }
+    }


 def set_rwkv_config(model: AbstractRWKV, body: ModelConfigBody):