Răsfoiți Sursa

feat(tui): show provider next to model

adamdottv 8 luni în urmă
părinte
comite
183ca64ef9

+ 4 - 1
packages/tui/internal/completions/manager.go

@@ -20,10 +20,13 @@ func NewCompletionManager(app *app.App) *CompletionManager {
 	}
 }
 
+func (m *CompletionManager) DefaultProvider() dialog.CompletionProvider {
+	return m.providers["commands"]
+}
+
 func (m *CompletionManager) GetProvider(input string) dialog.CompletionProvider {
 	if strings.HasPrefix(input, "/") {
 		return m.providers["commands"]
 	}
 	return m.providers["files"]
 }
-

+ 1 - 1
packages/tui/internal/components/chat/editor.go

@@ -126,7 +126,7 @@ func (m *editorComponent) View() string {
 
 	model := ""
 	if m.app.Model != nil {
-		model = base(m.app.Model.Name) + muted(" • /model")
+		model = muted(m.app.Provider.Name) + base(" "+m.app.Model.Name)
 	}
 
 	space := m.width - 2 - lipgloss.Width(model) - lipgloss.Width(hint)

+ 2 - 1
packages/tui/internal/tui/tui.go

@@ -208,6 +208,7 @@ func (a appModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		cmds = append(cmds, cmd)
 	case dialog.CompletionDialogCloseMsg:
 		a.showCompletionDialog = false
+		a.completions.SetProvider(a.completionManager.DefaultProvider())
 	case client.EventSessionUpdated:
 		if msg.Properties.Info.Id == a.app.Session.Id {
 			a.app.Session = &msg.Properties.Info
@@ -494,7 +495,7 @@ func (a appModel) executeCommand(command commands.Command) (tea.Model, tea.Cmd)
 
 func NewModel(app *app.App) tea.Model {
 	completionManager := completions.NewCompletionManager(app)
-	initialProvider := completionManager.GetProvider("")
+	initialProvider := completionManager.DefaultProvider()
 
 	messages := chat.NewMessagesComponent(app)
 	editor := chat.NewEditorComponent(app)