From ef94eafd1372c5516c5ef485053c20876814a30d Mon Sep 17 00:00:00 2001
From: Andrei Fajardo <92402603+nerdai@users.noreply.github.com>
Date: Tue, 13 Feb 2024 10:56:45 -0500
Subject: [PATCH] Add OpensearchVectorClient to __init__ (#10660)

* add OpensearchVectorClient

* fix OpensearchVectorClient import

* version bump; CHANGELOG
---
 docs/examples/vector_stores/OpensearchDemo.ipynb     | 12 ++++++++----
 .../CHANGELOG.md                                     |  9 +++++++++
 .../llama_index/vector_stores/opensearch/__init__.py |  7 +++++--
 .../pyproject.toml                                   |  2 +-
 4 files changed, 23 insertions(+), 7 deletions(-)
 create mode 100644 llama-index-integrations/vector_stores/llama-index-vector-stores-opensearch/CHANGELOG.md

diff --git a/docs/examples/vector_stores/OpensearchDemo.ipynb b/docs/examples/vector_stores/OpensearchDemo.ipynb
index 7f38185b8..a36d82629 100644
--- a/docs/examples/vector_stores/OpensearchDemo.ipynb
+++ b/docs/examples/vector_stores/OpensearchDemo.ipynb
@@ -79,8 +79,10 @@
    "source": [
     "from os import getenv\n",
     "from llama_index.core import SimpleDirectoryReader\n",
-    "from llama_index.vector_stores.opensearch import OpensearchVectorStore\n",
-    "from llama_index.core.vector_stores import OpensearchVectorClient\n",
+    "from llama_index.vector_stores.opensearch import (\n",
+    "    OpensearchVectorStore,\n",
+    "    OpensearchVectorClient,\n",
+    ")\n",
     "from llama_index.core import VectorStoreIndex, StorageContext\n",
     "\n",
     "# http endpoint for your cluster (opensearch required for vector index usage)\n",
@@ -333,8 +335,10 @@
    "outputs": [],
    "source": [
     "from os import getenv\n",
-    "from llama_index.vector_stores.opensearch import OpensearchVectorStore\n",
-    "from llama_index.core.vector_stores import OpensearchVectorClient\n",
+    "from llama_index.vector_stores.opensearch import (\n",
+    "    OpensearchVectorStore,\n",
+    "    OpensearchVectorClient,\n",
+    ")\n",
     "\n",
     "# http endpoint for your cluster (opensearch required for vector index usage)\n",
     "endpoint = getenv(\"OPENSEARCH_ENDPOINT\", \"http://localhost:9200\")\n",
diff --git a/llama-index-integrations/vector_stores/llama-index-vector-stores-opensearch/CHANGELOG.md b/llama-index-integrations/vector_stores/llama-index-vector-stores-opensearch/CHANGELOG.md
new file mode 100644
index 000000000..633b40f21
--- /dev/null
+++ b/llama-index-integrations/vector_stores/llama-index-vector-stores-opensearch/CHANGELOG.md
@@ -0,0 +1,9 @@
+# CHANGELOG — llama-index-vector-stores-opensearch
+
+## [0.1.2]
+
+- Adds OpensearchVectorClient as top-level import
+
+## [0.1.1]
+
+- Fixes strict equality in dependency of llama-index-core
diff --git a/llama-index-integrations/vector_stores/llama-index-vector-stores-opensearch/llama_index/vector_stores/opensearch/__init__.py b/llama-index-integrations/vector_stores/llama-index-vector-stores-opensearch/llama_index/vector_stores/opensearch/__init__.py
index 1e6c16962..ccff07288 100644
--- a/llama-index-integrations/vector_stores/llama-index-vector-stores-opensearch/llama_index/vector_stores/opensearch/__init__.py
+++ b/llama-index-integrations/vector_stores/llama-index-vector-stores-opensearch/llama_index/vector_stores/opensearch/__init__.py
@@ -1,3 +1,6 @@
-from llama_index.vector_stores.opensearch.base import OpensearchVectorStore
+from llama_index.vector_stores.opensearch.base import (
+    OpensearchVectorStore,
+    OpensearchVectorClient,
+)
 
-__all__ = ["OpensearchVectorStore"]
+__all__ = ["OpensearchVectorStore", "OpensearchVectorClient"]
diff --git a/llama-index-integrations/vector_stores/llama-index-vector-stores-opensearch/pyproject.toml b/llama-index-integrations/vector_stores/llama-index-vector-stores-opensearch/pyproject.toml
index a21651265..758666af0 100644
--- a/llama-index-integrations/vector_stores/llama-index-vector-stores-opensearch/pyproject.toml
+++ b/llama-index-integrations/vector_stores/llama-index-vector-stores-opensearch/pyproject.toml
@@ -19,7 +19,7 @@ description = "llama-index vector_stores opensearch integration"
 license = "MIT"
 name = "llama-index-vector-stores-opensearch"
 readme = "README.md"
-version = "0.1.1"
+version = "0.1.2"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<3.12"
-- 
GitLab