feat: allow vertex embedding connection settings to be configurable

Author: garethjevans

models/spring-ai-vertex-ai-embedding/src/main/java/org/springframework/ai/vertexai/embedding/VertexAiEmbeddingConnectionDetails.java

@@ -58,19 +58,14 @@ public class VertexAiEmbeddingConnectionDetails {
 
 	private final String publisher;
 
-	private PredictionServiceSettings predictionServiceSettings;
+	private final PredictionServiceSettings predictionServiceSettings;
 
-	public VertexAiEmbeddingConnectionDetails(String endpoint, String projectId, String location, String publisher) {
+	public VertexAiEmbeddingConnectionDetails(String projectId, String location, String publisher,
+			PredictionServiceSettings predictionServiceSettings) {
 		this.projectId = projectId;
 		this.location = location;
 		this.publisher = publisher;
-
-		try {
-			this.predictionServiceSettings = PredictionServiceSettings.newBuilder().setEndpoint(endpoint).build();
-		}
-		catch (IOException e) {
-			throw new RuntimeException(e);
-		}
+		this.predictionServiceSettings = predictionServiceSettings;
 	}
 
 	public static Builder builder() {
@@ -125,6 +120,11 @@ public static class Builder {
 		 */
 		private String publisher;
 
+		/**
+		 * Allows the connection settings to be customised
+		 */
+		private PredictionServiceSettings predictionServiceSettings;
+
 		public Builder withApiEndpoint(String endpoint) {
 			this.endpoint = endpoint;
 			return this;
@@ -145,6 +145,11 @@ public Builder withPublisher(String publisher) {
 			return this;
 		}
 
+		public Builder withPredictionServiceSettings(PredictionServiceSettings predictionServiceSettings) {
+			this.predictionServiceSettings = predictionServiceSettings;
+			return this;
+		}
+
 		public VertexAiEmbeddingConnectionDetails build() {
 			if (!StringUtils.hasText(this.endpoint)) {
 				if (!StringUtils.hasText(this.location)) {
@@ -160,7 +165,19 @@ public VertexAiEmbeddingConnectionDetails build() {
 				this.publisher = DEFAULT_PUBLISHER;
 			}
 
-			return new VertexAiEmbeddingConnectionDetails(this.endpoint, this.projectId, this.location, this.publisher);
+			if (this.predictionServiceSettings == null) {
+				try {
+					this.predictionServiceSettings = PredictionServiceSettings.newBuilder()
+						.setEndpoint(endpoint)
+						.build();
+				}
+				catch (IOException e) {
+					throw new RuntimeException(e);
+				}
+			}
+
+			return new VertexAiEmbeddingConnectionDetails(this.projectId, this.location, this.publisher,
+					this.predictionServiceSettings);
 		}
 
 	}