mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-04 18:40:10 +00:00
doc:ollama document (#1512)
This commit is contained in:
@@ -19,6 +19,7 @@ logger = logging.getLogger(__name__)
|
||||
|
||||
CHROMA_COLLECTION_NAME = "langchain"
|
||||
|
||||
|
||||
@register_resource(
|
||||
_("Chroma Vector Store"),
|
||||
"chroma_vector_store",
|
||||
@@ -152,7 +153,6 @@ class ChromaStore(VectorStoreBase):
|
||||
files = list(filter(lambda f: f != "chroma.sqlite3", files))
|
||||
return len(files) > 0
|
||||
|
||||
|
||||
def load_document(self, chunks: List[Chunk]) -> List[str]:
|
||||
"""Load document to vector store."""
|
||||
logger.info("ChromaStore load document")
|
||||
|
@@ -66,7 +66,7 @@ class PGVectorStore(VectorStoreBase):
|
||||
embedding_function=self.embeddings,
|
||||
collection_name=self.collection_name,
|
||||
connection_string=self.connection_string,
|
||||
) # mypy: ignore
|
||||
) # mypy: ignore
|
||||
|
||||
def similar_search(
|
||||
self, text: str, topk: int, filters: Optional[MetadataFilters] = None
|
||||
|
Reference in New Issue
Block a user