From 6a2da1026b9f0f5ac6adb14d0f421491a758b28e Mon Sep 17 00:00:00 2001 From: Almas Abdrazak Date: Wed, 10 Dec 2025 22:15:33 -0800 Subject: [PATCH] add framework name to mongodb driver metadata Signed-off-by: Almas Abdrazak --- .../MongoDBAtlasVectorStoreAutoConfiguration.java | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/auto-configurations/vector-stores/spring-ai-autoconfigure-vector-store-mongodb-atlas/src/main/java/org/springframework/ai/vectorstore/mongodb/autoconfigure/MongoDBAtlasVectorStoreAutoConfiguration.java b/auto-configurations/vector-stores/spring-ai-autoconfigure-vector-store-mongodb-atlas/src/main/java/org/springframework/ai/vectorstore/mongodb/autoconfigure/MongoDBAtlasVectorStoreAutoConfiguration.java index 8a302cd1c37..1e7208f2c74 100644 --- a/auto-configurations/vector-stores/spring-ai-autoconfigure-vector-store-mongodb-atlas/src/main/java/org/springframework/ai/vectorstore/mongodb/autoconfigure/MongoDBAtlasVectorStoreAutoConfiguration.java +++ b/auto-configurations/vector-stores/spring-ai-autoconfigure-vector-store-mongodb-atlas/src/main/java/org/springframework/ai/vectorstore/mongodb/autoconfigure/MongoDBAtlasVectorStoreAutoConfiguration.java @@ -16,6 +16,8 @@ package org.springframework.ai.vectorstore.mongodb.autoconfigure; +import com.mongodb.MongoDriverInformation; +import com.mongodb.client.MongoClient; import java.util.Arrays; import java.util.List; @@ -28,6 +30,7 @@ import org.springframework.ai.vectorstore.mongodb.atlas.MongoDBAtlasVectorStore; import org.springframework.ai.vectorstore.observation.VectorStoreObservationConvention; import org.springframework.beans.factory.ObjectProvider; +import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.autoconfigure.AutoConfiguration; import org.springframework.boot.autoconfigure.condition.ConditionalOnClass; import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean; @@ -65,11 +68,14 @@ BatchingStrategy batchingStrategy() { @Bean @ConditionalOnMissingBean - MongoDBAtlasVectorStore vectorStore(MongoTemplate mongoTemplate, EmbeddingModel embeddingModel, + MongoDBAtlasVectorStore vectorStore(@Autowired(required = false) MongoClient mongoClient, MongoTemplate mongoTemplate, EmbeddingModel embeddingModel, MongoDBAtlasVectorStoreProperties properties, ObjectProvider observationRegistry, ObjectProvider customObservationConvention, BatchingStrategy batchingStrategy) { - + if(mongoClient != null) { + // append framework name to the driver info + mongoClient.appendMetadata(MongoDriverInformation.builder().driverName("spring-ai").build()); + } MongoDBAtlasVectorStore.Builder builder = MongoDBAtlasVectorStore.builder(mongoTemplate, embeddingModel) .initializeSchema(properties.isInitializeSchema()) .observationRegistry(observationRegistry.getIfUnique(() -> ObservationRegistry.NOOP))