mlow/lmcli
1
0
Fork 0
Code Issues 3 Pull Requests Packages Projects Releases Wiki Activity

Renamed RequestParameters.ToolBag to Toolbox

Browse Source
This commit is contained in:
Matt Low 2024-06-23 19:10:03 +00:00
parent 8ddac2f820
commit 914d9ac0c1
6 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pkg/api/api.go
View File

@ -18,7 +18,7 @@ type RequestParameters struct {
Temperature float32 Temperature float32
TopP float32 TopP float32
ToolBag []ToolSpec Toolbox []ToolSpec
} }
type ChatCompletionProvider interface { type ChatCompletionProvider interface {

4
pkg/api/provider/anthropic/anthropic.go
View File

@ -183,8 +183,8 @@ func createChatCompletionRequest(
Temperature: params.Temperature, Temperature: params.Temperature,
} }
if len(params.ToolBag) > 0 { if len(params.Toolbox) > 0 {
request.Tools = convertTools(params.ToolBag) request.Tools = convertTools(params.Toolbox)
} }
var prefill string var prefill string

4
pkg/api/provider/google/google.go
View File

@ -253,8 +253,8 @@ func createGenerateContentRequest(
} }
} }
if len(params.ToolBag) > 0 { if len(params.Toolbox) > 0 {
request.Tools = convertTools(params.ToolBag) request.Tools = convertTools(params.Toolbox)
} }
return request, nil return request, nil

4
pkg/api/provider/openai/openai.go
View File

@ -177,8 +177,8 @@ func createChatCompletionRequest(
N: 1, // limit responses to 1 "choice". we use choices[0] to reference it N: 1, // limit responses to 1 "choice". we use choices[0] to reference it
} }
if len(params.ToolBag) > 0 { if len(params.Toolbox) > 0 {
request.Tools = convertTools(params.ToolBag) request.Tools = convertTools(params.Toolbox)
request.ToolChoice = "auto" request.ToolChoice = "auto"
} }

2
pkg/cmd/util/util.go
View File

@ -35,7 +35,7 @@ func Prompt(ctx *lmcli.Context, messages []api.Message, callback func(api.Messag
if agent.SystemPrompt != "" { if agent.SystemPrompt != "" {
system = agent.SystemPrompt system = agent.SystemPrompt
} }
params.ToolBag = agent.Toolbox params.Toolbox = agent.Toolbox
} }
if system != "" { if system != "" {

2
pkg/tui/views/chat/conversation.go
View File

@ -279,7 +279,7 @@ func (m *Model) promptLLM() tea.Cmd {
agent := m.Shared.Ctx.GetAgent(m.Shared.Ctx.Config.Defaults.Agent) agent := m.Shared.Ctx.GetAgent(m.Shared.Ctx.Config.Defaults.Agent)
if agent != nil { if agent != nil {
params.ToolBag = agent.Toolbox params.Toolbox = agent.Toolbox
} }
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())