15 lines
311 B
Go
15 lines
311 B
Go
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
|
|
}
|