import knowledgeFrame1 from '../../assets/image/management/knowledge-frame1.png';
import knowledgeFrame2 from '../../assets/image/management/knowledge-frame2.png';
import knowledgeFrame3 from '../../assets/image/management/knowledge-frame3.png';
import knowledgeFrame4 from '../../assets/image/management/knowledge-frame4.png';
import intelligentFrame1 from '../../assets/image/management/intelligent-frame1.png';
import intelligentFrame2 from '../../assets/image/management/intelligent-frame2.png';
import intelligentFrame3 from '../../assets/image/management/intelligent-frame3.png';
import intelligentFrame4 from '../../assets/image/management/intelligent-frame4.png';
import intelligentFrame5 from '../../assets/image/management/intelligent-frame5.png';
import intelligentFrame6 from '../../assets/image/management/intelligent-frame6.png';
import intelligentFrame7 from '../../assets/image/management/intelligent-frame7.png';
import intelligentFrame8 from '../../assets/image/management/intelligent-frame8.png';
import intellFrame1 from '../../assets/image/management/intellFrame1.png';
import intellFrame2 from '../../assets/image/management/intellFrame2.png';
import intellFrame3 from '../../assets/image/management/intellFrame3.png';
import intellFrame4 from '../../assets/image/management/intellFrame4.png';
import intellFrame5 from '../../assets/image/management/intellFrame5.png';
import intellFrame6 from '../../assets/image/management/intellFrame6.png';

const image_knowLedge_map = {
  knowledgeFrame1: knowledgeFrame1,
  knowledgeFrame2: knowledgeFrame2,
  knowledgeFrame3: knowledgeFrame3,
  knowledgeFrame4: knowledgeFrame4
};
const image_intelligent_map = {
  intelligentFrame1: intelligentFrame1,
  intelligentFrame2: intelligentFrame2,
  intelligentFrame3: intelligentFrame3,
  intelligentFrame4: intelligentFrame4,
  intelligentFrame5: intelligentFrame5,
  intelligentFrame6: intelligentFrame6,
  intelligentFrame7: intelligentFrame7,
  intelligentFrame8: intelligentFrame8
};
const icon_intelligent_map = {
  intellFrame1: intellFrame1,
  intellFrame2: intellFrame2,
  intellFrame3: intellFrame3,
  intellFrame4: intellFrame4,
  intellFrame5: intellFrame5,
  intellFrame6: intellFrame6
};
/**
 * @description: 智能体类型
 * */
export const dialogType = {
  RAGFLOW: '1',
  BS: '2',
  BASIC: '3',
  DIFY: '4'
};

/**
 * @description: 图片映射表
 * avatar: 图片名称
 * defaultImg: 默认图片
 */
export const getImageKnowLedge = (avatar) => {
  return image_knowLedge_map[avatar] || knowledgeFrame1;
};
export const getImageIntelligent = (avatar) => {
  return image_intelligent_map[avatar];
};
export const getIconIntelligent = (avatar) => {
  return icon_intelligent_map[avatar] || intellFrame1;
};

/**
 * @description: 默认智能体数据
 */
export const getDefaultIntelligent = () => {
  return [
    {
      name: '问答助理',
      description: 'EMBEDDING,SPEECH2TEXT,MODERATION',
      labels: 'LLM,TEXT',
      type: 'ragflow'
    },
    {
      name: '工作流助手',
      description: 'EMBEDDING,SPEECH2TEXT,MODERATION',
      labels: 'LLM,TEXT',
      type: 'advanced-chat'
    },
    {
      name: '智能Agent',
      description: 'EMBEDDING,SPEECH2TEXT,MODERATION',
      labels: 'LLM,TEXT',
      type: 'agent-chat'
    },
    {
      name: '工作流应用',
      description: 'EMBEDDING,SPEECH2TEXT,MODERATION',
      labels: 'LLM,TEXT',
      type: 'workflow'
    }
  ];
};