feat: advanced prompt (#1330)

Co-authored-by: Joel <iamjoel007@gmail.com>
Co-authored-by: JzoNg <jzongcode@gmail.com>
Co-authored-by: Gillian97 <jinling.sunshine@gmail.com>

web/app/components/app/configuration/hooks/use-advanced-prompt-config.ts

@@ -0,0 +1,176 @@
+import { useState } from 'react'
+import { clone } from 'lodash-es'
+import produce from 'immer'
+import type { ChatPromptConfig, CompletionPromptConfig, ConversationHistoriesRole, PromptItem } from '@/models/debug'
+import { PromptMode } from '@/models/debug'
+import { AppType, ModelModeType } from '@/types/app'
+import { DEFAULT_CHAT_PROMPT_CONFIG, DEFAULT_COMPLETION_PROMPT_CONFIG } from '@/config'
+import { PRE_PROMPT_PLACEHOLDER_TEXT, checkHasContextBlock, checkHasHistoryBlock, checkHasQueryBlock } from '@/app/components/base/prompt-editor/constants'
+import { fetchPromptTemplate } from '@/service/debug'
+
+type Param = {
+  appMode: string
+  modelModeType: ModelModeType
+  modelName: string
+  promptMode: PromptMode
+  prePrompt: string
+  onUserChangedPrompt: () => void
+  hasSetDataSet: boolean
+}
+
+const useAdvancedPromptConfig = ({
+  appMode,
+  modelModeType,
+  modelName,
+  promptMode,
+  prePrompt,
+  onUserChangedPrompt,
+  hasSetDataSet,
+}: Param) => {
+  const isAdvancedPrompt = promptMode === PromptMode.advanced
+  const [chatPromptConfig, setChatPromptConfig] = useState<ChatPromptConfig>(clone(DEFAULT_CHAT_PROMPT_CONFIG))
+  const [completionPromptConfig, setCompletionPromptConfig] = useState<CompletionPromptConfig>(clone(DEFAULT_COMPLETION_PROMPT_CONFIG))
+
+  const currentAdvancedPrompt = (() => {
+    if (!isAdvancedPrompt)
+      return []
+
+    return (modelModeType === ModelModeType.chat) ? chatPromptConfig.prompt : completionPromptConfig.prompt
+  })()
+
+  const setCurrentAdvancedPrompt = (prompt: PromptItem | PromptItem[], isUserChanged?: boolean) => {
+    if (!isAdvancedPrompt)
+      return
+
+    if (modelModeType === ModelModeType.chat) {
+      setChatPromptConfig({
+        ...chatPromptConfig,
+        prompt: prompt as PromptItem[],
+      })
+    }
+    else {
+      setCompletionPromptConfig({
+        ...completionPromptConfig,
+        prompt: prompt as PromptItem,
+      })
+    }
+    if (isUserChanged)
+      onUserChangedPrompt()
+  }
+
+  const setConversationHistoriesRole = (conversationHistoriesRole: ConversationHistoriesRole) => {
+    setCompletionPromptConfig({
+      ...completionPromptConfig,
+      conversation_histories_role: conversationHistoriesRole,
+    })
+  }
+
+  const hasSetBlockStatus = (() => {
+    if (!isAdvancedPrompt) {
+      return {
+        context: checkHasContextBlock(prePrompt),
+        history: false,
+        query: false,
+      }
+    }
+    if (modelModeType === ModelModeType.chat) {
+      return {
+        context: !!chatPromptConfig.prompt.find(p => checkHasContextBlock(p.text)),
+        history: false,
+        query: !!chatPromptConfig.prompt.find(p => checkHasQueryBlock(p.text)),
+      }
+    }
+    else {
+      const prompt = completionPromptConfig.prompt.text
+      return {
+        context: checkHasContextBlock(prompt),
+        history: checkHasHistoryBlock(prompt),
+        query: checkHasQueryBlock(prompt),
+      }
+    }
+  })()
+
+  /* prompt: simple to advanced process, or chat model to completion model
+  * 1. migrate prompt
+  * 2. change promptMode to advanced
+  */
+  const migrateToDefaultPrompt = async (isMigrateToCompetition?: boolean, toModelModeType?: ModelModeType) => {
+    const mode = modelModeType
+    const toReplacePrePrompt = prePrompt || ''
+    if (!isAdvancedPrompt) {
+      const { chat_prompt_config, completion_prompt_config } = await fetchPromptTemplate({
+        appMode,
+        mode,
+        modelName,
+        hasSetDataSet,
+      })
+      if (modelModeType === ModelModeType.chat) {
+        const newPromptConfig = produce(chat_prompt_config, (draft) => {
+          draft.prompt = draft.prompt.map((p) => {
+            return {
+              ...p,
+              text: p.text.replace(PRE_PROMPT_PLACEHOLDER_TEXT, toReplacePrePrompt),
+            }
+          })
+        })
+        setChatPromptConfig(newPromptConfig)
+      }
+
+      else {
+        const newPromptConfig = produce(completion_prompt_config, (draft) => {
+          draft.prompt.text = draft.prompt.text.replace(PRE_PROMPT_PLACEHOLDER_TEXT, toReplacePrePrompt)
+        })
+        setCompletionPromptConfig(newPromptConfig)
+      }
+      return
+    }
+
+    if (isMigrateToCompetition) {
+      const { completion_prompt_config, chat_prompt_config } = await fetchPromptTemplate({
+        appMode,
+        mode: toModelModeType as ModelModeType,
+        modelName,
+        hasSetDataSet,
+      })
+
+      if (toModelModeType === ModelModeType.completion) {
+        const newPromptConfig = produce(completion_prompt_config, (draft) => {
+          if (!completionPromptConfig.prompt.text)
+            draft.prompt.text = draft.prompt.text.replace(PRE_PROMPT_PLACEHOLDER_TEXT, toReplacePrePrompt)
+
+          else
+            draft.prompt.text = completionPromptConfig.prompt.text.replace(PRE_PROMPT_PLACEHOLDER_TEXT, toReplacePrePrompt)
+
+          if (appMode === AppType.chat && completionPromptConfig.conversation_histories_role.assistant_prefix && completionPromptConfig.conversation_histories_role.user_prefix)
+            draft.conversation_histories_role = completionPromptConfig.conversation_histories_role
+        })
+        setCompletionPromptConfig(newPromptConfig)
+      }
+      else {
+        const newPromptConfig = produce(chat_prompt_config, (draft) => {
+          draft.prompt = draft.prompt.map((p) => {
+            return {
+              ...p,
+              text: p.text.replace(PRE_PROMPT_PLACEHOLDER_TEXT, toReplacePrePrompt),
+            }
+          })
+        })
+        setChatPromptConfig(newPromptConfig)
+      }
+    }
+  }
+
+  return {
+    chatPromptConfig,
+    setChatPromptConfig,
+    completionPromptConfig,
+    setCompletionPromptConfig,
+    currentAdvancedPrompt,
+    setCurrentAdvancedPrompt,
+    hasSetBlockStatus,
+    setConversationHistoriesRole,
+    migrateToDefaultPrompt,
+  }
+}
+
+export default useAdvancedPromptConfig