Просмотр исходного кода

Merge pull request #6104 from Peter-De-Ath/hotfix/ollama-embed

fix: ollama embed support list[str] or str on input
Timothy Jaeryang Baek 6 месяцев назад
Родитель
Сommit
4adc57fd34
1 измененных файлов с 1 добавлено и 1 удалено
  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):
     model: str
-    input: list[str]
+    input: list[str]|str
     truncate: Optional[bool] = None
     options: Optional[dict] = None
     keep_alive: Optional[Union[int, str]] = None