Merge pull request #1231 from newrelic/add-langchain-vs

Instrument SQLiteVec vectorstore in LangChain
newrelic · Oct 9, 2024 · 2d17237 · 2d17237
2 parents 3a001b0 + 6a2a660
commit 2d17237
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/newrelic/config.py b/newrelic/config.py
@@ -2511,6 +2511,12 @@ def _process_module_builtin_defaults():
         "instrument_langchain_vectorstore_similarity_search",
     )
 
+    _process_module_definition(
+        "langchain_community.vectorstores.sqlitevec",
+        "newrelic.hooks.mlmodel_langchain",
+        "instrument_langchain_vectorstore_similarity_search",
+    )
+
     _process_module_definition(
         "langchain_community.vectorstores.sqlitevss",
         "newrelic.hooks.mlmodel_langchain",

diff --git a/newrelic/hooks/mlmodel_langchain.py b/newrelic/hooks/mlmodel_langchain.py
@@ -100,6 +100,7 @@
     "langchain_community.vectorstores.semadb": "SemaDB",
     "langchain_community.vectorstores.singlestoredb": "SingleStoreDB",
     "langchain_community.vectorstores.sklearn": "SKLearnVectorStore",
+    "langchain_community.vectorstores.sqlitevec": "SQLiteVec",
     "langchain_community.vectorstores.sqlitevss": "SQLiteVSS",
     "langchain_community.vectorstores.starrocks": "StarRocks",
     "langchain_community.vectorstores.supabase": "SupabaseVectorStore",
@@ -452,9 +453,7 @@ def _record_tool_success(
     try:
         result = str(response)
     except Exception:
-        _logger.debug(
-            f"Failed to convert tool response into a string.\n{traceback.format_exception(*sys.exc_info())}"
-        )
+        _logger.debug(f"Failed to convert tool response into a string.\n{traceback.format_exception(*sys.exc_info())}")
     if settings.ai_monitoring.record_content.enabled:
         full_tool_event_dict.update(
             {