From 7dc8c245670cbe13e569534c3290088ca267adc8 Mon Sep 17 00:00:00 2001 From: Lalleh Rafeei Date: Tue, 24 Dec 2024 11:22:03 -0800 Subject: [PATCH] Add Tablestore vectorstore to list --- newrelic/config.py | 6 ++++++ newrelic/hooks/mlmodel_langchain.py | 1 + 2 files changed, 7 insertions(+) diff --git a/newrelic/config.py b/newrelic/config.py index 9896d6a5e..2a4d4d044 100644 --- 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 index 2ae76ef2c..6277a2b02 100644 --- 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",