github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-07 03:50:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(rag): Fix CrossEncoderRanker bug of EmbeddingRetriever (#1504)

Browse Source 
Co-authored-by: aries_ckt <916701291@qq.com>
This commit is contained in:
ivanzhu
2024-05-10 13:32:58 +08:00
committed by GitHub
parent 87a67cce7d
commit 8eb64d7311
3 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dbgpt/storage/vector_store/chroma_store.py
View File

@@ -82,7 +82,7 @@ class ChromaStore(VectorStoreBase):
# client_settings=chroma_settings,
client=client,
collection_metadata=collection_metadata,
)
) # type: ignore
def similar_search(
self, text, topk, filters: Optional[MetadataFilters] = None

6
dbgpt/storage/vector_store/pgvector_store.py
View File

@@ -93,7 +93,8 @@ class PGVectorStore(VectorStoreBase):
List[str]: chunk ids.
"""
lc_documents = [Chunk.chunk2langchain(chunk) for chunk in chunks]
return self.vector_store_client.from_documents(lc_documents)
self.vector_store_client.from_documents(lc_documents)
return [str(chunk.chunk_id) for chunk in lc_documents]
def delete_vector_name(self, vector_name: str):
"""Delete vector by name.
@@ -109,4 +110,5 @@ class PGVectorStore(VectorStoreBase):
Args:
ids(str): vector ids, separated by comma.
"""
return self.vector_store_client.delete(ids)
delete_ids = ids.split(",")
return self.vector_store_client.delete(delete_ids)