姚章浩
19 lines
468 B
Go
19 lines
468 B
Go
package agent
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/Tencent/WeKnora/internal/models/chat"
|
|
)
|
|
|
|
func TestCollectAgentImageURLsForFinalSynthesis(t *testing.T) {
|
|
messages := []chat.Message{
|
|
{Role: "system", Content: "system"},
|
|
{Role: "user", Content: "question", Images: []string{"data:image/png;base64,YWJj"}},
|
|
}
|
|
got := collectAgentImageURLs(messages)
|
|
if len(got) != 1 || got[0] != "data:image/png;base64,YWJj" {
|
|
t.Fatalf("collectAgentImageURLs() = %#v", got)
|
|
}
|
|
}
|