feat: add gRPC client and endpoint options for GoogleAI

Author: asdek

llms/googleai/option.go

@@ -7,6 +7,7 @@ import (
 
 	"cloud.google.com/go/vertexai/genai"
 	"google.golang.org/api/option"
+	"google.golang.org/grpc"
 )
 
 // Options is a set of options for GoogleAI and Vertex clients.
@@ -90,6 +91,24 @@ func WithRest() Option {
 	}
 }
 
+// WithGRPCClient append a ClientOption that uses the provided gRPC client to
+// make requests.
+// This is useful for gemini clients.
+func WithGRPCClient(grpcClient *grpc.ClientConn) Option {
+	return func(opts *Options) {
+		opts.ClientOptions = append(opts.ClientOptions, option.WithGRPCConn(grpcClient))
+	}
+}
+
+// WithEndpoint append a ClientOption that uses the provided endpoint to
+// make requests.
+// This is useful for gemini clients.
+func WithEndpoint(endpoint string) Option {
+	return func(opts *Options) {
+		opts.ClientOptions = append(opts.ClientOptions, option.WithEndpoint(endpoint))
+	}
+}
+
 // WithHTTPClient append a ClientOption that uses the provided HTTP client to
 // make requests.
 // This is useful for vertex clients.