dify/web/features/agent-v2/agent-detail/configure/state.ts

import { atom } from 'jotai'

export type AgentConfigureRightPanelMode = 'build' | 'preview'
export type AgentConfigureConversationIds = Record<AgentConfigureRightPanelMode, string | null>
export type AgentConfigureSoulSource = 'draft' | 'build-draft' | 'view-version'

export const agentConfigureSelectedVersionIdAtom = atom<string | null>(null)
export const agentConfigureComposerRebaseRevisionAtom = atom(0)
export const agentConfigureSoulSourceOverrideAtom = atom<AgentConfigureSoulSource | null>(null)

export const agentConfigureShowChatFeaturesAtom = atom(false)
export const agentConfigureShowPreviewVersionsAtom = atom(false)
export const agentConfigureRightPanelModeAtom = atom<AgentConfigureRightPanelMode>('build')
export const agentConfigureConversationIdsAtom = atom<AgentConfigureConversationIds>({
  build: null,
  preview: null,
})

export const agentConfigureRightPanelChatModeAtom = atom((get): AgentConfigureRightPanelMode => {
  const mode = get(agentConfigureRightPanelModeAtom)

  return mode === 'preview' ? 'build' : mode
})

export const agentConfigureSelectVersionAtom = atom(null, (_get, set, versionId: string | null) => {
  set(agentConfigureSoulSourceOverrideAtom, versionId ? 'view-version' : null)
  set(agentConfigureSelectedVersionIdAtom, versionId)
})

export const rebaseAgentConfigureComposerAtom = atom(null, (get, set) => {
  set(agentConfigureComposerRebaseRevisionAtom, get(agentConfigureComposerRebaseRevisionAtom) + 1)
})

export const setAgentConfigureConversationIdAtom = atom(
  null,
  (
    get,
    set,
    {
      mode,
      conversationId,
    }: {
      mode: AgentConfigureRightPanelMode
      conversationId: string | null
    },
  ) => {
    set(agentConfigureConversationIdsAtom, {
      ...get(agentConfigureConversationIdsAtom),
      [mode]: conversationId,
    })
  },
)

export const resetAgentConfigureConversationAtom = atom(
  null,
  (get, set, mode: AgentConfigureRightPanelMode) => {
    set(agentConfigureConversationIdsAtom, {
      ...get(agentConfigureConversationIdsAtom),
      [mode]: null,
    })
  },
)

export const agentConfigureScopedAtoms = [
  agentConfigureSelectedVersionIdAtom,
  agentConfigureComposerRebaseRevisionAtom,
  agentConfigureSoulSourceOverrideAtom,
  agentConfigureShowChatFeaturesAtom,
  agentConfigureShowPreviewVersionsAtom,
  agentConfigureRightPanelModeAtom,
] as const