Function TestMixedFormatConversation

coderd/x/chatd/chatprompt/chatprompt_test.go:1656–1790

TestMixedFormatConversation verifies ConvertMessagesWithFiles handles a realistic post-deploy conversation where legacy and new storage formats coexist.

(t *testing.T)

// handles a realistic post-deploy conversation where legacy and new
// storage formats coexist.
func TestMixedFormatConversation(t *testing.T) {
	t.Parallel()

	fileID := uuid.New()
	resolvedFileData := []byte("resolved-png-bytes")

	resolver := func(_ context.Context, ids []uuid.UUID) (map[uuid.UUID]chatprompt.FileData, error) {
		out := make(map[uuid.UUID]chatprompt.FileData)
		for _, id := range ids {
			if id == fileID {
				out[id] = chatprompt.FileData{Data: resolvedFileData, MediaType: "image/png"}
			}
		}
		return out, nil
	}

	// 1. System (JSON string).
	systemRaw, err := json.Marshal("You are helpful.")
	require.NoError(t, err)

	// 2. Old user (fantasy envelope: text + file with file_id).
	oldUserRaw := mustJSON(t, []json.RawMessage{
		mustJSON(t, map[string]any{
			"type": "text",
			"data": map[string]any{"text": "Look at this image."},
		}),
		mustJSON(t, map[string]any{
			"type": "file",
			"data": map[string]any{
				"media_type": "image/png",
				"file_id":    fileID.String(),
			},
		}),
	})

	// 3. Old assistant (fantasy envelope: tool-call).
	oldAssistantRaw, err := chatprompt.MarshalContent([]fantasy.Content{
		fantasy.ToolCallContent{
			ToolCallID: "call_1",
			ToolName:   "analyze_image",
			Input:      `{"detail":"high"}`,
		},
	}, nil)
	require.NoError(t, err)

	// 4. Old tool (legacy result rows).
	oldToolRaw, err := chatprompt.MarshalToolResult(
		"call_1", "analyze_image",
		json.RawMessage(`{"description":"a cat"}`), false, false,
		false, nil,
	)
	require.NoError(t, err)

	// 5. New user (SDK parts: text + file-reference).
	newUserRaw, err := chatprompt.MarshalParts([]codersdk.ChatMessagePart{
		{Type: codersdk.ChatMessagePartTypeText, Text: "Check this diff."},
		{Type: codersdk.ChatMessagePartTypeFileReference, FileName: "main.go", StartLine: 5, EndLine: 15, Content: "func main() {}"},
	})