fix: expand the model after deploying from catalog

src/pages/llmodels/catalog.tsx

@@ -1,9 +1,11 @@
+import { modelsExpandKeysAtom } from '@/atoms/models';
 import PageTools from '@/components/page-tools';
 import { PageAction } from '@/config';
 import { SyncOutlined } from '@ant-design/icons';
 import { PageContainer } from '@ant-design/pro-components';
 import { useIntl, useNavigate } from '@umijs/max';
 import { Button, Input, Pagination, Select, Space, message } from 'antd';
+import { useAtom } from 'jotai';
 import _ from 'lodash';
 import React, { useCallback, useEffect, useState } from 'react';
 import { createModel, queryCatalogList } from './apis';
@@ -38,6 +40,7 @@ const Catalog: React.FC = () => {
     current: {},
     source: modelSourceMap.huggingface_value
   });
+  const [modelsExpandKeys, setModelsExpandKeys] = useAtom(modelsExpandKeysAtom);
   const cacheData = React.useRef<CatalogItemType[]>([]);
 
   const categoryOptions = [...modelCategories.filter((item) => item.value)];
@@ -53,7 +56,7 @@ const Catalog: React.FC = () => {
           );
         }
         if (search) {
-          return _.toLower(item.name).includes(search);
+          return _.toLower(item.name).includes(_.toLower(search));
         }
         if (categories.length > 0) {
           return categories.some((category) =>
@@ -134,6 +137,7 @@ const Catalog: React.FC = () => {
           show: false
         });
         message.success(intl.formatMessage({ id: 'common.message.success' }));
+        setModelsExpandKeys([modelData.id]);
         navigate('/models/list');
       } catch (error) {}
     },

src/pages/llmodels/components/table-list.tsx

@@ -1,3 +1,4 @@
+import { modelsExpandKeysAtom } from '@/atoms/models';
 import AutoTooltip from '@/components/auto-tooltip';
 import DeleteModal from '@/components/delete-modal';
 import DropdownButtons from '@/components/drop-down-buttons';
@@ -30,6 +31,7 @@ import { PageContainer } from '@ant-design/pro-components';
 import { Access, useAccess, useIntl, useNavigate } from '@umijs/max';
 import { Button, Dropdown, Input, Select, Space, Tag, message } from 'antd';
 import dayjs from 'dayjs';
+import { useAtom } from 'jotai';
 import _ from 'lodash';
 import { memo, useCallback, useEffect, useRef, useState } from 'react';
 import { useHotkeys } from 'react-hotkeys-hook';
@@ -125,6 +127,7 @@ const Models: React.FC<ModelsProps> = ({
   loading,
   total
 }) => {
+  const [expandAtom] = useAtom(modelsExpandKeysAtom);
   const access = useAccess();
   const intl = useIntl();
   const navigate = useNavigate();
@@ -134,7 +137,7 @@ const Models: React.FC<ModelsProps> = ({
     updateExpandedRowKeys,
     removeExpandedRowKey,
     expandedRowKeys
-  } = useExpandedRowKeys();
+  } = useExpandedRowKeys(expandAtom);
   const { sortOrder, setSortOrder } = useTableSort({
     defaultSortOrder: 'descend'
   });
@@ -309,6 +312,7 @@ const Models: React.FC<ModelsProps> = ({
         }
       });
       message.success(intl.formatMessage({ id: 'common.message.success' }));
+      updateExpandedRowKeys([row.id, ...expandedRowKeys]);
     } catch (error) {
       // ingore
     }
@@ -329,6 +333,7 @@ const Models: React.FC<ModelsProps> = ({
           replicas: 0
         }
       });
+      removeExpandedRowKey([row.id]);
     } catch (error) {
       // ingore
     }

src/pages/playground/components/ground-images.tsx

@@ -352,6 +352,7 @@ const GroundImages: React.FC<MessageProps> = forwardRef((props, ref) => {
           if (item.b64_json && stream_options.chunk_results) {
             imgItem.dataUrl += item.b64_json;
           } else if (item.b64_json) {
+            console.log('item.b64_json:', item.b64_json, item.index);
             imgItem.dataUrl = `data:image/png;base64,${item.b64_json}`;
           }
           const progress = _.round(item.progress, 0);