gemini

package

v1.16.0 Latest Latest Go to latest Published: Dec 30, 2025 License: Apache-2.0 Imports: 24 Imported by: 0

Details

Valid go.mod file
Redistributable license
Tagged version
Stable version
Learn more about best practices

Repository

github.com/docker/cagent

Links

Documentation ¶

Index ¶

func ConvertParametersToSchema(params any) (*genai.Schema, error)
type Client
- func NewClient(ctx context.Context, cfg *latest.ModelConfig, env environment.Provider, ...) (*Client, error)
- func (c *Client) CreateChatCompletionStream(ctx context.Context, messages []chat.Message, requestTools []tools.Tool) (chat.MessageStream, error)
- func (c *Client) Rerank(ctx context.Context, query string, documents []types.Document, criteria string) ([]float64, error)
type StreamAdapter
- func NewStreamAdapter(iter func(func(*genai.GenerateContentResponse, error) bool), model string, ...) *StreamAdapter
- func (g *StreamAdapter) Close()
- func (g *StreamAdapter) Recv() (chat.MessageStreamResponse, error)

Constants ¶

This section is empty.

Variables ¶

This section is empty.

Functions ¶

func ConvertParametersToSchema ¶ added in v1.7.0

func ConvertParametersToSchema(params any) (*genai.Schema, error)

ConvertParametersToSchema converts parameters to Gemini Schema format

Types ¶

type Client ¶

type Client struct {
	base.Config
	// contains filtered or unexported fields
}

Client represents a Gemini client wrapper It implements the provider.Provider interface

func NewClient ¶

func NewClient(ctx context.Context, cfg *latest.ModelConfig, env environment.Provider, opts ...options.Opt) (*Client, error)

NewClient creates a new Gemini client from the provided configuration

func (*Client) CreateChatCompletionStream ¶

func (c *Client) CreateChatCompletionStream(
	ctx context.Context,
	messages []chat.Message,
	requestTools []tools.Tool,
) (chat.MessageStream, error)

CreateChatCompletionStream creates a streaming chat completion request

func (*Client) Rerank ¶ added in v1.9.22

func (c *Client) Rerank(ctx context.Context, query string, documents []types.Document, criteria string) ([]float64, error)

Rerank scores documents by relevance to the query using Gemini's structured output feature. It returns relevance scores in the same order as input documents.

type StreamAdapter ¶

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

StreamAdapter adapts the Gemini streaming iterator to chat.MessageStream

func NewStreamAdapter ¶

func NewStreamAdapter(iter func(func(*genai.GenerateContentResponse, error) bool), model string, trackUsage bool) *StreamAdapter

NewStreamAdapter constructs a StreamAdapter from Gemini's iterator

func (*StreamAdapter) Close ¶

func (g *StreamAdapter) Close()

Close closes the stream

func (*StreamAdapter) Recv ¶

func (g *StreamAdapter) Recv() (chat.MessageStreamResponse, error)

Recv gets the next Gemini content chunk

Source Files ¶

View all Source files

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