ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-16 06:15:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

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

Browse source 
feat: allow pgvector to use large embedding models
This commit is contained in:
Tim Jaeryang Baek 2025-04-30 02:41:26 -07:00 committed by GitHub
commit c9d3bb35c3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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: