gemini

type Content struct {
	Role  string `json:"role"`
	Parts []Part `json:"parts"`
}

type FileData struct {
	MimeType string `json:"mimeType,omitempty"`
	FileUri  string `json:"fileUri,omitempty"`
}

type FileUploadResponse struct {
	File struct {
		MimeType string `json:"mimeType"`
		Uri      string `json:"uri"`
	} `json:"file"`
}

type GenerateContentRequest struct {
	Contents          []Content         `json:"contents"`
	SafetySettings    []SafetySetting   `json:"safetySettings"`
	GenerationConfig  GenerationConfig  `json:"generationConfig"`
	SystemInstruction SystemInstruction `json:"system_instruction"`
	Tools             []Tool            `json:"tools"`
}

type GenerateContentResponse struct {
	Candidates []struct {
		Content       Content `json:"content"`
		FinishReason  string  `json:"finishReason"`
		SafetyRatings []struct {
			Category    string `json:"category"`
			Probability string `json:"probability"`
		} `json:"safetyRatings"`
		GroundingMetadata GroundingMetadata `json:"groundingMetadata"`
	} `json:"candidates"`
}

type GenerationConfig struct {
	Temperature     float64        `json:"temperature"`
	TopK            int            `json:"topK"`
	TopP            int            `json:"topP"`
	MaxOutputTokens int            `json:"maxOutputTokens"`
	ThinkingConfig  ThinkingConfig `json:"thinkingConfig"`
}

type GroundingMetadata struct {
	// https://ai.google.dev/api/generate-content#GroundingChunk
	GroundingChunks []struct {
		// https://ai.google.dev/api/generate-content#Web
		Web struct {
			Uri   string `json:"uri"`
			Title string `json:"title"`
		} `json:"web"`
	} `json:"groundingChunks"`
	WebSearchQueries []string `json:"webSearchQueries"`
}

type Part struct {
	Text     string    `json:"text,omitempty"`
	FileData *FileData `json:"fileData,omitempty"`
}

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

type SafetySetting struct {
	Category  string `json:"category"`  // https://ai.google.dev/api/generate-content#v1beta.HarmCategory
	Threshold string `json:"threshold"` // https://ai.google.dev/api/generate-content#HarmBlockThreshold
}

type Service interface {
	GetHistory(chat *gotgbot.Chat) (model.GeminiData, error)
	ResetHistory(chat *gotgbot.Chat) error
	SetHistory(chat *gotgbot.Chat, history string) error
}

type SystemInstruction struct {
	Parts []Part `json:"parts"`
}

type ThinkingConfig struct {
	IncludeThoughts bool `json:"includeThoughts"`
	ThinkingBudget  int  `json:"thinkingBudget"`
}

type Tool struct {
	// GoogleSearch - https://ai.google.dev/api/caching#GoogleSearch (has no fields)
	GoogleSearch struct {
	} `json:"google_search"`
}