Spaces:
Running
Running
File size: 2,404 Bytes
fb4d8fe | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 | import { describe, expect, it } from "vitest";
import { formatMediaUnderstandingBody } from "./format.js";
describe("formatMediaUnderstandingBody", () => {
it("replaces placeholder body with transcript", () => {
const body = formatMediaUnderstandingBody({
body: "<media:audio>",
outputs: [
{
kind: "audio.transcription",
attachmentIndex: 0,
text: "hello world",
provider: "groq",
},
],
});
expect(body).toBe("[Audio]\nTranscript:\nhello world");
});
it("includes user text when body is meaningful", () => {
const body = formatMediaUnderstandingBody({
body: "caption here",
outputs: [
{
kind: "audio.transcription",
attachmentIndex: 0,
text: "transcribed",
provider: "groq",
},
],
});
expect(body).toBe("[Audio]\nUser text:\ncaption here\nTranscript:\ntranscribed");
});
it("strips leading media placeholders from user text", () => {
const body = formatMediaUnderstandingBody({
body: "<media:audio> caption here",
outputs: [
{
kind: "audio.transcription",
attachmentIndex: 0,
text: "transcribed",
provider: "groq",
},
],
});
expect(body).toBe("[Audio]\nUser text:\ncaption here\nTranscript:\ntranscribed");
});
it("keeps user text once when multiple outputs exist", () => {
const body = formatMediaUnderstandingBody({
body: "caption here",
outputs: [
{
kind: "audio.transcription",
attachmentIndex: 0,
text: "audio text",
provider: "groq",
},
{
kind: "video.description",
attachmentIndex: 1,
text: "video text",
provider: "google",
},
],
});
expect(body).toBe(
[
"User text:\ncaption here",
"[Audio]\nTranscript:\naudio text",
"[Video]\nDescription:\nvideo text",
].join("\n\n"),
);
});
it("formats image outputs", () => {
const body = formatMediaUnderstandingBody({
body: "<media:image>",
outputs: [
{
kind: "image.description",
attachmentIndex: 0,
text: "a cat",
provider: "openai",
},
],
});
expect(body).toBe("[Image]\nDescription:\na cat");
});
});
|