aboutsummaryrefslogtreecommitdiff
path: root/cmd/mimi/internal
diff options
context:
space:
mode:
authorXe Iaso <me@xeiaso.net>2024-07-27 08:56:01 -0400
committerXe Iaso <me@xeiaso.net>2024-07-27 08:56:01 -0400
commitd73a41507f8aab06b378fad1d30f4ccf827e65db (patch)
tree4753d52984e44a00cfda52da75e5b6e7f7e85169 /cmd/mimi/internal
parent7b9862f9410b2ca7d952d6665be02330f63c7174 (diff)
downloadx-d73a41507f8aab06b378fad1d30f4ccf827e65db.tar.xz
x-d73a41507f8aab06b378fad1d30f4ccf827e65db.zip
cmd/mimi: conversate
Signed-off-by: Xe Iaso <me@xeiaso.net>
Diffstat (limited to 'cmd/mimi/internal')
-rw-r--r--cmd/mimi/internal/flags.go6
1 files changed, 5 insertions, 1 deletions
diff --git a/cmd/mimi/internal/flags.go b/cmd/mimi/internal/flags.go
index 1f26fc4..747ea14 100644
--- a/cmd/mimi/internal/flags.go
+++ b/cmd/mimi/internal/flags.go
@@ -13,7 +13,7 @@ import (
var (
dataDir = flag.String("data-dir", "./var", "data directory for the bot")
- ollamaModel = flag.String("ollama-model", "llama3", "ollama model tag")
+ ollamaModel = flag.String("ollama-model", "llama3.1", "ollama model tag")
ollamaHost = flag.String("ollama-host", "http://xe-inference.flycast:80", "ollama host")
)
@@ -22,6 +22,10 @@ func DataDir() string {
return *dataDir
}
+func OllamaHost() string {
+ return *ollamaHost
+}
+
func OllamaClient() *ollama.Client {
return ollama.NewClient(*ollamaHost)
}