Function TestSamplingMessagesToChat

pkg/runtime/sampling_test.go:14–184

(t *testing.T)

)

func TestSamplingMessagesToChat(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name    string
		req     *mcp.CreateMessageParams
		want    []chat.Message
		wantErr bool
	}{
		{
			name: "single user text message",
			req: &mcp.CreateMessageParams{
				Messages: []*mcp.SamplingMessage{
					{Role: "user", Content: &mcp.TextContent{Text: "hello"}},
				},
			},
			want: []chat.Message{
				{Role: chat.MessageRoleUser, Content: "hello"},
			},
		},
		{
			name: "system prompt is prepended",
			req: &mcp.CreateMessageParams{
				SystemPrompt: "be terse",
				Messages: []*mcp.SamplingMessage{
					{Role: "user", Content: &mcp.TextContent{Text: "hi"}},
				},
			},
			want: []chat.Message{
				{Role: chat.MessageRoleSystem, Content: "be terse"},
				{Role: chat.MessageRoleUser, Content: "hi"},
			},
		},
		{
			name: "user and assistant turns are preserved",
			req: &mcp.CreateMessageParams{
				Messages: []*mcp.SamplingMessage{
					{Role: "user", Content: &mcp.TextContent{Text: "ping"}},
					{Role: "assistant", Content: &mcp.TextContent{Text: "pong"}},
					{Role: "user", Content: &mcp.TextContent{Text: "again"}},
				},
			},
			want: []chat.Message{
				{Role: chat.MessageRoleUser, Content: "ping"},
				{Role: chat.MessageRoleAssistant, Content: "pong"},
				{Role: chat.MessageRoleUser, Content: "again"},
			},
		},
		{
			name: "image content becomes a data URL multi-part",
			req: &mcp.CreateMessageParams{
				Messages: []*mcp.SamplingMessage{
					{
						Role:    "user",
						Content: &mcp.ImageContent{Data: []byte("PNGBYTES"), MIMEType: "image/png"},
					},
				},
			},