| | |
| | | import type { RefObject } from 'react' |
| | | import { createContext, useContext } from 'use-context-selector' |
| | | import { PromptMode } from '@/models/debug' |
| | | import type { |
| | |
| | | import { ANNOTATION_DEFAULT, DEFAULT_AGENT_SETTING, DEFAULT_CHAT_PROMPT_CONFIG, DEFAULT_COMPLETION_PROMPT_CONFIG } from '@/config' |
| | | import type { FormValue } from '@/app/components/header/account-setting/model-provider-page/declarations' |
| | | import type { Collection } from '@/app/components/tools/types' |
| | | import { noop } from 'lodash-es' |
| | | |
| | | type IDebugConfiguration = { |
| | | appId: string |
| | |
| | | showSelectDataSet: () => void |
| | | // dataset config |
| | | datasetConfigs: DatasetConfigs |
| | | datasetConfigsRef: RefObject<DatasetConfigs> |
| | | setDatasetConfigs: (config: DatasetConfigs) => void |
| | | hasSetContextVar: boolean |
| | | isShowVisionConfig: boolean |
| | |
| | | mode: '', |
| | | modelModeType: ModelModeType.chat, |
| | | promptMode: PromptMode.simple, |
| | | setPromptMode: noop, |
| | | setPromptMode: () => { }, |
| | | isAdvancedMode: false, |
| | | isAgent: false, |
| | | isFunctionCall: false, |
| | | isOpenAI: false, |
| | | collectionList: [], |
| | | canReturnToSimpleMode: false, |
| | | setCanReturnToSimpleMode: noop, |
| | | setCanReturnToSimpleMode: () => { }, |
| | | chatPromptConfig: DEFAULT_CHAT_PROMPT_CONFIG, |
| | | completionPromptConfig: DEFAULT_COMPLETION_PROMPT_CONFIG, |
| | | currentAdvancedPrompt: [], |
| | | showHistoryModal: noop, |
| | | showHistoryModal: () => { }, |
| | | conversationHistoriesRole: { |
| | | user_prefix: 'user', |
| | | assistant_prefix: 'assistant', |
| | | }, |
| | | setConversationHistoriesRole: noop, |
| | | setCurrentAdvancedPrompt: noop, |
| | | setConversationHistoriesRole: () => { }, |
| | | setCurrentAdvancedPrompt: () => { }, |
| | | hasSetBlockStatus: { |
| | | context: false, |
| | | history: false, |
| | | query: false, |
| | | }, |
| | | conversationId: '', |
| | | setConversationId: noop, |
| | | setConversationId: () => { }, |
| | | introduction: '', |
| | | setIntroduction: noop, |
| | | setIntroduction: () => { }, |
| | | suggestedQuestions: [], |
| | | setSuggestedQuestions: noop, |
| | | setSuggestedQuestions: () => { }, |
| | | controlClearChatMessage: 0, |
| | | setControlClearChatMessage: noop, |
| | | setControlClearChatMessage: () => { }, |
| | | prevPromptConfig: { |
| | | prompt_template: '', |
| | | prompt_variables: [], |
| | | }, |
| | | setPrevPromptConfig: noop, |
| | | setPrevPromptConfig: () => { }, |
| | | moreLikeThisConfig: { |
| | | enabled: false, |
| | | }, |
| | | setMoreLikeThisConfig: noop, |
| | | setMoreLikeThisConfig: () => { }, |
| | | suggestedQuestionsAfterAnswerConfig: { |
| | | enabled: false, |
| | | }, |
| | | setSuggestedQuestionsAfterAnswerConfig: noop, |
| | | setSuggestedQuestionsAfterAnswerConfig: () => { }, |
| | | speechToTextConfig: { |
| | | enabled: false, |
| | | }, |
| | | setSpeechToTextConfig: noop, |
| | | setSpeechToTextConfig: () => { }, |
| | | textToSpeechConfig: { |
| | | enabled: false, |
| | | voice: '', |
| | | language: '', |
| | | }, |
| | | setTextToSpeechConfig: noop, |
| | | setTextToSpeechConfig: () => { }, |
| | | citationConfig: { |
| | | enabled: false, |
| | | }, |
| | | setCitationConfig: noop, |
| | | setCitationConfig: () => { }, |
| | | moderationConfig: { |
| | | enabled: false, |
| | | }, |
| | |
| | | embedding_provider_name: '', |
| | | }, |
| | | }, |
| | | setAnnotationConfig: noop, |
| | | setModerationConfig: noop, |
| | | setAnnotationConfig: () => { }, |
| | | setModerationConfig: () => { }, |
| | | externalDataToolsConfig: [], |
| | | setExternalDataToolsConfig: noop, |
| | | setExternalDataToolsConfig: () => { }, |
| | | formattingChanged: false, |
| | | setFormattingChanged: noop, |
| | | setFormattingChanged: () => { }, |
| | | inputs: {}, |
| | | setInputs: noop, |
| | | setInputs: () => { }, |
| | | query: '', |
| | | setQuery: noop, |
| | | setQuery: () => { }, |
| | | completionParams: { |
| | | max_tokens: 16, |
| | | temperature: 1, // 0-2 |
| | |
| | | presence_penalty: 1, // -2-2 |
| | | frequency_penalty: 1, // -2-2 |
| | | }, |
| | | setCompletionParams: noop, |
| | | setCompletionParams: () => { }, |
| | | modelConfig: { |
| | | provider: 'OPENAI', // 'OPENAI' |
| | | model_id: 'gpt-3.5-turbo', // 'gpt-3.5-turbo' |
| | |
| | | dataSets: [], |
| | | agentConfig: DEFAULT_AGENT_SETTING, |
| | | }, |
| | | setModelConfig: noop, |
| | | setModelConfig: () => { }, |
| | | dataSets: [], |
| | | showSelectDataSet: noop, |
| | | setDataSets: noop, |
| | | showSelectDataSet: () => { }, |
| | | setDataSets: () => { }, |
| | | datasetConfigs: { |
| | | retrieval_model: RETRIEVE_TYPE.multiWay, |
| | | reranking_model: { |
| | |
| | | datasets: [], |
| | | }, |
| | | }, |
| | | datasetConfigsRef: { |
| | | current: null, |
| | | }, |
| | | setDatasetConfigs: noop, |
| | | setDatasetConfigs: () => { }, |
| | | hasSetContextVar: false, |
| | | isShowVisionConfig: false, |
| | | visionConfig: { |
| | |
| | | detail: Resolution.low, |
| | | transfer_methods: [TransferMethod.remote_url], |
| | | }, |
| | | setVisionConfig: noop, |
| | | setVisionConfig: () => { }, |
| | | isAllowVideoUpload: false, |
| | | isShowDocumentConfig: false, |
| | | rerankSettingModalOpen: false, |
| | | setRerankSettingModalOpen: noop, |
| | | setRerankSettingModalOpen: () => { }, |
| | | }) |
| | | |
| | | export const useDebugConfigurationContext = () => useContext(DebugConfigurationContext) |