소스 검색

Merge pull request #13342 from feddersen-group/feat/embedding_dim

feat: allow pgvector to use large embedding models
Tim Jaeryang Baek 3 달 전
부모
커밋
c9d3bb35c3
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: