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