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

~ truncate vectors for pgvector if too big

Alexander Grimm 3 месяцев назад
Родитель
Сommit
da9966aca1
1 измененных файлов с 2 добавлено и 3 удалено
  1. 2 3
      backend/open_webui/retrieval/vector/dbs/pgvector.py

+ 2 - 3
backend/open_webui/retrieval/vector/dbs/pgvector.py

@@ -141,9 +141,8 @@ class PgvectorClient(VectorDBBase):
             # Pad the vector with zeros
             vector += [0.0] * (VECTOR_LENGTH - current_length)
         elif current_length > VECTOR_LENGTH:
-            raise Exception(
-                f"Vector length {current_length} not supported. Max length must be <= {VECTOR_LENGTH}"
-            )
+            # Truncate the vector to VECTOR_LENGTH
+            vector = vector[:VECTOR_LENGTH]
         return vector
 
     def insert(self, collection_name: str, items: List[VectorItem]) -> None: