From fa2a61e06569bb600d36d7ea5fee2ab456b3434d Mon Sep 17 00:00:00 2001 From: Lucas Doyle Date: Sat, 29 Apr 2023 18:46:01 -0700 Subject: [PATCH] llama_cpp server: fields for the embedding endpoint --- llama_cpp/server/app.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/llama_cpp/server/app.py b/llama_cpp/server/app.py index ec5dbd3..9adddcd 100644 --- 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 = {