From 09361b94082a943a13b8a43fec9c688b70b631b7 Mon Sep 17 00:00:00 2001
From: Kujtim Hoxha <kujtimii.h@gmail.com>
Date: Wed, 7 Jan 2026 12:11:21 +0100
Subject: [PATCH] refactor: move model family

---
 internal/agent/{ => prompt}/model_family.go      | 2 +-
 internal/agent/{ => prompt}/model_family_test.go | 2 +-
 internal/agent/prompt/prompt.go                  | 6 +++---
 internal/agent/prompts.go                        | 3 +--
 4 files changed, 6 insertions(+), 7 deletions(-)
 rename internal/agent/{ => prompt}/model_family.go (98%)
 rename internal/agent/{ => prompt}/model_family_test.go (98%)

diff --git a/internal/agent/model_family.go b/internal/agent/prompt/model_family.go
similarity index 98%
rename from internal/agent/model_family.go
rename to internal/agent/prompt/model_family.go
index abc2f4ff20642b47418fdd1cbeca2ef8a428ab8c..b1d941c119fc8aef38f819ecdb5809dc365430b7 100644
--- a/internal/agent/model_family.go
+++ b/internal/agent/prompt/model_family.go
@@ -1,4 +1,4 @@
-package agent
+package prompt
 
 import "strings"
 
diff --git a/internal/agent/model_family_test.go b/internal/agent/prompt/model_family_test.go
similarity index 98%
rename from internal/agent/model_family_test.go
rename to internal/agent/prompt/model_family_test.go
index 851f8096ca52f4df3202f5d36e39c1356aa3db61..ac9764f5f8957ef71fe561bab4c8363cf93f3cd7 100644
--- a/internal/agent/model_family_test.go
+++ b/internal/agent/prompt/model_family_test.go
@@ -1,4 +1,4 @@
-package agent
+package prompt
 
 import (
 	"testing"
diff --git a/internal/agent/prompt/prompt.go b/internal/agent/prompt/prompt.go
index c0b0dac93526813179ec6f9d2fcddd89957a9c2c..a4b0020309c6b1a6aa2e9657e53635f538e6d12f 100644
--- a/internal/agent/prompt/prompt.go
+++ b/internal/agent/prompt/prompt.go
@@ -24,13 +24,13 @@ type Prompt struct {
 	now         func() time.Time
 	platform    string
 	workingDir  string
-	modelFamily string
+	modelFamily ModelFamily
 }
 
 type PromptDat struct {
 	Provider      string
 	Model         string
-	ModelFamily   string
+	ModelFamily   ModelFamily
 	Config        config.Config
 	WorkingDir    string
 	IsGitRepo     bool
@@ -66,7 +66,7 @@ func WithWorkingDir(workingDir string) Option {
 	}
 }
 
-func WithModelFamily(modelFamily string) Option {
+func WithModelFamily(modelFamily ModelFamily) Option {
 	return func(p *Prompt) {
 		p.modelFamily = modelFamily
 	}
diff --git a/internal/agent/prompts.go b/internal/agent/prompts.go
index c6acd52eb08d44a1a3a4e43228d36ffb054fa376..5f6635eba2ee94504713aa2eb86935789f05f6b3 100644
--- a/internal/agent/prompts.go
+++ b/internal/agent/prompts.go
@@ -18,8 +18,7 @@ var taskPromptTmpl []byte
 var initializePromptTmpl []byte
 
 func coderPrompt(modelName string, opts ...prompt.Option) (*prompt.Prompt, error) {
-	family := DetectModelFamily(modelName)
-	opts = append(opts, prompt.WithModelFamily(string(family)))
+	opts = append(opts, prompt.WithModelFamily(prompt.DetectModelFamily(modelName)))
 	return prompt.NewPrompt("coder", string(coderPromptTmpl), opts...)
 }