mm-qa-mcp 3.0.2__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

@@ -2,6 +2,7 @@
 import asyncio
 import sys
 import os
+import json
 
 # 将项目根目录添加到Python路径中
 current_dir = os.path.dirname(os.path.abspath(__file__))
@@ -18,8 +19,11 @@ from minimax_qa_mcp.src.generator_case.generator_case import GeneratorCase
 from minimax_qa_mcp.src.get_weaviate_info.get_weaviate_info import GetWeaviateInfo
 from minimax_qa_mcp.src.grafana.service import GetFromGrafana, GetApiFromGrafana
 from minimax_qa_mcp.src.gateway_case.get_case import CaseGrafanaService
-from minimax_qa_mcp.src.query_segments.query_segments import query_main, TYPE_API, TYPE_FUNC, TYPE_CODE, TYPE_ANY, TYPE_FUNC_DETAIL
+from minimax_qa_mcp.src.query_segments.query_segments import query_main, TYPE_API, TYPE_FUNC, TYPE_CODE, TYPE_ANY, \
+    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")
@@ -162,6 +166,24 @@ async def get_weaviate_info(input_data: str) -> dict:
     return result
 
 
+@mcp.tool()
+async def get_auto_case(file_path: str, ref_case_name: str, use_moxing: bool = True) -> dict:
+    """
+    根据给定的文件路径和参考case名称，自动生成case
+
+    Args:
+        file_path: 需要处理的文件路径
+        ref_case_name: 参考的case名称
+        use_moxing: 是否使用模型，开关默认开启    
+    Returns:
+        包含自动生成case的相关信息的字典
+    """
+    # 调用 PDFWeaviateInfo.get_pdf_and_weaviate_info 方法
+    prd_case = PDFWeaviateInfo(file_path, ref_case_name, use_moxing=use_moxing)
+    case_response = prd_case.get_pdf_and_weaviate_info()
+    return case_response
+
+
 @mcp.tool()
 async def get_full_api_call_chain(api_path: str) -> dict:
     """
@@ -171,6 +193,9 @@ async def get_full_api_call_chain(api_path: str) -> dict:
     Return:
         API调用链信息
     """
+    # 判断api_path是不是'/'开头 不是的话 拼接
+    if not api_path.startswith('/'):
+        api_path = '/' + api_path
     logger.info(f"===== The input params is :{api_path}")
 
     api_call_chain = GetFullApiCallChain(api_path)
@@ -178,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")
@@ -190,10 +362,10 @@ def run_server():
     # 确保当前工作目录在sys.path中
     if os.getcwd() not in sys.path:
         sys.path.insert(0, os.getcwd())
-    
+
     # 输出启动信息
     print("Starting Minimax QA MCP server from CLI")
-    
+
     # 调用主函数
     main()
 

minimax_qa_mcp/src/auto_case/case_write.py

@@ -0,0 +1,182 @@
+import functools
+import re
+from minimax_qa_mcp.src.get_weaviate_info.get_weaviate_info import GetWeaviateInfo
+from minimax_qa_mcp.src.auto_case.pdf_jiexi import read_pdf_text
+import logging
+import requests
+import json
+import time
+
+# 设置日志
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+logger = logging.getLogger('生成日志case')
+
+
+def retry(max_attempts=3, wait=2):
+    def decorator(func):
+        @functools.wraps(func)
+        def wrapper(*args, **kwargs):
+            last_exception = None
+            for attempt in range(1, max_attempts + 1):
+                try:
+                    return func(*args, **kwargs)
+                except Exception as e:
+                    logger.warning(f"第{attempt}次请求异常：{e}")
+                    last_exception = e
+                    if attempt < max_attempts:
+                        time.sleep(wait)
+            logger.error(f"请求失败，已重试{max_attempts}次。")
+            return f"请求失败，已重试{max_attempts}次，最后异常：{last_exception}"
+
+        return wrapper
+
+    return decorator
+
+
+class PDFWeaviateInfo(GetWeaviateInfo):
+    def __init__(self, prd_path, input_question,use_moxing, is_need_module=False):
+        super().__init__(input_question, is_need_module)
+        self.prd_path = prd_path
+        self.timeout = 120
+        self.use_moxing = use_moxing
+
+    def get_pdf_and_weaviate_info(self):
+        # 解析PDF内容
+        pdf_text = read_pdf_text(self.prd_path)
+        # 获取Weaviate信息
+        weaviate_info = self.get_knowledge()
+        # 拼接结果
+        result = {
+            'pdf_text': pdf_text,
+            'weaviate_info': weaviate_info
+        }
+
+        case_response = self.call_model(result)
+        return case_response
+
+    def call_model(self, prd_content, max_attempts=5, wait=6):
+        """
+            调用模型，给出参考case
+        Args:
+            prd_content: prd信息+参考case
+            max_attempts: 最大重试次数
+            wait: 重试等待时间（秒）
+
+        Returns:
+            content内容拼接后的字符串，或错误信息
+        """
+        prd_case = prd_content.get("pdf_text", "")
+        prompt = f"需求内容：{prd_case}"
+        cankao_case = prd_content.get("weaviate_info", {}).get("results", [])
+        for i, content in enumerate(cankao_case):
+            case_content = content.get('content', '').strip()
+            case_content_clean = re.sub(r'\s+', ' ', case_content)
+            prompt += f"参考用例{i + 1}、用例名称：{content.get('title', 'N/A')}、用例内容：{case_content_clean}\n\n"
+            break
+
+        prompt += " 帮我写一份测试用例，参考用例若与本次需求相关可借鉴其设计思路和细节，不相关则忽略，用例设计需覆盖功能、边界、异常、安全、兼容性、性能、数据一致性、UI等多个维度，每个测试用例需包含：用例名称、用例编号、前置条件、测试步骤、预期结果、测试环境，用例内容要具体、细致，测试步骤要可操作，预期结果要明确"
+        # prompt += "帮我写一份测试用例，参考用例若与本次需求相关可借鉴其设计思路和细节，不相关则忽略"
+        clean_params = prompt.replace('\\"', "'")
+        prd_prompt = clean_params.replace("\n", " ").strip()
+
+        if not self.use_moxing:
+            return prd_prompt
+
+        payload = {
+            "scene": "qa_agent",
+            "params": {
+                "user_content": prd_prompt
+            }
+        }
+
+        logger.info(f"==== 发送请求调用模型 ======")
+        # print(payload)
+        last_exception = None
+        for attempt in range(1, max_attempts + 1):
+            try:
+                response = requests.post(
+                    self.api_url,
+                    json=payload,
+                    headers={'Content-Type': 'application/json'},
+                    verify=False,
+                    timeout=self.timeout
+                )
+
+                # logger.info(f"API响应状态码: {response.status_code}")
+                # logger.info(f"API响应内容: {response.text}")
+
+                if response.status_code != 200:
+                    logger.error(f"API请求失败，状态码: {response.status_code}")
+                    raise Exception(f"API请求失败，状态码: {response.status_code}")
+
+                try:
+                    resp_json = response.json()
+                    if 'response' in resp_json:
+                        try:
+                            model_response = json.loads(resp_json['response'])
+                            if 'choices' in model_response and isinstance(model_response['choices'], list):
+                                # 兼容OpenAI/Claude等大模型返回结构
+                                for choice in model_response['choices']:
+                                    message = choice.get('message', {})
+                                    content = message.get('content')
+                                    if content:
+                                        content += "\n- 需求内容" + f"{prd_case}"
+                                        return content
+                                # 如果没找到content，兜底返回整个choices
+                                return str(model_response['choices'])
+                            elif 'content' in model_response and isinstance(model_response['content'], list):
+                                # 兼容content直接在顶层的情况
+                                text_content = ""
+                                for item in model_response['content']:
+                                    if item.get('type') == 'text':
+                                        text_content += item.get('text', '')
+                                text_content += "\n- 需求内容" + f"{prd_case}"
+                                return text_content
+                            else:
+                                return str(model_response)
+                        except json.JSONDecodeError as json_e:
+                            logger.error(f"解析二层JSON失败: {json_e}")
+                            # 抛出异常，触发重试机制
+                            raise Exception(f"解析二层JSON失败: {json_e}")
+                        except Exception as e:
+                            logger.error(f"处理二层JSON时发生未知错误: {e}")
+                            # 抛出异常，触发重试机制  
+                            raise Exception(f"处理二层JSON时发生未知错误: {e}")
+                    
+                    # 如果没有response字段，直接返回原始响应
+                    return response.text
+                    
+                except json.JSONDecodeError as json_e:
+                    logger.error(f"解析一层JSON失败: {json_e}")
+                    # 抛出异常，触发重试机制
+                    raise Exception(f"解析一层JSON失败: {json_e}")
+                except Exception as e:
+                    logger.error(f"处理响应时发生未知错误: {e}")
+                    # 抛出异常，触发重试机制
+                    raise Exception(f"处理响应时发生未知错误: {e}")
+                    
+            except requests.RequestException as e:
+                logger.warning(f"第{attempt}次网络请求异常：{e}")
+                last_exception = e
+                if attempt < max_attempts:
+                    time.sleep(wait)
+                    continue
+                else:
+                    break
+            except Exception as e:
+                logger.warning(f"第{attempt}次请求处理异常：{e}")
+                last_exception = e
+                if attempt < max_attempts:
+                    time.sleep(wait)
+                    continue
+                else:
+                    break
+                    
+        logger.error(f"请求失败，已重试{max_attempts}次。最后异常：{last_exception}")
+        return f"请求失败，已重试{max_attempts}次，最后异常：{last_exception}"
+
+
+if __name__ == '__main__':
+    a = PDFWeaviateInfo(prd_path="支持搜索用户功能.pdf", input_question="星野支持搜索用户功能的相关用例")
+    print(a.get_pdf_and_weaviate_info())
+    # read_pdf_text(prd_path = "注销流程优化.pdf")

minimax_qa_mcp/src/auto_case/pdf_jiexi.py

@@ -0,0 +1,28 @@
+import pdfplumber
+import warnings
+import re
+warnings.filterwarnings("ignore")
+def read_pdf_text(pdf_path):
+    """
+    读取指定路径的PDF文件内容，返回所有文本内容的字符串。
+    :param pdf_path: PDF文件的本地路径
+    :return: PDF中的全部文本内容
+    """
+    all_text = ""
+    with pdfplumber.open(pdf_path) as pdf:
+        for page in pdf.pages:
+            all_text += page.extract_text() or ""
+    # 清除特殊不可见字符
+    all_text = re.sub(r'[\x00-\x1F\x7F]', '', all_text)
+    # all_text = clean_encoding(all_text)
+    return all_text
+
+# def clean_encoding(text):
+#     # 先转换为bytes，再解码回字符串
+#     text_bytes = text.encode('utf-8', errors='ignore')
+#     cleaned_text = text_bytes.decode('utf-8')
+#     return cleaned_text
+if __name__ == "__main__":
+    pdf_path = "创作输入框改版.pdf"
+    text = read_pdf_text(pdf_path)
+    print(text)

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']