index.tsx 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596
  1. 'use client'
  2. import type { FC } from 'react'
  3. import React from 'react'
  4. import { useContext } from 'use-context-selector'
  5. import produce from 'immer'
  6. import { useFormattingChangedDispatcher } from '../debug/hooks'
  7. import DatasetConfig from '../dataset-config'
  8. import HistoryPanel from '../config-prompt/conversation-history/history-panel'
  9. import ConfigVision from '../config-vision'
  10. import AgentTools from './agent/agent-tools'
  11. import ConfigContext from '@/context/debug-configuration'
  12. import ConfigPrompt from '@/app/components/app/configuration/config-prompt'
  13. import ConfigVar from '@/app/components/app/configuration/config-var'
  14. import { type ModelConfig, type PromptVariable } from '@/models/debug'
  15. import type { AppType } from '@/types/app'
  16. import { ModelModeType } from '@/types/app'
  17. const Config: FC = () => {
  18. const {
  19. mode,
  20. isAdvancedMode,
  21. modelModeType,
  22. isAgent,
  23. hasSetBlockStatus,
  24. showHistoryModal,
  25. modelConfig,
  26. setModelConfig,
  27. setPrevPromptConfig,
  28. } = useContext(ConfigContext)
  29. const isChatApp = ['advanced-chat', 'agent-chat', 'chat'].includes(mode)
  30. const formattingChangedDispatcher = useFormattingChangedDispatcher()
  31. const promptTemplate = modelConfig.configs.prompt_template
  32. const promptVariables = modelConfig.configs.prompt_variables
  33. // simple mode
  34. const handlePromptChange = (newTemplate: string, newVariables: PromptVariable[]) => {
  35. const newModelConfig = produce(modelConfig, (draft: ModelConfig) => {
  36. draft.configs.prompt_template = newTemplate
  37. draft.configs.prompt_variables = [...draft.configs.prompt_variables, ...newVariables]
  38. })
  39. if (modelConfig.configs.prompt_template !== newTemplate)
  40. formattingChangedDispatcher()
  41. setPrevPromptConfig(modelConfig.configs)
  42. setModelConfig(newModelConfig)
  43. }
  44. const handlePromptVariablesNameChange = (newVariables: PromptVariable[]) => {
  45. setPrevPromptConfig(modelConfig.configs)
  46. const newModelConfig = produce(modelConfig, (draft: ModelConfig) => {
  47. draft.configs.prompt_variables = newVariables
  48. })
  49. setModelConfig(newModelConfig)
  50. }
  51. return (
  52. <>
  53. <div
  54. className="grow h-0 relative px-6 pb-[50px] overflow-y-auto"
  55. >
  56. {/* Template */}
  57. <ConfigPrompt
  58. mode={mode as AppType}
  59. promptTemplate={promptTemplate}
  60. promptVariables={promptVariables}
  61. onChange={handlePromptChange}
  62. />
  63. {/* Variables */}
  64. <ConfigVar
  65. promptVariables={promptVariables}
  66. onPromptVariablesChange={handlePromptVariablesNameChange}
  67. />
  68. {/* Dataset */}
  69. <DatasetConfig />
  70. {/* Tools */}
  71. {isAgent && (
  72. <AgentTools />
  73. )}
  74. <ConfigVision />
  75. {/* Chat History */}
  76. {isAdvancedMode && isChatApp && modelModeType === ModelModeType.completion && (
  77. <HistoryPanel
  78. showWarning={!hasSetBlockStatus.history}
  79. onShowEditModal={showHistoryModal}
  80. />
  81. )}
  82. </div>
  83. </>
  84. )
  85. }
  86. export default React.memo(Config)