adk

type AgentCompressionConfig struct {
	CompactionInterval *int   `json:"compaction_interval,omitempty"`
	OverlapSize        *int   `json:"overlap_size,omitempty"`
	SummarizerModel    Model  `json:"summarizer_model,omitempty"`
	PromptTemplate     string `json:"prompt_template,omitempty"`
	TokenThreshold     *int   `json:"token_threshold,omitempty"`
	EventRetentionSize *int   `json:"event_retention_size,omitempty"`
}

type AgentConfig struct {
	Model         Model                 `json:"model"`
	Description   string                `json:"description"`
	Instruction   string                `json:"instruction"`
	HttpTools     []HttpMcpServerConfig `json:"http_tools"`
	SseTools      []SseMcpServerConfig  `json:"sse_tools"`
	RemoteAgents  []RemoteAgentConfig   `json:"remote_agents"`
	ExecuteCode   bool                  `json:"execute_code,omitempty"`
	Stream        bool                  `json:"stream"`
	Memory        *MemoryConfig         `json:"memory,omitempty"`
	ContextConfig *AgentContextConfig   `json:"context_config,omitempty"`
}

type AgentContextConfig struct {
	Compaction *AgentCompressionConfig `json:"compaction,omitempty"`
}

type Anthropic struct {
	BaseModel
	BaseUrl string `json:"base_url"`
}

type AzureOpenAI struct {
	BaseModel
}

type BaseModel struct {
	Type    string            `json:"type"`
	Model   string            `json:"model"`
	Headers map[string]string `json:"headers,omitempty"`

	// TLS/SSL configuration (applies to all model types)
	TLSDisableVerify    *bool   `json:"tls_disable_verify,omitempty"`
	TLSCACertPath       *string `json:"tls_ca_cert_path,omitempty"`
	TLSDisableSystemCAs *bool   `json:"tls_disable_system_cas,omitempty"`

	// APIKeyPassthrough enables forwarding the Bearer token from incoming requests
	// as the LLM API key instead of using a static secret.
	APIKeyPassthrough bool `json:"api_key_passthrough,omitempty"`
}

type Bedrock struct {
	BaseModel
	// Region is the AWS region where the model is available
	Region string `json:"region,omitempty"`
}

type EmbeddingConfig struct {
	Provider string `json:"provider"`
	Model    string `json:"model"`
	BaseUrl  string `json:"base_url,omitempty"`
}

type Gemini struct {
	BaseModel
}

type GeminiAnthropic struct {
	BaseModel
}

type GeminiVertexAI struct {
	BaseModel
}

type HttpMcpServerConfig struct {
	Params         StreamableHTTPConnectionParams `json:"params"`
	Tools          []string                       `json:"tools"`
	AllowedHeaders []string                       `json:"allowed_headers,omitempty"`
}

type MemoryConfig struct {
	TTLDays   int              `json:"ttl_days,omitempty"`
	Embedding *EmbeddingConfig `json:"embedding,omitempty"`
}

type Model interface {
	GetType() string
}

type Ollama struct {
	BaseModel
	Options map[string]string `json:"options,omitempty"`
}

type OpenAI struct {
	BaseModel
	BaseUrl          string   `json:"base_url"`
	FrequencyPenalty *float64 `json:"frequency_penalty,omitempty"`
	MaxTokens        *int     `json:"max_tokens,omitempty"`
	N                *int     `json:"n,omitempty"`
	PresencePenalty  *float64 `json:"presence_penalty,omitempty"`
	ReasoningEffort  *string  `json:"reasoning_effort,omitempty"`
	Seed             *int     `json:"seed,omitempty"`
	Temperature      *float64 `json:"temperature,omitempty"`
	Timeout          *int     `json:"timeout,omitempty"`
	TopP             *float64 `json:"top_p,omitempty"`
}

type RemoteAgentConfig struct {
	Name        string            `json:"name"`
	Url         string            `json:"url"`
	Headers     map[string]string `json:"headers,omitempty"`
	Description string            `json:"description,omitempty"`
}

type SseConnectionParams struct {
	Url            string            `json:"url"`
	Headers        map[string]string `json:"headers"`
	Timeout        *float64          `json:"timeout,omitempty"`
	SseReadTimeout *float64          `json:"sse_read_timeout,omitempty"`
}

type SseMcpServerConfig struct {
	Params         SseConnectionParams `json:"params"`
	Tools          []string            `json:"tools"`
	AllowedHeaders []string            `json:"allowed_headers,omitempty"`
}

type StreamableHTTPConnectionParams struct {
	Url              string            `json:"url"`
	Headers          map[string]string `json:"headers"`
	Timeout          *float64          `json:"timeout,omitempty"`
	SseReadTimeout   *float64          `json:"sse_read_timeout,omitempty"`
	TerminateOnClose *bool             `json:"terminate_on_close,omitempty"`
}