From c2c90f6d2b766bbba30c8ea8087f799a6c21f525 Mon Sep 17 00:00:00 2001 From: Grail Finder Date: Mon, 9 Mar 2026 08:50:33 +0300 Subject: Enha: pw agent --- agent/request.go | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'agent/request.go') diff --git a/agent/request.go b/agent/request.go index 4ca619d..754f16e 100644 --- a/agent/request.go +++ b/agent/request.go @@ -80,6 +80,20 @@ func (ag *AgentClient) FormMsg(msg string) (io.Reader, error) { return bytes.NewReader(b), nil } +func (ag *AgentClient) FormMsgWithToolCallID(msg, toolCallID string) (io.Reader, error) { + m := models.RoleMsg{ + Role: "tool", + Content: msg, + ToolCallID: toolCallID, + } + ag.chatBody.Messages = append(ag.chatBody.Messages, m) + b, err := ag.buildRequest() + if err != nil { + return nil, err + } + return bytes.NewReader(b), nil +} + // buildRequest creates the appropriate LLM request based on the current API endpoint. func (ag *AgentClient) buildRequest() ([]byte, error) { isCompletion, isChat, isDeepSeek, isOpenRouter := detectAPI(ag.cfg.CurrentAPI) -- cgit v1.2.3