diff options
-rw-r--r-- | bot.go | 2 | ||||
-rw-r--r-- | llm.go | 4 |
2 files changed, 5 insertions, 1 deletions
@@ -442,7 +442,7 @@ func findCall(msg, toolCall string, tv *tview.TextView) { // call a func f, ok := fnMap[fc.Name] if !ok { - m := fc.Name + "%s is not implemented" + m := fc.Name + " is not implemented" chatRound(m, cfg.ToolRole, tv, false, false) return } @@ -146,6 +146,10 @@ func (op OpenAIer) ParseChunk(data []byte) (*models.TextChunk, error) { } if len(llmchunk.Choices[len(llmchunk.Choices)-1].Delta.ToolCalls) > 0 { resp.ToolChunk = llmchunk.Choices[len(llmchunk.Choices)-1].Delta.ToolCalls[0].Function.Arguments + fname := llmchunk.Choices[len(llmchunk.Choices)-1].Delta.ToolCalls[0].Function.Name + if fname != "" { + resp.FuncName = fname + } } if llmchunk.Choices[len(llmchunk.Choices)-1].FinishReason == "stop" { if resp.Chunk != "" { |