mirror of https://github.com/hwchase17/langchain
Create VectorStore interface (#92)
parent
b9f61390e9
commit
61f12229df
@ -0,0 +1,6 @@
|
||||
"""Wrappers on top of vector stores."""
|
||||
from langchain.vectorstores.base import VectorStore
|
||||
from langchain.vectorstores.elastic_vector_search import ElasticVectorSearch
|
||||
from langchain.vectorstores.faiss import FAISS
|
||||
|
||||
__all__ = ["ElasticVectorSearch", "FAISS", "VectorStore"]
|
@ -0,0 +1,13 @@
|
||||
"""Interface for vector stores."""
|
||||
from abc import ABC, abstractmethod
|
||||
from typing import List
|
||||
|
||||
from langchain.docstore.document import Document
|
||||
|
||||
|
||||
class VectorStore(ABC):
|
||||
"""Interface for vector stores."""
|
||||
|
||||
@abstractmethod
|
||||
def similarity_search(self, query: str, k: int = 4) -> List[Document]:
|
||||
"""Return docs most similar to query."""
|
Loading…
Reference in New Issue