llama_cpp server: fields for the embedding endpoint

2023-04-29 18:46:01 -07:00 · 2023-04-29 18:46:01 -07:00 · fa2a61e065
parent 8dcbf65a45
commit fa2a61e065
1 changed files with 3 additions and 1 deletions
--- a/llama_cpp/server/app.py
+++ b/llama_cpp/server/app.py
@ -189,7 +189,9 @@ def create_completion(
 class CreateEmbeddingRequest(BaseModel):
    # ignored, but marked as required for the sake of compatibility with openai's api
    model: str = model_field
-    input: str
+    input: str = Field(
+        description="The input to embed."
+    )

    class Config:
        schema_extra = {