googlegenai

type CodeExecutionResult struct {
	Outcome string `json:"outcome"`
	Output  string `json:"output"`
}

type EmbedOptions struct {
	// Document title.
	Title string `json:"title,omitempty"`
	// Task type: RETRIEVAL_QUERY, RETRIEVAL_DOCUMENT, and so forth.
	// See the Vertex AI text embedding docs.
	TaskType string `json:"task_type,omitempty"`
}

type ExecutableCode struct {
	Language string `json:"language"`
	Code     string `json:"code"`
}

type GeminiConfig struct {
	// MaxOutputTokens is the maximum number of tokens to generate.
	MaxOutputTokens int `json:"maxOutputTokens,omitempty"`
	// StopSequences is the list of sequences where the model will stop generating further tokens.
	StopSequences []string `json:"stopSequences,omitempty"`
	// Temperature is the temperature to use for the model.
	Temperature float64 `json:"temperature,omitempty"`
	// TopK is the number of top tokens to consider for the model.
	TopK int `json:"topK,omitempty"`
	// TopP is the top-p value to use for the model.
	TopP float64 `json:"topP,omitempty"`
	// Version is the version of the model to use.
	Version string `json:"version,omitempty"`
	// SafetySettings is the list of safety settings to use for the model.
	SafetySettings []*SafetySetting `json:"safetySettings,omitempty"`
	// CodeExecution is whether to allow executing of code generated by the model.
	CodeExecution bool `json:"codeExecution,omitempty"`
	// Response modalities for returned model messages
	ResponseModalities []Modality `json:"responseModalities,omitempty"`
	// Thinking configuration controls the model's internal reasoning process
	ThinkingConfig *ThinkingConfig `json:"thinkingConfig,omitempty"`
}

type GoogleAI struct {
	APIKey string // API key to access the service. If empty, the values of the environment variables GEMINI_API_KEY or GOOGLE_API_KEY will be consulted, in that order.
	// contains filtered or unexported fields
}

type HarmBlockMethod string

type HarmBlockThreshold string

type HarmCategory string

type Modality string

type SafetySetting struct {
	// Determines if the harm block method uses probability or probability
	// and severity scores.
	Method HarmBlockMethod `json:"method,omitempty"`
	// Required. Harm category.
	Category HarmCategory `json:"category,omitempty"`
	// Required. The harm block threshold.
	Threshold HarmBlockThreshold `json:"threshold,omitempty"`
}

type ThinkingConfig struct {
	// Indicates whether the response should include thoughts (if available and supported)
	IncludeThoughts bool `json:"includeThoughts,omitempty"`
	// Thinking budget in tokens. If set to zero, thinking gets disabled
	ThinkingBudget int32 `json:"thinkingBudget,omitempty"`
}

type VertexAI struct {
	ProjectID string // Google Cloud project to use for Vertex AI. If empty, the value of the environment variable GOOGLE_CLOUD_PROJECT will be consulted.
	Location  string // Location of the Vertex AI service. If empty, GOOGLE_CLOUD_LOCATION and GOOGLE_CLOUD_REGION environment variables will be consulted, in that order.
	// contains filtered or unexported fields
}