implement two tool functions in the backend interface

Author: sawka

pkg/aiusechat/openaicomp/openaicomp-convertmessage.go

@@ -14,6 +14,7 @@ import (
 	"strings"
 
 	"github.com/wavetermdev/waveterm/pkg/aiusechat/aiutil"
+	"github.com/wavetermdev/waveterm/pkg/aiusechat/chatstore"
 	"github.com/wavetermdev/waveterm/pkg/aiusechat/uctypes"
 	"github.com/wavetermdev/waveterm/pkg/wavebase"
 )
@@ -292,3 +293,54 @@ func ConvertAIChatToUIChat(aiChat uctypes.AIChat) (*uctypes.UIChat, error) {
 
 	return uiChat, nil
 }
+
+// GetFunctionCallInputByToolCallId searches for a tool call by ID in the chat history
+func GetFunctionCallInputByToolCallId(aiChat uctypes.AIChat, toolCallId string) *uctypes.AIFunctionCallInput {
+	for _, genMsg := range aiChat.NativeMessages {
+		compMsg, ok := genMsg.(*CompletionsChatMessage)
+		if !ok {
+			continue
+		}
+		idx := compMsg.Message.FindToolCallIndex(toolCallId)
+		if idx == -1 {
+			continue
+		}
+		toolCall := compMsg.Message.ToolCalls[idx]
+		return &uctypes.AIFunctionCallInput{
+			CallId:      toolCall.ID,
+			Name:        toolCall.Function.Name,
+			Arguments:   toolCall.Function.Arguments,
+			ToolUseData: toolCall.ToolUseData,
+		}
+	}
+	return nil
+}
+
+// UpdateToolUseData updates the ToolUseData for a specific tool call in the chat history
+func UpdateToolUseData(chatId string, callId string, newToolUseData *uctypes.UIMessageDataToolUse) error {
+	chat := chatstore.DefaultChatStore.Get(chatId)
+	if chat == nil {
+		return fmt.Errorf("chat not found: %s", chatId)
+	}
+
+	for _, genMsg := range chat.NativeMessages {
+		compMsg, ok := genMsg.(*CompletionsChatMessage)
+		if !ok {
+			continue
+		}
+		idx := compMsg.Message.FindToolCallIndex(callId)
+		if idx == -1 {
+			continue
+		}
+		updatedMsg := compMsg.Copy()
+		updatedMsg.Message.ToolCalls[idx].ToolUseData = newToolUseData
+		aiOpts := &uctypes.AIOptsType{
+			APIType:    chat.APIType,
+			Model:      chat.Model,
+			APIVersion: chat.APIVersion,
+		}
+		return chatstore.DefaultChatStore.PostMessage(chatId, aiOpts, updatedMsg)
+	}
+
+	return fmt.Errorf("tool call with callId %s not found in chat %s", callId, chatId)
+}

pkg/aiusechat/openaicomp/openaicomp-types.go

@@ -40,6 +40,15 @@ func (cm *CompletionsMessage) clean() *CompletionsMessage {
 	return &rtn
 }
 
+func (cm *CompletionsMessage) FindToolCallIndex(toolCallId string) int {
+	for i, tc := range cm.ToolCalls {
+		if tc.ID == toolCallId {
+			return i
+		}
+	}
+	return -1
+}
+
 type ToolDefinition struct {
 	Type     string          `json:"type"` // "function"
 	Function ToolFunctionDef `json:"function"`
@@ -138,3 +147,25 @@ func (m *CompletionsChatMessage) GetUsage() *uctypes.AIUsage {
 		OutputTokens: m.Usage.CompletionTokens,
 	}
 }
+
+func (m *CompletionsChatMessage) Copy() *CompletionsChatMessage {
+	if m == nil {
+		return nil
+	}
+	copy := *m
+	if len(m.Message.ToolCalls) > 0 {
+		copy.Message.ToolCalls = make([]ToolCall, len(m.Message.ToolCalls))
+		for i, tc := range m.Message.ToolCalls {
+			copy.Message.ToolCalls[i] = tc
+			if tc.ToolUseData != nil {
+				toolUseDataCopy := *tc.ToolUseData
+				copy.Message.ToolCalls[i].ToolUseData = &toolUseDataCopy
+			}
+		}
+	}
+	if m.Usage != nil {
+		usageCopy := *m.Usage
+		copy.Usage = &usageCopy
+	}
+	return &copy
+}

pkg/aiusechat/usechat-backend.go

@@ -141,7 +141,7 @@ func (b *openaiCompletionsBackend) RunChatStep(
 }
 
 func (b *openaiCompletionsBackend) UpdateToolUseData(chatId string, toolCallId string, toolUseData *uctypes.UIMessageDataToolUse) error {
-	return fmt.Errorf("tools not supported in openai-comp backend")
+	return openaicomp.UpdateToolUseData(chatId, toolCallId, toolUseData)
 }
 
 func (b *openaiCompletionsBackend) ConvertToolResultsToNativeChatMessage(toolResults []uctypes.AIToolResult) ([]uctypes.GenAIMessage, error) {
@@ -153,7 +153,7 @@ func (b *openaiCompletionsBackend) ConvertAIMessageToNativeChatMessage(message u
 }
 
 func (b *openaiCompletionsBackend) GetFunctionCallInputByToolCallId(aiChat uctypes.AIChat, toolCallId string) *uctypes.AIFunctionCallInput {
-	return nil
+	return openaicomp.GetFunctionCallInputByToolCallId(aiChat, toolCallId)
 }
 
 func (b *openaiCompletionsBackend) ConvertAIChatToUIChat(aiChat uctypes.AIChat) (*uctypes.UIChat, error) {