googlegenai

package
v0.3.0 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Mar 28, 2025 License: Apache-2.0 Imports: 9 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func GoogleAIEmbedder

func GoogleAIEmbedder(g *genkit.Genkit, name string) ai.Embedder

GoogleAIEmbedder returns the ai.Embedder with the given name. It returns nil if the embedder was not defined.

func GoogleAIModel

func GoogleAIModel(g *genkit.Genkit, name string) ai.Model

GoogleAIModel returns the ai.Model with the given name. It returns nil if the model was not defined.

func VertexAIEmbedder

func VertexAIEmbedder(g *genkit.Genkit, name string) ai.Embedder

VertexAIEmbedder returns the ai.Embedder with the given name. It returns nil if the embedder was not defined.

func VertexAIModel

func VertexAIModel(g *genkit.Genkit, name string) ai.Model

VertexAIModel returns the ai.Model with the given name. It returns nil if the model was not defined.

Types

type GoogleAI

type GoogleAI struct {
	APIKey string // API key to access the service. If empty, the values of the environment variables GEMINI_API_KEY or GOOGLE_API_KEY will be consulted, in that order.
	// contains filtered or unexported fields
}

GoogleAI is a Genkit plugin for interacting with the Google AI service.

func (*GoogleAI) DefineEmbedder

func (ga *GoogleAI) DefineEmbedder(g *genkit.Genkit, name string) (ai.Embedder, error)

DefineEmbedder defines an embedder with a given name.

func (*GoogleAI) DefineModel

func (ga *GoogleAI) DefineModel(g *genkit.Genkit, name string, info *ai.ModelInfo) (ai.Model, error)

DefineModel defines an unknown model with the given name. The second argument describes the capability of the model. Use [IsDefinedModel] to determine if a model is already defined. After [Init] is called, only the known models are defined.

func (*GoogleAI) Init

func (ga *GoogleAI) Init(ctx context.Context, g *genkit.Genkit) (err error)

Init initializes the Google AI plugin and all known models and embedders. After calling Init, you may call [DefineModel] and [DefineEmbedder] to create and register any additional generative models and embedders

func (*GoogleAI) IsDefinedEmbedder

func (ga *GoogleAI) IsDefinedEmbedder(g *genkit.Genkit, name string) bool

IsDefinedEmbedder reports whether the named [Embedder] is defined by this plugin.

func (*GoogleAI) Name

func (ga *GoogleAI) Name() string

Name returns the name of the plugin.

type VertexAI

type VertexAI struct {
	ProjectID string // Google Cloud project to use for Vertex AI. If empty, the value of the environment variable GOOGLE_CLOUD_PROJECT will be consulted.
	Location  string // Location of the Vertex AI service. If empty, GOOGLE_CLOUD_LOCATION and GOOGLE_CLOUD_REGION environment variables will be consulted, in that order.
	// contains filtered or unexported fields
}

VertexAI is a Genkit plugin for interacting with the Google Vertex AI service.

func (*VertexAI) DefineEmbedder

func (v *VertexAI) DefineEmbedder(g *genkit.Genkit, name string) (ai.Embedder, error)

DefineEmbedder defines an embedder with a given name.

func (*VertexAI) DefineModel

func (v *VertexAI) DefineModel(g *genkit.Genkit, name string, info *ai.ModelInfo) (ai.Model, error)

DefineModel defines an unknown model with the given name. The second argument describes the capability of the model. Use [IsDefinedModel] to determine if a model is already defined. After [Init] is called, only the known models are defined.

func (*VertexAI) Init

func (v *VertexAI) Init(ctx context.Context, g *genkit.Genkit) (err error)

Init initializes the VertexAI plugin and all known models and embedders. After calling Init, you may call [DefineModel] and [DefineEmbedder] to create and register any additional generative models and embedders

func (*VertexAI) IsDefinedEmbedder

func (v *VertexAI) IsDefinedEmbedder(g *genkit.Genkit, name string) bool

IsDefinedEmbedder reports whether the named [Embedder] is defined by this plugin.

func (*VertexAI) Name

func (v *VertexAI) Name() string

Name returns the name of the plugin.

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL