doc:ollama document (#1512)

This commit is contained in:
Aries-ckt
2024-05-11 17:13:25 +08:00
committed by GitHub
parent d3131552d3
commit f389a0c32d
4 changed files with 47 additions and 2 deletions

View File

@@ -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")

View File

@@ -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