package ollama

type Ollama struct {
	baseUrl       string
	customHeaders map[string]string
}

func New(baseUrl string) *Ollama {
	return &Ollama{baseUrl: baseUrl, customHeaders: make(map[string]string)}
}

func (o *Ollama) SetCustomHeaders(customHeaders map[string]string) {
	o.customHeaders = customHeaders
}