feat: 启动 AI 功能优化

safe1ine · MonkeyCode-AI · safe1ine · commit 7c8650f734e6 · 2026-03-09T13:36:53.000Z
- 去掉“生成项目文档”选项，点击“启动 AI”直接弹出对话框
- 添加仓库分支选择功能，调用 API 获取分支列表
- 添加项目名称和仓库地址展示（只读输入框样式）
- internal 平台不显示分支选择框，branch 传空字符串
- 使用 project.full_name 字段获取仓库名称
- 优先选择 main/master 分支，无则选择第一个
- 添加分支加载状态显示
- 去掉对话框子标题和上下边距
- 分支选择框撑满一行，宽度与上方输入框一致
- 修复生成项目文档对话框的构建错误
Co-authored-by: monkeycode-ai &lt;monkeycode-ai@chaitin.com&gt;
diff --git a/frontend/src/api/Api.ts b/frontend/src/api/Api.ts
@@ -827,6 +827,8 @@ export interface DomainProject {
   created_at?: number;
   /** 项目描述 */
   description?: string;
+  /** 仓库 full_name */
+  full_name?: string;
   /** 项目关联的 git identity id */
   git_identity_id?: string;
   /** 项目ID */
diff --git a/frontend/src/components/console/project/generate-doc-dialog.tsx b/frontend/src/components/console/project/generate-doc-dialog.tsx
@@ -1,40 +1,113 @@
-import { ConstsCliName, ConstsTaskSubType, ConstsTaskType } from "@/api/Api"
+import { ConstsCliName, ConstsTaskSubType, ConstsTaskType, ConstsGitPlatform, type DomainProject, type DomainBranch } from "@/api/Api"
 import { useCommonData } from "@/components/console/data-provider"
 import { Button } from "@/components/ui/button"
-import { Dialog, DialogContent, DialogDescription, DialogFooter, DialogHeader, DialogTitle } from "@/components/ui/dialog"
+import { Dialog, DialogContent, DialogFooter, DialogHeader, DialogTitle } from "@/components/ui/dialog"
+import { Input } from "@/components/ui/input"
+import { Label } from "@/components/ui/label"
+import { Select, SelectContent, SelectItem, SelectTrigger, SelectValue } from "@/components/ui/select"
 import { Spinner } from "@/components/ui/spinner"
 import { packAndUploadFilesAsZip, selectHost, selectImage, selectModel } from "@/utils/common"
 import { apiRequest } from "@/utils/requestUtils"
 import { IconSparkles } from "@tabler/icons-react"
-import { useState } from "react"
+import { useState, useEffect } from "react"
 import { toast } from "sonner"
 
 interface GenerateDocDialogProps {
   open: boolean
   onOpenChange: (open: boolean) => void
   projectId: string
   projectName: string
+  project?: DomainProject
 }
 
 export default function GenerateDocDialog({
   open,
   onOpenChange,
   projectId,
-  projectName
+  projectName,
+  project
 }: GenerateDocDialogProps) {
   const [selectedFiles, setSelectedFiles] = useState<File[]>([])
   const [submitting, setSubmitting] = useState<boolean>(false)
+  const [branches, setBranches] = useState<string[]>([])
+  const [selectedBranch, setSelectedBranch] = useState<string>('')
+  const [loadingBranches, setLoadingBranches] = useState<boolean>(false)
   const { images, models, hosts } = useCommonData()
 
+  const fetchBranches = async () => {
+    if (!project?.git_identity_id || !project?.repo_url) {
+      return
+    }
+
+    // internal 平台不需要获取分支列表
+    if (project.platform === ConstsGitPlatform.GitPlatformInternal) {
+      setSelectedBranch('')
+      setBranches([])
+      return
+    }
+
+    setLoadingBranches(true)
+    
+    try {
+      // 直接使用 full_name 字段
+      const escapedRepoFullName = project?.full_name || ''
+      
+      if (!escapedRepoFullName) {
+        toast.error('无法获取仓库信息')
+        setLoadingBranches(false)
+        return
+      }
+
+      // URL 编码仓库名称
+      const encodedRepoName = encodeURIComponent(escapedRepoFullName)
+
+      await apiRequest('v1UsersGitIdentitiesBranchesDetail', {}, [project.git_identity_id, encodedRepoName], (resp) => {
+        if (resp.code === 0 && resp.data) {
+          const branchList = resp.data.map((b: DomainBranch) => b.name || '').filter(Boolean)
+          setBranches(branchList)
+          
+          // 优先选择 main 或 master，否则选择第一个
+          if (branchList.includes('main')) {
+            setSelectedBranch('main')
+          } else if (branchList.includes('master')) {
+            setSelectedBranch('master')
+          } else if (branchList.length > 0) {
+            setSelectedBranch(branchList[0])
+          }
+        } else {
+          toast.error('获取分支列表失败: ' + resp.message)
+        }
+      })
+    } catch (error) {
+      console.error('Fetch branches error:', error)
+      toast.error('获取分支列表失败')
+    } finally {
+      setLoadingBranches(false)
+    }
+  }
+
+  useEffect(() => {
+    if (open) {
+      fetchBranches()
+    }
+  }, [open, project])
+
   const handleOpenChange = (open: boolean) => {
     onOpenChange(open)
     if (!open) {
       // 重置表单状态
       setSelectedFiles([])
+      setSelectedBranch('')
+      setBranches([])
     }
   }
 
   const handleSubmit = async () => {
+    if (project?.platform !== ConstsGitPlatform.GitPlatformInternal && !selectedBranch) {
+      toast.error('请选择分支')
+      return
+    }
+
     setSubmitting(true)
 
     let repoInfo: { zip_url?: string; repo_filename?: string } = {}
@@ -68,6 +141,7 @@ export default function GenerateDocDialog({
       },
       extra: {
         project_id: projectId,
+        branch: project?.platform === ConstsGitPlatform.GitPlatformInternal ? '' : selectedBranch,
       },
       task_type: ConstsTaskType.TaskTypeDesign,
       sub_type: ConstsTaskSubType.TaskSubTypeGenerateDocs,
@@ -89,10 +163,43 @@ export default function GenerateDocDialog({
       <DialogContent>
         <DialogHeader>
           <DialogTitle>生成项目文档</DialogTitle>
-          <DialogDescription>
-            AI 将根据你的需求为你生成项目架构设计文档
-          </DialogDescription>
         </DialogHeader>
+        
+        <div className="space-y-4 py-4">
+          <div className="space-y-2">
+            <Label>仓库地址</Label>
+            <Input 
+              value={project?.repo_url || '-'} 
+              readOnly 
+              className="bg-muted"
+            />
+          </div>
+          {(project?.platform !== ConstsGitPlatform.GitPlatformInternal) && (
+            <div className="space-y-2">
+              <Label>选择分支</Label>
+              <Select value={selectedBranch} onValueChange={setSelectedBranch} disabled={loadingBranches || branches.length === 0}>
+                <SelectTrigger>
+                  {loadingBranches ? (
+                    <div className="flex items-center gap-2">
+                      <Spinner className="size-4" />
+                      <span>加载中...</span>
+                    </div>
+                  ) : (
+                    <SelectValue placeholder="请选择分支" />
+                  )}
+                </SelectTrigger>
+                <SelectContent>
+                  {branches.map((branch) => (
+                    <SelectItem key={branch} value={branch}>
+                      {branch}
+                    </SelectItem>
+                  ))}
+                </SelectContent>
+              </Select>
+            </div>
+          )}
+        </div>
+        
         <DialogFooter>
           <Button 
             onClick={handleSubmit}
@@ -106,4 +213,3 @@ export default function GenerateDocDialog({
     </Dialog>
   )
 }
-
diff --git a/frontend/src/components/console/project/project-info.tsx b/frontend/src/components/console/project/project-info.tsx
@@ -2,7 +2,6 @@ import { type DomainProject } from "@/api/Api"
 import { useCommonData } from "@/components/console/data-provider"
 import EditCollaboratorsDialog from "@/components/console/project/edit-collaborators"
 import EditProjectNameDialog from "@/components/console/project/edit-project-name"
-import GenerateDocDialog from "@/components/console/project/generate-doc-dialog"
 import StartDevelopTaskDialog from "@/components/console/project/start-develop-task-dialog"
 import { AlertDialog, AlertDialogAction, AlertDialogCancel, AlertDialogContent, AlertDialogDescription, AlertDialogFooter, AlertDialogHeader, AlertDialogTitle } from "@/components/ui/alert-dialog"
 import { Avatar, AvatarFallback, AvatarImage } from "@/components/ui/avatar"
@@ -11,7 +10,7 @@ import { DropdownMenu, DropdownMenuContent, DropdownMenuItem, DropdownMenuTrigge
 import { Item, ItemActions, ItemContent, ItemDescription, ItemMedia, ItemTitle } from "@/components/ui/item"
 import { isProjectRepoUnbound } from "@/utils/project"
 import { apiRequest } from "@/utils/requestUtils"
-import { IconBook, IconBrandGithub, IconLoader, IconPencil, IconSparkles, IconTrash, IconUsers } from "@tabler/icons-react"
+import { IconBrandGithub, IconLoader, IconPencil, IconSparkles, IconTrash, IconUsers } from "@tabler/icons-react"
 import { MoreVertical } from "lucide-react"
 import { useState } from "react"
 import { useNavigate } from "react-router-dom"
@@ -32,7 +31,6 @@ const ProjectInfo = ({
   const [deleteDialogOpen, setDeleteDialogOpen] = useState(false)
   const [deletingProject, setDeletingProject] = useState<DomainProject | undefined>(undefined)
   const [deleteLoading, setDeleteLoading] = useState(false)
-  const [emptyDocDialogOpen, setEmptyDocDialogOpen] = useState(false)
   const [conversationDialogOpen, setConversationDialogOpen] = useState(false)
   const navigate = useNavigate()
   const { projects, reloadProjects } = useCommonData()
@@ -79,11 +77,6 @@ const ProjectInfo = ({
     setDeleteLoading(false)
   }
 
-  const handleGenerateDoc = () => {
-    if (!project) return
-    setEmptyDocDialogOpen(true)
-  }
-
   const handleStartConversation = () => {
     if (!project || isRepoUnbound) return
     setConversationDialogOpen(true)
@@ -118,24 +111,16 @@ const ProjectInfo = ({
         </ItemContent>
         <ItemActions>
           
-        <DropdownMenu>
-            <DropdownMenuTrigger asChild>
-              <Button variant="secondary" size="sm" disabled={isRepoUnbound}>
-                <IconSparkles className="size-4" />
-                启动 AI
-              </Button>
-            </DropdownMenuTrigger>
-            <DropdownMenuContent align="end">
-              <DropdownMenuItem onClick={handleStartConversation}>
-                <IconSparkles />
-                发起对话
-              </DropdownMenuItem>
-              <DropdownMenuItem onClick={handleGenerateDoc}>
-                <IconBook />
-                生成项目文档
-              </DropdownMenuItem>
-            </DropdownMenuContent>
-          </DropdownMenu>
+        <Button 
+            variant="secondary" 
+            size="sm" 
+            disabled={isRepoUnbound}
+            onClick={handleStartConversation}
+          >
+            <IconSparkles className="size-4" />
+            启动 AI
+          </Button>
+          
           <DropdownMenu>
             <DropdownMenuTrigger asChild>
               <Button variant="ghost" size="icon-sm">
@@ -200,13 +185,6 @@ const ProjectInfo = ({
         </AlertDialogContent>
       </AlertDialog>
 
-      <GenerateDocDialog
-        open={emptyDocDialogOpen}
-        onOpenChange={setEmptyDocDialogOpen}
-        projectId={project?.id || ''}
-        projectName={project?.name || ''}
-      />
-
       <StartDevelopTaskDialog
         open={conversationDialogOpen}
         onOpenChange={setConversationDialogOpen}
diff --git a/frontend/src/components/console/project/start-develop-task-dialog.tsx b/frontend/src/components/console/project/start-develop-task-dialog.tsx
