feature: add topic prompt (#1696)

* feat: 新增话题补充提示词

* feat: 新增话题补充提示词

* feat: 新增话题补充提示词

* feat: 新增话题补充提示词

* feat: 新增话题补充提示词

src/renderer/src/hooks/useTopic.ts

@@ -17,8 +17,10 @@ export function useActiveTopic(_assistant: Assistant, topic?: Topic) {
 
   useEffect(() => {
     // activeTopic not in assistant.topics
-    if (assistant && !find(assistant.topics, { id: activeTopic?.id })) {
+    const topicTarget = find(assistant.topics, { id: activeTopic?.id })
-      setActiveTopic(assistant.topics[0])
+    // if (assistant && !find(assistant.topics, { id: activeTopic?.id })) {
+    if (assistant) {
+      setActiveTopic(topicTarget || assistant.topics[0])
     }
   }, [activeTopic?.id, assistant])
 

src/renderer/src/i18n/locales/en-us.json

@@ -136,7 +136,10 @@
       "topics.pinned": "Pinned Topics",
       "topics.title": "Topics",
       "topics.unpinned": "Unpinned Topics",
-      "translate": "Translate"
+      "translate": "Translate",
+      "topics.prompt": "Topic Prompts",
+      "topics.prompt.tips": "Topic Prompts: Additional supplementary prompts provided for the current topic",
+      "topics.prompt.edit.title": "Edit Topic Prompts"
     },
     "common": {
       "add": "Add",

src/renderer/src/i18n/locales/ja-jp.json

@@ -136,7 +136,10 @@
       "topics.pinned": "トピックを固定",
       "topics.title": "トピック",
       "topics.unpinned": "固定解除",
-      "translate": "翻訳"
+      "translate": "翻訳",
+      "topics.prompt": "トピック提示語",
+      "topics.prompt.tips": "トピック提示語：現在のトピックに対して追加の補足提示語を提供",
+      "topics.prompt.edit.title": "トピック提示語を編集する"
     },
     "common": {
       "add": "追加",

src/renderer/src/i18n/locales/ru-ru.json

@@ -136,7 +136,10 @@
       "topics.pinned": "Закрепленные темы",
       "topics.title": "Топики",
       "topics.unpinned": "Открепленные темы",
-      "translate": "Перевести"
+      "translate": "Перевести",
+      "topics.prompt": "Тематические подсказки",
+      "topics.prompt.tips": "Тематические подсказки: Дополнительные подсказки, предоставленные для текущей темы",
+      "topics.prompt.edit.title": "Редактировать подсказки темы"
     },
     "common": {
       "add": "Добавить",

src/renderer/src/i18n/locales/zh-cn.json

@@ -136,7 +136,10 @@
       "topics.pinned": "固定话题",
       "topics.title": "话题",
       "topics.unpinned": "取消固定",
-      "translate": "翻译"
+      "translate": "翻译",
+      "topics.prompt": "话题提示词",
+      "topics.prompt.tips": "话题提示词: 针对当前话题提供额外的补充提示词",
+      "topics.prompt.edit.title": "编辑话题提示词"
     },
     "common": {
       "add": "添加",

src/renderer/src/i18n/locales/zh-tw.json

@@ -136,7 +136,10 @@
       "topics.pinned": "固定話題",
       "topics.title": "話題",
       "topics.unpinned": "取消固定",
-      "translate": "翻譯"
+      "translate": "翻譯",
+      "topics.prompt": "話題提示詞",
+      "topics.prompt.tips": "話題提示詞：針對目前話題提供額外的補充提示詞",
+      "topics.prompt.edit.title": "編輯話題提示詞"
     },
     "common": {
       "add": "添加",

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

@@ -118,8 +118,12 @@ const MessageItem: FC<Props> = ({
   useEffect(() => {
     if (topic && onGetMessages && onSetMessages) {
       if (message.status === 'sending') {
+        const _assistant = {
+          ...assistant,
+          prompt: `${assistant?.prompt || ''}\n${topic?.prompt || ''}`
+        }
         const messages = onGetMessages()
-        const assistantWithModel = message.model ? { ...assistant, model: message.model } : assistant
+        const assistantWithModel = message.model ? { ..._assistant, model: message.model } : _assistant
 
         fetchChatCompletion({
           message,

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

@@ -315,7 +315,7 @@ const Messages: FC<Props> = ({ assistant, topic, setActiveTopic }) => {
             ))}
           </ScrollContainer>
         </InfiniteScroll>
-        <Prompt assistant={assistant} key={assistant.prompt} />
+        <Prompt assistant={assistant} key={assistant.prompt} topic={topic} />
       </NarrowLayout>
     </Container>
   )
@@ -349,8 +349,7 @@ interface ContainerProps {
 const Container = styled(Scrollbar)<ContainerProps>`
   display: flex;
   flex-direction: column-reverse;
-  padding: 10px 0;
+  padding: 10px 0 20px;
-  padding-bottom: 20px;
   overflow-x: hidden;
   background-color: var(--color-background);
 `

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

@@ -1,22 +1,22 @@
 import AssistantSettingsPopup from '@renderer/pages/settings/AssistantSettings'
-import { Assistant } from '@renderer/types'
+import { Assistant, Topic } from '@renderer/types'
 import { FC } from 'react'
 import { useTranslation } from 'react-i18next'
 import styled from 'styled-components'
 
 interface Props {
   assistant: Assistant
+  topic?: Topic
 }
 
-const Prompt: FC<Props> = ({ assistant }) => {
+const Prompt: FC<Props> = ({ assistant, topic }) => {
   const { t } = useTranslation()
 
   const prompt = assistant.prompt || t('chat.default.description')
-
+  const topicPrompt = topic?.prompt || ''
-  if (!prompt) {
+  if (!prompt && !topicPrompt) {
     return null
   }
-
   return (
     <Container className="system-prompt" onClick={() => AssistantSettingsPopup.show({ assistant })}>
       <Text>{prompt}</Text>

src/renderer/src/pages/home/Tabs/TopicsTab.tsx

@@ -5,6 +5,7 @@ import {
   EditOutlined,
   FolderOutlined,
   PushpinOutlined,
+  QuestionCircleOutlined,
   UploadOutlined
 } from '@ant-design/icons'
 import DragableList from '@renderer/components/DragableList'
@@ -20,7 +21,7 @@ import store from '@renderer/store'
 import { setGenerating } from '@renderer/store/runtime'
 import { Assistant, Topic } from '@renderer/types'
 import { exportTopicAsMarkdown, exportTopicToNotion, topicToMarkdown } from '@renderer/utils/export'
-import { Dropdown, MenuProps } from 'antd'
+import { Dropdown, MenuProps, Tooltip } from 'antd'
 import dayjs from 'dayjs'
 import { findIndex } from 'lodash'
 import { FC, useCallback } from 'react'
@@ -115,6 +116,28 @@ const Topics: FC<Props> = ({ assistant: _assistant, activeTopic, setActiveTopic
             }
           }
         },
+        {
+          label: t('chat.topics.prompt'),
+          key: 'topic-prompt',
+          icon: <i className="iconfont icon-ai-model1" style={{ fontSize: '14px' }} />,
+          extra: (
+            <Tooltip title={t('chat.topics.prompt.tips')}>
+              <QuestionIcon />
+            </Tooltip>
+          ),
+          async onClick() {
+            const prompt = await PromptPopup.show({
+              title: t('chat.topics.prompt.edit.title'),
+              message: '',
+              defaultValue: topic?.prompt || '',
+              inputProps: {
+                rows: 8,
+                allowClear: true
+              }
+            })
+            if (prompt !== null) updateTopic({ ...topic, prompt })
+          }
+        },
         {
           label: topic.pinned ? t('chat.topics.unpinned') : t('chat.topics.pinned'),
           key: 'pin',
@@ -211,6 +234,11 @@ const Topics: FC<Props> = ({ assistant: _assistant, activeTopic, setActiveTopic
                 onClick={() => onSwitchTopic(topic)}
                 style={{ borderRadius }}>
                 <TopicName className="name">{topic.name.replace('`', '')}</TopicName>
+                {topic.prompt && (
+                  <TopicPromptText className="prompt">
+                    {t('common.prompt')}: {topic.prompt}
+                  </TopicPromptText>
+                )}
                 {showTopicTime && (
                   <TopicTime className="time">{dayjs(topic.createdAt).format('MM/DD HH:mm')}</TopicTime>
                 )}
@@ -291,6 +319,18 @@ const TopicName = styled.div`
   font-size: 13px;
 `
 
+const TopicPromptText = styled.div`
+  color: var(--color-text-2);
+  font-size: 12px;
+  display: -webkit-box;
+  -webkit-line-clamp: 2;
+  -webkit-box-orient: vertical;
+  overflow: hidden;
+  ~ .prompt-text {
+    margin-top: 10px;
+  }
+`
+
 const TopicTime = styled.div`
   color: var(--color-text-3);
   font-size: 11px;
@@ -310,5 +350,10 @@ const MenuButton = styled.div`
     font-size: 12px;
   }
 `
+const QuestionIcon = styled(QuestionCircleOutlined)`
+  font-size: 14px;
+  cursor: pointer;
+  color: var(--color-text-3);
+`
 
 export default Topics

src/renderer/src/types/index.ts

@@ -89,6 +89,7 @@ export type Topic = {
   updatedAt: string
   messages: Message[]
   pinned?: boolean
+  prompt?: string
 }
 
 export type User = {