From c748d3319224a00bcd39fa1bd515855700abedd4 Mon Sep 17 00:00:00 2001
From: Peter De-Ath <peter@de-ath.co.uk>
Date: Thu, 10 Oct 2024 22:41:57 +0100
Subject: [PATCH] support list[str] | str as input

---
 backend/open_webui/apps/ollama/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/backend/open_webui/apps/ollama/main.py b/backend/open_webui/apps/ollama/main.py
index eed74064f..f835e3175 100644
--- a/backend/open_webui/apps/ollama/main.py
+++ b/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