vertexai

Documentation

Index

• type AI
  • func NewAI(ctx context.Context, projectID, locationID, modelID string) *AI
  • func (ai *AI) GenerateImageFromText(ctx context.Context, prompt string) (image.Image, error)
  • func (ai *AI) SimpleQuery(ctx context.Context, prompt string) (string, error)
  • func (ai *AI) SimpleQueryPng(ctx context.Context, prompt string, image []byte) (string, error)

Types

type AI

type AI struct {
	Client *genai.Client
	Gemini *genai.GenerativeModel
	Token  string
	// contains filtered or unexported fields
}

func NewAI

func NewAI(ctx context.Context, projectID, locationID, modelID string) *AI

func (*AI) GenerateImageFromText

func (ai *AI) GenerateImageFromText(ctx context.Context, prompt string) (image.Image, error)

GenerateImageFromText generates an image from a given text prompt. The prompt parameter provides the description of the image to generate. It returns the generated image as an image.Image object, and any error encountered.

func (*AI) SimpleQuery

func (ai *AI) SimpleQuery(ctx context.Context, prompt string) (string, error)

SimpleQuery to the AI model

func (*AI) SimpleQueryPng

func (ai *AI) SimpleQueryPng(ctx context.Context, prompt string, image []byte) (string, error)

SimpleQuery to the AI model