diff --git a/newrelic/config.py b/newrelic/config.py index dbf1037a2d..729c0739c5 100644 --- a/newrelic/config.py +++ b/newrelic/config.py @@ -2408,6 +2408,12 @@ def _process_module_builtin_defaults(): "instrument_langchain_vectorstore_similarity_search", ) + _process_module_definition( + "langchain_community.vectorstores.oraclevs", + "newrelic.hooks.mlmodel_langchain", + "instrument_langchain_vectorstore_similarity_search", + ) + _process_module_definition( "langchain_community.vectorstores.pathway", "newrelic.hooks.mlmodel_langchain", @@ -2444,6 +2450,12 @@ def _process_module_builtin_defaults(): "instrument_langchain_vectorstore_similarity_search", ) + _process_module_definition( + "langchain_community.vectorstores.relyt", + "newrelic.hooks.mlmodel_langchain", + "instrument_langchain_vectorstore_similarity_search", + ) + _process_module_definition( "langchain_community.vectorstores.rocksetdb", "newrelic.hooks.mlmodel_langchain", @@ -2546,6 +2558,12 @@ def _process_module_builtin_defaults(): "instrument_langchain_vectorstore_similarity_search", ) + _process_module_definition( + "langchain_community.vectorstores.upstash", + "newrelic.hooks.mlmodel_langchain", + "instrument_langchain_vectorstore_similarity_search", + ) + _process_module_definition( "langchain_community.vectorstores.usearch", "newrelic.hooks.mlmodel_langchain", diff --git a/newrelic/hooks/mlmodel_langchain.py b/newrelic/hooks/mlmodel_langchain.py index d234b4d3d0..551d6c2909 100644 --- a/newrelic/hooks/mlmodel_langchain.py +++ b/newrelic/hooks/mlmodel_langchain.py @@ -81,6 +81,7 @@ "langchain_community.vectorstores.neo4j_vector": "Neo4jVector", "langchain_community.vectorstores.thirdai_neuraldb": "NeuralDBVectorStore", "langchain_community.vectorstores.nucliadb": "NucliaDB", + "langchain_community.vectorstores.oraclevs": "OracleVS", "langchain_community.vectorstores.opensearch_vector_search": "OpenSearchVectorSearch", "langchain_community.vectorstores.pathway": "PathwayVectorClient", "langchain_community.vectorstores.pgembedding": "PGEmbedding", @@ -89,6 +90,7 @@ "langchain_community.vectorstores.pinecone": "Pinecone", "langchain_community.vectorstores.qdrant": "Qdrant", "langchain_community.vectorstores.redis.base": "Redis", + "langchain_community.vectorstores.relyt": "Relyt", "langchain_community.vectorstores.rocksetdb": "Rockset", "langchain_community.vectorstores.scann": "ScaNN", "langchain_community.vectorstores.semadb": "SemaDB", @@ -105,6 +107,7 @@ "langchain_community.vectorstores.tiledb": "TileDB", "langchain_community.vectorstores.timescalevector": "TimescaleVector", "langchain_community.vectorstores.typesense": "Typesense", + "langchain_community.vectorstores.upstash": "UpstashVectorStore", "langchain_community.vectorstores.usearch": "USearch", "langchain_community.vectorstores.vald": "Vald", "langchain_community.vectorstores.vdms": "VDMS", diff --git a/tests/mlmodel_langchain/test_chain.py b/tests/mlmodel_langchain/test_chain.py index 69c54b9509..26cbb2a424 100644 --- a/tests/mlmodel_langchain/test_chain.py +++ b/tests/mlmodel_langchain/test_chain.py @@ -1464,7 +1464,7 @@ def _test(): "create_function,call_function,input_", ( (create_structured_output_runnable, "ainvoke", {"input": "Sally is 13"}), - (create_structured_output_chain, "arun", "Sally is 13"), + (create_structured_output_chain, "arun", "Sally is 13"), # Deprecated in 0.2.0 ), ) @reset_core_stats_engine() diff --git a/tox.ini b/tox.ini index 81872f5e47..86ec4b0a2b 100644 --- a/tox.ini +++ b/tox.ini @@ -355,7 +355,8 @@ deps = mlmodel_openai-openailatest: openai[datalib] ; Required for openai testing mlmodel_openai: protobuf - mlmodel_langchain: langchain + ; Pinning to 0.1.16 while adding support for with_structured_output in chain tests + mlmodel_langchain: langchain<0.1.17 mlmodel_langchain: langchain-community mlmodel_langchain: openai[datalib] ; Required for langchain testing