diff --git a/webapp/packages/supersonic-fe/src/app.tsx b/webapp/packages/supersonic-fe/src/app.tsx
index e10676f8e..107a7ea68 100644
--- a/webapp/packages/supersonic-fe/src/app.tsx
+++ b/webapp/packages/supersonic-fe/src/app.tsx
@@ -129,7 +129,7 @@ export const layout: RunTimeLayoutConfig = (params) => {
           color="#1672fa"
           style={{ display: 'inline-block', marginTop: 8 }}
         />
-        <div className="logo" style={{ color: '#262626' }}>
+        <div className="logo" style={{ position: 'relative', top: '-2px' }}>
           SuperSonic
         </div>
       </Space>
diff --git a/webapp/packages/supersonic-fe/src/global.less b/webapp/packages/supersonic-fe/src/global.less
index 7aa35e672..aa9cffb97 100644
--- a/webapp/packages/supersonic-fe/src/global.less
+++ b/webapp/packages/supersonic-fe/src/global.less
@@ -172,8 +172,9 @@ ol {
 }
 
 .logo {
+  color: var(--tme-primary-color);
+  font-family: var(--tencent-font-family);
   position: relative;
-  color: #fff;
   font-size: 20px;
   font-weight: 700;
   padding-right: 50px;
diff --git a/webapp/packages/supersonic-fe/src/pages/Agent/ToolModal.tsx b/webapp/packages/supersonic-fe/src/pages/Agent/ToolModal.tsx
index 5c3c03949..e1990d655 100644
--- a/webapp/packages/supersonic-fe/src/pages/Agent/ToolModal.tsx
+++ b/webapp/packages/supersonic-fe/src/pages/Agent/ToolModal.tsx
@@ -31,13 +31,25 @@ const ToolModal: React.FC<Props> = ({ editTool, onSaveTool, onCancel }) => {
   const [plugins, setPlugins] = useState<PluginType[]>([]);
   const [form] = Form.useForm();
 
+  // const filterTree = (treeData: any[]) => {
+  //   treeData.forEach((node) => {
+  //     if (Array.isArray(node.children) && node.children?.length > 0) {
+  //       node.children = node.children.filter((item: any) => item.type !== 'DOMAIN');
+  //       filterTree(node.children);
+  //     }
+  //   });
+  //   return treeData;
+  // };
+
   const initModelList = async () => {
-    const { code, data } = await getModelList();
-    if (code === 200) {
-      setModelList([{ name: '默认', id: -1 }, ...data]);
-    } else {
-      message.error('获取模型列表失败!');
-    }
+    const res = await getModelList();
+    const treeData = traverseTree(res.data, (node: any) => {
+      node.title = node.name;
+      node.value = node.type === 'DOMAIN' ? `DOMAIN_${node.id}` : node.id;
+      node.checkable =
+        node.type === 'DATASET' || (node.type === 'DOMAIN' && node.children?.length > 0);
+    });
+    setModelList([{ title: '默认', value: -1, type: 'DATASET' }, ...treeData]);
   };
 
   const initPluginList = async () => {
@@ -104,10 +116,11 @@ const ToolModal: React.FC<Props> = ({ editTool, onSaveTool, onCancel }) => {
         {(toolType === AgentToolTypeEnum.NL2SQL_RULE ||
           toolType === AgentToolTypeEnum.NL2SQL_LLM) && (
           <FormItem name="dataSetIds" label="数据集">
-            <Select
-              options={modelList.map((item) => ({ label: item.name, value: item.id }))}
+            <TreeSelect
+              treeData={modelList}
               placeholder="请选择数据集"
-              mode="multiple"
+              multiple
+              treeCheckable
               allowClear
             />
           </FormItem>
diff --git a/webapp/packages/supersonic-fe/src/pages/SemanticModel/components/DomainList.tsx b/webapp/packages/supersonic-fe/src/pages/SemanticModel/components/DomainList.tsx
index f4cfbd806..0aa205575 100644
--- a/webapp/packages/supersonic-fe/src/pages/SemanticModel/components/DomainList.tsx
+++ b/webapp/packages/supersonic-fe/src/pages/SemanticModel/components/DomainList.tsx
@@ -194,7 +194,7 @@ const DomainListTree: FC<DomainListProps> = ({
             <Search
               allowClear
               className={styles.search}
-              placeholder="请输入主题域"
+              placeholder="请输入名称搜索"
               onSearch={onSearch}
             />
           </Col>