Custom Topic Tags Using LLM Gateway

const baseUrl = "https://api.assemblyai.com";
const headers = { authorization: "<YOUR_API_KEY>" };

// Step 1: Transcribe the audio
const audioUrl = "https://storage.googleapis.com/aai-web-samples/meeting.mp4";
const data = { audio_url: audioUrl, speech_models: ["universal-3-pro"] };

let res = await fetch(baseUrl + "/v2/transcript", {
  method: "POST",
  headers: { ...headers, "Content-Type": "application/json" },
  body: JSON.stringify(data),
});
if (!res.ok) throw new Error(`Error: ${res.status}`);
const transcriptResponse = await res.json();
const transcriptId = transcriptResponse.id;
const pollingEndpoint = baseUrl + "/v2/transcript/" + transcriptId;

let transcriptionResult;
while (true) {
    res = await fetch(pollingEndpoint, { headers });
    if (!res.ok) throw new Error(`Error: ${res.status}`);
    transcriptionResult = await res.json();
    if (transcriptionResult.status === "completed") {
        break;
    } else if (transcriptionResult.status === "error") {
        throw new Error(`Transcription failed: ${transcriptionResult.error}`);
    } else {
        await new Promise((resolve) => setTimeout(resolve, 3000));
    }
}

// Step 2: Generate topic tags with LLM Gateway
const tagList = {
    Sports: "News and updates on various athletic events, teams, and sports personalities.",
    Politics: "Coverage and discussion of government activities, policies, and political events.",
    Entertainment: "Information on movies, music, television, celebrities, and arts.",
    Technology: "News and reviews on gadgets, software, tech advancements, and trends.",
    Health: "Articles focusing on medical news, wellness, and health-related topics.",
    Business: "Updates on markets, industries, companies, and economic trends.",
    Science: "News and insights into scientific discoveries, research, and innovations.",
    Education: "Coverage of topics related to schools, educational policies, and learning.",
    Travel: "Information on destinations, travel tips, and tourism news.",
    Lifestyle: "Articles on fashion, hobbies, personal interests, and daily life.",
    Environment: "News and discussion about environmental issues and sustainability.",
    Finance: "Information on personal finance, investments, banking, and economic news.",
    "World News": "International news covering global events and issues.",
    Crime: "Reports and updates on criminal activities, law enforcement, and legal cases.",
    Culture: "Coverage of cultural events, traditions, and societal norms.",
};

const prompt = `
You are a helpful assistant designed to label video content with topic tags.

I will give you a list of topics and definitions. Select the most relevant topic from the list. Return your selection and nothing else.

<topics_list>
${JSON.stringify(tagList, null, 2)}
</topics_list>
`;

const llmGatewayData = {
    model: "claude-sonnet-4-5-20250929",
    messages: [
        { role: "user", content: `${prompt}\n\n{{ transcript }}` },
    ],
    transcript_id: transcriptId,
    max_tokens: 500,
};

res = await fetch("https://llm-gateway.assemblyai.com/v1/chat/completions", {
  method: "POST",
  headers: { ...headers, "Content-Type": "application/json" },
  body: JSON.stringify(llmGatewayData),
});
if (!res.ok) throw new Error(`Error: ${res.status}`);
const response = await res.json();

const result = response.choices[0].message.content;
console.log(result.trim());

const baseUrl = "https://api.assemblyai.com";
const headers = { authorization: "<YOUR_API_KEY>" };

// Transcribe the audio
const audioUrl = "https://storage.googleapis.com/aai-web-samples/meeting.mp4"; // You can also use a URL to an audio or video file on the web
const data = { audio_url: audioUrl, speech_models: ["universal-3-pro"] };

let res = await fetch(baseUrl + "/v2/transcript", {
  method: "POST",
  headers: { ...headers, "Content-Type": "application/json" },
  body: JSON.stringify(data),
});
if (!res.ok) throw new Error(`Error: ${res.status}`);
const transcriptResponse = await res.json();
const transcriptId = transcriptResponse.id;
const pollingEndpoint = baseUrl + "/v2/transcript/" + transcriptId;

let transcriptionResult;
while (true) {
    res = await fetch(pollingEndpoint, { headers });
    if (!res.ok) throw new Error(`Error: ${res.status}`);
    transcriptionResult = await res.json();
    if (transcriptionResult.status === "completed") {
        break;
    } else if (transcriptionResult.status === "error") {
        throw new Error(`Transcription failed: ${transcriptionResult.error}`);
    } else {
        await new Promise((resolve) => setTimeout(resolve, 3000));
    }
}

const prompt = `
You are a helpful assistant designed to label video content with topic tags.

I will give you a list of topics and definitions. Select the most relevant topic from the list. Return your selection and nothing else.

<topics_list>
${JSON.stringify(tagList, null, 2)}
</topics_list>
`;

const llmGatewayData = {
    model: "claude-sonnet-4-5-20250929",
    messages: [
        { role: "user", content: `${prompt}\n\n{{ transcript }}` },
    ],
    transcript_id: transcriptId,
    max_tokens: 500,
};

let res = await fetch("https://llm-gateway.assemblyai.com/v1/chat/completions", {
  method: "POST",
  headers: { ...headers, "Content-Type": "application/json" },
  body: JSON.stringify(llmGatewayData),
});
if (!res.ok) throw new Error(`Error: ${res.status}`);
const response = await res.json();

const result = response.choices[0].message.content;
console.log(result.trim());