diff --git a/modules/programs/aichat.nix b/modules/programs/aichat.nix
index 38895b26..cba7b0b0 100644
--- a/modules/programs/aichat.nix
+++ b/modules/programs/aichat.nix
@@ -29,15 +29,17 @@ in
       defaultText = lib.literalExpression "{ }";
       example = lib.literalExpression ''
         {
-          model = "Ollama:mistral-small:latest";
+          model = "ollama:mistral-small3.1:latest";
           clients = [
             {
               type = "openai-compatible";
-              name = "Ollama";
+              name = "ollama";
               api_base = "http://localhost:11434/v1";
               models = [
                 {
-                  name = "llama3.2:latest";
+                  name = "mistral-small3.1:latest";
+                  supports_function_calling = true;
+                  supports_vision = true;
                 }
               ];
             }
diff --git a/tests/modules/programs/aichat/settings.nix b/tests/modules/programs/aichat/settings.nix
index 899e115d..d8d5085a 100644
--- a/tests/modules/programs/aichat/settings.nix
+++ b/tests/modules/programs/aichat/settings.nix
@@ -3,15 +3,16 @@
   programs.aichat = {
     enable = true;
     settings = {
-      model = "Ollama:mistral-small:latest";
+      model = "ollama:llama3.2:latest";
       clients = [
         {
           type = "openai-compatible";
-          name = "Ollama";
+          name = "ollama";
           api_base = "http://localhost:11434/v1";
           models = [
             {
               name = "llama3.2:latest";
+              supports_function_calling = true;
             }
           ];
         }
diff --git a/tests/modules/programs/aichat/settings.yml b/tests/modules/programs/aichat/settings.yml
index e43464d6..eecf51ec 100644
--- a/tests/modules/programs/aichat/settings.yml
+++ b/tests/modules/programs/aichat/settings.yml
@@ -2,6 +2,7 @@ clients:
 - api_base: http://localhost:11434/v1
   models:
   - name: llama3.2:latest
-  name: Ollama
+    supports_function_calling: true
+  name: ollama
   type: openai-compatible
-model: Ollama:mistral-small:latest
+model: ollama:llama3.2:latest