From 914d9ac0c17bf3ea595c15c32d44ecc88b7db440 Mon Sep 17 00:00:00 2001
From: Matt Low <matt@mlow.ca>
Date: Sun, 23 Jun 2024 19:10:03 +0000
Subject: [PATCH] Renamed RequestParameters.ToolBag to Toolbox

---
 pkg/api/api.go                          | 2 +-
 pkg/api/provider/anthropic/anthropic.go | 4 ++--
 pkg/api/provider/google/google.go       | 4 ++--
 pkg/api/provider/openai/openai.go       | 4 ++--
 pkg/cmd/util/util.go                    | 2 +-
 pkg/tui/views/chat/conversation.go      | 2 +-
 6 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/pkg/api/api.go b/pkg/api/api.go
index 717458d..c8c54ad 100644
--- a/pkg/api/api.go
+++ b/pkg/api/api.go
@@ -18,7 +18,7 @@ type RequestParameters struct {
 	Temperature float32
 	TopP        float32
 
-	ToolBag []ToolSpec
+	Toolbox []ToolSpec
 }
 
 type ChatCompletionProvider interface {
diff --git a/pkg/api/provider/anthropic/anthropic.go b/pkg/api/provider/anthropic/anthropic.go
index 157d34b..9b3f25d 100644
--- a/pkg/api/provider/anthropic/anthropic.go
+++ b/pkg/api/provider/anthropic/anthropic.go
@@ -183,8 +183,8 @@ func createChatCompletionRequest(
 		Temperature: params.Temperature,
 	}
 
-	if len(params.ToolBag) > 0 {
-		request.Tools = convertTools(params.ToolBag)
+	if len(params.Toolbox) > 0 {
+		request.Tools = convertTools(params.Toolbox)
 	}
 
 	var prefill string
diff --git a/pkg/api/provider/google/google.go b/pkg/api/provider/google/google.go
index a2c4b2e..3290f80 100644
--- a/pkg/api/provider/google/google.go
+++ b/pkg/api/provider/google/google.go
@@ -253,8 +253,8 @@ func createGenerateContentRequest(
 		}
 	}
 
-	if len(params.ToolBag) > 0 {
-		request.Tools = convertTools(params.ToolBag)
+	if len(params.Toolbox) > 0 {
+		request.Tools = convertTools(params.Toolbox)
 	}
 
 	return request, nil
diff --git a/pkg/api/provider/openai/openai.go b/pkg/api/provider/openai/openai.go
index 3acd24c..fecb76f 100644
--- a/pkg/api/provider/openai/openai.go
+++ b/pkg/api/provider/openai/openai.go
@@ -177,8 +177,8 @@ func createChatCompletionRequest(
 		N:           1, // limit responses to 1 "choice". we use choices[0] to reference it
 	}
 
-	if len(params.ToolBag) > 0 {
-		request.Tools = convertTools(params.ToolBag)
+	if len(params.Toolbox) > 0 {
+		request.Tools = convertTools(params.Toolbox)
 		request.ToolChoice = "auto"
 	}
 
diff --git a/pkg/cmd/util/util.go b/pkg/cmd/util/util.go
index fe98123..f6944bb 100644
--- a/pkg/cmd/util/util.go
+++ b/pkg/cmd/util/util.go
@@ -35,7 +35,7 @@ func Prompt(ctx *lmcli.Context, messages []api.Message, callback func(api.Messag
 		if agent.SystemPrompt != "" {
 			system = agent.SystemPrompt
 		}
-		params.ToolBag = agent.Toolbox
+		params.Toolbox = agent.Toolbox
 	}
 
 	if system != "" {
diff --git a/pkg/tui/views/chat/conversation.go b/pkg/tui/views/chat/conversation.go
index 00de5ee..c4ff808 100644
--- a/pkg/tui/views/chat/conversation.go
+++ b/pkg/tui/views/chat/conversation.go
@@ -279,7 +279,7 @@ func (m *Model) promptLLM() tea.Cmd {
 
 		agent := m.Shared.Ctx.GetAgent(m.Shared.Ctx.Config.Defaults.Agent)
 		if agent != nil {
-			params.ToolBag = agent.Toolbox
+			params.Toolbox = agent.Toolbox
 		}
 
 		ctx, cancel := context.WithCancel(context.Background())