com.microsoft.semantickernel.aiservices.google.GeminiServiceBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of semantickernel-aiservices-google Show documentation

Google services for Semantic Kernel

There is a newer version: 1.4.2

Show newest version

// Copyright (c) Microsoft. All rights reserved.
package com.microsoft.semantickernel.aiservices.google;

import com.google.cloud.vertexai.VertexAI;
import com.microsoft.semantickernel.builders.SemanticKernelBuilder;
import javax.annotation.Nullable;

/**
 * Builder for a Gemini service.
 */
public abstract class GeminiServiceBuilder> implements
    SemanticKernelBuilder {

    @Nullable
    protected String modelId;
    @Nullable
    protected VertexAI client;

    /**
     * Sets the model ID for the service
     *
     * @param modelId The model ID
     * @return The builder
     */
    public U withModelId(String modelId) {
        this.modelId = modelId;
        return (U) this;
    }

    /**
     * Sets the VertexAI client for the service
     *
     * @param client The VertexAI client
     * @return The builder
     */
    public U withVertexAIClient(VertexAI client) {
        this.client = client;
        return (U) this;
    }

    @Override
    public abstract T build();
}