genai

Documentation

Index

• Variables
• type ClientOptions
• type GeminiClient
  • func NewGeminiClient(apiKey string) *GeminiClient
  • func (gc *GeminiClient) Summarize(s string) (string, error)
  • func (gc *GeminiClient) TokenCount(s string) int
• type GenClient
• type VertexClient
  • func NewVertexClient(opts ClientOptions) *VertexClient
  • func (gc *VertexClient) Summarize(s string) (string, error)
  • func (gc *VertexClient) TokenCount(s string) int

Variables

var MODEL_NAME = "gemini-1.5-flash-001"

var MODEL_NAME = "gemini-pro" var MODEL_NAME = "gemini-1.5-pro-preview-0409"

Types

type ClientOptions

type ClientOptions struct {
	ApiKey    string
	Region    string
	ProjectId string
}

type GeminiClient

type GeminiClient struct {
	GenClient
	ApiKey string
}

func NewGeminiClient

func NewGeminiClient(apiKey string) *GeminiClient

func (*GeminiClient) Summarize

func (gc *GeminiClient) Summarize(s string) (string, error)

func (*GeminiClient) TokenCount

func (gc *GeminiClient) TokenCount(s string) int

type GenClient

type GenClient interface {
	TokenCount(s string) int
	Summarize(s string) (string, error)
}

type VertexClient

type VertexClient struct {
	GenClient
	Opts ClientOptions
}

func NewVertexClient

func NewVertexClient(opts ClientOptions) *VertexClient

func (*VertexClient) Summarize

func (gc *VertexClient) Summarize(s string) (string, error)

func (*VertexClient) TokenCount

func (gc *VertexClient) TokenCount(s string) int