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

refactor(agent): Refactor resource of agents (#1518)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-05-15 09:57:19 +08:00
committed by GitHub
parent db4d318a5f
commit 559affe87d
102 changed files with 2633 additions and 2549 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,4 +1,5 @@
"""Postgres vector store."""
import logging
from typing import List, Optional
@@ -67,10 +68,10 @@ class PGVectorStore(VectorStoreBase):
self.collection_name = vector_store_config.name
self.vector_store_client = PGVector(
embedding_function=self.embeddings,
embedding_function=self.embeddings, # type: ignore
collection_name=self.collection_name,
connection_string=self.connection_string,
) # mypy: ignore
)
def similar_search(
self, text: str, topk: int, filters: Optional[MetadataFilters] = None
@@ -97,7 +98,7 @@ class PGVectorStore(VectorStoreBase):
List[str]: chunk ids.
"""
lc_documents = [Chunk.chunk2langchain(chunk) for chunk in chunks]
self.vector_store_client.from_documents(lc_documents)
self.vector_store_client.from_documents(lc_documents) # type: ignore
return [str(chunk.chunk_id) for chunk in lc_documents]
def delete_vector_name(self, vector_name: str):