From c4964aa0dd2a93b04b6b32cfb51f64d9384dc03e Mon Sep 17 00:00:00 2001 From: Lalleh Rafeei Date: Tue, 19 Dec 2023 13:23:38 -0800 Subject: [PATCH] Add SurrealDBStore to vectorstore 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 0f436f69e6..fd80ac88ee 100644 --- a/newrelic/config.py +++ b/newrelic/config.py @@ -2378,6 +2378,12 @@ def _process_module_builtin_defaults(): "instrument_langchain_vectorstore_similarity_search", ) + _process_module_definition( + "langchain_community.vectorstores.surrealdb", + "newrelic.hooks.mlmodel_langchain", + "instrument_langchain_vectorstore_similarity_search", + ) + _process_module_definition( "langchain_community.vectorstores.tair", "newrelic.hooks.mlmodel_langchain", diff --git a/newrelic/hooks/mlmodel_langchain.py b/newrelic/hooks/mlmodel_langchain.py index 68c1f1335c..2b2e5d232d 100644 --- a/newrelic/hooks/mlmodel_langchain.py +++ b/newrelic/hooks/mlmodel_langchain.py @@ -75,6 +75,7 @@ "langchain_community.vectorstores.sqlitevss": "SQLiteVSS", "langchain_community.vectorstores.starrocks": "StarRocks", "langchain_community.vectorstores.supabase": "SupabaseVectorStore", + "langchain_community.vectorstores.surrealdb": "SurrealDBStore", "langchain_community.vectorstores.tair": "Tair", "langchain_community.vectorstores.tencentvectordb": "TencentVectorDB", "langchain_community.vectorstores.tigris": "Tigris",