mirror of
https://github.com/ollama/ollama.git
synced 2026-03-27 02:58:43 +07:00
launch: set default model as active selection in vscode copilot chat
This commit is contained in:
@@ -361,9 +361,11 @@ func (v *VSCode) statePath() string {
|
||||
}
|
||||
|
||||
// ShowInModelPicker ensures the given models are visible in VS Code's Copilot
|
||||
// Chat model picker. It sets the configured models to true in the picker
|
||||
// preferences so they appear in the dropdown. Models use the VS Code identifier
|
||||
// format "ollama/Ollama/<name>".
|
||||
// Chat model picker and sets the primary model as the active selection. It sets
|
||||
// the configured models to true in the picker preferences so they appear in the
|
||||
// dropdown, and writes the first model as the selected model for both the panel
|
||||
// and editor chat views. Models use the VS Code identifier format
|
||||
// "ollama/Ollama/<name>".
|
||||
func (v *VSCode) ShowInModelPicker(models []string) error {
|
||||
if len(models) == 0 {
|
||||
return nil
|
||||
@@ -440,6 +442,18 @@ func (v *VSCode) ShowInModelPicker(models []string) error {
|
||||
return err
|
||||
}
|
||||
|
||||
// Set the primary model as the active selection in Copilot Chat so it
|
||||
// doesn't default to "auto" or whatever the user last picked manually.
|
||||
primaryID := v.modelVSCodeIDs(models[0], nameToID)[0]
|
||||
for _, key := range []string{"chat.currentLanguageModel.panel", "chat.currentLanguageModel.editor"} {
|
||||
if _, err := db.Exec("INSERT OR REPLACE INTO ItemTable (key, value) VALUES (?, ?)", key, primaryID); err != nil {
|
||||
return err
|
||||
}
|
||||
if _, err := db.Exec("INSERT OR REPLACE INTO ItemTable (key, value) VALUES (?, ?)", key+".isDefault", "false"); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
@@ -388,6 +388,71 @@ func TestShowInModelPicker(t *testing.T) {
|
||||
}
|
||||
})
|
||||
|
||||
// helper to read a string value from the state DB
|
||||
readValue := func(t *testing.T, dbPath, key string) string {
|
||||
t.Helper()
|
||||
db, err := sql.Open("sqlite3", dbPath)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
defer db.Close()
|
||||
var val string
|
||||
if err := db.QueryRow("SELECT value FROM ItemTable WHERE key = ?", key).Scan(&val); err != nil {
|
||||
return ""
|
||||
}
|
||||
return val
|
||||
}
|
||||
|
||||
t.Run("sets primary model as active selection", func(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
setTestHome(t, tmpDir)
|
||||
t.Setenv("XDG_CONFIG_HOME", "")
|
||||
setupDB(t, testVSCodePath(t, tmpDir, ""), nil, nil)
|
||||
|
||||
err := v.ShowInModelPicker([]string{"llama3.2", "qwen3:8b"})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
dbPath := testVSCodePath(t, tmpDir, filepath.Join("globalStorage", "state.vscdb"))
|
||||
panelModel := readValue(t, dbPath, "chat.currentLanguageModel.panel")
|
||||
if panelModel != "ollama/Ollama/llama3.2" {
|
||||
t.Errorf("expected panel model ollama/Ollama/llama3.2, got %q", panelModel)
|
||||
}
|
||||
editorModel := readValue(t, dbPath, "chat.currentLanguageModel.editor")
|
||||
if editorModel != "ollama/Ollama/llama3.2" {
|
||||
t.Errorf("expected editor model ollama/Ollama/llama3.2, got %q", editorModel)
|
||||
}
|
||||
panelDefault := readValue(t, dbPath, "chat.currentLanguageModel.panel.isDefault")
|
||||
if panelDefault != "false" {
|
||||
t.Errorf("expected panel isDefault false, got %q", panelDefault)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("sets cached numeric ID as active selection", func(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
setTestHome(t, tmpDir)
|
||||
t.Setenv("XDG_CONFIG_HOME", "")
|
||||
cache := []map[string]any{
|
||||
{
|
||||
"identifier": "ollama/Ollama/4",
|
||||
"metadata": map[string]any{"vendor": "ollama", "name": "llama3.2"},
|
||||
},
|
||||
}
|
||||
setupDB(t, testVSCodePath(t, tmpDir, ""), nil, cache)
|
||||
|
||||
err := v.ShowInModelPicker([]string{"llama3.2"})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
dbPath := testVSCodePath(t, tmpDir, filepath.Join("globalStorage", "state.vscdb"))
|
||||
panelModel := readValue(t, dbPath, "chat.currentLanguageModel.panel")
|
||||
if panelModel != "ollama/Ollama/4" {
|
||||
t.Errorf("expected panel model to use cached numeric ID ollama/Ollama/4, got %q", panelModel)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("previously hidden model is re-shown when configured", func(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
setTestHome(t, tmpDir)
|
||||
|
||||
Reference in New Issue
Block a user