From 1b3b94fb669aec824a00c6c21bba52368b56cff7 Mon Sep 17 00:00:00 2001
From: ben
Date: Wed, 22 Jan 2025 21:13:04 +0100
Subject: add Gemma2 which is better for language-related prompts.

---
 docker-compose.yml                     | 2 +-
 src/aichat/roles/english-translator.md | 2 +-
 src/aichat/roles/spell-checker.md      | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index 7cdcfdd..97b141c 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -45,7 +45,7 @@ services:
     build:
       dockerfile: src/llm_provision/Dockerfile
     environment:
-      - MODELS=qwen2.5:latest,qwen2.5-coder:32b,nomic-embed-text:latest
+      - MODELS=qwen2.5:latest,qwen2.5-coder:32b,nomic-embed-text:latest,gemma2
     restart: no
     depends_on:
       ollama:
diff --git a/src/aichat/roles/english-translator.md b/src/aichat/roles/english-translator.md
index 8f8c4bb..63bbd13 100644
--- a/src/aichat/roles/english-translator.md
+++ b/src/aichat/roles/english-translator.md
@@ -1,5 +1,5 @@
 ---
-model: ollama:qwen2.5
+model: ollama:gemma2
 temperature: 0
 top_p: 0
 
diff --git a/src/aichat/roles/spell-checker.md b/src/aichat/roles/spell-checker.md
index f4d6787..23e9c2e 100644
--- a/src/aichat/roles/spell-checker.md
+++ b/src/aichat/roles/spell-checker.md
@@ -1,5 +1,5 @@
 ---
-model: ollama:qwen2.5
+model: ollama:gemma2
 temperature: 0
 top_p: 0
 
-- 
cgit v1.2.3