fix: correct model selection index in models dialog

Kujtim Hoxha created 4 months ago

Change summary

internal/tui/components/dialogs/models/models.go | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)

Detailed changes

internal/tui/components/dialogs/models/models.go

@@ -97,13 +97,13 @@ func (m *modelDialogCmp) Init() tea.Cmd {
 		}
 		modelItems = append(modelItems, commands.NewItemSection(name))
 		for _, model := range provider.Models {
-			if model.ID == agentModel.ID && provider.ID == agentProvider.ID {
-				selectIndex = len(modelItems) // Set the selected index to the current model
-			}
 			modelItems = append(modelItems, completions.NewCompletionItem(model.Name, ModelOption{
 				Provider: provider,
 				Model:    model,
 			}))
+			if model.ID == agentModel.ID && provider.ID == agentProvider.ID {
+				selectIndex = len(modelItems) - 1 // Set the selected index to the current model
+			}
 		}
 	}