gemini

Documentation

Index

• func NewGemini(ctx context.Context, modelName string, systemPrompt string) (llms.Model, error)
• type Gemini
  • func (g *Gemini) Chat(ctx context.Context, message string) (string, error)
  • func (g *Gemini) CloseBackend() error
  • func (g *Gemini) ContentToString(content *genai.Content) string
  • func (g *Gemini) GenerateContent(ctx context.Context, prompt string) (string, error)
  • func (g *Gemini) SetSystemPrompt(ctx context.Context, prompt string) error

Functions

func NewGemini

func NewGemini(ctx context.Context, modelName string, systemPrompt string) (llms.Model, error)

NewGemini returns a new Gemini client.

Types

type Gemini

type Gemini struct {
	Client *genai.Client
	Model  *genai.GenerativeModel

	SystemPrompt string
	// contains filtered or unexported fields
}

Gemini is a wrapper around the Gemini API.

func (*Gemini) Chat

func (g *Gemini) Chat(ctx context.Context, message string) (string, error)

Chat with the model.

func (*Gemini) CloseBackend

func (g *Gemini) CloseBackend() error

Close the client.

func (*Gemini) ContentToString

func (g *Gemini) ContentToString(content *genai.Content) string

Cotent to string

func (*Gemini) GenerateContent

func (g *Gemini) GenerateContent(ctx context.Context, prompt string) (string, error)

GenerateContent generates content from a prompt.

func (*Gemini) SetSystemPrompt

func (g *Gemini) SetSystemPrompt(ctx context.Context, prompt string) error

Set system prompt