github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-04-20 03:02:16 -04:00
Code Issues Packages Projects Releases Wiki Activity

Embedding function should always return a list of a list of vectors (#3570)

Browse Source 
Co-authored-by: patrickaubin-abbott <78225817+patrickaubin-abbott@users.noreply.github.com>
Co-authored-by: Li Jiang <bnujli@gmail.com>
This commit is contained in:
paubins2
2024-09-26 01:23:53 -05:00
committed by GitHub
parent 0d5163b78a
commit 94064da472
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
autogen/agentchat/contrib/vectordb/pgvectordb.py
View File

@@ -415,7 +415,8 @@ class Collection:
cursor = self.client.cursor()
results = []
for query_text in query_texts:
vector = self.embedding_function(query_text, convert_to_tensor=False).tolist()
vector = self.embedding_function(query_text)
if distance_type.lower() == "cosine":
index_function = "<=>"
elif distance_type.lower() == "euclidean":
@@ -619,7 +620,7 @@ class PGVectorDB(VectorDB):
if embedding_function:
self.embedding_function = embedding_function
else:
self.embedding_function = SentenceTransformer("all-MiniLM-L6-v2").encode
self.embedding_function = lambda s: SentenceTransformer("all-MiniLM-L6-v2").encode(s).tolist()
self.metadata = metadata
register_vector(self.client)
self.active_collection = None