improve api docs

2023-06-15 21:52:22 +08:00
parent 3f77762fda
commit 21c3009945
4 changed files with 55 additions and 3 deletions
--- a/backend-python/utils/rwkv.py
+++ b/backend-python/utils/rwkv.py
@@ -201,6 +201,17 @@ 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:
+        schema_extra = {
+            "example": {
+                "max_tokens": 1000,
+                "temperature": 1.2,
+                "top_p": 0.5,
+                "presence_penalty": 0.4,
+                "frequency_penalty": 0.4,
+            }
+        }
+

 def set_rwkv_config(model: RWKV, body: ModelConfigBody):
    if body.max_tokens is not None: