diff --git a/newrelic/config.py b/newrelic/config.py index 093c178bd..7fd6f31b8 100644 --- 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 index 48d281a24..c8353e74f 100644 --- 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( {