mm-qa-mcp 3.0.1__tar.gz → 3.0.3__tar.gz

{mm_qa_mcp-3.0.1/mm_qa_mcp.egg-info → mm_qa_mcp-3.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mm_qa_mcp
-Version: 3.0.1
+Version: 3.0.3
 Summary: QA agent service 集合
 Author-email: xingyun <xingyun@minimaxi.com>
 License-Expression: MIT
@@ -60,6 +60,7 @@ Requires-Dist: validators<1.0.0,>=0.21.2
 Requires-Dist: wcwidth>=0.2.13
 Requires-Dist: weaviate-client<4.0.0,>=3.25.0
 Requires-Dist: zstandard>=0.23.0
+Requires-Dist: pdfplumber>=0.11.6
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0.0; extra == "dev"
 Requires-Dist: mypy>=1.0.0; extra == "dev"

{mm_qa_mcp-3.0.1 → mm_qa_mcp-3.0.3}/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,7 +19,9 @@ 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
 
 # Initialize FastMCP server
@@ -162,6 +165,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 +192,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)
@@ -190,10 +214,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()
 

mm_qa_mcp-3.0.3/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")

mm_qa_mcp-3.0.3/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)

{mm_qa_mcp-3.0.1 → mm_qa_mcp-3.0.3/mm_qa_mcp.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mm_qa_mcp
-Version: 3.0.1
+Version: 3.0.3
 Summary: QA agent service 集合
 Author-email: xingyun <xingyun@minimaxi.com>
 License-Expression: MIT
@@ -60,6 +60,7 @@ Requires-Dist: validators<1.0.0,>=0.21.2
 Requires-Dist: wcwidth>=0.2.13
 Requires-Dist: weaviate-client<4.0.0,>=3.25.0
 Requires-Dist: zstandard>=0.23.0
+Requires-Dist: pdfplumber>=0.11.6
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0.0; extra == "dev"
 Requires-Dist: mypy>=1.0.0; extra == "dev"

{mm_qa_mcp-3.0.1 → mm_qa_mcp-3.0.3}/mm_qa_mcp.egg-info/SOURCES.txt

@@ -7,6 +7,9 @@ minimax_qa_mcp/server.py
 minimax_qa_mcp/conf/__init__.py
 minimax_qa_mcp/conf/conf.ini
 minimax_qa_mcp/src/__init__.py
+minimax_qa_mcp/src/auto_case/__init__.py
+minimax_qa_mcp/src/auto_case/case_write.py
+minimax_qa_mcp/src/auto_case/pdf_jiexi.py
 minimax_qa_mcp/src/gateway_case/__init__.py
 minimax_qa_mcp/src/gateway_case/get_case.py
 minimax_qa_mcp/src/generator_case/__init__.py

{mm_qa_mcp-3.0.1 → mm_qa_mcp-3.0.3}/mm_qa_mcp.egg-info/requires.txt

@@ -47,6 +47,7 @@ validators<1.0.0,>=0.21.2
 wcwidth>=0.2.13
 weaviate-client<4.0.0,>=3.25.0
 zstandard>=0.23.0
+pdfplumber>=0.11.6
 
 [dev]
 pytest>=7.0.0

{mm_qa_mcp-3.0.1 → mm_qa_mcp-3.0.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "mm_qa_mcp"
-version = "3.0.1"
+version = "3.0.3"
 description = "QA agent service 集合"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -68,7 +68,8 @@ dependencies = [
     "validators>=0.21.2,<1.0.0",
     "wcwidth>=0.2.13",
     "weaviate-client>=3.25.0,<4.0.0",
-    "zstandard>=0.23.0"
+    "zstandard>=0.23.0",
+    "pdfplumber>=0.11.6"
 ]
 
 [project.optional-dependencies]