feat: improved translation features and settings

src/renderer/src/config/prompts.ts

@@ -47,8 +47,21 @@ export const AGENT_PROMPT = `
 export const SUMMARIZE_PROMPT =
   '你是一名擅长会话的助理，你需要将用户的会话总结为 10 个字以内的标题，标题语言与用户的首要语言一致，不要使用标点符号和其他特殊符号'
 
-export const TRANSLATE_PROMPT =
-  'You are a translation expert. Translate from input language to {{target_language}}, provide the translation result directly without any explanation and keep original format. Do not translate if the target language is the same as the source language.'
+export const TRANSLATE_PROMPT = `You are a translation expert.
+
+Translate from input language to {{target_language}}
+
+Requirements:
+
+1. provide the translation result directly without any explanation and keep original format.
+2. Do not translate if the target language is the same as the source language.
+
+Example:
+
+input: 你好，今天天气怎么样？
+target_language: english
+output: Hello, how is the weather today?
+`
 
 export const REFERENCE_PROMPT = `请根据参考资料回答问题，并使用脚注格式引用数据来源。请忽略无关的参考资料。
 

src/renderer/src/pages/home/Messages/MessageMenubar.tsx

@@ -156,51 +156,9 @@ const MessageMenubar: FC<Props> = (props) => {
         key: 'edit',
         icon: <EditOutlined />,
         onClick: onEdit
-      },
-      {
-        label: t('chat.translate'),
-        key: 'translate',
-        icon: isTranslating ? <SyncOutlined spin /> : <TranslationOutlined />,
-        children: [
-          {
-            label: '🇨🇳 ' + t('languages.chinese'),
-            key: 'translate-chinese',
-            onClick: () => handleTranslate('chinese')
-          },
-          {
-            label: '🇭🇰 ' + t('languages.chinese-traditional'),
-            key: 'translate-chinese-traditional',
-            onClick: () => handleTranslate('chinese-traditional')
-          },
-          {
-            label: '🇬🇧 ' + t('languages.english'),
-            key: 'translate-english',
-            onClick: () => handleTranslate('english')
-          },
-          {
-            label: '🇯🇵 ' + t('languages.japanese'),
-            key: 'translate-japanese',
-            onClick: () => handleTranslate('japanese')
-          },
-          {
-            label: '🇰🇷 ' + t('languages.korean'),
-            key: 'translate-korean',
-            onClick: () => handleTranslate('korean')
-          },
-          {
-            label: '🇷🇺 ' + t('languages.russian'),
-            key: 'translate-russian',
-            onClick: () => handleTranslate('russian')
-          },
-          {
-            label: '✖ ' + t('translate.close'),
-            key: 'translate-close',
-            onClick: () => onEditMessage?.({ ...message, translatedContent: undefined })
-          }
-        ]
       }
     ],
-    [handleTranslate, isTranslating, message, onEdit, onEditMessage, t]
+    [message, onEdit, t]
   )
 
   const onAtModelRegenerate = async () => {
@@ -261,6 +219,57 @@ const MessageMenubar: FC<Props> = (props) => {
           </ActionButton>
         </Tooltip>
       )}
+      {!isUserMessage && (
+        <Dropdown
+          menu={{
+            items: [
+              {
+                label: '🇨🇳 ' + t('languages.chinese'),
+                key: 'translate-chinese',
+                onClick: () => handleTranslate('chinese')
+              },
+              {
+                label: '🇭🇰 ' + t('languages.chinese-traditional'),
+                key: 'translate-chinese-traditional',
+                onClick: () => handleTranslate('chinese-traditional')
+              },
+              {
+                label: '🇬🇧 ' + t('languages.english'),
+                key: 'translate-english',
+                onClick: () => handleTranslate('english')
+              },
+              {
+                label: '🇯🇵 ' + t('languages.japanese'),
+                key: 'translate-japanese',
+                onClick: () => handleTranslate('japanese')
+              },
+              {
+                label: '🇰🇷 ' + t('languages.korean'),
+                key: 'translate-korean',
+                onClick: () => handleTranslate('korean')
+              },
+              {
+                label: '🇷🇺 ' + t('languages.russian'),
+                key: 'translate-russian',
+                onClick: () => handleTranslate('russian')
+              },
+              {
+                label: '✖ ' + t('translate.close'),
+                key: 'translate-close',
+                onClick: () => onEditMessage?.({ ...message, translatedContent: undefined })
+              }
+            ]
+          }}
+          trigger={['click']}
+          placement="topRight"
+          arrow>
+          <Tooltip title={t('chat.translate')} mouseEnterDelay={1.2}>
+            <ActionButton className="message-action-button">
+              <TranslationOutlined />
+            </ActionButton>
+          </Tooltip>
+        </Dropdown>
+      )}
       <Popconfirm
         title={t('message.message.delete.content')}
         okButtonProps={{ danger: true }}

src/renderer/src/providers/OpenAIProvider.ts

@@ -198,7 +198,8 @@ export default class OpenAIProvider extends BaseProvider {
       model: model.id,
       messages: messages as ChatCompletionMessageParam[],
       stream: false,
-      keep_alive: this.keepAliveTime
+      keep_alive: this.keepAliveTime,
+      temperature: assistant?.settings?.temperature
     })
 
     return response.choices[0].message?.content || ''

src/renderer/src/services/AssistantService.ts

@@ -23,6 +23,11 @@ export function getDefaultTranslateAssistant(targetLanguage: string, text: strin
   const translateModel = getTranslateModel()
   const assistant: Assistant = getDefaultAssistant()
   assistant.model = translateModel
+
+  assistant.settings = {
+    temperature: 0.7
+  }
+
   assistant.prompt = store
     .getState()
     .settings.translateModelPrompt.replace('{{target_language}}', targetLanguage)