From 0a9a0c75eb8a4668ef85b6cfbccaa5f9ff7de099 Mon Sep 17 00:00:00 2001
From: Joel <iamjoel007@gmail.com>
Date: Tue, 3 Sep 2024 11:05:32 +0800
Subject: [PATCH] fix: dataset not has type create btn ui problem

---
 web/app/components/datasets/documents/index.tsx | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/web/app/components/datasets/documents/index.tsx b/web/app/components/datasets/documents/index.tsx
index 32c045eccd..eecbd58c51 100644
--- a/web/app/components/datasets/documents/index.tsx
+++ b/web/app/components/datasets/documents/index.tsx
@@ -20,6 +20,7 @@ import type { NotionPage } from '@/models/common'
 import type { CreateDocumentReq } from '@/models/datasets'
 import { DataSourceType } from '@/models/datasets'
 import RetryButton from '@/app/components/base/retry-button'
+import cn from '@/utils/classnames'
 // Custom page count is not currently supported.
 const limit = 15
 
@@ -211,10 +212,14 @@ const Documents: FC<IDocumentsProps> = ({ datasetId }) => {
             <RetryButton datasetId={datasetId} />
             {embeddingAvailable && (
               <Button variant='primary' onClick={routeToDocCreate} className='shrink-0'>
-                <PlusIcon className='h-4 w-4 mr-2 stroke-current' />
-                {isDataSourceNotion && t('datasetDocuments.list.addPages')}
-                {isDataSourceWeb && t('datasetDocuments.list.addUrl')}
-                {isDataSourceFile && t('datasetDocuments.list.addFile')}
+                <PlusIcon className={cn('h-4 w-4 stroke-current', dataset?.data_source_type && 'mr-2')} />
+                {dataset?.data_source_type && (
+                  <>
+                    {isDataSourceNotion && t('datasetDocuments.list.addPages')}
+                    {isDataSourceWeb && t('datasetDocuments.list.addUrl')}
+                    {isDataSourceFile && t('datasetDocuments.list.addFile')}
+                  </>
+                )}
               </Button>
             )}
           </div>