Merge pull request #1 from thepeterstone/claude/local-oss-model-agents-MEBqj

Local OSS models as a third runner (epic)

2 files changed, 72 insertions, 12 deletions

diff --git a/internal/config/config.go b/internal/config/config.go
index ce3b53f..5801239 100644
--- a/internal/config/config.go
+++ b/internal/config/config.go
@@ -15,19 +15,49 @@ type Project struct {
 	Dir  string `toml:"dir"`
 }
 
+// LocalModel configures an OpenAI-compatible local LLM endpoint used for
+// internal helpers (classifier, elaboration, future summarization) and as
+// the backend for the "local" runner. If Endpoint is empty, the LocalRunner
+// is not registered and the classifier falls back to the Gemini CLI.
+//
+// PreferForElaborate gates whether the API server's elaboration handler
+// uses this client. It defaults to true when Endpoint is set; users with a
+// slow or low-quality local model can disable it.
+type LocalModel struct {
+	Endpoint           string  `toml:"endpoint"`              // e.g. "http://localhost:11434/v1"
+	Model              string  `toml:"model"`                 // e.g. "llama3.1:8b"
+	TimeoutSeconds     int     `toml:"timeout_seconds"`       // default 60
+	DefaultTemperature float64 `toml:"default_temperature"`   // default 0.2
+	APIKey             string  `toml:"api_key"`               // optional bearer token
+	PreferForElaborate *bool   `toml:"prefer_for_elaborate"`  // pointer so default-true survives parse
+}
+
+// UseForElaborate returns true when elaboration should try this local model
+// before falling back to Claude/Gemini. Default is true when Endpoint is set.
+func (m LocalModel) UseForElaborate() bool {
+	if m.Endpoint == "" {
+		return false
+	}
+	if m.PreferForElaborate == nil {
+		return true
+	}
+	return *m.PreferForElaborate
+}
+
 type Config struct {
-	DataDir          string    `toml:"data_dir"`
-	DBPath           string    `toml:"-"`
-	LogDir           string    `toml:"-"`
-	ClaudeBinaryPath string    `toml:"claude_binary_path"`
-	GeminiBinaryPath string    `toml:"gemini_binary_path"`
-	MaxConcurrent    int       `toml:"max_concurrent"`
-	DefaultTimeout   string    `toml:"default_timeout"`
-	ServerAddr       string    `toml:"server_addr"`
-	WebhookURL       string    `toml:"webhook_url"`
-	WorkspaceRoot    string    `toml:"workspace_root"`
-	WebhookSecret    string    `toml:"webhook_secret"`
-	Projects         []Project `toml:"projects"`
+	DataDir          string     `toml:"data_dir"`
+	DBPath           string     `toml:"-"`
+	LogDir           string     `toml:"-"`
+	ClaudeBinaryPath string     `toml:"claude_binary_path"`
+	GeminiBinaryPath string     `toml:"gemini_binary_path"`
+	MaxConcurrent    int        `toml:"max_concurrent"`
+	DefaultTimeout   string     `toml:"default_timeout"`
+	ServerAddr       string     `toml:"server_addr"`
+	WebhookURL       string     `toml:"webhook_url"`
+	WorkspaceRoot    string     `toml:"workspace_root"`
+	WebhookSecret    string     `toml:"webhook_secret"`
+	Projects         []Project  `toml:"projects"`
+	LocalModel       LocalModel `toml:"local_model"`
 }
 
 func Default() (*Config, error) {
diff --git a/internal/config/config_test.go b/internal/config/config_test.go
index 2bba2c4..e4f1a5d 100644
--- a/internal/config/config_test.go
+++ b/internal/config/config_test.go
@@ -53,3 +53,33 @@ func TestLoadFile_MissingFile_ReturnsError(t *testing.T) {
 		t.Fatal("expected error for missing file, got nil")
 	}
 }
+
+func TestLocalModel_UseForElaborate_EmptyEndpoint(t *testing.T) {
+	m := LocalModel{}
+	if m.UseForElaborate() {
+		t.Error("empty endpoint should never opt into elaborate")
+	}
+}
+
+func TestLocalModel_UseForElaborate_DefaultTrue(t *testing.T) {
+	m := LocalModel{Endpoint: "http://localhost:11434/v1"}
+	if !m.UseForElaborate() {
+		t.Error("endpoint set + default flag should opt in")
+	}
+}
+
+func TestLocalModel_UseForElaborate_ExplicitFalse(t *testing.T) {
+	f := false
+	m := LocalModel{Endpoint: "http://localhost:11434/v1", PreferForElaborate: &f}
+	if m.UseForElaborate() {
+		t.Error("explicit false should opt out")
+	}
+}
+
+func TestLocalModel_UseForElaborate_ExplicitTrue(t *testing.T) {
+	tr := true
+	m := LocalModel{Endpoint: "http://localhost:11434/v1", PreferForElaborate: &tr}
+	if !m.UseForElaborate() {
+		t.Error("explicit true should opt in")
+	}
+}