Selaa lähdekoodia

support list[str] | str as input

Peter De-Ath 6 kuukautta sitten
vanhempi
commit
c748d33192
1 muutettua tiedostoa jossa 1 lisäystä ja 1 poistoa
  1. 1 1
      backend/open_webui/apps/ollama/main.py

+ 1 - 1
backend/open_webui/apps/ollama/main.py

@@ -547,7 +547,7 @@ class GenerateEmbeddingsForm(BaseModel):
 
 
 class GenerateEmbedForm(BaseModel):
 class GenerateEmbedForm(BaseModel):
     model: str
     model: str
-    input: list[str]
+    input: list[str]|str
     truncate: Optional[bool] = None
     truncate: Optional[bool] = None
     options: Optional[dict] = None
     options: Optional[dict] = None
     keep_alive: Optional[Union[int, str]] = None
     keep_alive: Optional[Union[int, str]] = None