72 lines
1.9 KiB
TypeScript
72 lines
1.9 KiB
TypeScript
|
import { pageAssistModel } from "@/models"
|
|||
|
|
import { Storage } from "@plasmohq/storage"
|
|||
|
|
import { getOllamaURL } from "./ollama"
|
|||
|
|
import { cleanUrl } from "@/libs/clean-url"
|
|||
|
|
import { HumanMessage } from "langchain/schema"
|
|||
|
|
const storage = new Storage()
|
|||
|
|
|
|||
|
|
// this prompt is copied from the OpenWebUI codebase
|
|||
|
|
export const DEFAULT_TITLE_GEN_PROMPT = `Here is the query:
|
|||
|
|
|
|||
|
|
--------------
|
|||
|
|
|
|||
|
|
{{query}}
|
|||
|
|
|
|||
|
|
--------------
|
|||
|
|
|
|||
|
|
Create a concise, 3-5 word phrase as a title for the previous query. Suitable emojis for the summary can be used to enhance understanding. Avoid quotation marks or special formatting. RESPOND ONLY WITH THE TITLE TEXT. ANSWER USING THE SAME LANGUAGE AS THE QUERY.
|
|||
|
|
|
|||
|
|
|
|||
|
|
Examples of titles:
|
|||
|
|
|
|||
|
|
Stellar Achievement Celebration
|
|||
|
|
Family Bonding Activities
|
|||
|
|
🇫🇷 Voyage à Paris
|
|||
|
|
🍜 Receta de Ramen Casero
|
|||
|
|
Shakespeare Analyse Literarische
|
|||
|
|
日本の春祭り体験
|
|||
|
|
Древнегреческая Философия Обзор
|
|||
|
|
|
|||
|
|
Response:`
|
|||
|
|
|
|||
|
|
|
|||
|
|
export const isTitleGenEnabled = async () => {
|
|||
|
|
const enabled = await storage.get<boolean | undefined>("titleGenEnabled")
|
|||
|
|
return enabled ?? false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export const setTitleGenEnabled = async (enabled: boolean) => {
|
|||
|
|
await storage.set("titleGenEnabled", enabled)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
export const generateTitle = async (model: string, query: string, fallBackTitle: string) => {
|
|||
|
|
|
|||
|
|
const isEnabled = await isTitleGenEnabled()
|
|||
|
|
|
|||
|
|
if (!isEnabled) {
|
|||
|
|
return fallBackTitle
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
try {
|
|||
|
|
const url = await getOllamaURL()
|
|||
|
|
|
|||
|
|
const titleModel = await pageAssistModel({
|
|||
|
|
baseUrl: cleanUrl(url),
|
|||
|
|
model
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
const prompt = DEFAULT_TITLE_GEN_PROMPT.replace("{{query}}", query)
|
|||
|
|
|
|||
|
|
const title = await titleModel.invoke([
|
|||
|
|
new HumanMessage({
|
|||
|
|
content: prompt
|
|||
|
|
})
|
|||
|
|
])
|
|||
|
|
|
|||
|
|
return title.content.toString()
|
|||
|
|
} catch (error) {
|
|||
|
|
console.log(`Error generating title: ${error}`)
|
|||
|
|
return fallBackTitle
|
|||
|
|
}
|
|||
|
|
}
|