diff options
| -rw-r--r-- | defaults.go | 4 | ||||
| -rw-r--r-- | types.go | 4 |
2 files changed, 6 insertions, 2 deletions
diff --git a/defaults.go b/defaults.go index 60217ae..b81cfc0 100644 --- a/defaults.go +++ b/defaults.go @@ -132,4 +132,8 @@ func AddSaneDefaults(config *TomlConfig) { if config.DbBackOffMaxInterval == 0 { config.DbBackOffMaxInterval = 60 } + + if config.OllamaThink == "" { + config.OllamaThink = "false" + } } @@ -122,7 +122,7 @@ type TomlConfig struct { OllamaSeed int `toml:"ollamaSeed"` OllamaNumPredict int `toml:"ollamaNumPredict"` OllamaMinP float64 `toml:"ollamaMinP"` - OllamaThink bool `toml:"ollamaThink"` + OllamaThink string `toml:"ollamaThink"` TopP float32 `toml:"topP"` TopK int32 `toml:"topK"` IrcBackOffInitialInterval int `toml:"ircBackOffInitialInterval"` @@ -239,7 +239,7 @@ type OllamaChatMessagesResponse struct { type OllamaChatRequest struct { Model string `json:"model"` Stream bool `json:"stream"` - Think bool `json:"think"` + Think string `json:"think"` KeepAlive time.Duration `json:"keep_alive"` Options OllamaRequestOptions `json:"options"` System string `json:"system"` |