diff --git a/web/app/components/workflow/skill/utils.ts b/web/app/components/workflow/skill/utils.ts new file mode 100644 index 0000000000..4746d75194 --- /dev/null +++ b/web/app/components/workflow/skill/utils.ts @@ -0,0 +1,13 @@ +import { FileAppearanceTypeEnum } from '@/app/components/base/file-uploader/types' + +export const getFileIconType = (name: string) => { + const extension = name.split('.').pop()?.toLowerCase() ?? '' + + if (['md', 'markdown', 'mdx'].includes(extension)) + return FileAppearanceTypeEnum.markdown + + if (['json', 'yaml', 'yml', 'toml', 'js', 'jsx', 'ts', 'tsx', 'py', 'schema'].includes(extension)) + return FileAppearanceTypeEnum.code + + return FileAppearanceTypeEnum.document +}