From 4a49fc5a9563a3f57e59265fde5498fd93395dfe Mon Sep 17 00:00:00 2001 From: "Chaunte W. Lacewell" Date: Thu, 28 Mar 2024 12:54:24 -0700 Subject: [PATCH] community[patch]: Fix bug in vdms (#19728) **Description:** Fix embedding check in vdms **Contribution maintainer:** [@cwlacewe](https://github.com/cwlacewe) --- libs/community/langchain_community/vectorstores/vdms.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/libs/community/langchain_community/vectorstores/vdms.py b/libs/community/langchain_community/vectorstores/vdms.py index d367dce649d..d07c57469b0 100644 --- a/libs/community/langchain_community/vectorstores/vdms.py +++ b/libs/community/langchain_community/vectorstores/vdms.py @@ -746,7 +746,8 @@ class VDMS(VectorStore): ) -> Tuple[List[Dict[str, Any]], List]: all_blobs: List[Any] = [] blob = embedding2bytes(query_embedding) - all_blobs.append(blob) + if blob is not None: + all_blobs.append(blob) if constraints is None: # K results returned @@ -1534,7 +1535,7 @@ def _check_descriptor_exists_by_id( def embedding2bytes(embedding: Union[List[float], None]) -> Union[bytes, None]: blob = None - if embedding: + if embedding is not None: emb = np.array(embedding, dtype="float32") blob = emb.tobytes() return blob