Skip to content

Commit 9de3592

Browse files
Nguyen Tran (Mark)ilayaperumalg
authored andcommitted
fix: VertixAiEmbeddingConnectionDetails not injected when using vertex-ai-embedding starter
Signed-off-by: Nguyen Tran (Mark) <[email protected]>
1 parent a8b3982 commit 9de3592

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

auto-configurations/models/spring-ai-autoconfigure-model-vertex-ai/src/main/java/org/springframework/ai/model/vertexai/autoconfigure/embedding/VertexAiEmbeddingConnectionAutoConfiguration.java

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616

1717
package org.springframework.ai.model.vertexai.autoconfigure.embedding;
1818

19-
import com.google.cloud.vertexai.VertexAI;
19+
import com.google.cloud.aiplatform.v1.PredictionServiceSettings;
2020

2121
import org.springframework.ai.vertexai.embedding.VertexAiEmbeddingConnectionDetails;
2222
import org.springframework.boot.autoconfigure.AutoConfiguration;
@@ -33,10 +33,11 @@
3333
* @author Christian Tzolov
3434
* @author Mark Pollack
3535
* @author Ilayaperumal Gopinathan
36+
* @author Nguyen Tran
3637
* @since 1.0.0
3738
*/
3839
@AutoConfiguration
39-
@ConditionalOnClass(VertexAI.class)
40+
@ConditionalOnClass(PredictionServiceSettings.class)
4041
@EnableConfigurationProperties(VertexAiEmbeddingConnectionProperties.class)
4142
public class VertexAiEmbeddingConnectionAutoConfiguration {
4243

0 commit comments

Comments
 (0)