|
21 | 21 | import org.springframework.ai.embedding.BatchingStrategy; |
22 | 22 | import org.springframework.ai.embedding.EmbeddingModel; |
23 | 23 | import org.springframework.ai.embedding.TokenCountBatchingStrategy; |
24 | | -import org.springframework.ai.vectorstore.PineconeVectorStore; |
25 | | -import org.springframework.ai.vectorstore.PineconeVectorStore.PineconeVectorStoreConfig; |
| 24 | +import org.springframework.ai.vectorstore.pinecone.PineconeVectorStore; |
| 25 | +import org.springframework.ai.vectorstore.pinecone.PineconeVectorStore.PineconeVectorStoreConfig; |
26 | 26 | import org.springframework.ai.vectorstore.observation.VectorStoreObservationConvention; |
27 | 27 | import org.springframework.beans.factory.ObjectProvider; |
28 | 28 | import org.springframework.boot.autoconfigure.AutoConfiguration; |
@@ -55,20 +55,20 @@ public PineconeVectorStore vectorStore(EmbeddingModel embeddingModel, PineconeVe |
55 | 55 | ObjectProvider<VectorStoreObservationConvention> customObservationConvention, |
56 | 56 | BatchingStrategy batchingStrategy) { |
57 | 57 |
|
58 | | - var config = PineconeVectorStoreConfig.builder() |
59 | | - .withApiKey(properties.getApiKey()) |
60 | | - .withEnvironment(properties.getEnvironment()) |
61 | | - .withProjectId(properties.getProjectId()) |
62 | | - .withIndexName(properties.getIndexName()) |
63 | | - .withNamespace(properties.getNamespace()) |
64 | | - .withContentFieldName(properties.getContentFieldName()) |
65 | | - .withDistanceMetadataFieldName(properties.getDistanceMetadataFieldName()) |
66 | | - .withServerSideTimeout(properties.getServerSideTimeout()) |
| 58 | + return PineconeVectorStore.builder() |
| 59 | + .embeddingModel(embeddingModel) |
| 60 | + .apiKey(properties.getApiKey()) |
| 61 | + .environment(properties.getEnvironment()) |
| 62 | + .projectId(properties.getProjectId()) |
| 63 | + .indexName(properties.getIndexName()) |
| 64 | + .namespace(properties.getNamespace()) |
| 65 | + .contentFieldName(properties.getContentFieldName()) |
| 66 | + .distanceMetadataFieldName(properties.getDistanceMetadataFieldName()) |
| 67 | + .serverSideTimeout(properties.getServerSideTimeout()) |
| 68 | + .observationRegistry(observationRegistry.getIfUnique(() -> ObservationRegistry.NOOP)) |
| 69 | + .customObservationConvention(customObservationConvention.getIfAvailable(() -> null)) |
| 70 | + .batchingStrategy(batchingStrategy) |
67 | 71 | .build(); |
68 | | - |
69 | | - return new PineconeVectorStore(config, embeddingModel, |
70 | | - observationRegistry.getIfUnique(() -> ObservationRegistry.NOOP), |
71 | | - customObservationConvention.getIfAvailable(() -> null), batchingStrategy); |
72 | 72 | } |
73 | 73 |
|
74 | 74 | } |
0 commit comments