소스 검색

~ truncate vectors for pgvector if too big

Alexander Grimm 3 달 전
부모
커밋
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: