protocol

type Chat ¶

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

Chat implements the minimal Core Model and optional Streamer capabilities.

func NewChat(ctx context.Context, config ChatConfig) (*Chat, error)

func (c *Chat) Call(ctx context.Context, req *corechat.Request) (*corechat.Response, error)

func (c *Chat) Stream(ctx context.Context, req *corechat.Request) iter.Seq2[*corechat.ResponseDelta, error]

type ChatConfig ¶

type ChatConfig struct {
	Provider       string
	Client         ClientConfig
	DefaultOptions corechat.Options
}

func (c ChatConfig) Validate() error

type ClientConfig ¶

type ClientConfig struct {
	APIKey string

	// Backend selects the genai backend. Zero value falls back to
	// [genai.BackendGeminiAPI] — the public Gemini API. Set to
	// [genai.BackendVertexAI] for GCP-hosted enterprise deployments;
	// Project and Location become required in that mode and APIKey
	// is ignored in favor of the supplied [genai.ClientConfig.Credentials]
	// (or ADC).
	Backend genai.Backend

	// Project is the GCP project id, required when Backend ==
	// BackendVertexAI. Ignored otherwise.
	Project string

	// Location is the GCP region (e.g. "us-central1"), required when
	// Backend == BackendVertexAI. Ignored otherwise.
	Location string

	// BaseURL overrides the genai client endpoint. Optional —
	// production users should leave it empty (the SDK picks the right
	// host per Backend). Useful for mock servers / corporate proxies.
	BaseURL    string
	APIVersion string

	// HTTPClient remains owned by the caller; the SDK borrows a shallow copy.
	HTTPClient *http.Client
}

func (c ClientConfig) Validate() error

type EmbeddingModel ¶

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

EmbeddingModel admits the SDK's dimensionality representation; the selected model owns its supported dimensions. Core's text-only request cannot carry multimodal embedding inputs through a hidden native extension.

func NewEmbeddingModel(ctx context.Context, config EmbeddingModelConfig) (*EmbeddingModel, error)

func (e *EmbeddingModel) Call(ctx context.Context, req *embedding.Request) (*embedding.Response, error)

type EmbeddingModelConfig ¶

type EmbeddingModelConfig struct {
	Provider       string
	Client         ClientConfig
	DefaultOptions embedding.Options
}

func (e EmbeddingModelConfig) Validate() error

type GenerationClient ¶ added in v0.54.0

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

GenerationClient owns GenerateContent's native request and response boundary. The SDK supplies authentication, DTOs, and generation-config projection; its unchecked response converters are never entered.

func NewGenerationClient(ctx context.Context, config ClientConfig) (*GenerationClient, error)

func (g *GenerationClient) GenerateContent(ctx context.Context, model string, contents []*genai.Content, config *genai.GenerateContentConfig) (*genai.GenerateContentResponse, error)

func (g *GenerationClient) GenerateContentStream(ctx context.Context, model string, contents []*genai.Content, config *genai.GenerateContentConfig) iter.Seq2[*genai.GenerateContentResponse, error]

type SpeechModel ¶ added in v0.41.0

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

SpeechModel owns Gemini 2.5's unary-only synthesis protocol. Gemini 3.1 uses StreamingSpeechModel for both aggregate and streaming output.

Speed and OutputFormat are not honored: Gemini's TTS has no playback-rate knob. GenerateContent returns 24 kHz signed 16-bit little-endian PCM; callers choose their own container at the application boundary.

func NewSpeechModel(ctx context.Context, config SpeechModelConfig) (*SpeechModel, error)

func (s *SpeechModel) Call(ctx context.Context, req *speech.Request) (*speech.Response, error)

type SpeechModelConfig ¶ added in v0.41.0

type SpeechModelConfig struct {
	Provider       string
	Client         ClientConfig
	DefaultOptions speech.Options
}

func (s SpeechModelConfig) Validate() error

type StreamingSpeechModel ¶ added in v0.41.0

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

StreamingSpeechModel exposes the native incremental audio protocol.

func NewStreamingSpeechModel(ctx context.Context, config SpeechModelConfig) (*StreamingSpeechModel, error)

func (s *StreamingSpeechModel) Call(ctx context.Context, req *speech.Request) (*speech.Response, error)

func (s *StreamingSpeechModel) Stream(ctx context.Context, req *speech.Request) iter.Seq2[*speech.Response, error]

type TextCounter ¶ added in v0.21.0

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

TextCounter reports input-token counts via Gemini's count_tokens endpoint. Implements tokenizer.TextCounter so it drops into code paths gating on token budgets (RAG chunking, prompt-window checks).

func NewTextCounter(ctx context.Context, config TextCounterConfig) (*TextCounter, error)

func (t *TextCounter) CountText(ctx context.Context, text string) (int, error)

type TextCounterConfig ¶ added in v0.21.0

type TextCounterConfig struct {
	Client ClientConfig
	Model  string
}

Token counts vary across model families — supply the same Model name you intend to send chat requests under so the count matches the real billing.

func (t TextCounterConfig) Validate() error

type TranscriptionModel ¶ added in v0.41.0

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

TranscriptionModel exposes Gemini's multimodal chat through the transcription interface. Gemini has no /transcribe endpoint — any audio-accepting model returns a transcript when prompted. This adapter uses the stable instruction "Transcribe this audio.".

func NewTranscriptionModel(ctx context.Context, config TranscriptionModelConfig) (*TranscriptionModel, error)

func (t *TranscriptionModel) Call(ctx context.Context, req *transcription.Request) (*transcription.Response, error)

type TranscriptionModelConfig ¶ added in v0.41.0

type TranscriptionModelConfig struct {
	Provider       string
	Client         ClientConfig
	DefaultOptions transcription.Options
}

func (t TranscriptionModelConfig) Validate() error