OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

~ truncate vectors for pgvector if too big

Browse Source
This commit is contained in:
Alexander Grimm 2025-04-30 05:35:17 +00:00
parent 4ee5dd58b7
commit da9966aca1
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend/open_webui/retrieval/vector/dbs/pgvector.py
View File

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