community: use NeuralDB object to initialize NeuralDBVectorStore (#17272)

**Description:** This PR adds an `__init__` method to the
NeuralDBVectorStore class, which takes in a NeuralDB object to
instantiate the state of NeuralDBVectorStore.
**Issue:** N/A
**Dependencies:** N/A
**Twitter handle:** N/A
This commit is contained in:
kartikTAI 2024-02-22 11:05:01 -06:00 committed by GitHub
parent a51a257575
commit 9cf6661dc5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -12,7 +12,22 @@ from langchain_core.vectorstores import VectorStore
class NeuralDBVectorStore(VectorStore):
"""Vectorstore that uses ThirdAI's NeuralDB."""
"""Vectorstore that uses ThirdAI's NeuralDB.
To use, you should have the ``thirdai[neural_db]`` python package installed.
Example:
.. code-block:: python
from langchain_community.vectorstores import NeuralDBVectorStore
from thirdai import neural_db as ndb
db = ndb.NeuralDB()
vectorstore = NeuralDBVectorStore(db=db)
"""
def __init__(self, db):
self.db = db
db: Any = None #: :meta private:
"""NeuralDB instance"""
@ -322,7 +337,6 @@ class NeuralDBVectorStore(VectorStore):
metadata={
"id": ref.id,
"upvote_ids": ref.upvote_ids,
"text": ref.text,
"source": ref.source,
"metadata": ref.metadata,
"score": ref.score,