mm-qa-mcp 3.0.3__py3-none-any.whl → 3.0.4__py3-none-any.whl

minimax_qa_mcp/conf/conf.ini

@@ -66,6 +66,18 @@ url = http://10.11.8.37
 port = 8080
 grpc_port = 50051
 
+[git_info]
+# GitLab服务器地址
+gitlab_url = https://gitlab.xaminim.com
+# GitLab API访问令牌（Private Token）
+access_token = SrFXdGjtd3AtRVJCkpDn
+
+[apollo_info]
+# Apollo配置获取API基础URL
+apollo_base_url = http://swing-babel-ali-prod.xaminim.com/swing/api/get_apollo_value_by_key
+# 存储产品线与GitLab Group映射的Apollo Key
+# Apollo中配置格式: {"qa_group": [{"group_id": 1117, "group_name": "qa"}]}
+product_line_gitlab_key = product_line_gitlab_projects
 
 [generator_case_conf]
 # 模型API URL，用于ModuleClient类调用

minimax_qa_mcp/server.py

@@ -23,6 +23,7 @@ from minimax_qa_mcp.src.query_segments.query_segments import query_main, TYPE_AP
     TYPE_FUNC_DETAIL
 from minimax_qa_mcp.src.auto_case.case_write import PDFWeaviateInfo
 from minimax_qa_mcp.src.get_full_api_call_chain.get_full_api_call_chain import GetFullApiCallChain
+from minimax_qa_mcp.src.gitlab_branch_analyzer.gitlab_branch_service import GitlabBranchService, GitlabRepoTypeDetector, GitlabBedrockResolver
 
 # Initialize FastMCP server
 mcp = FastMCP("mcp")
@@ -202,6 +203,153 @@ async def get_full_api_call_chain(api_path: str) -> dict:
     logger.info(f"===== The result of get_full_api_call_chain is :{result}")
     return result
 
+@mcp.tool()
+async def git_repo_clone(repo_url: str, clone_path: str) -> str:
+    """
+    克隆Git仓库到指定路径
+
+    Args:
+        repo_url: Git仓库的URL
+        clone_path: 本地克隆路径
+
+    Returns:
+        克隆结果的字符串信息
+    """
+    import git
+    try:
+        git.Repo.clone_from(repo_url, clone_path)
+        return f"Successfully cloned {repo_url} to {clone_path}"
+    except Exception as e:
+        return f"Failed to clone repository: {str(e)}"
+
+
+@mcp.tool()
+async def get_recent_branches_by_product_line(product_line: str, days: int = 3) -> list:
+    """
+    获取产品线最近N天新创建的分支列表
+    
+    根据产品线名称，从Apollo获取对应的GitLab项目列表，然后查询这些项目最近N天内新创建的分支。
+    
+    Args:
+        product_line: 产品线名称，如 xingye（星野）, talkie, hailuo（海螺）等
+        days: 查询最近多少天的分支，默认3天
+    
+    Returns:
+        列表，包含新分支信息:
+        [
+            {
+                "repo_url": "https://gitlab.xaminim.com/qa/project_name",
+                "project_name": "project_name",
+                "branch_name": "feature/xxx",
+                "created_at": "2026-02-03T10:30:00+08:00",
+                "author": "developer_name"
+            },
+            ...
+        ]
+    """
+    import concurrent.futures
+    loop = asyncio.get_running_loop()
+    
+    logger.info(f"===== 获取产品线 {product_line} 最近 {days} 天的新分支 =====")
+    
+    with concurrent.futures.ThreadPoolExecutor() as pool:
+        service = GitlabBranchService(product_line)
+        result = await loop.run_in_executor(pool, service.analyze_all_projects, days)
+    
+    logger.info(f"===== 产品线 {product_line} 共找到 {len(result)} 个新分支 =====")
+    return result
+
+
+@mcp.tool()
+async def detect_repo_type(project_id: int, branch: str = None) -> dict:
+    """
+    检测GitLab仓库类型（Server/FE/混合）
+    
+    通过分析仓库的文件结构和依赖，判断项目是后端服务(Server)还是前端项目(FE)。
+    
+    检测规则：
+    - Server特征：go.mod, pom.xml, requirements.txt, Cargo.toml, cmd/, pkg/, internal/
+    - FE特征：package.json + react/vue/angular依赖, src/components/, pages/
+    
+    Args:
+        project_id: GitLab项目ID，可以从分支查询结果中获取
+        branch: 分支名，可选，默认使用仓库默认分支
+    
+    Returns:
+        字典，包含仓库类型信息:
+        {
+            "type": "server" | "fe" | "mixed" | "unknown",
+            "language": "go" | "java" | "python" | "react" | "vue" | ...,
+            "framework": 框架名称（FE项目会返回）,
+            "confidence": 0.0-1.0 置信度,
+            "indicators": ["go.mod", "cmd/", ...] 检测到的特征
+        }
+    """
+    import concurrent.futures
+    loop = asyncio.get_running_loop()
+    
+    logger.info(f"===== 检测项目 {project_id} 的仓库类型 =====")
+    
+    with concurrent.futures.ThreadPoolExecutor() as pool:
+        detector = GitlabRepoTypeDetector()
+        result = await loop.run_in_executor(pool, detector.detect_repo_type, project_id, branch)
+    
+    logger.info(f"===== 项目 {project_id} 类型检测结果: {result['type']} =====")
+    return result
+
+
+@mcp.tool()
+async def get_bedrock_info_by_repo(project_id: int, branch: str = None) -> dict:
+    """
+    获取Git仓库对应的Bedrock部署信息
+    
+    通过解析仓库的配置文件（.gitlab-ci.yml、Makefile等），
+    自动提取Bedrock部署相关信息，包括PSM、项目名、应用名等。
+    
+    解析优先级：
+    1. .gitlab-ci.yml（最可靠）
+    2. Makefile
+    3. deploy.yaml / bedrock.yaml
+    4. 根据项目名推断（兜底方案）
+    
+    Args:
+        project_id: GitLab项目ID，可以从分支查询结果中获取
+        branch: 分支名，可选，默认使用仓库默认分支
+    
+    Returns:
+        字典，包含Bedrock部署信息:
+        {
+            "found": true/false,  # 是否找到配置
+            "source": "gitlab-ci" | "makefile" | "inferred",  # 配置来源
+            "psm": "xxx.yyy.zzz",  # PSM 服务标识
+            "bedrock_project": "project-name",  # Bedrock 项目名
+            "bedrock_app": "app-name",  # Bedrock 应用名
+            "image_repo": "txharbor.xaminim.com/...",  # 镜像仓库地址
+            "deploy_stages": [...],  # 部署阶段配置
+            "confidence": "high" | "medium" | "low",  # 置信度
+            "project_info": {  # 项目基本信息
+                "name": "项目名",
+                "path": "group/project",
+                "web_url": "https://..."
+            }
+        }
+    """
+    import concurrent.futures
+    loop = asyncio.get_running_loop()
+    
+    logger.info(f"===== 获取项目 {project_id} 的 Bedrock 部署信息 =====")
+    
+    with concurrent.futures.ThreadPoolExecutor() as pool:
+        resolver = GitlabBedrockResolver()
+        result = await loop.run_in_executor(pool, resolver.get_bedrock_info, project_id, branch)
+    
+    if result.get('found'):
+        logger.info(f"===== 项目 {project_id} 找到 Bedrock 配置: PSM={result.get('psm')}, App={result.get('bedrock_app')} =====")
+    else:
+        logger.warning(f"===== 项目 {project_id} 未找到 Bedrock 配置 =====")
+    
+    return result
+
 
 def main():
     print("Starting Minimax QA MCP server")

minimax_qa_mcp/src/gitlab_branch_analyzer/__init__.py

@@ -0,0 +1,11 @@
+"""
+coding:utf-8
+@Software: PyCharm
+@Time: 2026/2/4
+@Author: xingyun
+@Desc: GitLab分支分析模块
+"""
+
+from minimax_qa_mcp.src.gitlab_branch_analyzer.gitlab_branch_service import GitlabBranchService
+
+__all__ = ['GitlabBranchService']