()
+ const { t } = useTranslation(["playground"])
const [dropState, setDropState] = React.useState<
"idle" | "dragging" | "error"
>("idle")
- const { chatMode, messages, setHistory, setHistoryId, setMessages } =
- useMessage()
+ const {
+ chatMode,
+ streaming,
+ onSubmit,
+ messages,
+ setHistory,
+ setHistoryId,
+ setMessages,
+ selectedModel
+ } = useMessage()
+
+ const bgMsg = useBackgroundMessage()
const setRecentMessagesOnLoad = async () => {
-
- const isEnabled = await copilotResumeLastChat();
+ const isEnabled = await copilotResumeLastChat()
if (!isEnabled) {
- return;
+ return
}
if (messages.length === 0) {
const recentChat = await getRecentChatFromCopilot()
@@ -92,11 +105,26 @@ const SidepanelChat = () => {
}
}, [])
-
React.useEffect(() => {
setRecentMessagesOnLoad()
}, [])
+ React.useEffect(() => {
+ if (bgMsg && !streaming) {
+ if (selectedModel) {
+ onSubmit({
+ message: bgMsg.text,
+ messageType: bgMsg.type,
+ image: ""
+ })
+ } else {
+ notification.error({
+ message: t("formError.noModel")
+ })
+ }
+ }
+ }, [bgMsg])
+
return (
{
+ return (await storage.get("copilotSummaryPrompt")) || DEFAULT_SUMMARY_PROMPT
+}
+
+export const setSummaryPrompt = async (prompt: string) => {
+ await storage.set("copilotSummaryPrompt", prompt)
+}
+
+export const getRephrasePrompt = async () => {
+ return (await storage.get("copilotRephrasePrompt")) || DEFAULT_REPHRASE_PROMPT
+}
+
+export const setRephrasePrompt = async (prompt: string) => {
+ await storage.set("copilotRephrasePrompt", prompt)
+}
+
+export const getTranslatePrompt = async () => {
+ return (
+ (await storage.get("copilotTranslatePrompt")) || DEFAULT_TRANSLATE_PROMPT
+ )
+}
+
+export const setTranslatePrompt = async (prompt: string) => {
+ await storage.set("copilotTranslatePrompt", prompt)
+}
+
+export const getExplainPrompt = async () => {
+ return (await storage.get("copilotExplainPrompt")) || DEFAULT_EXPLAIN_PROMPT
+}
+
+export const setExplainPrompt = async (prompt: string) => {
+ await storage.set("copilotExplainPrompt", prompt)
+}
+
+export const getCustomPrompt = async () => {
+ return (await storage.get("copilotCustomPrompt")) || DEFAULT_CUSTOM_PROMPT
+}
+
+export const setCustomPrompt = async (prompt: string) => {
+ await storage.set("copilotCustomPrompt", prompt)
+}
+
+export const getAllCopilotPrompts = async () => {
+ const [
+ summaryPrompt,
+ rephrasePrompt,
+ translatePrompt,
+ explainPrompt,
+ customPrompt
+ ] = await Promise.all([
+ getSummaryPrompt(),
+ getRephrasePrompt(),
+ getTranslatePrompt(),
+ getExplainPrompt(),
+ getCustomPrompt()
+ ])
+
+ return [
+ { key: "summary", prompt: summaryPrompt },
+ { key: "rephrase", prompt: rephrasePrompt },
+ { key: "translate", prompt: translatePrompt },
+ { key: "explain", prompt: explainPrompt },
+ { key: "custom", prompt: customPrompt }
+ ]
+}
+
+export const setAllCopilotPrompts = async (
+ prompts: { key: string; prompt: string }[]
+) => {
+ for (const { key, prompt } of prompts) {
+ switch (key) {
+ case "summary":
+ await setSummaryPrompt(prompt)
+ break
+ case "rephrase":
+ await setRephrasePrompt(prompt)
+ break
+ case "translate":
+ await setTranslatePrompt(prompt)
+ break
+ case "explain":
+ await setExplainPrompt(prompt)
+ break
+ case "custom":
+ await setCustomPrompt(prompt)
+ break
+
+ }
+ }
+}
+
+export const getPrompt = async (key: string) => {
+ switch (key) {
+ case "summary":
+ return await getSummaryPrompt()
+ case "rephrase":
+ return await getRephrasePrompt()
+ case "translate":
+ return await getTranslatePrompt()
+ case "explain":
+ return await getExplainPrompt()
+ case "custom":
+ return await getCustomPrompt()
+ default:
+ return ""
+ }
+}
diff --git a/src/services/title.ts b/src/services/title.ts
new file mode 100644
index 0000000..ac4b8e8
--- /dev/null
+++ b/src/services/title.ts
@@ -0,0 +1,72 @@
+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. 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("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
+ }
+}
\ No newline at end of file
diff --git a/src/store/index.tsx b/src/store/index.tsx
index 8910705..c1929d3 100644
--- a/src/store/index.tsx
+++ b/src/store/index.tsx
@@ -12,6 +12,7 @@ export type ChatHistory = {
role: "user" | "assistant" | "system"
content: string
image?: string
+ messageType?: string
}[]
type State = {
diff --git a/src/store/option.tsx b/src/store/option.tsx
index 8e1e845..5184b8f 100644
--- a/src/store/option.tsx
+++ b/src/store/option.tsx
@@ -18,12 +18,14 @@ export type Message = {
images?: string[]
search?: WebSearch
id?: string
+ messageType?: string
}
export type ChatHistory = {
role: "user" | "assistant" | "system"
content: string
- image?: string
+ image?: string,
+ messageType?: string
}[]
type State = {
diff --git a/src/types/message.ts b/src/types/message.ts
index 66e954e..3be7cdc 100644
--- a/src/types/message.ts
+++ b/src/types/message.ts
@@ -14,5 +14,6 @@ type WebSearch = {
sources: any[]
images?: string[]
search?: WebSearch
+ messageType?: string
id?: string
}
\ No newline at end of file
diff --git a/src/utils/color.ts b/src/utils/color.ts
new file mode 100644
index 0000000..b0579a8
--- /dev/null
+++ b/src/utils/color.ts
@@ -0,0 +1,8 @@
+export const tagColors = {
+ summary: "blue",
+ explain: "green",
+ translate: "purple",
+ custom: "orange",
+ rephrase: "yellow"
+ }
+
\ No newline at end of file
diff --git a/src/utils/pull-ollama.ts b/src/utils/pull-ollama.ts
new file mode 100644
index 0000000..d15ff2b
--- /dev/null
+++ b/src/utils/pull-ollama.ts
@@ -0,0 +1,77 @@
+import { setBadgeBackgroundColor, setBadgeText, setTitle } from "@/utils/action"
+
+
+export const progressHuman = (completed: number, total: number) => {
+ return ((completed / total) * 100).toFixed(0) + "%"
+}
+
+export const clearBadge = () => {
+ setBadgeText({ text: "" })
+ setTitle({ title: "" })
+}
+export const streamDownload = async (url: string, model: string) => {
+ url += "/api/pull"
+ const response = await fetch(url, {
+ method: "POST",
+ headers: {
+ "Content-Type": "application/json"
+ },
+ body: JSON.stringify({ model, stream: true })
+ })
+
+ const reader = response.body?.getReader()
+
+ const decoder = new TextDecoder()
+
+ let isSuccess = true
+ while (true) {
+ if (!reader) {
+ break
+ }
+ const { done, value } = await reader.read()
+
+ if (done) {
+ break
+ }
+
+ const text = decoder.decode(value)
+ try {
+ const json = JSON.parse(text.trim()) as {
+ status: string
+ total?: number
+ completed?: number
+ }
+ if (json.total && json.completed) {
+ setBadgeText({
+ text: progressHuman(json.completed, json.total)
+ })
+ setBadgeBackgroundColor({ color: "#0000FF" })
+ } else {
+ setBadgeText({ text: "🏋️♂️" })
+ setBadgeBackgroundColor({ color: "#FFFFFF" })
+ }
+
+ setTitle({ title: json.status })
+
+ if (json.status === "success") {
+ isSuccess = true
+ }
+ } catch (e) {
+ console.error(e)
+ }
+ }
+
+ if (isSuccess) {
+ setBadgeText({ text: "✅" })
+ setBadgeBackgroundColor({ color: "#00FF00" })
+ setTitle({ title: "Model pulled successfully" })
+ } else {
+ setBadgeText({ text: "❌" })
+ setBadgeBackgroundColor({ color: "#FF0000" })
+ setTitle({ title: "Model pull failed" })
+ }
+
+ setTimeout(() => {
+ clearBadge()
+ }, 5000)
+}
\ No newline at end of file
diff --git a/wxt.config.ts b/wxt.config.ts
index 74fda20..10b2eff 100644
--- a/wxt.config.ts
+++ b/wxt.config.ts
@@ -50,7 +50,7 @@ export default defineConfig({
outDir: "build",
manifest: {
- version: "1.1.16",
+ version: "1.2.0",
name:
process.env.TARGET === "firefox"
? "Page Assist - A Web UI for Local AI Models"