Skip to content

Commit 428701a

Browse files
authored
Merge pull request #298 from rumpl/cleanup-tool
Cleanup the tool component
2 parents 2e35bfe + 69d170c commit 428701a

1 file changed

Lines changed: 12 additions & 20 deletions

File tree

pkg/tui/components/tool/tool.go

Lines changed: 12 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@ import (
66
"strings"
77

88
"github.com/charmbracelet/bubbles/v2/spinner"
9-
"github.com/charmbracelet/bubbles/v2/viewport"
109
tea "github.com/charmbracelet/bubbletea/v2"
1110
"github.com/charmbracelet/glamour/v2"
1211

@@ -18,13 +17,11 @@ import (
1817

1918
// toolModel implements Model
2019
type toolModel struct {
21-
message *types.Message
22-
renderer *glamour.TermRenderer
23-
width int
24-
height int
25-
focused bool
26-
spinner spinner.Model
27-
app *app.App
20+
message *types.Message
21+
width int
22+
height int
23+
spinner spinner.Model
24+
app *app.App
2825
}
2926

3027
// SetSize implements Model.
@@ -39,17 +36,15 @@ func New(msg *types.Message, a *app.App, renderer *glamour.TermRenderer) layout.
3936
if msg.ToolCall.Function.Name == "transfer_task" {
4037
return &transferTaskModel{
4138
msg: msg,
42-
// renderer: renderer,
4339
}
4440
}
41+
4542
return &toolModel{
46-
message: msg,
47-
width: 80, // Default width
48-
height: 1, // Will be calculated
49-
focused: false,
50-
spinner: spinner.New(spinner.WithSpinner(spinner.Points)),
51-
app: a,
52-
renderer: renderer,
43+
message: msg,
44+
width: 80,
45+
height: 1,
46+
spinner: spinner.New(spinner.WithSpinner(spinner.Points)),
47+
app: a,
5348
}
5449
}
5550

@@ -105,10 +100,7 @@ func (mv *toolModel) Render(width int) string {
105100
content += " " + render_search_files(msg.ToolCall)
106101
} else {
107102
lines := wrapLines(msg.ToolCall.Function.Arguments, mv.width-2)
108-
argsViewport := viewport.New(viewport.WithWidth(mv.width), viewport.WithHeight(len(lines)))
109-
argsViewport.SetContent(styles.MutedStyle.Render(strings.Join(lines, "\n")))
110-
argsViewport.GotoBottom()
111-
content += "\n" + argsViewport.View()
103+
content += "\n" + strings.Join(lines, "\n")
112104
}
113105
}
114106

0 commit comments

Comments
 (0)