From 94503a4ab6315d0b9b01129711889a9bbe4fa954 Mon Sep 17 00:00:00 2001
From: ningyv <1793599591@qq.com>
Date: Thu, 20 Feb 2025 14:19:34 +0800
Subject: [PATCH] feat: feature/1.5-Data Integration

---
 .../common/src/components/aoplatform/PageList.tsx      |  3 +++
 .../core/src/pages/aiSetting/LocalModelList.tsx        | 10 +++++++++-
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/frontend/packages/common/src/components/aoplatform/PageList.tsx b/frontend/packages/common/src/components/aoplatform/PageList.tsx
index 761366cb..dab31d82 100644
--- a/frontend/packages/common/src/components/aoplatform/PageList.tsx
+++ b/frontend/packages/common/src/components/aoplatform/PageList.tsx
@@ -36,6 +36,7 @@ interface PageListProps<T> extends ProTableProps<T, unknown>, RefAttributes<Acti
   primaryKey?: string
   addNewBtnTitle?: string
   addNewBtnAccess?: string
+  addNewBtnDisabled?: boolean
   tableClickAccess?: string
   onAddNewBtnClick?: () => void
   beforeSearchNode?: React.ReactNode[]
@@ -75,6 +76,7 @@ const PageList = <T extends Record<string, unknown>>(
     primaryKey = 'id',
     addNewBtnTitle,
     addNewBtnAccess,
+    addNewBtnDisabled = false,
     tableClickAccess,
     tableClass,
     onAddNewBtnClick,
@@ -202,6 +204,7 @@ const PageList = <T extends Record<string, unknown>>(
             <Button
               type="primary"
               className={`mr-btnrbase my-btnbase ${addNewBtnWrapperClass}`}
+              disabled={addNewBtnDisabled}
               onClick={onAddNewBtnClick}
             >
               {addNewBtnTitle}
diff --git a/frontend/packages/core/src/pages/aiSetting/LocalModelList.tsx b/frontend/packages/core/src/pages/aiSetting/LocalModelList.tsx
index 7c2cba59..792f4577 100644
--- a/frontend/packages/core/src/pages/aiSetting/LocalModelList.tsx
+++ b/frontend/packages/core/src/pages/aiSetting/LocalModelList.tsx
@@ -140,7 +140,6 @@ const LocalModelList: React.FC = () => {
         return ConfigureOllamaServiceRef.current?.save().then((res) => {
           if (res === true) {
             getOllamaData()
-            pageListRef.current?.reload()
           }
         })
       },
@@ -191,6 +190,7 @@ const LocalModelList: React.FC = () => {
 
     if (response.code === STATUS_CODE.SUCCESS) {
       setOllamaAddress(response.data?.config?.address || '')
+      pageListRef.current?.reload()
     } else {
       message.error(response.msg || $t(RESPONSE_TIPS.error))
     }
@@ -286,6 +286,13 @@ const LocalModelList: React.FC = () => {
 
   const requestList = async (params: any) => {
     try {
+      if (!ollamaAddress) {
+        return {
+          data: [],
+          success: true,
+          total: 0
+        }
+      }
       const response = await fetchData<BasicResponse<{ data: ModelListData[] }>>('model/local/list', {
         method: 'GET',
         eoParams: {
@@ -449,6 +456,7 @@ const LocalModelList: React.FC = () => {
       columns={columns}
       addNewBtnTitle={$t('部署模型')}
       onAddNewBtnClick={handleAdd}
+      addNewBtnDisabled={!ollamaAddress}
     />
   )
 }