modelbuilder

type ArkConnectionInfo struct {
	Region string `json:"region,omitempty" yaml:"region,omitempty"`
}

type BuildOption func(p *LLMParams)

type Builder interface {
	Build(ctx context.Context, params *LLMParams) (model.ToolCallingChatModel, error)
}

type ConnectionInfo struct {
	APIKey  string `json:"api_key" yaml:"api_key"`
	BaseURL string `json:"base_url,omitempty" yaml:"base_url,omitempty"`
	Model   string `json:"model" yaml:"model"`

	Ark    *ArkConnectionInfo    `json:"ark,omitempty" yaml:"ark,omitempty"`
	OpenAI *OpenAIConnectionInfo `json:"openai,omitempty" yaml:"openai,omitempty"`
	Gemini *GeminiConnectionInfo `json:"gemini,omitempty" yaml:"gemini,omitempty"`
}

type GeminiConnectionInfo struct {
	Backend  string `json:"backend,omitempty" yaml:"backend,omitempty"`
	Project  string `json:"project,omitempty" yaml:"project,omitempty"`
	Location string `json:"location,omitempty" yaml:"location,omitempty"`
}

type LLMParams struct {
	Temperature      *float32       `json:"temperature,omitempty" yaml:"temperature,omitempty"`
	FrequencyPenalty *float32       `json:"frequency_penalty,omitempty" yaml:"frequency_penalty,omitempty"`
	PresencePenalty  *float32       `json:"presence_penalty,omitempty" yaml:"presence_penalty,omitempty"`
	MaxTokens        *int           `json:"max_tokens,omitempty" yaml:"max_tokens,omitempty"`
	TopP             *float32       `json:"top_p,omitempty" yaml:"top_p,omitempty"`
	TopK             *int32         `json:"top_k,omitempty" yaml:"top_k,omitempty"`
	ResponseFormat   ResponseFormat `json:"response_format,omitempty" yaml:"response_format,omitempty"`
	EnableThinking   *bool          `json:"enable_thinking,omitempty" yaml:"enable_thinking,omitempty"`
}

type ModelClass string

type ModelConfig struct {
	ModelClass   ModelClass      `json:"model_class" yaml:"model_class"`
	Connection   *ConnectionInfo `json:"connection" yaml:"connection"`
	ThinkingType ThinkingType    `json:"thinking_type,omitempty" yaml:"thinking_type,omitempty"`
}

type OpenAIConnectionInfo struct {
	ByAzure    bool   `json:"by_azure,omitempty" yaml:"by_azure,omitempty"`
	APIVersion string `json:"api_version,omitempty" yaml:"api_version,omitempty"`
}

type ResponseFormat string

type ThinkingType string