You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

index.tsx 39KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018
  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. } = useModelListAndDefaultModelAndCurrentProviderAndModel(ModelTypeEnum.rerank)
  214. const handleSelect = (data: DataSet[]) => {
  215. if (isEqual(data.map(item => item.id), dataSets.map(item => item.id))) {
  216. hideSelectDataSet()
  217. return
  218. }
  219. formattingChangedDispatcher()
  220. let newDatasets = data
  221. if (data.find(item => !item.name)) { // has not loaded selected dataset
  222. const newSelected = produce(data, (draft: any) => {
  223. data.forEach((item, index) => {
  224. if (!item.name) { // not fetched database
  225. const newItem = dataSets.find(i => i.id === item.id)
  226. if (newItem)
  227. draft[index] = newItem
  228. }
  229. })
  230. })
  231. setDataSets(newSelected)
  232. newDatasets = newSelected
  233. }
  234. else {
  235. setDataSets(data)
  236. }
  237. hideSelectDataSet()
  238. const {
  239. allExternal,
  240. allInternal,
  241. mixtureInternalAndExternal,
  242. mixtureHighQualityAndEconomic,
  243. inconsistentEmbeddingModel,
  244. } = getSelectedDatasetsMode(newDatasets)
  245. if (
  246. (allInternal && (mixtureHighQualityAndEconomic || inconsistentEmbeddingModel))
  247. || mixtureInternalAndExternal
  248. || allExternal
  249. )
  250. setRerankSettingModalOpen(true)
  251. const { datasets, retrieval_model, score_threshold_enabled, ...restConfigs } = datasetConfigs
  252. const retrievalConfig = getMultipleRetrievalConfig({
  253. top_k: restConfigs.top_k,
  254. score_threshold: restConfigs.score_threshold,
  255. reranking_model: restConfigs.reranking_model && {
  256. provider: restConfigs.reranking_model.reranking_provider_name,
  257. model: restConfigs.reranking_model.reranking_model_name,
  258. },
  259. reranking_mode: restConfigs.reranking_mode,
  260. weights: restConfigs.weights,
  261. reranking_enable: restConfigs.reranking_enable,
  262. }, newDatasets, dataSets, !!currentRerankModel)
  263. setDatasetConfigs({
  264. ...retrievalConfig,
  265. reranking_model: restConfigs.reranking_model && {
  266. reranking_provider_name: restConfigs.reranking_model.reranking_provider_name,
  267. reranking_model_name: restConfigs.reranking_model.reranking_model_name,
  268. },
  269. retrieval_model,
  270. score_threshold_enabled,
  271. datasets,
  272. })
  273. }
  274. const [isShowHistoryModal, { setTrue: showHistoryModal, setFalse: hideHistoryModal }] = useBoolean(false)
  275. const syncToPublishedConfig = (_publishedConfig: PublishConfig) => {
  276. const modelConfig = _publishedConfig.modelConfig
  277. setModelConfig(_publishedConfig.modelConfig)
  278. setCompletionParams(_publishedConfig.completionParams)
  279. setDataSets(modelConfig.dataSets || [])
  280. // reset feature
  281. setIntroduction(modelConfig.opening_statement!)
  282. setMoreLikeThisConfig(modelConfig.more_like_this || {
  283. enabled: false,
  284. })
  285. setSuggestedQuestionsAfterAnswerConfig(modelConfig.suggested_questions_after_answer || {
  286. enabled: false,
  287. })
  288. setSpeechToTextConfig(modelConfig.speech_to_text || {
  289. enabled: false,
  290. })
  291. setTextToSpeechConfig(modelConfig.text_to_speech || {
  292. enabled: false,
  293. voice: '',
  294. language: '',
  295. })
  296. setCitationConfig(modelConfig.retriever_resource || {
  297. enabled: false,
  298. })
  299. }
  300. const { isAPIKeySet } = useProviderContext()
  301. const {
  302. currentModel: currModel,
  303. textGenerationModelList,
  304. } = useTextGenerationCurrentProviderAndModelAndModelList(
  305. {
  306. provider: modelConfig.provider,
  307. model: modelConfig.model_id,
  308. },
  309. )
  310. const isFunctionCall = (() => {
  311. const features = currModel?.features
  312. if (!features)
  313. return false
  314. return features.includes(ModelFeatureEnum.toolCall) || features.includes(ModelFeatureEnum.multiToolCall)
  315. })()
  316. // Fill old app data missing model mode.
  317. useEffect(() => {
  318. if (hasFetchedDetail && !modelModeType) {
  319. const mode = currModel?.model_properties.mode as (ModelModeType | undefined)
  320. if (mode) {
  321. const newModelConfig = produce(modelConfig, (draft: ModelConfig) => {
  322. draft.mode = mode
  323. })
  324. setModelConfig(newModelConfig)
  325. }
  326. }
  327. }, [textGenerationModelList, hasFetchedDetail, modelModeType, currModel, modelConfig])
  328. const [promptMode, doSetPromptMode] = useState(PromptMode.simple)
  329. const isAdvancedMode = promptMode === PromptMode.advanced
  330. const [canReturnToSimpleMode, setCanReturnToSimpleMode] = useState(true)
  331. const setPromptMode = async (mode: PromptMode) => {
  332. if (mode === PromptMode.advanced) {
  333. // eslint-disable-next-line @typescript-eslint/no-use-before-define
  334. await migrateToDefaultPrompt()
  335. setCanReturnToSimpleMode(true)
  336. }
  337. doSetPromptMode(mode)
  338. }
  339. const [visionConfig, doSetVisionConfig] = useState({
  340. enabled: false,
  341. number_limits: 2,
  342. detail: Resolution.low,
  343. transfer_methods: [TransferMethod.local_file],
  344. })
  345. const handleSetVisionConfig = (config: VisionSettings, notNoticeFormattingChanged?: boolean) => {
  346. doSetVisionConfig({
  347. enabled: config.enabled || false,
  348. number_limits: config.number_limits || 2,
  349. detail: config.detail || Resolution.low,
  350. transfer_methods: config.transfer_methods || [TransferMethod.local_file],
  351. })
  352. if (!notNoticeFormattingChanged)
  353. formattingChangedDispatcher()
  354. }
  355. const {
  356. chatPromptConfig,
  357. setChatPromptConfig,
  358. completionPromptConfig,
  359. setCompletionPromptConfig,
  360. currentAdvancedPrompt,
  361. setCurrentAdvancedPrompt,
  362. hasSetBlockStatus,
  363. setConversationHistoriesRole,
  364. migrateToDefaultPrompt,
  365. } = useAdvancedPromptConfig({
  366. appMode: mode,
  367. modelName: modelConfig.model_id,
  368. promptMode,
  369. modelModeType,
  370. prePrompt: modelConfig.configs.prompt_template,
  371. hasSetDataSet: dataSets.length > 0,
  372. onUserChangedPrompt: () => {
  373. setCanReturnToSimpleMode(false)
  374. },
  375. completionParams,
  376. setCompletionParams,
  377. setStop: setTempStop,
  378. })
  379. const setModel = async ({
  380. modelId,
  381. provider,
  382. mode: modeMode,
  383. features,
  384. }: { modelId: string; provider: string; mode: string; features: string[] }) => {
  385. if (isAdvancedMode) {
  386. const appMode = mode
  387. if (modeMode === ModelModeType.completion) {
  388. if (appMode !== AppType.completion) {
  389. if (!completionPromptConfig.prompt?.text || !completionPromptConfig.conversation_histories_role.assistant_prefix || !completionPromptConfig.conversation_histories_role.user_prefix)
  390. await migrateToDefaultPrompt(true, ModelModeType.completion)
  391. }
  392. else {
  393. if (!completionPromptConfig.prompt?.text)
  394. await migrateToDefaultPrompt(true, ModelModeType.completion)
  395. }
  396. }
  397. if (modeMode === ModelModeType.chat) {
  398. if (chatPromptConfig.prompt.length === 0)
  399. await migrateToDefaultPrompt(true, ModelModeType.chat)
  400. }
  401. }
  402. const newModelConfig = produce(modelConfig, (draft: ModelConfig) => {
  403. draft.provider = provider
  404. draft.model_id = modelId
  405. draft.mode = modeMode as ModelModeType
  406. })
  407. setModelConfig(newModelConfig)
  408. const supportVision = features && features.includes(ModelFeatureEnum.vision)
  409. handleSetVisionConfig({
  410. ...visionConfig,
  411. enabled: supportVision,
  412. }, true)
  413. setCompletionParams({})
  414. }
  415. const isShowVisionConfig = !!currModel?.features?.includes(ModelFeatureEnum.vision)
  416. // *** web app features ***
  417. const featuresData: FeaturesData = useMemo(() => {
  418. return {
  419. moreLikeThis: modelConfig.more_like_this || { enabled: false },
  420. opening: {
  421. enabled: !!modelConfig.opening_statement,
  422. opening_statement: modelConfig.opening_statement || '',
  423. suggested_questions: modelConfig.suggested_questions || [],
  424. },
  425. moderation: modelConfig.sensitive_word_avoidance || { enabled: false },
  426. speech2text: modelConfig.speech_to_text || { enabled: false },
  427. text2speech: modelConfig.text_to_speech || { enabled: false },
  428. file: {
  429. image: {
  430. detail: modelConfig.file_upload?.image?.detail || Resolution.high,
  431. enabled: !!modelConfig.file_upload?.image?.enabled,
  432. number_limits: modelConfig.file_upload?.image?.number_limits || 3,
  433. transfer_methods: modelConfig.file_upload?.image?.transfer_methods || ['local_file', 'remote_url'],
  434. },
  435. enabled: !!(modelConfig.file_upload?.enabled || modelConfig.file_upload?.image?.enabled),
  436. allowed_file_types: modelConfig.file_upload?.allowed_file_types || [SupportUploadFileTypes.image, SupportUploadFileTypes.video],
  437. allowed_file_extensions: modelConfig.file_upload?.allowed_file_extensions || [...FILE_EXTS[SupportUploadFileTypes.image], ...FILE_EXTS[SupportUploadFileTypes.video]].map(ext => `.${ext}`),
  438. allowed_file_upload_methods: modelConfig.file_upload?.allowed_file_upload_methods || modelConfig.file_upload?.image?.transfer_methods || ['local_file', 'remote_url'],
  439. number_limits: modelConfig.file_upload?.number_limits || modelConfig.file_upload?.image?.number_limits || 3,
  440. fileUploadConfig: fileUploadConfigResponse,
  441. } as FileUpload,
  442. suggested: modelConfig.suggested_questions_after_answer || { enabled: false },
  443. citation: modelConfig.retriever_resource || { enabled: false },
  444. annotationReply: modelConfig.annotation_reply || { enabled: false },
  445. }
  446. }, [fileUploadConfigResponse, modelConfig])
  447. const handleFeaturesChange = useCallback((flag: any) => {
  448. setShowAppConfigureFeaturesModal(true)
  449. if (flag)
  450. formattingChangedDispatcher()
  451. }, [formattingChangedDispatcher, setShowAppConfigureFeaturesModal])
  452. const handleAddPromptVariable = useCallback((variable: PromptVariable[]) => {
  453. const newModelConfig = produce(modelConfig, (draft: ModelConfig) => {
  454. draft.configs.prompt_variables = variable
  455. })
  456. setModelConfig(newModelConfig)
  457. }, [modelConfig])
  458. useEffect(() => {
  459. (async () => {
  460. const collectionList = await fetchCollectionList()
  461. setCollectionList(collectionList)
  462. fetchAppDetail({ url: '/apps', id: appId }).then(async (res: any) => {
  463. setMode(res.mode)
  464. const modelConfig = res.model_config
  465. const promptMode = modelConfig.prompt_type === PromptMode.advanced ? PromptMode.advanced : PromptMode.simple
  466. doSetPromptMode(promptMode)
  467. if (promptMode === PromptMode.advanced) {
  468. if (modelConfig.chat_prompt_config && modelConfig.chat_prompt_config.prompt.length > 0)
  469. setChatPromptConfig(modelConfig.chat_prompt_config)
  470. else
  471. setChatPromptConfig(clone(DEFAULT_CHAT_PROMPT_CONFIG) as any)
  472. setCompletionPromptConfig(modelConfig.completion_prompt_config || clone(DEFAULT_COMPLETION_PROMPT_CONFIG) as any)
  473. setCanReturnToSimpleMode(false)
  474. }
  475. const model = res.model_config.model
  476. let datasets: any = null
  477. // old dataset struct
  478. if (modelConfig.agent_mode?.tools?.find(({ dataset }: any) => dataset?.enabled))
  479. datasets = modelConfig.agent_mode?.tools.filter(({ dataset }: any) => dataset?.enabled)
  480. // new dataset struct
  481. else if (modelConfig.dataset_configs.datasets?.datasets?.length > 0)
  482. datasets = modelConfig.dataset_configs?.datasets?.datasets
  483. if (dataSets && datasets?.length && datasets?.length > 0) {
  484. const { data: dataSetsWithDetail } = await fetchDatasets({ url: '/datasets', params: { page: 1, ids: datasets.map(({ dataset }: any) => dataset.id) } })
  485. datasets = dataSetsWithDetail
  486. setDataSets(datasets)
  487. }
  488. setIntroduction(modelConfig.opening_statement)
  489. setSuggestedQuestions(modelConfig.suggested_questions || [])
  490. if (modelConfig.more_like_this)
  491. setMoreLikeThisConfig(modelConfig.more_like_this)
  492. if (modelConfig.suggested_questions_after_answer)
  493. setSuggestedQuestionsAfterAnswerConfig(modelConfig.suggested_questions_after_answer)
  494. if (modelConfig.speech_to_text)
  495. setSpeechToTextConfig(modelConfig.speech_to_text)
  496. if (modelConfig.text_to_speech)
  497. setTextToSpeechConfig(modelConfig.text_to_speech)
  498. if (modelConfig.retriever_resource)
  499. setCitationConfig(modelConfig.retriever_resource)
  500. if (modelConfig.annotation_reply)
  501. setAnnotationConfig(modelConfig.annotation_reply, true)
  502. if (modelConfig.sensitive_word_avoidance)
  503. setModerationConfig(modelConfig.sensitive_word_avoidance)
  504. if (modelConfig.external_data_tools)
  505. setExternalDataToolsConfig(modelConfig.external_data_tools)
  506. const config = {
  507. modelConfig: {
  508. provider: model.provider,
  509. model_id: model.name,
  510. mode: model.mode,
  511. configs: {
  512. prompt_template: modelConfig.pre_prompt || '',
  513. prompt_variables: userInputsFormToPromptVariables(
  514. [
  515. ...modelConfig.user_input_form,
  516. ...(
  517. modelConfig.external_data_tools?.length
  518. ? modelConfig.external_data_tools.map((item: any) => {
  519. return {
  520. external_data_tool: {
  521. variable: item.variable as string,
  522. label: item.label as string,
  523. enabled: item.enabled,
  524. type: item.type as string,
  525. config: item.config,
  526. required: true,
  527. icon: item.icon,
  528. icon_background: item.icon_background,
  529. },
  530. }
  531. })
  532. : []
  533. ),
  534. ],
  535. modelConfig.dataset_query_variable,
  536. ),
  537. },
  538. more_like_this: modelConfig.more_like_this,
  539. opening_statement: modelConfig.opening_statement,
  540. suggested_questions: modelConfig.suggested_questions,
  541. sensitive_word_avoidance: modelConfig.sensitive_word_avoidance,
  542. speech_to_text: modelConfig.speech_to_text,
  543. text_to_speech: modelConfig.text_to_speech,
  544. file_upload: modelConfig.file_upload,
  545. suggested_questions_after_answer: modelConfig.suggested_questions_after_answer,
  546. retriever_resource: modelConfig.retriever_resource,
  547. annotation_reply: modelConfig.annotation_reply,
  548. external_data_tools: modelConfig.external_data_tools,
  549. dataSets: datasets || [],
  550. // eslint-disable-next-line multiline-ternary
  551. agentConfig: res.mode === 'agent-chat' ? {
  552. max_iteration: DEFAULT_AGENT_SETTING.max_iteration,
  553. ...modelConfig.agent_mode,
  554. // remove dataset
  555. enabled: true, // modelConfig.agent_mode?.enabled is not correct. old app: the value of app with dataset's is always true
  556. tools: modelConfig.agent_mode?.tools.filter((tool: any) => {
  557. return !tool.dataset
  558. }).map((tool: any) => {
  559. return {
  560. ...tool,
  561. isDeleted: res.deleted_tools?.includes(tool.tool_name),
  562. notAuthor: collectionList.find(c => tool.provider_id === c.id)?.is_team_authorization === false,
  563. }
  564. }),
  565. } : DEFAULT_AGENT_SETTING,
  566. },
  567. completionParams: model.completion_params,
  568. }
  569. if (modelConfig.file_upload)
  570. handleSetVisionConfig(modelConfig.file_upload.image, true)
  571. syncToPublishedConfig(config)
  572. setPublishedConfig(config)
  573. const retrievalConfig = getMultipleRetrievalConfig(modelConfig.dataset_configs, datasets, datasets, !!currentRerankModel)
  574. setDatasetConfigs({
  575. retrieval_model: RETRIEVE_TYPE.multiWay,
  576. ...modelConfig.dataset_configs,
  577. ...retrievalConfig,
  578. })
  579. setHasFetchedDetail(true)
  580. })
  581. })()
  582. // eslint-disable-next-line react-hooks/exhaustive-deps
  583. }, [appId])
  584. const promptEmpty = (() => {
  585. if (mode !== AppType.completion)
  586. return false
  587. if (isAdvancedMode) {
  588. if (modelModeType === ModelModeType.chat)
  589. return chatPromptConfig.prompt.every(({ text }: any) => !text)
  590. else
  591. return !completionPromptConfig.prompt?.text
  592. }
  593. else { return !modelConfig.configs.prompt_template }
  594. })()
  595. const cannotPublish = (() => {
  596. if (mode !== AppType.completion) {
  597. if (!isAdvancedMode)
  598. return false
  599. if (modelModeType === ModelModeType.completion) {
  600. if (!hasSetBlockStatus.history || !hasSetBlockStatus.query)
  601. return true
  602. return false
  603. }
  604. return false
  605. }
  606. else { return promptEmpty }
  607. })()
  608. const contextVarEmpty = mode === AppType.completion && dataSets.length > 0 && !hasSetContextVar
  609. const onPublish = async (modelAndParameter?: ModelAndParameter, features?: FeaturesData) => {
  610. const modelId = modelAndParameter?.model || modelConfig.model_id
  611. const promptTemplate = modelConfig.configs.prompt_template
  612. const promptVariables = modelConfig.configs.prompt_variables
  613. if (promptEmpty) {
  614. notify({ type: 'error', message: t('appDebug.otherError.promptNoBeEmpty') })
  615. return
  616. }
  617. if (isAdvancedMode && mode !== AppType.completion) {
  618. if (modelModeType === ModelModeType.completion) {
  619. if (!hasSetBlockStatus.history) {
  620. notify({ type: 'error', message: t('appDebug.otherError.historyNoBeEmpty') })
  621. return
  622. }
  623. if (!hasSetBlockStatus.query) {
  624. notify({ type: 'error', message: t('appDebug.otherError.queryNoBeEmpty') })
  625. return
  626. }
  627. }
  628. }
  629. if (contextVarEmpty) {
  630. notify({ type: 'error', message: t('appDebug.feature.dataSet.queryVariable.contextVarNotEmpty') })
  631. return
  632. }
  633. const postDatasets = dataSets.map(({ id }) => ({
  634. dataset: {
  635. enabled: true,
  636. id,
  637. },
  638. }))
  639. const fileUpload = { ...features?.file }
  640. delete fileUpload?.fileUploadConfig
  641. // new model config data struct
  642. const data: BackendModelConfig = {
  643. // Simple Mode prompt
  644. pre_prompt: !isAdvancedMode ? promptTemplate : '',
  645. prompt_type: promptMode,
  646. chat_prompt_config: {},
  647. completion_prompt_config: {},
  648. user_input_form: promptVariablesToUserInputsForm(promptVariables),
  649. dataset_query_variable: contextVar || '',
  650. // features
  651. more_like_this: features?.moreLikeThis as any,
  652. opening_statement: features?.opening?.enabled ? (features.opening?.opening_statement || '') : '',
  653. suggested_questions: features?.opening?.enabled ? (features.opening?.suggested_questions || []) : [],
  654. sensitive_word_avoidance: features?.moderation as any,
  655. speech_to_text: features?.speech2text as any,
  656. text_to_speech: features?.text2speech as any,
  657. file_upload: fileUpload as any,
  658. suggested_questions_after_answer: features?.suggested as any,
  659. retriever_resource: features?.citation as any,
  660. agent_mode: {
  661. ...modelConfig.agentConfig,
  662. strategy: isFunctionCall ? AgentStrategy.functionCall : AgentStrategy.react,
  663. },
  664. model: {
  665. provider: modelAndParameter?.provider || modelConfig.provider,
  666. name: modelId,
  667. mode: modelConfig.mode,
  668. completion_params: modelAndParameter?.parameters || completionParams as any,
  669. },
  670. dataset_configs: {
  671. ...datasetConfigs,
  672. datasets: {
  673. datasets: [...postDatasets],
  674. } as any,
  675. },
  676. }
  677. if (isAdvancedMode) {
  678. data.chat_prompt_config = chatPromptConfig
  679. data.completion_prompt_config = completionPromptConfig
  680. }
  681. await updateAppModelConfig({ url: `/apps/${appId}/model-config`, body: data })
  682. const newModelConfig = produce(modelConfig, (draft: any) => {
  683. draft.opening_statement = introduction
  684. draft.more_like_this = moreLikeThisConfig
  685. draft.suggested_questions_after_answer = suggestedQuestionsAfterAnswerConfig
  686. draft.speech_to_text = speechToTextConfig
  687. draft.text_to_speech = textToSpeechConfig
  688. draft.retriever_resource = citationConfig
  689. draft.dataSets = dataSets
  690. })
  691. setPublishedConfig({
  692. modelConfig: newModelConfig,
  693. completionParams,
  694. })
  695. notify({ type: 'success', message: t('common.api.success') })
  696. setCanReturnToSimpleMode(false)
  697. return true
  698. }
  699. const [showUseGPT4Confirm, setShowUseGPT4Confirm] = useState(false)
  700. const {
  701. debugWithMultipleModel,
  702. multipleModelConfigs,
  703. handleMultipleModelConfigsChange,
  704. } = useDebugWithSingleOrMultipleModel(appId)
  705. const handleDebugWithMultipleModelChange = () => {
  706. handleMultipleModelConfigsChange(
  707. true,
  708. [
  709. { id: `${Date.now()}`, model: modelConfig.model_id, provider: modelConfig.provider, parameters: completionParams },
  710. { id: `${Date.now()}-no-repeat`, model: '', provider: '', parameters: {} },
  711. ],
  712. )
  713. setAppSiderbarExpand('collapse')
  714. }
  715. if (isLoading) {
  716. return <div className='flex items-center justify-center h-full'>
  717. <Loading type='area' />
  718. </div>
  719. }
  720. return (
  721. <ConfigContext.Provider value={{
  722. appId,
  723. isAPIKeySet,
  724. isTrailFinished: false,
  725. mode,
  726. modelModeType,
  727. promptMode,
  728. isAdvancedMode,
  729. isAgent,
  730. isOpenAI,
  731. isFunctionCall,
  732. collectionList,
  733. setPromptMode,
  734. canReturnToSimpleMode,
  735. setCanReturnToSimpleMode,
  736. chatPromptConfig,
  737. completionPromptConfig,
  738. currentAdvancedPrompt,
  739. setCurrentAdvancedPrompt,
  740. conversationHistoriesRole: completionPromptConfig.conversation_histories_role,
  741. showHistoryModal,
  742. setConversationHistoriesRole,
  743. hasSetBlockStatus,
  744. conversationId,
  745. introduction,
  746. setIntroduction,
  747. suggestedQuestions,
  748. setSuggestedQuestions,
  749. setConversationId,
  750. controlClearChatMessage,
  751. setControlClearChatMessage,
  752. prevPromptConfig,
  753. setPrevPromptConfig,
  754. moreLikeThisConfig,
  755. setMoreLikeThisConfig,
  756. suggestedQuestionsAfterAnswerConfig,
  757. setSuggestedQuestionsAfterAnswerConfig,
  758. speechToTextConfig,
  759. setSpeechToTextConfig,
  760. textToSpeechConfig,
  761. setTextToSpeechConfig,
  762. citationConfig,
  763. setCitationConfig,
  764. annotationConfig,
  765. setAnnotationConfig,
  766. moderationConfig,
  767. setModerationConfig,
  768. externalDataToolsConfig,
  769. setExternalDataToolsConfig,
  770. formattingChanged,
  771. setFormattingChanged,
  772. inputs,
  773. setInputs,
  774. query,
  775. setQuery,
  776. completionParams,
  777. setCompletionParams,
  778. modelConfig,
  779. setModelConfig,
  780. showSelectDataSet,
  781. dataSets,
  782. setDataSets,
  783. datasetConfigs,
  784. setDatasetConfigs,
  785. hasSetContextVar,
  786. isShowVisionConfig,
  787. visionConfig,
  788. setVisionConfig: handleSetVisionConfig,
  789. rerankSettingModalOpen,
  790. setRerankSettingModalOpen,
  791. }}
  792. >
  793. <FeaturesProvider features={featuresData}>
  794. <>
  795. <div className="flex flex-col h-full">
  796. <div className='relative flex grow h-[200px] pt-14'>
  797. {/* Header */}
  798. <div className='absolute top-0 left-0 w-full bg-white h-14'>
  799. <div className='flex items-center justify-between px-6 h-14'>
  800. <div className='flex items-center'>
  801. <div className='text-base font-semibold leading-6 text-gray-900'>{t('appDebug.orchestrate')}</div>
  802. <div className='flex items-center h-[14px] space-x-1 text-xs'>
  803. {isAdvancedMode && (
  804. <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>
  805. )}
  806. </div>
  807. </div>
  808. <div className='flex items-center'>
  809. {/* Agent Setting */}
  810. {isAgent && (
  811. <AgentSettingButton
  812. isChatModel={modelConfig.mode === ModelModeType.chat}
  813. agentConfig={modelConfig.agentConfig}
  814. isFunctionCall={isFunctionCall}
  815. onAgentSettingChange={(config) => {
  816. const nextConfig = produce(modelConfig, (draft: ModelConfig) => {
  817. draft.agentConfig = config
  818. })
  819. setModelConfig(nextConfig)
  820. }}
  821. />
  822. )}
  823. {/* Model and Parameters */}
  824. {!debugWithMultipleModel && (
  825. <>
  826. <ModelParameterModal
  827. isAdvancedMode={isAdvancedMode}
  828. mode={mode}
  829. provider={modelConfig.provider}
  830. completionParams={completionParams}
  831. modelId={modelConfig.model_id}
  832. setModel={setModel as any}
  833. onCompletionParamsChange={(newParams: FormValue) => {
  834. setCompletionParams(newParams)
  835. }}
  836. debugWithMultipleModel={debugWithMultipleModel}
  837. onDebugWithMultipleModelChange={handleDebugWithMultipleModelChange}
  838. />
  839. <div className='mx-2 w-[1px] h-[14px] bg-gray-200'></div>
  840. </>
  841. )}
  842. {isMobile && (
  843. <Button className='!h-8 !text-[13px] font-medium' onClick={showDebugPanel}>
  844. <span className='mr-1'>{t('appDebug.operation.debugConfig')}</span>
  845. <CodeBracketIcon className="w-4 h-4 text-gray-500" />
  846. </Button>
  847. )}
  848. <AppPublisher {...{
  849. publishDisabled: cannotPublish,
  850. publishedAt: (latestPublishedAt || 0) * 1000,
  851. debugWithMultipleModel,
  852. multipleModelConfigs,
  853. onPublish,
  854. publishedConfig: publishedConfig!,
  855. resetAppConfig: () => syncToPublishedConfig(publishedConfig!),
  856. }} />
  857. </div>
  858. </div>
  859. </div>
  860. <div className={`w-full sm:w-1/2 shrink-0 flex flex-col h-full ${debugWithMultipleModel && 'max-w-[560px]'}`}>
  861. <Config />
  862. </div>
  863. {!isMobile && <div className="relative flex flex-col w-1/2 h-full overflow-y-auto grow " style={{ borderColor: 'rgba(0, 0, 0, 0.02)' }}>
  864. <div className='grow flex flex-col border-t-[0.5px] border-l-[0.5px] rounded-tl-2xl border-components-panel-border bg-chatbot-bg '>
  865. <Debug
  866. isAPIKeySet={isAPIKeySet}
  867. onSetting={() => setShowAccountSettingModal({ payload: 'provider' })}
  868. inputs={inputs}
  869. modelParameterParams={{
  870. setModel: setModel as any,
  871. onCompletionParamsChange: setCompletionParams,
  872. }}
  873. debugWithMultipleModel={debugWithMultipleModel}
  874. multipleModelConfigs={multipleModelConfigs}
  875. onMultipleModelConfigsChange={handleMultipleModelConfigsChange}
  876. />
  877. </div>
  878. </div>}
  879. </div>
  880. </div>
  881. {showUseGPT4Confirm && (
  882. <Confirm
  883. title={t('appDebug.trailUseGPT4Info.title')}
  884. content={t('appDebug.trailUseGPT4Info.description')}
  885. isShow={showUseGPT4Confirm}
  886. onConfirm={() => {
  887. setShowAccountSettingModal({ payload: 'provider' })
  888. setShowUseGPT4Confirm(false)
  889. }}
  890. onCancel={() => setShowUseGPT4Confirm(false)}
  891. />
  892. )}
  893. {isShowSelectDataSet && (
  894. <SelectDataSet
  895. isShow={isShowSelectDataSet}
  896. onClose={hideSelectDataSet}
  897. selectedIds={selectedIds}
  898. onSelect={handleSelect}
  899. />
  900. )}
  901. {isShowHistoryModal && (
  902. <EditHistoryModal
  903. isShow={isShowHistoryModal}
  904. saveLoading={false}
  905. onClose={hideHistoryModal}
  906. data={completionPromptConfig.conversation_histories_role}
  907. onSave={(data) => {
  908. setConversationHistoriesRole(data)
  909. hideHistoryModal()
  910. }}
  911. />
  912. )}
  913. {isMobile && (
  914. <Drawer showClose isOpen={isShowDebugPanel} onClose={hideDebugPanel} mask footer={null} panelClassname='!bg-gray-50'>
  915. <Debug
  916. isAPIKeySet={isAPIKeySet}
  917. onSetting={() => setShowAccountSettingModal({ payload: 'provider' })}
  918. inputs={inputs}
  919. modelParameterParams={{
  920. setModel: setModel as any,
  921. onCompletionParamsChange: setCompletionParams,
  922. }}
  923. debugWithMultipleModel={debugWithMultipleModel}
  924. multipleModelConfigs={multipleModelConfigs}
  925. onMultipleModelConfigsChange={handleMultipleModelConfigsChange}
  926. />
  927. </Drawer>
  928. )}
  929. {showAppConfigureFeaturesModal && (
  930. <NewFeaturePanel
  931. show
  932. inWorkflow={false}
  933. showFileUpload={false}
  934. isChatMode={mode !== 'completion'}
  935. disabled={false}
  936. onChange={handleFeaturesChange}
  937. onClose={() => setShowAppConfigureFeaturesModal(false)}
  938. promptVariables={modelConfig.configs.prompt_variables}
  939. onAutoAddPromptVariable={handleAddPromptVariable}
  940. />
  941. )}
  942. </>
  943. </FeaturesProvider>
  944. </ConfigContext.Provider>
  945. )
  946. }
  947. export default React.memo(Configuration)