diff --git a/src/main/java/com/liuzhihang/doc/view/service/impl/YApiServiceImpl.java b/src/main/java/com/liuzhihang/doc/view/service/impl/YApiServiceImpl.java
index 8c0c2e6..d9b023b 100644
--- a/src/main/java/com/liuzhihang/doc/view/service/impl/YApiServiceImpl.java
+++ b/src/main/java/com/liuzhihang/doc/view/service/impl/YApiServiceImpl.java
@@ -10,6 +10,7 @@
 import com.intellij.psi.PsiType;
 import com.intellij.psi.util.InheritanceUtil;
 import com.liuzhihang.doc.view.DocViewBundle;
+import com.liuzhihang.doc.view.config.Settings;
 import com.liuzhihang.doc.view.config.YApiSettings;
 import com.liuzhihang.doc.view.constant.FieldTypeConstant;
 import com.liuzhihang.doc.view.dto.Body;
@@ -57,14 +58,19 @@ public void upload(@NotNull Project project, @NotNull List<DocView> docViewList)
     @Override
     public void upload(@NotNull Project project, @NotNull DocView docView) {
 
-        String catName = docView.getPsiClass().getName();
+        String catName = docView.getDocTitle();
+
+        // 全类名过长
+        if (Settings.getInstance(project).getTitleUseFullClassName() && catName.contains(".")) {
+            catName = catName.substring(catName.lastIndexOf("."));
+        }
+
 
         try {
             YApiSettings settings = YApiSettings.getInstance(project);
 
             YApiFacadeService facadeService = ServiceManager.getService(YApiFacadeServiceImpl.class);
 
-            assert catName != null;
             YApiCat cat = getOrAddCat(settings, catName);
 
             YapiSave save = new YapiSave();
diff --git a/src/main/java/com/liuzhihang/doc/view/ui/PreviewForm.java b/src/main/java/com/liuzhihang/doc/view/ui/PreviewForm.java
index 90b1d2b..1649330 100644
--- a/src/main/java/com/liuzhihang/doc/view/ui/PreviewForm.java
+++ b/src/main/java/com/liuzhihang/doc/view/ui/PreviewForm.java
@@ -519,7 +519,7 @@ private void buildDoc() {
 
             currentDocView = docViewMap.get(selectedValue);
 
-            docNameLabel.setText(currentDocView.getPsiClass().getQualifiedName());
+            docNameLabel.setText(currentDocView.getDocTitle());
 
             // 将 docView 按照模版转换
             currentMarkdownText = DocViewData.buildMarkdownText(project, currentDocView);