aistudio

package

v0.0.0-...-cdc1604 Latest Latest Go to latest Published: Sep 17, 2026 License: MIT Imports: 10 Imported by: 1

Details

This section is empty.

This section is empty.

This section is empty.

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

func New(opts ...Opt) (*AiStudio, error)

func (p *AiStudio) ChatCompletion(ctx context.Context, llm internal.Adapter, requester llmberjack.Requester) (*llmberjack.InnerResponse, error)

func (p *AiStudio) CloseThread(threadId *llmberjack.ThreadId)

func (p *AiStudio) CopyThread(threadId *llmberjack.ThreadId) *llmberjack.ThreadId

func (p *AiStudio) Init(adapter internal.Adapter) error

func (*AiStudio) RequestOptionsType() reflect.Type

func (p *AiStudio) ResetThread(threadId *llmberjack.ThreadId)

type Opt func(*AiStudio)

func WithApiKey(apiKey string) Opt

func WithBackend(backend genai.Backend) Opt

WithBackend represents which Google GenAI backend to use (VertexAI or Gemini).

It only accepts values of `genai.GeminiAPI` or `genai.VertexAI`.

func WithDefaultModel(model string) Opt

func WithLocation(location string) Opt

WithLocation defines the Google Cloud Platform region to use to connect to VertexAI.

It is only taken into account when using the VertexAI backend.

func WithProject(project string) Opt

WithProject defines the Google Cloud Platform project to use to connect to VertexAI.

It is only taken into account when using the VertexAI backend.

type RequestOptions struct {
	GoogleSearch *bool
	TopK         *float64
	Thinking     *ThinkingConfig
}

func (RequestOptions) ProviderRequestOptions()

type ThinkingConfig struct {
	IncludeThoughts bool
	ThinkingLevel   string
}