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

fix: serialize picture description parameters to JSON in DoclingLoader

Vaclav Cerny 3 месяцев назад
Родитель
Сommit
4bbc32efa6
1 измененных файлов с 4 добавлено и 4 удалено
  1. 4 4
      backend/open_webui/retrieval/loaders/main.py

+ 4 - 4
backend/open_webui/retrieval/loaders/main.py

@@ -162,15 +162,15 @@ class DoclingLoader:
                     if picture_description_mode == "local" and self.params.get(
                         "picture_description_local", {}
                     ):
-                        params["picture_description_local"] = self.params.get(
-                            "picture_description_local", {}
+                        params["picture_description_local"] = json.dumps(
+                            self.params.get("picture_description_local", {})
                         )
 
                     elif picture_description_mode == "api" and self.params.get(
                         "picture_description_api", {}
                     ):
-                        params["picture_description_api"] = self.params.get(
-                            "picture_description_api", {}
+                        params["picture_description_api"] = json.dumps(
+                            self.params.get("picture_description_api", {})
                         )
 
                 if self.params.get("ocr_engine") and self.params.get("ocr_lang"):