gemini

package

v1.0.21 Latest Latest Go to latest Published: Feb 10, 2025 License: MIT Imports: 14 Imported by: 0

Details

This section is empty.

This section is empty.

func New(ctx context.Context, opts ...ai.ClientOption) (ai.AI, error)

func NewWithClient(client *genai.Client, model string) ai.AI

type ChatResponse struct {
	*genai.GenerateContentResponse
}

func (resp *ChatResponse) FunctionCalls() (res []ai.FunctionCall)

func (resp *ChatResponse) Raw() any

func (resp *ChatResponse) Results() (res []string)

func (resp *ChatResponse) String() string

func (resp *ChatResponse) TokenCount() (res ai.TokenCount)

type ChatSession struct {
	// contains filtered or unexported fields
}

func (session *ChatSession) Chat(ctx context.Context, parts ...ai.Part) (ai.ChatResponse, error)

func (session *ChatSession) ChatStream(ctx context.Context, parts ...ai.Part) (ai.ChatStream, error)

func (session *ChatSession) History() (history []ai.Content)

type ChatStream struct {
	// contains filtered or unexported fields
}

func (stream *ChatStream) Close() error

func (stream *ChatStream) Next() (ai.ChatResponse, error)

type Gemini struct {
	*genai.Client
	// contains filtered or unexported fields
}

func (ai *Gemini) Chat(ctx context.Context, parts ...ai.Part) (ai.ChatResponse, error)

func (ai *Gemini) ChatSession() ai.ChatSession

func (ai *Gemini) ChatStream(ctx context.Context, parts ...ai.Part) (ai.ChatStream, error)

func (Gemini) LLMs() ai.LLMs

func (gemini *Gemini) Limit() (rpm int64)

func (gemini *Gemini) Model(ctx context.Context) (string, error)

func (ai *Gemini) SetCount(i int64)

func (gemini *Gemini) SetFunctionCall(f []ai.Function, mode ai.FunctionCallingMode)

func (ai *Gemini) SetJSONResponse(set bool, schema *ai.JSONSchema)

func (gemini *Gemini) SetLimit(rpm int64)

func (ai *Gemini) SetMaxTokens(i int64)

func (ai *Gemini) SetModel(model string)

func (ai *Gemini) SetTemperature(f float64)

func (ai *Gemini) SetTopP(f float64)