Commit 44c8b0bb authored by Jun Siang Cheah's avatar Jun Siang Cheah
Browse files

feat: rename title generation model to task model

parent 466b3e36
......@@ -6,6 +6,7 @@
"{{modelName}} is thinking...": "{{modelName}} myśli...",
"{{user}}'s Chats": "",
"{{webUIName}} Backend Required": "Backend {{webUIName}} wymagane",
"A task model is used when performing tasks such as generating titles for chats and web search queries": "",
"a user": "użytkownik",
"About": "O nas",
"Account": "Konto",
......@@ -395,7 +396,7 @@
"Set Model": "Ustaw model",
"Set reranking model (e.g. {{model}})": "",
"Set Steps": "Ustaw kroki",
"Set Title Auto-Generation Model": "Ustaw model automatycznego generowania tytułów",
"Set Task Model": "",
"Set Voice": "Ustaw głos",
"Settings": "Ustawienia",
"Settings saved successfully!": "Ustawienia zapisane pomyślnie!",
......
......@@ -6,6 +6,7 @@
"{{modelName}} is thinking...": "{{modelName}} está pensando...",
"{{user}}'s Chats": "",
"{{webUIName}} Backend Required": "{{webUIName}} Backend Necessário",
"A task model is used when performing tasks such as generating titles for chats and web search queries": "",
"a user": "um usuário",
"About": "Sobre",
"Account": "Conta",
......@@ -395,7 +396,7 @@
"Set Model": "Definir Modelo",
"Set reranking model (e.g. {{model}})": "",
"Set Steps": "Definir Etapas",
"Set Title Auto-Generation Model": "Definir Modelo de Geração Automática de Título",
"Set Task Model": "",
"Set Voice": "Definir Voz",
"Settings": "Configurações",
"Settings saved successfully!": "Configurações salvas com sucesso!",
......
......@@ -6,6 +6,7 @@
"{{modelName}} is thinking...": "{{modelName}} está pensando...",
"{{user}}'s Chats": "",
"{{webUIName}} Backend Required": "{{webUIName}} Backend Necessário",
"A task model is used when performing tasks such as generating titles for chats and web search queries": "",
"a user": "um usuário",
"About": "Sobre",
"Account": "Conta",
......@@ -395,7 +396,7 @@
"Set Model": "Definir Modelo",
"Set reranking model (e.g. {{model}})": "",
"Set Steps": "Definir Etapas",
"Set Title Auto-Generation Model": "Definir Modelo de Geração Automática de Título",
"Set Task Model": "",
"Set Voice": "Definir Voz",
"Settings": "Configurações",
"Settings saved successfully!": "Configurações salvas com sucesso!",
......
......@@ -6,6 +6,7 @@
"{{modelName}} is thinking...": "{{modelName}} думает...",
"{{user}}'s Chats": "",
"{{webUIName}} Backend Required": "{{webUIName}} бэкенд требуемый",
"A task model is used when performing tasks such as generating titles for chats and web search queries": "",
"a user": "пользователь",
"About": "Об",
"Account": "Аккаунт",
......@@ -395,7 +396,7 @@
"Set Model": "Установить модель",
"Set reranking model (e.g. {{model}})": "",
"Set Steps": "Установить шаги",
"Set Title Auto-Generation Model": "Установить модель автогенерации заголовков",
"Set Task Model": "",
"Set Voice": "Установить голос",
"Settings": "Настройки",
"Settings saved successfully!": "Настройки успешно сохранены!",
......
......@@ -6,6 +6,7 @@
"{{modelName}} is thinking...": "{{modelName}} tänker...",
"{{user}}'s Chats": "",
"{{webUIName}} Backend Required": "{{webUIName}} Backend krävs",
"A task model is used when performing tasks such as generating titles for chats and web search queries": "",
"a user": "en användare",
"About": "Om",
"Account": "Konto",
......@@ -395,7 +396,7 @@
"Set Model": "Ställ in modell",
"Set reranking model (e.g. {{model}})": "",
"Set Steps": "Ange steg",
"Set Title Auto-Generation Model": "Ange modell för automatisk generering av titel",
"Set Task Model": "",
"Set Voice": "Ange röst",
"Settings": "Inställningar",
"Settings saved successfully!": "Inställningar sparades framgångsrikt!",
......
......@@ -6,6 +6,7 @@
"{{modelName}} is thinking...": "{{modelName}} düşünüyor...",
"{{user}}'s Chats": "{{user}} Sohbetleri",
"{{webUIName}} Backend Required": "{{webUIName}} Arkayüz Gerekli",
"A task model is used when performing tasks such as generating titles for chats and web search queries": "",
"a user": "bir kullanıcı",
"About": "Hakkında",
"Account": "Hesap",
......@@ -395,7 +396,7 @@
"Set Model": "Model Ayarla",
"Set reranking model (e.g. {{model}})": "",
"Set Steps": "Adımları Ayarla",
"Set Title Auto-Generation Model": "Otomatik Başlık Oluşturma Modelini Ayarla",
"Set Task Model": "",
"Set Voice": "Ses Ayarla",
"Settings": "Ayarlar",
"Settings saved successfully!": "Ayarlar başarıyla kaydedildi!",
......
......@@ -6,6 +6,7 @@
"{{modelName}} is thinking...": "{{modelName}} думає...",
"{{user}}'s Chats": "Чати {{user}}а",
"{{webUIName}} Backend Required": "Необхідно підключення бекенду {{webUIName}}",
"A task model is used when performing tasks such as generating titles for chats and web search queries": "",
"a user": "користувача",
"About": "Про програму",
"Account": "Обліковий запис",
......@@ -395,7 +396,7 @@
"Set Model": "Встановити модель",
"Set reranking model (e.g. {{model}})": "Встановити модель переранжування (напр., {{model}})",
"Set Steps": "Встановити кроки",
"Set Title Auto-Generation Model": "Встановити модель автогенерації заголовків",
"Set Task Model": "",
"Set Voice": "Встановити голос",
"Settings": "Налаштування",
"Settings saved successfully!": "Налаштування успішно збережено!",
......
......@@ -6,6 +6,7 @@
"{{modelName}} is thinking...": "{{modelName}} đang suy nghĩ...",
"{{user}}'s Chats": "",
"{{webUIName}} Backend Required": "{{webUIName}} Yêu cầu Backend",
"A task model is used when performing tasks such as generating titles for chats and web search queries": "",
"a user": "người sử dụng",
"About": "Giới thiệu",
"Account": "Tài khoản",
......@@ -395,7 +396,7 @@
"Set Model": "Thiết lập mô hình",
"Set reranking model (e.g. {{model}})": "",
"Set Steps": "Đặt Số Bước",
"Set Title Auto-Generation Model": "Đặt tiêu đề tự động",
"Set Task Model": "",
"Set Voice": "Đặt Giọng nói",
"Settings": "Cài đặt",
"Settings saved successfully!": "Cài đặt đã được lưu thành công!",
......
......@@ -6,6 +6,7 @@
"{{modelName}} is thinking...": "{{modelName}} 正在思考...",
"{{user}}'s Chats": "",
"{{webUIName}} Backend Required": "需要 {{webUIName}} 后端",
"A task model is used when performing tasks such as generating titles for chats and web search queries": "",
"a user": "用户",
"About": "关于",
"Account": "账户",
......@@ -395,7 +396,7 @@
"Set Model": "设置模型",
"Set reranking model (e.g. {{model}})": "",
"Set Steps": "设置步骤",
"Set Title Auto-Generation Model": "设置标题自动生成模型",
"Set Task Model": "",
"Set Voice": "设置声音",
"Settings": "设置",
"Settings saved successfully!": "设置已保存",
......
......@@ -6,6 +6,7 @@
"{{modelName}} is thinking...": "{{modelName}} 正在思考...",
"{{user}}'s Chats": "",
"{{webUIName}} Backend Required": "需要 {{webUIName}} 後台",
"A task model is used when performing tasks such as generating titles for chats and web search queries": "",
"a user": "使用者",
"About": "關於",
"Account": "帳號",
......@@ -395,7 +396,7 @@
"Set Model": "設定模型",
"Set reranking model (e.g. {{model}})": "",
"Set Steps": "設定步數",
"Set Title Auto-Generation Model": "設定自動生成標題用模型",
"Set Task Model": "",
"Set Voice": "設定語音",
"Settings": "設定",
"Settings saved successfully!": "成功儲存設定",
......
......@@ -860,30 +860,27 @@
}
};
// TODO: Add support for adding all the user's messages as context, and not just the last message
const generateChatSearchQuery = async (messageId: string) => {
const model = $models.find((model) => model.id === selectedModels[0]);
// TODO: rename titleModel to taskModel - this is the model used for non-chat tasks (e.g. title generation, search query generation)
const titleModelId =
const taskModelId =
model?.external ?? false
? $settings?.title?.modelExternal ?? selectedModels[0]
: $settings?.title?.model ?? selectedModels[0];
const titleModel = $models.find((model) => model.id === titleModelId);
const taskModel = $models.find((model) => model.id === taskModelId);
const userMessage = history.messages[messageId];
const userPrompt = userMessage.content;
const previousMessages = messages.filter((message) => message.role === 'user').map((message) => message.content);
console.log(titleModel);
return await generateSearchQuery(
localStorage.token,
titleModelId,
taskModelId,
previousMessages,
userPrompt,
titleModel?.external ?? false
? titleModel?.source?.toLowerCase() === 'litellm'
taskModel?.external ?? false
? taskModel?.source?.toLowerCase() === 'litellm'
? `${LITELLM_API_BASE_URL}/v1`
: `${OPENAI_API_BASE_URL}`
: `${OLLAMA_API_BASE_URL}/v1`
......
......@@ -865,30 +865,27 @@
}
};
// TODO: Add support for adding all the user's messages as context, and not just the last message
const generateChatSearchQuery = async (messageId: string) => {
const model = $models.find((model) => model.id === selectedModels[0]);
// TODO: rename titleModel to taskModel - this is the model used for non-chat tasks (e.g. title generation, search query generation)
const titleModelId =
const taskModelId =
model?.external ?? false
? $settings?.title?.modelExternal ?? selectedModels[0]
: $settings?.title?.model ?? selectedModels[0];
const titleModel = $models.find((model) => model.id === titleModelId);
const taskModel = $models.find((model) => model.id === taskModelId);
const userMessage = history.messages[messageId];
const userPrompt = userMessage.content;
const previousMessages = messages.filter((message) => message.role === 'user').map((message) => message.content);
console.log(titleModel);
return await generateSearchQuery(
localStorage.token,
titleModelId,
taskModelId,
previousMessages,
userPrompt,
titleModel?.external ?? false
? titleModel?.source?.toLowerCase() === 'litellm'
taskModel?.external ?? false
? taskModel?.source?.toLowerCase() === 'litellm'
? `${LITELLM_API_BASE_URL}/v1`
: `${OPENAI_API_BASE_URL}`
: `${OLLAMA_API_BASE_URL}/v1`
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment