Source code
Revision control
Copy as Markdown
Other Tools
/**
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
*/
import {
openAIEngine,
renderPrompt,
MODEL_FEATURES,
} from "moz-src:///browser/components/aiwindow/models/Utils.sys.mjs";
/**
* Generate a default title from the first four words of a message.
*
* @param {string} message - The user's message
* @returns {string} The default title
*/
function generateDefaultTitle(message) {
if (!message || typeof message !== "string") {
return "New Chat";
}
const words = message
.trim()
.split(/\s+/)
.filter(word => !!word.length);
if (words.length === 0) {
return "New Chat";
}
const titleWords = words.slice(0, 4);
return titleWords.join(" ") + "...";
}
/**
* Generate a chat title based on the user's message and current tab information.
*
* @param {string} message - The user's message
* @param {object} current_tab - Object containing current tab information
* @returns {Promise<string>} The generated chat title
*/
export async function generateChatTitle(message, current_tab) {
try {
// Build the OpenAI engine
const engine = await openAIEngine.build(MODEL_FEATURES.TITLE_GENERATION);
const tabInfo = current_tab || { url: "", title: "", description: "" };
// Load and render the prompt with actual values
const rawPrompt = await engine.loadPrompt(MODEL_FEATURES.TITLE_GENERATION);
const systemPrompt = await renderPrompt(rawPrompt, {
current_tab: JSON.stringify(tabInfo),
});
// Prepare messages for the LLM
const messages = [
{ role: "system", content: systemPrompt },
{ role: "user", content: message },
];
// Get config for inference parameters if exists
const config = engine.getConfig(engine.feature);
const inferenceParams = config?.parameters || {};
// Call the LLM
const response = await engine.run({
messages,
fxAccountToken: await openAIEngine.getFxAccountToken(),
...inferenceParams,
});
// Extract the generated title from the response
const title =
response?.choices?.[0]?.message?.content?.trim() ||
generateDefaultTitle(message);
return title;
} catch (error) {
console.error("Failed to generate chat title:", error);
return generateDefaultTitle(message);
}
}