From 024118218e3b44347d8b80c784324a2c415b888a Mon Sep 17 00:00:00 2001
From: Adrian Cole <64215+codefromthecrypt@users.noreply.github.com>
Date: Wed, 25 Sep 2024 10:12:59 +1000
Subject: [PATCH] fix: default ollama to tested model (#88)

Signed-off-by: Adrian Cole <adrian.cole@elastic.co>
---
 src/goose/cli/config.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/goose/cli/config.py b/src/goose/cli/config.py
index ee901941e5..b76835a6f2 100644
--- a/src/goose/cli/config.py
+++ b/src/goose/cli/config.py
@@ -9,6 +9,8 @@ from rich.prompt import Confirm
 from rich.text import Text
 from ruamel.yaml import YAML
 
+from exchange.providers.ollama import OLLAMA_MODEL
+
 from goose.profile import Profile
 from goose.utils import load_plugins
 from goose.utils.diff import pretty_diff
@@ -127,6 +129,7 @@ def default_model_configuration() -> Tuple[str, str, str]:
         )
 
     recommended = {
+        "ollama": (OLLAMA_MODEL, OLLAMA_MODEL),
         "openai": ("gpt-4o", "gpt-4o-mini"),
         "anthropic": (
             "claude-3-5-sonnet-20240620",