refac: add suffix field to ollama generate form

2024-11-22 08:07:55 +00:00 · 2024-11-21 08:12:19 -08:00 · 2024-11-21 08:12:19 -08:00 · d587206929
commit d587206929
parent 1c0327ed7f
1 changed files with 1 additions and 0 deletions
--- a/backend/open_webui/apps/ollama/main.py
+++ b/backend/open_webui/apps/ollama/main.py
@ -837,6 +837,7 @@ async def generate_ollama_batch_embeddings(
 class GenerateCompletionForm(BaseModel):
    model: str
    prompt: str
+    suffix: Optional[str] = None
    images: Optional[list[str]] = None
    format: Optional[str] = None
    options: Optional[dict] = None