Parcourir la source

Fix: features of agent-chat (#3178)

KVOJJJin il y a 1 an
Parent
commit
48d4d55ecc

+ 3 - 2
web/app/components/app/configuration/config/index.tsx

@@ -20,7 +20,8 @@ import ConfigContext from '@/context/debug-configuration'
 import ConfigPrompt from '@/app/components/app/configuration/config-prompt'
 import ConfigVar from '@/app/components/app/configuration/config-var'
 import { type CitationConfig, type ModelConfig, type ModerationConfig, type MoreLikeThisConfig, type PromptVariable, type SpeechToTextConfig, type SuggestedQuestionsAfterAnswerConfig, type TextToSpeechConfig } from '@/models/debug'
-import { AppType, ModelModeType } from '@/types/app'
+import type { AppType } from '@/types/app'
+import { ModelModeType } from '@/types/app'
 import { useModalContext } from '@/context/modal-context'
 import ConfigParamModal from '@/app/components/app/configuration/toolbox/annotation/config-param-modal'
 import AnnotationFullModal from '@/app/components/billing/annotation-full/modal'
@@ -60,7 +61,7 @@ const Config: FC = () => {
     moderationConfig,
     setModerationConfig,
   } = useContext(ConfigContext)
-  const isChatApp = mode === AppType.chat
+  const isChatApp = ['advanced-chat', 'agent-chat', 'chat'].includes(mode)
   const { data: speech2textDefaultModel } = useDefaultModel(ModelTypeEnum.speech2text)
   const { data: text2speechDefaultModel } = useDefaultModel(ModelTypeEnum.tts)
   const { setShowModerationSettingModal } = useModalContext()

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

@@ -3,7 +3,7 @@ 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 { 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'
@@ -152,7 +152,7 @@ const useAdvancedPromptConfig = ({
           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)
+          if (['advanced-chat', 'agent-chat', 'chat'].includes(appMode) && completionPromptConfig.conversation_histories_role.assistant_prefix && completionPromptConfig.conversation_histories_role.user_prefix)
             draft.conversation_histories_role = completionPromptConfig.conversation_histories_role
         })
         setCompletionPromptConfig(newPromptConfig)