Pārlūkot izejas kodu

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 mēneši atpakaļ
vecāks
revīzija
4adc57fd34
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  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