feat: set search system message as config

Signed-off-by: Bob Du <i@bobdu.cc>

Author: BobDu

service/src/chatgpt/index.ts

@@ -17,74 +17,8 @@ import type { ChatMessage, RequestOptions } from './types'
 
 dotenv.config()
 
-function systemMessageWithSearchResult(currentTime: string): string {
-  return `You are an intelligent assistant that needs to answer user questions based on search results.
-
-**Search Results Format Description:**
-- Search results may contain irrelevant information, please filter and use accordingly
-
-**Context Information:**
-- Current time: ${currentTime}
-
-**Response Requirements:**
-
-1. **Content Processing**
-   - Screen and filter search results, selecting content most relevant to the question
-   - Synthesize information from multiple web pages, avoiding repetitive citations from a single source
-   - When using web page information, please indicate the source in your answer (post it back to the application in the form of a link) to facilitate users' verification of the information source
-
-2. **Response Strategy**
-   - **Listing questions**: Limit to within 10 key points, prioritize providing the most relevant and complete information
-   - **Creative questions**: Make full use of search results to generate in-depth professional long-form answers
-   - **Objective Q&A**: Brief answers may appropriately supplement 1-2 sentences of related information
-
-3. **Format Requirements**
-   - Respond using markdown (latex start with $).
-   - Use structured, paragraph-based answer format
-   - When answering in points, limit to within 5 points, merging related content
-   - Ensure answers are aesthetically pleasing and highly readable
-
-4. **Language Standards**
-   - Keep answer language consistent with user's question language
-   - Do not change language unless specifically requested by the user
-
-**Notes:**
-- Not all search results are relevant, need to judge based on the question
-- For listing questions, inform users they can check search sources for complete information
-- Creative answers need to be multi-perspective, information-rich, and thoroughly discussed`
-}
-
-function systemMessageGetSearchQuery(currentTime: string): string {
-  return `You are an intelligent search assistant.
-Current time: ${currentTime}
-
-Before formally answering user questions, you need to analyze the user's questions and conversation context to determine whether you need to obtain more information through internet search to provide accurate answers.
-
-**Task Flow:**
-1. Carefully analyze the user's question content and previous conversation history
-2. Based on the current time, determine whether the question involves time-sensitive information. If it involves time-sensitive issues, please inform the specific date to be searched in the returned results instead of referring to pronouns such as today or yesterday
-3. Evaluate whether existing knowledge is sufficient to answer the question
-4. If search is needed, generate a precise search query
-5. If search is not needed, return empty result
-
-**Output Format Requirements:**
-- If search is needed: return <search_query>example search query keywords</search_query>
-- If search is not needed: return <search_query></search_query>
-- Do not include any other explanations or answer content
-- Search query should be concise and clear, able to obtain the most relevant information
-
-**Judgment Criteria:**
-- Time-sensitive information (such as latest news, stock prices, weather, real-time data, etc.): search needed
-- Latest policies, regulations, technological developments: may need search
-- Common sense questions, historical facts, basic knowledge: usually no search needed
-- Latest research or developments in professional fields: search recommended
-
-**Notes:**
-- Search query should target the core needs of user questions
-- Consider the timeliness and accuracy requirements of information
-- Prioritize obtaining the latest and most authoritative information sources
-
-Please strictly return results according to the above format.`
+function renderSystemMessage(template: string, currentTime: string): string {
+  return template.replace(/{current_time}/g, currentTime)
 }
 
 const ErrorCodeMessage: Record<string, string> = {
@@ -167,7 +101,7 @@ async function chatReplyProcess(options: RequestOptions) {
     let hasSearchResult = false
     const searchConfig = globalConfig.searchConfig
     if (searchConfig.enabled && searchConfig?.options?.apiKey && searchEnabled) {
-      messages[0].content = systemMessageGetSearchQuery(dayjs().format('YYYY-MM-DD HH:mm:ss'))
+      messages[0].content = renderSystemMessage(searchConfig.systemMessageGetSearchQuery, dayjs().format('YYYY-MM-DD HH:mm:ss'))
       const completion = await openai.chat.completions.create({
         model,
         messages,
@@ -201,7 +135,7 @@ search query: <search_query>${searchQuery}</search_query>
 search result: <search_result>${searchResult}</search_result>`,
         })
 
-        messages[0].content = systemMessageWithSearchResult(dayjs().format('YYYY-MM-DD HH:mm:ss'))
+        messages[0].content = renderSystemMessage(searchConfig.systemMessageWithSearchResult, dayjs().format('YYYY-MM-DD HH:mm:ss'))
         hasSearchResult = true
       }
     }

service/src/storage/model.ts

@@ -177,6 +177,8 @@ export class SearchConfig {
   public enabled: boolean
   public provider?: SearchServiceProvider
   public options?: SearchServiceOptions
+  public systemMessageWithSearchResult?: string
+  public systemMessageGetSearchQuery?: string
 }
 
 export enum SearchServiceProvider {

src/components/common/Setting/Search.vue

@@ -24,7 +24,7 @@ async function fetchConfig() {
     loading.value = true
     const { data } = await fetchChatConfig<ConfigState>()
     if (!data.searchConfig)
-      data.searchConfig = new SearchConfig(false, '', { apiKey: '' })
+      data.searchConfig = new SearchConfig(false, '', { apiKey: '' }, '', '')
     if (!data.searchConfig.options)
       data.searchConfig.options = { apiKey: '' }
     config.value = data.searchConfig
@@ -107,6 +107,18 @@ onMounted(() => {
             />
           </div>
         </div>
+        <div v-if="config && config.enabled" class="flex items-center space-x-4">
+          <span class="shrink-0 w-[100px]">{{ $t('setting.systemMessageWithSearchResult') }}</span>
+          <div class="flex-1">
+            <NInput v-model:value="config.systemMessageWithSearchResult" type="textarea" :autosize="{ minRows: 2 }" :placeholder="t('setting.systemMessageWithSearchResultPlaceholder')" />
+          </div>
+        </div>
+        <div v-if="config && config.enabled" class="flex items-center space-x-4">
+          <span class="shrink-0 w-[100px]">{{ $t('setting.systemMessageGetSearchQuery') }}</span>
+          <div class="flex-1">
+            <NInput v-model:value="config.systemMessageGetSearchQuery" type="textarea" :autosize="{ minRows: 2 }" :placeholder="t('setting.systemMessageGetSearchQueryPlaceholder')" />
+          </div>
+        </div>
         <div class="flex items-center space-x-4">
           <span class="shrink-0 w-[100px]" />
           <div class="flex flex-wrap items-center gap-4">

src/components/common/Setting/model.ts

@@ -192,9 +192,13 @@ export class SearchConfig {
   enabled: boolean
   provider: SearchServiceProvider
   options: SearchServiceOptions
-  constructor(enabled: boolean, provider: SearchServiceProvider, options: SearchServiceOptions) {
+  systemMessageWithSearchResult: string
+  systemMessageGetSearchQuery: string
+  constructor(enabled: boolean, provider: SearchServiceProvider, options: SearchServiceOptions, systemMessageWithSearchResult: string, systemMessageGetSearchQuery: string) {
     this.enabled = enabled
     this.provider = provider
     this.options = options
+    this.systemMessageWithSearchResult = systemMessageWithSearchResult
+    this.systemMessageGetSearchQuery = systemMessageGetSearchQuery
   }
 }

src/locales/en-US.ts

@@ -167,6 +167,10 @@ export default {
     searchEnabled: 'Search Enabled',
     searchProvider: 'Search Provider',
     searchApiKey: 'Search API Key',
+    systemMessageWithSearchResult: 'System message for conversations with search results',
+    systemMessageGetSearchQuery: 'System message for getting search query',
+    systemMessageWithSearchResultPlaceholder: 'System message template when with search results. Use {\'{current_time}\'} as placeholder for current time.',
+    systemMessageGetSearchQueryPlaceholder: 'System message template for generating search query word. Use {\'{current_time}\'} as placeholder for current time. Require LLM to return the query word in <search_query>example search query</search_query> tags or return empty <search_query></search_query> tag if not need search.',
     searchTest: 'Test Search',
     accessTokenExpiredTime: 'Expired Time',
     userConfig: 'Users',

src/locales/zh-CN.ts

@@ -167,6 +167,10 @@ export default {
     searchEnabled: '搜索功能',
     searchProvider: '搜索提供商',
     searchApiKey: '搜索 API 密钥',
+    systemMessageWithSearchResult: '包含搜索结果时的对话系统提示词',
+    systemMessageGetSearchQuery: '用于获取搜索查询词的系统提示词',
+    systemMessageWithSearchResultPlaceholder: '携带搜索结果时的系统消息模板, 使用 {\'{current_time}\'} 作为当前时间的占位符',
+    systemMessageGetSearchQueryPlaceholder: '用于生成搜索查询的系统消息模板, 使用 {\'{current_time}\'} 作为当前时间的占位符, 要求 LLM 在 <search_query>example search query</search_query> 标签中返回查询词，如果不需要搜索则返回空<search_query></search_query>标签',
     searchTest: '测试搜索',
     accessTokenExpiredTime: '过期时间',
     userConfig: '用户管理',