Documentation
¶
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
Types ¶
type Client ¶
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 ¶
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) Recv ¶
func (g *StreamAdapter) Recv() (chat.MessageStreamResponse, error)
Recv gets the next Gemini content chunk
Source Files
Click to show internal directories.
Click to hide internal directories.