mirror of
https://github.com/Ed1s0nZ/CyberStrikeAI.git
synced 2026-08-21 18:37:33 +02:00
公明andCursor
24d06c5220
Merge streaming tool outputs by CallID with ConcatMessages, pair same-name historical results, and FIFO-match duplicate IDs so concurrent nmap 1/2 and 2/2 stay distinct. Co-authored-by: Cursor <cursoragent@cursor.com>
42 lines
1.3 KiB
Go
42 lines
1.3 KiB
Go
package multiagent
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/cloudwego/eino/schema"
|
|
)
|
|
|
|
func TestConcatToolResultChunksUsesEinoConcatForSingleCall(t *testing.T) {
|
|
got, err := concatToolResultChunks([]*schema.Message{
|
|
schema.ToolMessage("hel", "call-1", schema.WithToolName("execute")),
|
|
schema.ToolMessage("lo", "call-1", schema.WithToolName("execute")),
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("concat: %v", err)
|
|
}
|
|
if len(got) != 1 || got[0].ToolCallID != "call-1" || got[0].Content != "hello" || got[0].ToolName != "execute" {
|
|
t.Fatalf("got = %#v, want one ConcatMessages result", got)
|
|
}
|
|
}
|
|
|
|
func TestConcatToolResultChunksSplitsParallelCalls(t *testing.T) {
|
|
got, err := concatToolResultChunks([]*schema.Message{
|
|
schema.ToolMessage("nmap 1/2 ", "call-1", schema.WithToolName("nmap")),
|
|
schema.ToolMessage("nmap 2/2 ", "call-2", schema.WithToolName("nmap")),
|
|
schema.ToolMessage("22/tcp", "call-1", schema.WithToolName("nmap")),
|
|
schema.ToolMessage("80/tcp", "call-2", schema.WithToolName("nmap")),
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("concat: %v", err)
|
|
}
|
|
if len(got) != 2 {
|
|
t.Fatalf("got = %#v, want two calls", got)
|
|
}
|
|
if got[0].ToolCallID != "call-1" || got[0].Content != "nmap 1/2 22/tcp" {
|
|
t.Fatalf("call-1 = %#v", got[0])
|
|
}
|
|
if got[1].ToolCallID != "call-2" || got[1].Content != "nmap 2/2 80/tcp" {
|
|
t.Fatalf("call-2 = %#v", got[1])
|
|
}
|
|
}
|