From dcab991d44066ccba1104c91efe0eea6b8f6490b Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Tue, 8 Oct 2024 11:05:57 -0700 Subject: [PATCH] fix: model dump issue --- backend/open_webui/apps/retrieval/utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/backend/open_webui/apps/retrieval/utils.py b/backend/open_webui/apps/retrieval/utils.py index b7d746bfa..174e37495 100644 --- 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: