Package org.apache.beam.sdk.ml.inference.gemini

Class GeminiModelHandler<InputT extends BaseInput,OutputT extends BaseResponse>

java.lang.Object

org.apache.beam.sdk.ml.inference.gemini.GeminiModelHandler<InputT,OutputT>

All Implemented Interfaces:

BaseModelHandler<GeminiModelParameters<InputT,OutputT>,InputT,OutputT>

public class GeminiModelHandler<InputT extends BaseInput,OutputT extends BaseResponse>
extends Object
implements BaseModelHandler<GeminiModelParameters<InputT,OutputT>,InputT,OutputT>

Model handler for Google Gemini API inference requests.

This handler manages communication with Google's Gemini API, including client initialization, request formatting, and response parsing. It allows executing a custom GeminiRequestFunction against a batch of inputs.

Constructor Summary

Constructors

Constructor	Description
GeminiModelHandler()

Method Summary

Modifier and Type	Method	Description
void	createClient(GeminiModelParameters<InputT,OutputT> parameters)	Initializes the remote model client with the provided parameters.
Iterable<PredictionResult<InputT,OutputT>>	request(List<InputT> input)	Performs inference on a batch of inputs and returns the results.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

GeminiModelHandler

public GeminiModelHandler()

Method Details

createClient

public void createClient(GeminiModelParameters<InputT,OutputT> parameters)

Description copied from interface: BaseModelHandler

Initializes the remote model client with the provided parameters.

Specified by:

createClient in interface BaseModelHandler<GeminiModelParameters<InputT extends BaseInput,OutputT extends BaseResponse>,InputT extends BaseInput,OutputT extends BaseResponse>

request

public Iterable<PredictionResult<InputT,OutputT>> request(List<InputT> input)

Description copied from interface: BaseModelHandler

Performs inference on a batch of inputs and returns the results.

Specified by:

request in interface BaseModelHandler<GeminiModelParameters<InputT extends BaseInput,OutputT extends BaseResponse>,InputT extends BaseInput,OutputT extends BaseResponse>