Skip to content
Snippets Groups Projects
Unverified Commit 52383c7f authored by Ravi Theja's avatar Ravi Theja Committed by GitHub
Browse files

Update opensearch vectorstore to PydanticVectorStore class (#11400)

parent 4077feea
Branches
Tags
No related merge requests found
...@@ -2,11 +2,12 @@ ...@@ -2,11 +2,12 @@
import json import json
import uuid import uuid
from typing import Any, Dict, Iterable, List, Optional, Union, cast from typing import Any, Dict, Iterable, List, Optional, Union, cast
from llama_index.core.bridge.pydantic import PrivateAttr
from llama_index.core.schema import BaseNode, MetadataMode, TextNode from llama_index.core.schema import BaseNode, MetadataMode, TextNode
from llama_index.core.vector_stores.types import ( from llama_index.core.vector_stores.types import (
MetadataFilters, MetadataFilters,
VectorStore, BasePydanticVectorStore,
VectorStoreQuery, VectorStoreQuery,
VectorStoreQueryMode, VectorStoreQueryMode,
VectorStoreQueryResult, VectorStoreQueryResult,
...@@ -419,7 +420,7 @@ class OpensearchVectorClient: ...@@ -419,7 +420,7 @@ class OpensearchVectorClient:
return VectorStoreQueryResult(nodes=nodes, ids=ids, similarities=scores) return VectorStoreQueryResult(nodes=nodes, ids=ids, similarities=scores)
class OpensearchVectorStore(VectorStore): class OpensearchVectorStore(BasePydanticVectorStore):
"""Elasticsearch/Opensearch vector store. """Elasticsearch/Opensearch vector store.
Args: Args:
...@@ -428,12 +429,14 @@ class OpensearchVectorStore(VectorStore): ...@@ -428,12 +429,14 @@ class OpensearchVectorStore(VectorStore):
""" """
stores_text: bool = True stores_text: bool = True
_client: OpensearchVectorClient = PrivateAttr(default=None)
def __init__( def __init__(
self, self,
client: OpensearchVectorClient, client: OpensearchVectorClient,
) -> None: ) -> None:
"""Initialize params.""" """Initialize params."""
super().__init__()
self._client = client self._client = client
@property @property
......
from llama_index.core.vector_stores.types import VectorStore from llama_index.core.vector_stores.types import BasePydanticVectorStore
from llama_index.vector_stores.opensearch import OpensearchVectorStore from llama_index.vector_stores.opensearch import OpensearchVectorStore
def test_class(): def test_class():
names_of_base_classes = [b.__name__ for b in OpensearchVectorStore.__mro__] names_of_base_classes = [b.__name__ for b in OpensearchVectorStore.__mro__]
assert VectorStore.__name__ in names_of_base_classes assert BasePydanticVectorStore.__name__ in names_of_base_classes
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment