index.tsx 40 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028
  1. 'use client'
  2. import type { FC } from 'react'
  3. import React, { useCallback, useEffect, useMemo, useRef, useState } from 'react'
  4. import useSWR from 'swr'
  5. import { useTranslation } from 'react-i18next'
  6. import { useContext } from 'use-context-selector'
  7. import { usePathname } from 'next/navigation'
  8. import produce from 'immer'
  9. import { useBoolean, useGetState } from 'ahooks'
  10. import { clone, isEqual } from 'lodash-es'
  11. import { CodeBracketIcon } from '@heroicons/react/20/solid'
  12. import { useShallow } from 'zustand/react/shallow'
  13. import AgentSettingButton from '@/app/components/app/configuration/config/agent-setting-button'
  14. import useAdvancedPromptConfig from '@/app/components/app/configuration/hooks/use-advanced-prompt-config'
  15. import EditHistoryModal from '@/app/components/app/configuration/config-prompt/conversation-history/edit-modal'
  16. import {
  17. useDebugWithSingleOrMultipleModel,
  18. useFormattingChangedDispatcher,
  19. } from '@/app/components/app/configuration/debug/hooks'
  20. import type { ModelAndParameter } from '@/app/components/app/configuration/debug/types'
  21. import Button from '@/app/components/base/button'
  22. import Loading from '@/app/components/base/loading'
  23. import AppPublisher from '@/app/components/app/app-publisher/features-wrapper'
  24. import type {
  25. AnnotationReplyConfig,
  26. DatasetConfigs,
  27. Inputs,
  28. ModelConfig,
  29. ModerationConfig,
  30. MoreLikeThisConfig,
  31. PromptConfig,
  32. PromptVariable,
  33. TextToSpeechConfig,
  34. } from '@/models/debug'
  35. import type { ExternalDataTool } from '@/models/common'
  36. import type { DataSet } from '@/models/datasets'
  37. import type { ModelConfig as BackendModelConfig, VisionSettings } from '@/types/app'
  38. import ConfigContext from '@/context/debug-configuration'
  39. import Config from '@/app/components/app/configuration/config'
  40. import Debug from '@/app/components/app/configuration/debug'
  41. import Confirm from '@/app/components/base/confirm'
  42. import { ModelFeatureEnum, ModelTypeEnum } from '@/app/components/header/account-setting/model-provider-page/declarations'
  43. import { ToastContext } from '@/app/components/base/toast'
  44. import { fetchAppDetail, updateAppModelConfig } from '@/service/apps'
  45. import { promptVariablesToUserInputsForm, userInputsFormToPromptVariables } from '@/utils/model-config'
  46. import { fetchDatasets } from '@/service/datasets'
  47. import { useProviderContext } from '@/context/provider-context'
  48. import { AgentStrategy, AppType, ModelModeType, RETRIEVE_TYPE, Resolution, TransferMethod } from '@/types/app'
  49. import { PromptMode } from '@/models/debug'
  50. import { ANNOTATION_DEFAULT, DATASET_DEFAULT, DEFAULT_AGENT_SETTING, DEFAULT_CHAT_PROMPT_CONFIG, DEFAULT_COMPLETION_PROMPT_CONFIG } from '@/config'
  51. import SelectDataSet from '@/app/components/app/configuration/dataset-config/select-dataset'
  52. import { useModalContext } from '@/context/modal-context'
  53. import useBreakpoints, { MediaType } from '@/hooks/use-breakpoints'
  54. import Drawer from '@/app/components/base/drawer'
  55. import ModelParameterModal from '@/app/components/header/account-setting/model-provider-page/model-parameter-modal'
  56. import type { FormValue } from '@/app/components/header/account-setting/model-provider-page/declarations'
  57. import {
  58. useModelListAndDefaultModelAndCurrentProviderAndModel,
  59. useTextGenerationCurrentProviderAndModelAndModelList,
  60. } from '@/app/components/header/account-setting/model-provider-page/hooks'
  61. import { fetchCollectionList } from '@/service/tools'
  62. import { type Collection } from '@/app/components/tools/types'
  63. import { useStore as useAppStore } from '@/app/components/app/store'
  64. import {
  65. getMultipleRetrievalConfig,
  66. getSelectedDatasetsMode,
  67. } from '@/app/components/workflow/nodes/knowledge-retrieval/utils'
  68. import { FeaturesProvider } from '@/app/components/base/features'
  69. import type { Features as FeaturesData, FileUpload } from '@/app/components/base/features/types'
  70. import { FILE_EXTS } from '@/app/components/base/prompt-editor/constants'
  71. import { SupportUploadFileTypes } from '@/app/components/workflow/types'
  72. import NewFeaturePanel from '@/app/components/base/features/new-feature-panel'
  73. import { fetchFileUploadConfig } from '@/service/common'
  74. type PublishConfig = {
  75. modelConfig: ModelConfig
  76. completionParams: FormValue
  77. }
  78. const Configuration: FC = () => {
  79. const { t } = useTranslation()
  80. const { notify } = useContext(ToastContext)
  81. const { appDetail, showAppConfigureFeaturesModal, setAppSiderbarExpand, setShowAppConfigureFeaturesModal } = useAppStore(useShallow(state => ({
  82. appDetail: state.appDetail,
  83. setAppSiderbarExpand: state.setAppSiderbarExpand,
  84. showAppConfigureFeaturesModal: state.showAppConfigureFeaturesModal,
  85. setShowAppConfigureFeaturesModal: state.setShowAppConfigureFeaturesModal,
  86. })))
  87. const { data: fileUploadConfigResponse } = useSWR({ url: '/files/upload' }, fetchFileUploadConfig)
  88. const latestPublishedAt = useMemo(() => appDetail?.model_config.updated_at, [appDetail])
  89. const [formattingChanged, setFormattingChanged] = useState(false)
  90. const { setShowAccountSettingModal } = useModalContext()
  91. const [hasFetchedDetail, setHasFetchedDetail] = useState(false)
  92. const isLoading = !hasFetchedDetail
  93. const pathname = usePathname()
  94. const matched = pathname.match(/\/app\/([^/]+)/)
  95. const appId = (matched?.length && matched[1]) ? matched[1] : ''
  96. const [mode, setMode] = useState('')
  97. const [publishedConfig, setPublishedConfig] = useState<PublishConfig | null>(null)
  98. const [conversationId, setConversationId] = useState<string | null>('')
  99. const media = useBreakpoints()
  100. const isMobile = media === MediaType.mobile
  101. const [isShowDebugPanel, { setTrue: showDebugPanel, setFalse: hideDebugPanel }] = useBoolean(false)
  102. const [introduction, setIntroduction] = useState<string>('')
  103. const [suggestedQuestions, setSuggestedQuestions] = useState<string[]>([])
  104. const [controlClearChatMessage, setControlClearChatMessage] = useState(0)
  105. const [prevPromptConfig, setPrevPromptConfig] = useState<PromptConfig>({
  106. prompt_template: '',
  107. prompt_variables: [],
  108. })
  109. const [moreLikeThisConfig, setMoreLikeThisConfig] = useState<MoreLikeThisConfig>({
  110. enabled: false,
  111. })
  112. const [suggestedQuestionsAfterAnswerConfig, setSuggestedQuestionsAfterAnswerConfig] = useState<MoreLikeThisConfig>({
  113. enabled: false,
  114. })
  115. const [speechToTextConfig, setSpeechToTextConfig] = useState<MoreLikeThisConfig>({
  116. enabled: false,
  117. })
  118. const [textToSpeechConfig, setTextToSpeechConfig] = useState<TextToSpeechConfig>({
  119. enabled: false,
  120. voice: '',
  121. language: '',
  122. })
  123. const [citationConfig, setCitationConfig] = useState<MoreLikeThisConfig>({
  124. enabled: false,
  125. })
  126. const [annotationConfig, doSetAnnotationConfig] = useState<AnnotationReplyConfig>({
  127. id: '',
  128. enabled: false,
  129. score_threshold: ANNOTATION_DEFAULT.score_threshold,
  130. embedding_model: {
  131. embedding_provider_name: '',
  132. embedding_model_name: '',
  133. },
  134. })
  135. const formattingChangedDispatcher = useFormattingChangedDispatcher()
  136. const setAnnotationConfig = (config: AnnotationReplyConfig, notSetFormatChanged?: boolean) => {
  137. doSetAnnotationConfig(config)
  138. if (!notSetFormatChanged)
  139. formattingChangedDispatcher()
  140. }
  141. const [moderationConfig, setModerationConfig] = useState<ModerationConfig>({
  142. enabled: false,
  143. })
  144. const [externalDataToolsConfig, setExternalDataToolsConfig] = useState<ExternalDataTool[]>([])
  145. const [inputs, setInputs] = useState<Inputs>({})
  146. const [query, setQuery] = useState('')
  147. const [completionParams, doSetCompletionParams] = useState<FormValue>({})
  148. const [_, setTempStop, getTempStop] = useGetState<string[]>([])
  149. const setCompletionParams = (value: FormValue) => {
  150. const params = { ...value }
  151. // eslint-disable-next-line @typescript-eslint/no-use-before-define
  152. if ((!params.stop || params.stop.length === 0) && (modeModeTypeRef.current === ModelModeType.completion)) {
  153. params.stop = getTempStop()
  154. setTempStop([])
  155. }
  156. doSetCompletionParams(params)
  157. }
  158. const [modelConfig, doSetModelConfig] = useState<ModelConfig>({
  159. provider: 'openai',
  160. model_id: 'gpt-3.5-turbo',
  161. mode: ModelModeType.unset,
  162. configs: {
  163. prompt_template: '',
  164. prompt_variables: [] as PromptVariable[],
  165. },
  166. more_like_this: null,
  167. opening_statement: '',
  168. suggested_questions: [],
  169. sensitive_word_avoidance: null,
  170. speech_to_text: null,
  171. text_to_speech: null,
  172. file_upload: null,
  173. suggested_questions_after_answer: null,
  174. retriever_resource: null,
  175. annotation_reply: null,
  176. dataSets: [],
  177. agentConfig: DEFAULT_AGENT_SETTING,
  178. })
  179. const isAgent = mode === 'agent-chat'
  180. const isOpenAI = modelConfig.provider === 'openai'
  181. const [collectionList, setCollectionList] = useState<Collection[]>([])
  182. useEffect(() => {
  183. }, [])
  184. const [datasetConfigs, setDatasetConfigs] = useState<DatasetConfigs>({
  185. retrieval_model: RETRIEVE_TYPE.multiWay,
  186. reranking_model: {
  187. reranking_provider_name: '',
  188. reranking_model_name: '',
  189. },
  190. top_k: DATASET_DEFAULT.top_k,
  191. score_threshold_enabled: false,
  192. score_threshold: DATASET_DEFAULT.score_threshold,
  193. datasets: {
  194. datasets: [],
  195. },
  196. })
  197. const setModelConfig = (newModelConfig: ModelConfig) => {
  198. doSetModelConfig(newModelConfig)
  199. }
  200. const modelModeType = modelConfig.mode
  201. const modeModeTypeRef = useRef(modelModeType)
  202. useEffect(() => {
  203. modeModeTypeRef.current = modelModeType
  204. }, [modelModeType])
  205. const [dataSets, setDataSets] = useState<DataSet[]>([])
  206. const contextVar = modelConfig.configs.prompt_variables.find((item: any) => item.is_context_var)?.key
  207. const hasSetContextVar = !!contextVar
  208. const [isShowSelectDataSet, { setTrue: showSelectDataSet, setFalse: hideSelectDataSet }] = useBoolean(false)
  209. const selectedIds = dataSets.map(item => item.id)
  210. const [rerankSettingModalOpen, setRerankSettingModalOpen] = useState(false)
  211. const {
  212. currentModel: currentRerankModel,
  213. currentProvider: currentRerankProvider,
  214. } = useModelListAndDefaultModelAndCurrentProviderAndModel(ModelTypeEnum.rerank)
  215. const handleSelect = (data: DataSet[]) => {
  216. if (isEqual(data.map(item => item.id), dataSets.map(item => item.id))) {
  217. hideSelectDataSet()
  218. return
  219. }
  220. formattingChangedDispatcher()
  221. let newDatasets = data
  222. if (data.find(item => !item.name)) { // has not loaded selected dataset
  223. const newSelected = produce(data, (draft: any) => {
  224. data.forEach((item, index) => {
  225. if (!item.name) { // not fetched database
  226. const newItem = dataSets.find(i => i.id === item.id)
  227. if (newItem)
  228. draft[index] = newItem
  229. }
  230. })
  231. })
  232. setDataSets(newSelected)
  233. newDatasets = newSelected
  234. }
  235. else {
  236. setDataSets(data)
  237. }
  238. hideSelectDataSet()
  239. const {
  240. allExternal,
  241. allInternal,
  242. mixtureInternalAndExternal,
  243. mixtureHighQualityAndEconomic,
  244. inconsistentEmbeddingModel,
  245. } = getSelectedDatasetsMode(newDatasets)
  246. if (
  247. (allInternal && (mixtureHighQualityAndEconomic || inconsistentEmbeddingModel))
  248. || mixtureInternalAndExternal
  249. || allExternal
  250. )
  251. setRerankSettingModalOpen(true)
  252. const { datasets, retrieval_model, score_threshold_enabled, ...restConfigs } = datasetConfigs
  253. const retrievalConfig = getMultipleRetrievalConfig({
  254. top_k: restConfigs.top_k,
  255. score_threshold: restConfigs.score_threshold,
  256. reranking_model: restConfigs.reranking_model && {
  257. provider: restConfigs.reranking_model.reranking_provider_name,
  258. model: restConfigs.reranking_model.reranking_model_name,
  259. },
  260. reranking_mode: restConfigs.reranking_mode,
  261. weights: restConfigs.weights,
  262. reranking_enable: restConfigs.reranking_enable,
  263. }, newDatasets, dataSets, {
  264. provider: currentRerankProvider?.provider,
  265. model: currentRerankModel?.model,
  266. })
  267. setDatasetConfigs({
  268. ...retrievalConfig,
  269. reranking_model: {
  270. reranking_provider_name: retrievalConfig?.reranking_model?.provider || '',
  271. reranking_model_name: retrievalConfig?.reranking_model?.model || '',
  272. },
  273. retrieval_model,
  274. score_threshold_enabled,
  275. datasets,
  276. })
  277. }
  278. const [isShowHistoryModal, { setTrue: showHistoryModal, setFalse: hideHistoryModal }] = useBoolean(false)
  279. const syncToPublishedConfig = (_publishedConfig: PublishConfig) => {
  280. const modelConfig = _publishedConfig.modelConfig
  281. setModelConfig(_publishedConfig.modelConfig)
  282. setCompletionParams(_publishedConfig.completionParams)
  283. setDataSets(modelConfig.dataSets || [])
  284. // reset feature
  285. setIntroduction(modelConfig.opening_statement!)
  286. setMoreLikeThisConfig(modelConfig.more_like_this || {
  287. enabled: false,
  288. })
  289. setSuggestedQuestionsAfterAnswerConfig(modelConfig.suggested_questions_after_answer || {
  290. enabled: false,
  291. })
  292. setSpeechToTextConfig(modelConfig.speech_to_text || {
  293. enabled: false,
  294. })
  295. setTextToSpeechConfig(modelConfig.text_to_speech || {
  296. enabled: false,
  297. voice: '',
  298. language: '',
  299. })
  300. setCitationConfig(modelConfig.retriever_resource || {
  301. enabled: false,
  302. })
  303. }
  304. const { isAPIKeySet } = useProviderContext()
  305. const {
  306. currentModel: currModel,
  307. textGenerationModelList,
  308. } = useTextGenerationCurrentProviderAndModelAndModelList(
  309. {
  310. provider: modelConfig.provider,
  311. model: modelConfig.model_id,
  312. },
  313. )
  314. const isFunctionCall = (() => {
  315. const features = currModel?.features
  316. if (!features)
  317. return false
  318. return features.includes(ModelFeatureEnum.toolCall) || features.includes(ModelFeatureEnum.multiToolCall)
  319. })()
  320. // Fill old app data missing model mode.
  321. useEffect(() => {
  322. if (hasFetchedDetail && !modelModeType) {
  323. const mode = currModel?.model_properties.mode as (ModelModeType | undefined)
  324. if (mode) {
  325. const newModelConfig = produce(modelConfig, (draft: ModelConfig) => {
  326. draft.mode = mode
  327. })
  328. setModelConfig(newModelConfig)
  329. }
  330. }
  331. }, [textGenerationModelList, hasFetchedDetail, modelModeType, currModel, modelConfig])
  332. const [promptMode, doSetPromptMode] = useState(PromptMode.simple)
  333. const isAdvancedMode = promptMode === PromptMode.advanced
  334. const [canReturnToSimpleMode, setCanReturnToSimpleMode] = useState(true)
  335. const setPromptMode = async (mode: PromptMode) => {
  336. if (mode === PromptMode.advanced) {
  337. // eslint-disable-next-line @typescript-eslint/no-use-before-define
  338. await migrateToDefaultPrompt()
  339. setCanReturnToSimpleMode(true)
  340. }
  341. doSetPromptMode(mode)
  342. }
  343. const [visionConfig, doSetVisionConfig] = useState({
  344. enabled: false,
  345. number_limits: 2,
  346. detail: Resolution.low,
  347. transfer_methods: [TransferMethod.local_file],
  348. })
  349. const handleSetVisionConfig = (config: VisionSettings, notNoticeFormattingChanged?: boolean) => {
  350. doSetVisionConfig({
  351. enabled: config.enabled || false,
  352. number_limits: config.number_limits || 2,
  353. detail: config.detail || Resolution.low,
  354. transfer_methods: config.transfer_methods || [TransferMethod.local_file],
  355. })
  356. if (!notNoticeFormattingChanged)
  357. formattingChangedDispatcher()
  358. }
  359. const {
  360. chatPromptConfig,
  361. setChatPromptConfig,
  362. completionPromptConfig,
  363. setCompletionPromptConfig,
  364. currentAdvancedPrompt,
  365. setCurrentAdvancedPrompt,
  366. hasSetBlockStatus,
  367. setConversationHistoriesRole,
  368. migrateToDefaultPrompt,
  369. } = useAdvancedPromptConfig({
  370. appMode: mode,
  371. modelName: modelConfig.model_id,
  372. promptMode,
  373. modelModeType,
  374. prePrompt: modelConfig.configs.prompt_template,
  375. hasSetDataSet: dataSets.length > 0,
  376. onUserChangedPrompt: () => {
  377. setCanReturnToSimpleMode(false)
  378. },
  379. completionParams,
  380. setCompletionParams,
  381. setStop: setTempStop,
  382. })
  383. const setModel = async ({
  384. modelId,
  385. provider,
  386. mode: modeMode,
  387. features,
  388. }: { modelId: string; provider: string; mode: string; features: string[] }) => {
  389. if (isAdvancedMode) {
  390. const appMode = mode
  391. if (modeMode === ModelModeType.completion) {
  392. if (appMode !== AppType.completion) {
  393. if (!completionPromptConfig.prompt?.text || !completionPromptConfig.conversation_histories_role.assistant_prefix || !completionPromptConfig.conversation_histories_role.user_prefix)
  394. await migrateToDefaultPrompt(true, ModelModeType.completion)
  395. }
  396. else {
  397. if (!completionPromptConfig.prompt?.text)
  398. await migrateToDefaultPrompt(true, ModelModeType.completion)
  399. }
  400. }
  401. if (modeMode === ModelModeType.chat) {
  402. if (chatPromptConfig.prompt.length === 0)
  403. await migrateToDefaultPrompt(true, ModelModeType.chat)
  404. }
  405. }
  406. const newModelConfig = produce(modelConfig, (draft: ModelConfig) => {
  407. draft.provider = provider
  408. draft.model_id = modelId
  409. draft.mode = modeMode as ModelModeType
  410. })
  411. setModelConfig(newModelConfig)
  412. const supportVision = features && features.includes(ModelFeatureEnum.vision)
  413. handleSetVisionConfig({
  414. ...visionConfig,
  415. enabled: supportVision,
  416. }, true)
  417. setCompletionParams({})
  418. }
  419. const isShowVisionConfig = !!currModel?.features?.includes(ModelFeatureEnum.vision)
  420. const isShowDocumentConfig = !!currModel?.features?.includes(ModelFeatureEnum.document)
  421. const isAllowVideoUpload = !!currModel?.features?.includes(ModelFeatureEnum.video)
  422. // *** web app features ***
  423. const featuresData: FeaturesData = useMemo(() => {
  424. return {
  425. moreLikeThis: modelConfig.more_like_this || { enabled: false },
  426. opening: {
  427. enabled: !!modelConfig.opening_statement,
  428. opening_statement: modelConfig.opening_statement || '',
  429. suggested_questions: modelConfig.suggested_questions || [],
  430. },
  431. moderation: modelConfig.sensitive_word_avoidance || { enabled: false },
  432. speech2text: modelConfig.speech_to_text || { enabled: false },
  433. text2speech: modelConfig.text_to_speech || { enabled: false },
  434. file: {
  435. image: {
  436. detail: modelConfig.file_upload?.image?.detail || Resolution.high,
  437. enabled: !!modelConfig.file_upload?.image?.enabled,
  438. number_limits: modelConfig.file_upload?.image?.number_limits || 3,
  439. transfer_methods: modelConfig.file_upload?.image?.transfer_methods || ['local_file', 'remote_url'],
  440. },
  441. enabled: !!(modelConfig.file_upload?.enabled || modelConfig.file_upload?.image?.enabled),
  442. allowed_file_types: modelConfig.file_upload?.allowed_file_types || [],
  443. allowed_file_extensions: modelConfig.file_upload?.allowed_file_extensions || [...FILE_EXTS[SupportUploadFileTypes.image], ...FILE_EXTS[SupportUploadFileTypes.video]].map(ext => `.${ext}`),
  444. allowed_file_upload_methods: modelConfig.file_upload?.allowed_file_upload_methods || modelConfig.file_upload?.image?.transfer_methods || ['local_file', 'remote_url'],
  445. number_limits: modelConfig.file_upload?.number_limits || modelConfig.file_upload?.image?.number_limits || 3,
  446. fileUploadConfig: fileUploadConfigResponse,
  447. } as FileUpload,
  448. suggested: modelConfig.suggested_questions_after_answer || { enabled: false },
  449. citation: modelConfig.retriever_resource || { enabled: false },
  450. annotationReply: modelConfig.annotation_reply || { enabled: false },
  451. }
  452. }, [fileUploadConfigResponse, modelConfig])
  453. const handleFeaturesChange = useCallback((flag: any) => {
  454. setShowAppConfigureFeaturesModal(true)
  455. if (flag)
  456. formattingChangedDispatcher()
  457. }, [formattingChangedDispatcher, setShowAppConfigureFeaturesModal])
  458. const handleAddPromptVariable = useCallback((variable: PromptVariable[]) => {
  459. const newModelConfig = produce(modelConfig, (draft: ModelConfig) => {
  460. draft.configs.prompt_variables = [...draft.configs.prompt_variables, ...variable]
  461. })
  462. setModelConfig(newModelConfig)
  463. }, [modelConfig])
  464. useEffect(() => {
  465. (async () => {
  466. const collectionList = await fetchCollectionList()
  467. setCollectionList(collectionList)
  468. fetchAppDetail({ url: '/apps', id: appId }).then(async (res: any) => {
  469. setMode(res.mode)
  470. const modelConfig = res.model_config
  471. const promptMode = modelConfig.prompt_type === PromptMode.advanced ? PromptMode.advanced : PromptMode.simple
  472. doSetPromptMode(promptMode)
  473. if (promptMode === PromptMode.advanced) {
  474. if (modelConfig.chat_prompt_config && modelConfig.chat_prompt_config.prompt.length > 0)
  475. setChatPromptConfig(modelConfig.chat_prompt_config)
  476. else
  477. setChatPromptConfig(clone(DEFAULT_CHAT_PROMPT_CONFIG) as any)
  478. setCompletionPromptConfig(modelConfig.completion_prompt_config || clone(DEFAULT_COMPLETION_PROMPT_CONFIG) as any)
  479. setCanReturnToSimpleMode(false)
  480. }
  481. const model = res.model_config.model
  482. let datasets: any = null
  483. // old dataset struct
  484. if (modelConfig.agent_mode?.tools?.find(({ dataset }: any) => dataset?.enabled))
  485. datasets = modelConfig.agent_mode?.tools.filter(({ dataset }: any) => dataset?.enabled)
  486. // new dataset struct
  487. else if (modelConfig.dataset_configs.datasets?.datasets?.length > 0)
  488. datasets = modelConfig.dataset_configs?.datasets?.datasets
  489. if (dataSets && datasets?.length && datasets?.length > 0) {
  490. const { data: dataSetsWithDetail } = await fetchDatasets({ url: '/datasets', params: { page: 1, ids: datasets.map(({ dataset }: any) => dataset.id) } })
  491. datasets = dataSetsWithDetail
  492. setDataSets(datasets)
  493. }
  494. setIntroduction(modelConfig.opening_statement)
  495. setSuggestedQuestions(modelConfig.suggested_questions || [])
  496. if (modelConfig.more_like_this)
  497. setMoreLikeThisConfig(modelConfig.more_like_this)
  498. if (modelConfig.suggested_questions_after_answer)
  499. setSuggestedQuestionsAfterAnswerConfig(modelConfig.suggested_questions_after_answer)
  500. if (modelConfig.speech_to_text)
  501. setSpeechToTextConfig(modelConfig.speech_to_text)
  502. if (modelConfig.text_to_speech)
  503. setTextToSpeechConfig(modelConfig.text_to_speech)
  504. if (modelConfig.retriever_resource)
  505. setCitationConfig(modelConfig.retriever_resource)
  506. if (modelConfig.annotation_reply)
  507. setAnnotationConfig(modelConfig.annotation_reply, true)
  508. if (modelConfig.sensitive_word_avoidance)
  509. setModerationConfig(modelConfig.sensitive_word_avoidance)
  510. if (modelConfig.external_data_tools)
  511. setExternalDataToolsConfig(modelConfig.external_data_tools)
  512. const config = {
  513. modelConfig: {
  514. provider: model.provider,
  515. model_id: model.name,
  516. mode: model.mode,
  517. configs: {
  518. prompt_template: modelConfig.pre_prompt || '',
  519. prompt_variables: userInputsFormToPromptVariables(
  520. [
  521. ...modelConfig.user_input_form,
  522. ...(
  523. modelConfig.external_data_tools?.length
  524. ? modelConfig.external_data_tools.map((item: any) => {
  525. return {
  526. external_data_tool: {
  527. variable: item.variable as string,
  528. label: item.label as string,
  529. enabled: item.enabled,
  530. type: item.type as string,
  531. config: item.config,
  532. required: true,
  533. icon: item.icon,
  534. icon_background: item.icon_background,
  535. },
  536. }
  537. })
  538. : []
  539. ),
  540. ],
  541. modelConfig.dataset_query_variable,
  542. ),
  543. },
  544. more_like_this: modelConfig.more_like_this,
  545. opening_statement: modelConfig.opening_statement,
  546. suggested_questions: modelConfig.suggested_questions,
  547. sensitive_word_avoidance: modelConfig.sensitive_word_avoidance,
  548. speech_to_text: modelConfig.speech_to_text,
  549. text_to_speech: modelConfig.text_to_speech,
  550. file_upload: modelConfig.file_upload,
  551. suggested_questions_after_answer: modelConfig.suggested_questions_after_answer,
  552. retriever_resource: modelConfig.retriever_resource,
  553. annotation_reply: modelConfig.annotation_reply,
  554. external_data_tools: modelConfig.external_data_tools,
  555. dataSets: datasets || [],
  556. // eslint-disable-next-line multiline-ternary
  557. agentConfig: res.mode === 'agent-chat' ? {
  558. max_iteration: DEFAULT_AGENT_SETTING.max_iteration,
  559. ...modelConfig.agent_mode,
  560. // remove dataset
  561. enabled: true, // modelConfig.agent_mode?.enabled is not correct. old app: the value of app with dataset's is always true
  562. tools: modelConfig.agent_mode?.tools.filter((tool: any) => {
  563. return !tool.dataset
  564. }).map((tool: any) => {
  565. return {
  566. ...tool,
  567. isDeleted: res.deleted_tools?.includes(tool.tool_name),
  568. notAuthor: collectionList.find(c => tool.provider_id === c.id)?.is_team_authorization === false,
  569. }
  570. }),
  571. } : DEFAULT_AGENT_SETTING,
  572. },
  573. completionParams: model.completion_params,
  574. }
  575. if (modelConfig.file_upload)
  576. handleSetVisionConfig(modelConfig.file_upload.image, true)
  577. syncToPublishedConfig(config)
  578. setPublishedConfig(config)
  579. const retrievalConfig = getMultipleRetrievalConfig(modelConfig.dataset_configs, datasets, datasets, {
  580. provider: currentRerankProvider?.provider,
  581. model: currentRerankModel?.model,
  582. })
  583. setDatasetConfigs({
  584. retrieval_model: RETRIEVE_TYPE.multiWay,
  585. ...modelConfig.dataset_configs,
  586. ...retrievalConfig,
  587. })
  588. setHasFetchedDetail(true)
  589. })
  590. })()
  591. // eslint-disable-next-line react-hooks/exhaustive-deps
  592. }, [appId])
  593. const promptEmpty = (() => {
  594. if (mode !== AppType.completion)
  595. return false
  596. if (isAdvancedMode) {
  597. if (modelModeType === ModelModeType.chat)
  598. return chatPromptConfig.prompt.every(({ text }: any) => !text)
  599. else
  600. return !completionPromptConfig.prompt?.text
  601. }
  602. else { return !modelConfig.configs.prompt_template }
  603. })()
  604. const cannotPublish = (() => {
  605. if (mode !== AppType.completion) {
  606. if (!isAdvancedMode)
  607. return false
  608. if (modelModeType === ModelModeType.completion) {
  609. if (!hasSetBlockStatus.history || !hasSetBlockStatus.query)
  610. return true
  611. return false
  612. }
  613. return false
  614. }
  615. else { return promptEmpty }
  616. })()
  617. const contextVarEmpty = mode === AppType.completion && dataSets.length > 0 && !hasSetContextVar
  618. const onPublish = async (modelAndParameter?: ModelAndParameter, features?: FeaturesData) => {
  619. const modelId = modelAndParameter?.model || modelConfig.model_id
  620. const promptTemplate = modelConfig.configs.prompt_template
  621. const promptVariables = modelConfig.configs.prompt_variables
  622. if (promptEmpty) {
  623. notify({ type: 'error', message: t('appDebug.otherError.promptNoBeEmpty') })
  624. return
  625. }
  626. if (isAdvancedMode && mode !== AppType.completion) {
  627. if (modelModeType === ModelModeType.completion) {
  628. if (!hasSetBlockStatus.history) {
  629. notify({ type: 'error', message: t('appDebug.otherError.historyNoBeEmpty') })
  630. return
  631. }
  632. if (!hasSetBlockStatus.query) {
  633. notify({ type: 'error', message: t('appDebug.otherError.queryNoBeEmpty') })
  634. return
  635. }
  636. }
  637. }
  638. if (contextVarEmpty) {
  639. notify({ type: 'error', message: t('appDebug.feature.dataSet.queryVariable.contextVarNotEmpty') })
  640. return
  641. }
  642. const postDatasets = dataSets.map(({ id }) => ({
  643. dataset: {
  644. enabled: true,
  645. id,
  646. },
  647. }))
  648. const fileUpload = { ...features?.file }
  649. delete fileUpload?.fileUploadConfig
  650. // new model config data struct
  651. const data: BackendModelConfig = {
  652. // Simple Mode prompt
  653. pre_prompt: !isAdvancedMode ? promptTemplate : '',
  654. prompt_type: promptMode,
  655. chat_prompt_config: {},
  656. completion_prompt_config: {},
  657. user_input_form: promptVariablesToUserInputsForm(promptVariables),
  658. dataset_query_variable: contextVar || '',
  659. // features
  660. more_like_this: features?.moreLikeThis as any,
  661. opening_statement: features?.opening?.enabled ? (features.opening?.opening_statement || '') : '',
  662. suggested_questions: features?.opening?.enabled ? (features.opening?.suggested_questions || []) : [],
  663. sensitive_word_avoidance: features?.moderation as any,
  664. speech_to_text: features?.speech2text as any,
  665. text_to_speech: features?.text2speech as any,
  666. file_upload: fileUpload as any,
  667. suggested_questions_after_answer: features?.suggested as any,
  668. retriever_resource: features?.citation as any,
  669. agent_mode: {
  670. ...modelConfig.agentConfig,
  671. strategy: isFunctionCall ? AgentStrategy.functionCall : AgentStrategy.react,
  672. },
  673. model: {
  674. provider: modelAndParameter?.provider || modelConfig.provider,
  675. name: modelId,
  676. mode: modelConfig.mode,
  677. completion_params: modelAndParameter?.parameters || completionParams as any,
  678. },
  679. dataset_configs: {
  680. ...datasetConfigs,
  681. datasets: {
  682. datasets: [...postDatasets],
  683. } as any,
  684. },
  685. }
  686. if (isAdvancedMode) {
  687. data.chat_prompt_config = chatPromptConfig
  688. data.completion_prompt_config = completionPromptConfig
  689. }
  690. await updateAppModelConfig({ url: `/apps/${appId}/model-config`, body: data })
  691. const newModelConfig = produce(modelConfig, (draft: any) => {
  692. draft.opening_statement = introduction
  693. draft.more_like_this = moreLikeThisConfig
  694. draft.suggested_questions_after_answer = suggestedQuestionsAfterAnswerConfig
  695. draft.speech_to_text = speechToTextConfig
  696. draft.text_to_speech = textToSpeechConfig
  697. draft.retriever_resource = citationConfig
  698. draft.dataSets = dataSets
  699. })
  700. setPublishedConfig({
  701. modelConfig: newModelConfig,
  702. completionParams,
  703. })
  704. notify({ type: 'success', message: t('common.api.success') })
  705. setCanReturnToSimpleMode(false)
  706. return true
  707. }
  708. const [showUseGPT4Confirm, setShowUseGPT4Confirm] = useState(false)
  709. const {
  710. debugWithMultipleModel,
  711. multipleModelConfigs,
  712. handleMultipleModelConfigsChange,
  713. } = useDebugWithSingleOrMultipleModel(appId)
  714. const handleDebugWithMultipleModelChange = () => {
  715. handleMultipleModelConfigsChange(
  716. true,
  717. [
  718. { id: `${Date.now()}`, model: modelConfig.model_id, provider: modelConfig.provider, parameters: completionParams },
  719. { id: `${Date.now()}-no-repeat`, model: '', provider: '', parameters: {} },
  720. ],
  721. )
  722. setAppSiderbarExpand('collapse')
  723. }
  724. if (isLoading) {
  725. return <div className='flex items-center justify-center h-full'>
  726. <Loading type='area' />
  727. </div>
  728. }
  729. return (
  730. <ConfigContext.Provider value={{
  731. appId,
  732. isAPIKeySet,
  733. isTrailFinished: false,
  734. mode,
  735. modelModeType,
  736. promptMode,
  737. isAdvancedMode,
  738. isAgent,
  739. isOpenAI,
  740. isFunctionCall,
  741. collectionList,
  742. setPromptMode,
  743. canReturnToSimpleMode,
  744. setCanReturnToSimpleMode,
  745. chatPromptConfig,
  746. completionPromptConfig,
  747. currentAdvancedPrompt,
  748. setCurrentAdvancedPrompt,
  749. conversationHistoriesRole: completionPromptConfig.conversation_histories_role,
  750. showHistoryModal,
  751. setConversationHistoriesRole,
  752. hasSetBlockStatus,
  753. conversationId,
  754. introduction,
  755. setIntroduction,
  756. suggestedQuestions,
  757. setSuggestedQuestions,
  758. setConversationId,
  759. controlClearChatMessage,
  760. setControlClearChatMessage,
  761. prevPromptConfig,
  762. setPrevPromptConfig,
  763. moreLikeThisConfig,
  764. setMoreLikeThisConfig,
  765. suggestedQuestionsAfterAnswerConfig,
  766. setSuggestedQuestionsAfterAnswerConfig,
  767. speechToTextConfig,
  768. setSpeechToTextConfig,
  769. textToSpeechConfig,
  770. setTextToSpeechConfig,
  771. citationConfig,
  772. setCitationConfig,
  773. annotationConfig,
  774. setAnnotationConfig,
  775. moderationConfig,
  776. setModerationConfig,
  777. externalDataToolsConfig,
  778. setExternalDataToolsConfig,
  779. formattingChanged,
  780. setFormattingChanged,
  781. inputs,
  782. setInputs,
  783. query,
  784. setQuery,
  785. completionParams,
  786. setCompletionParams,
  787. modelConfig,
  788. setModelConfig,
  789. showSelectDataSet,
  790. dataSets,
  791. setDataSets,
  792. datasetConfigs,
  793. setDatasetConfigs,
  794. hasSetContextVar,
  795. isShowVisionConfig,
  796. visionConfig,
  797. setVisionConfig: handleSetVisionConfig,
  798. isAllowVideoUpload,
  799. isShowDocumentConfig,
  800. rerankSettingModalOpen,
  801. setRerankSettingModalOpen,
  802. }}
  803. >
  804. <FeaturesProvider features={featuresData}>
  805. <>
  806. <div className="flex flex-col h-full">
  807. <div className='relative flex grow h-[200px] pt-14'>
  808. {/* Header */}
  809. <div className='absolute top-0 left-0 w-full bg-white h-14'>
  810. <div className='flex items-center justify-between px-6 h-14'>
  811. <div className='flex items-center'>
  812. <div className='text-base font-semibold leading-6 text-gray-900'>{t('appDebug.orchestrate')}</div>
  813. <div className='flex items-center h-[14px] space-x-1 text-xs'>
  814. {isAdvancedMode && (
  815. <div className='ml-1 flex items-center h-5 px-1.5 border border-gray-100 rounded-md text-[11px] font-medium text-gray-500 uppercase'>{t('appDebug.promptMode.advanced')}</div>
  816. )}
  817. </div>
  818. </div>
  819. <div className='flex items-center'>
  820. {/* Agent Setting */}
  821. {isAgent && (
  822. <AgentSettingButton
  823. isChatModel={modelConfig.mode === ModelModeType.chat}
  824. agentConfig={modelConfig.agentConfig}
  825. isFunctionCall={isFunctionCall}
  826. onAgentSettingChange={(config) => {
  827. const nextConfig = produce(modelConfig, (draft: ModelConfig) => {
  828. draft.agentConfig = config
  829. })
  830. setModelConfig(nextConfig)
  831. }}
  832. />
  833. )}
  834. {/* Model and Parameters */}
  835. {!debugWithMultipleModel && (
  836. <>
  837. <ModelParameterModal
  838. isAdvancedMode={isAdvancedMode}
  839. mode={mode}
  840. provider={modelConfig.provider}
  841. completionParams={completionParams}
  842. modelId={modelConfig.model_id}
  843. setModel={setModel as any}
  844. onCompletionParamsChange={(newParams: FormValue) => {
  845. setCompletionParams(newParams)
  846. }}
  847. debugWithMultipleModel={debugWithMultipleModel}
  848. onDebugWithMultipleModelChange={handleDebugWithMultipleModelChange}
  849. />
  850. <div className='mx-2 w-[1px] h-[14px] bg-gray-200'></div>
  851. </>
  852. )}
  853. {isMobile && (
  854. <Button className='!h-8 !text-[13px] font-medium' onClick={showDebugPanel}>
  855. <span className='mr-1'>{t('appDebug.operation.debugConfig')}</span>
  856. <CodeBracketIcon className="w-4 h-4 text-gray-500" />
  857. </Button>
  858. )}
  859. <AppPublisher {...{
  860. publishDisabled: cannotPublish,
  861. publishedAt: (latestPublishedAt || 0) * 1000,
  862. debugWithMultipleModel,
  863. multipleModelConfigs,
  864. onPublish,
  865. publishedConfig: publishedConfig!,
  866. resetAppConfig: () => syncToPublishedConfig(publishedConfig!),
  867. }} />
  868. </div>
  869. </div>
  870. </div>
  871. <div className={`w-full sm:w-1/2 shrink-0 flex flex-col h-full ${debugWithMultipleModel && 'max-w-[560px]'}`}>
  872. <Config />
  873. </div>
  874. {!isMobile && <div className="relative flex flex-col w-1/2 h-full overflow-y-auto grow " style={{ borderColor: 'rgba(0, 0, 0, 0.02)' }}>
  875. <div className='grow flex flex-col border-t-[0.5px] border-l-[0.5px] rounded-tl-2xl border-components-panel-border bg-chatbot-bg '>
  876. <Debug
  877. isAPIKeySet={isAPIKeySet}
  878. onSetting={() => setShowAccountSettingModal({ payload: 'provider' })}
  879. inputs={inputs}
  880. modelParameterParams={{
  881. setModel: setModel as any,
  882. onCompletionParamsChange: setCompletionParams,
  883. }}
  884. debugWithMultipleModel={debugWithMultipleModel}
  885. multipleModelConfigs={multipleModelConfigs}
  886. onMultipleModelConfigsChange={handleMultipleModelConfigsChange}
  887. />
  888. </div>
  889. </div>}
  890. </div>
  891. </div>
  892. {showUseGPT4Confirm && (
  893. <Confirm
  894. title={t('appDebug.trailUseGPT4Info.title')}
  895. content={t('appDebug.trailUseGPT4Info.description')}
  896. isShow={showUseGPT4Confirm}
  897. onConfirm={() => {
  898. setShowAccountSettingModal({ payload: 'provider' })
  899. setShowUseGPT4Confirm(false)
  900. }}
  901. onCancel={() => setShowUseGPT4Confirm(false)}
  902. />
  903. )}
  904. {isShowSelectDataSet && (
  905. <SelectDataSet
  906. isShow={isShowSelectDataSet}
  907. onClose={hideSelectDataSet}
  908. selectedIds={selectedIds}
  909. onSelect={handleSelect}
  910. />
  911. )}
  912. {isShowHistoryModal && (
  913. <EditHistoryModal
  914. isShow={isShowHistoryModal}
  915. saveLoading={false}
  916. onClose={hideHistoryModal}
  917. data={completionPromptConfig.conversation_histories_role}
  918. onSave={(data) => {
  919. setConversationHistoriesRole(data)
  920. hideHistoryModal()
  921. }}
  922. />
  923. )}
  924. {isMobile && (
  925. <Drawer showClose isOpen={isShowDebugPanel} onClose={hideDebugPanel} mask footer={null} panelClassname='!bg-gray-50'>
  926. <Debug
  927. isAPIKeySet={isAPIKeySet}
  928. onSetting={() => setShowAccountSettingModal({ payload: 'provider' })}
  929. inputs={inputs}
  930. modelParameterParams={{
  931. setModel: setModel as any,
  932. onCompletionParamsChange: setCompletionParams,
  933. }}
  934. debugWithMultipleModel={debugWithMultipleModel}
  935. multipleModelConfigs={multipleModelConfigs}
  936. onMultipleModelConfigsChange={handleMultipleModelConfigsChange}
  937. />
  938. </Drawer>
  939. )}
  940. {showAppConfigureFeaturesModal && (
  941. <NewFeaturePanel
  942. show
  943. inWorkflow={false}
  944. showFileUpload={false}
  945. isChatMode={mode !== 'completion'}
  946. disabled={false}
  947. onChange={handleFeaturesChange}
  948. onClose={() => setShowAppConfigureFeaturesModal(false)}
  949. promptVariables={modelConfig.configs.prompt_variables}
  950. onAutoAddPromptVariable={handleAddPromptVariable}
  951. />
  952. )}
  953. </>
  954. </FeaturesProvider>
  955. </ConfigContext.Provider>
  956. )
  957. }
  958. export default React.memo(Configuration)