Merge pull request #1278 from newrelic/add-tablestore-vectorstore

Add Tablestore VectorStore
newrelic · Dec 28, 2024 · 4c06d26 · 4c06d26
2 parents 25e1f1e + 7dc8c24
commit 4c06d26
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/newrelic/config.py b/newrelic/config.py
@@ -2719,6 +2719,12 @@ def _process_module_builtin_defaults():
         "instrument_langchain_vectorstore_similarity_search",
     )
 
+    _process_module_definition(
+        "langchain_community.vectorstores.tablestore",
+        "newrelic.hooks.mlmodel_langchain",
+        "instrument_langchain_vectorstore_similarity_search",
+    )
+
     _process_module_definition(
         "langchain_core.tools",
         "newrelic.hooks.mlmodel_langchain",

diff --git a/newrelic/hooks/mlmodel_langchain.py b/newrelic/hooks/mlmodel_langchain.py
@@ -105,6 +105,7 @@
     "langchain_community.vectorstores.starrocks": "StarRocks",
     "langchain_community.vectorstores.supabase": "SupabaseVectorStore",
     "langchain_community.vectorstores.surrealdb": "SurrealDBStore",
+    "langchain_community.vectorstores.tablestore": "TablestoreVectorStore",
     "langchain_community.vectorstores.tair": "Tair",
     "langchain_community.vectorstores.tencentvectordb": "TencentVectorDB",
     "langchain_community.vectorstores.tidb_vector": "TiDBVectorStore",