diff --git a/newrelic/config.py b/newrelic/config.py index 0f436f69e..fd80ac88e 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 68c1f1335..2b2e5d232 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",