fix: model dump issue

2024-10-08 11:05:57 -07:00 · 2024-10-08 11:05:57 -07:00 · dcab991d44
parent 457360dae7
commit dcab991d44
1 changed files with 2 additions and 1 deletions
--- a/backend/open_webui/apps/retrieval/utils.py
+++ b/backend/open_webui/apps/retrieval/utils.py
@ -193,7 +193,8 @@ def query_collection(
                    k=k,
                    query_embedding=query_embedding,
                )
-                results.append(result.model_dump())
+                if result is not None:
+                    results.append(result.model_dump())
            except Exception as e:
                log.exception(f"Error when querying the collection: {e}")
        else: