github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-04 18:40:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

style: code format (#1513)

Browse Source
This commit is contained in:
Aries-ckt
2024-05-13 09:39:15 +08:00
committed by GitHub
parent f389a0c32d
commit c8fb508fe4
5 changed files with 25 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -56,8 +56,12 @@ class PGVectorStore(VectorStoreBase):
def __init__(self, vector_store_config: PGVectorConfig) -> None:
"""Create a PGVectorStore instance."""
from langchain.vectorstores import PGVector
try:
from langchain.vectorstores import PGVector # mypy: ignore
except ImportError:
raise ImportError(
"Please install the `langchain` package to use the PGVector."
)
self.connection_string = vector_store_config.connection_string
self.embeddings = vector_store_config.embedding_fn
self.collection_name = vector_store_config.name