mm-qa-mcp 0.2.0__tar.gz

mm_qa_mcp-0.2.0/MANIFEST.in

@@ -0,0 +1,19 @@
+include README.md
+include pyproject.toml
+include setup.py
+
+# 包含所有配置文件
+recursive-include minimax_qa_mcp *.ini
+recursive-include minimax_qa_mcp *.json
+recursive-include minimax_qa_mcp *.yaml
+recursive-include minimax_qa_mcp *.yml
+recursive-include minimax_qa_mcp *.txt
+
+# 包含所有Python文件
+recursive-include minimax_qa_mcp *.py
+
+# 排除编译文件
+global-exclude *.py[cod]
+global-exclude __pycache__
+global-exclude *.so
+global-exclude .DS_Store 

mm_qa_mcp-0.2.0/PKG-INFO

@@ -0,0 +1,167 @@
+Metadata-Version: 2.4
+Name: mm_qa_mcp
+Version: 0.2.0
+Summary: QA agent service 集合
+Author-email: xingyun <xingyun@minimaxi.com>
+License-Expression: MIT
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: annotated-types>=0.7.0
+Requires-Dist: anyio>=4.9.0
+Requires-Dist: Authlib>=1.5.2
+Requires-Dist: certifi>=2025.1.31
+Requires-Dist: cffi>=1.17.1
+Requires-Dist: charset-normalizer>=3.4.1
+Requires-Dist: classes>=0.4.1
+Requires-Dist: click>=8.1.8
+Requires-Dist: cryptography>=44.0.2
+Requires-Dist: h11>=0.14.0
+Requires-Dist: httpcore>=1.0.7
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: httpx-sse>=0.4.0
+Requires-Dist: idna>=3.10
+Requires-Dist: jsonpatch>=1.33
+Requires-Dist: jsonpointer>=3.0.0
+Requires-Dist: langchain-core>=0.3.51
+Requires-Dist: langsmith>=0.3.27
+Requires-Dist: mcp>=1.6.0
+Requires-Dist: numpy>=2.2.4
+Requires-Dist: orjson>=3.10.16
+Requires-Dist: packaging>=24.2
+Requires-Dist: pandas>=2.2.3
+Requires-Dist: prettytable>=3.16.0
+Requires-Dist: pycparser>=2.22
+Requires-Dist: pydantic>=2.11.3
+Requires-Dist: pydantic-settings>=2.8.1
+Requires-Dist: pydantic_core>=2.33.1
+Requires-Dist: python-dateutil>=2.9.0.post0
+Requires-Dist: python-dotenv>=1.1.0
+Requires-Dist: pytz>=2025.2
+Requires-Dist: PyYAML>=6.0.2
+Requires-Dist: requests>=2.32.3
+Requires-Dist: requests-toolbelt>=1.0.0
+Requires-Dist: six>=1.17.0
+Requires-Dist: sniffio>=1.3.1
+Requires-Dist: sse-starlette>=2.2.1
+Requires-Dist: starlette>=0.46.1
+Requires-Dist: tabulate>=0.9.0
+Requires-Dist: tenacity>=9.1.2
+Requires-Dist: typing-inspection>=0.4.0
+Requires-Dist: typing_extensions>=4.13.1
+Requires-Dist: tzdata>=2025.2
+Requires-Dist: urllib3>=2.3.0
+Requires-Dist: uvicorn>=0.34.0
+Requires-Dist: validators>=0.34.0
+Requires-Dist: wcwidth>=0.2.13
+Requires-Dist: weaviate>=0.1.2
+Requires-Dist: weaviate-client>=3.26.7
+Requires-Dist: zstandard>=0.23.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: mypy>=1.0.0; extra == "dev"
+Requires-Dist: black>=23.0.0; extra == "dev"
+Requires-Dist: isort>=5.12.0; extra == "dev"
+Requires-Dist: twine>=4.0.0; extra == "dev"
+
+# QA - agent service 集合
+
+## 项目结构
+- `server.py`: MCP接口定义和服务启动入口
+- `src/`: 核心功能模块实现
+- `conf/`: 配置文件
+- `utils/`: 工具函数
+- `logs/`: 日志文件夹
+
+## 功能特性
+1. 天气服务(demo)：
+   - 获取美国各州天气警报信息
+   - 基于经纬度获取天气预报
+
+2. 日志检索服务：
+   - 从Grafana获取各业务服务的日志
+   - 支持按关键字、时间等条件筛选
+   - 支持获取服务接口列表
+
+## 环境配置与启动
+
+项目现在使用ux进行构建和打包，提供了多种安装运行方式。
+
+### 1. 从PyPI安装
+
+```shell
+# 直接从PyPI安装
+pip install minimax-qa-mcp
+
+# 运行服务
+minimax-qa-mcp
+```
+
+### 2. 开发模式
+
+```shell
+# 安装依赖
+pip install ux
+pip install -r requirements.txt
+
+# 开发模式运行
+bash run.sh
+# 或
+ux run
+```
+
+### 3. 构建与发布
+
+```shell
+# 构建项目
+bash build.sh
+
+# 发布到PyPI
+bash publish.sh
+```
+
+### 4. 查看启动
+```
+本地生成 log/agent_log.log文件，且打印 
+INFO:     Uvicorn running on http://0.0.0.0:8888 (Press CTRL+C to quit)
+```
+
+### 5. 集成到MCP客户端
+
+#### curl
+ ```shell
+http://0.0.0.0:8888/sse
+ ```
+
+#### Python模块方式
+```shell
+python -m minimax_qa_mcp.server
+```
+
+#### Claude Desktop 配置
+```json
+{
+    "agent_name": {
+      "command": "minimax-qa-mcp",
+      "args": ["--host", "0.0.0.0", "--port", "8888"]
+    }
+}
+```
+
+## API 说明
+
+### 天气服务(demo)
+- `get_alerts(state)`: 获取指定州的天气警报
+- `get_forecast(latitude, longitude)`: 获取指定位置的天气预报
+
+### 日志服务
+- `get_grafana_data(scene, psm, msg, from_time, to_time)`: 获取业务的grafana日志
+- `get_top_methods(scene, psm)`: 获取服务存在调用的接口列表
+- `get_http_data(scene, psm, api_path, from_time, to_time)`: 获取业务http接口流量日志
+- `query_code_segments(query, query_type, limit, exact, advanced, depth, direction, output)`: 查询代码片段
+- `gen_case(input_data, pwd)`: 生成link_case
+- `get_weaviate_info(input_data)`: 获取业务相关信息

mm_qa_mcp-0.2.0/README.md

@@ -0,0 +1,97 @@
+# QA - agent service 集合
+
+## 项目结构
+- `server.py`: MCP接口定义和服务启动入口
+- `src/`: 核心功能模块实现
+- `conf/`: 配置文件
+- `utils/`: 工具函数
+- `logs/`: 日志文件夹
+
+## 功能特性
+1. 天气服务(demo)：
+   - 获取美国各州天气警报信息
+   - 基于经纬度获取天气预报
+
+2. 日志检索服务：
+   - 从Grafana获取各业务服务的日志
+   - 支持按关键字、时间等条件筛选
+   - 支持获取服务接口列表
+
+## 环境配置与启动
+
+项目现在使用ux进行构建和打包，提供了多种安装运行方式。
+
+### 1. 从PyPI安装
+
+```shell
+# 直接从PyPI安装
+pip install minimax-qa-mcp
+
+# 运行服务
+minimax-qa-mcp
+```
+
+### 2. 开发模式
+
+```shell
+# 安装依赖
+pip install ux
+pip install -r requirements.txt
+
+# 开发模式运行
+bash run.sh
+# 或
+ux run
+```
+
+### 3. 构建与发布
+
+```shell
+# 构建项目
+bash build.sh
+
+# 发布到PyPI
+bash publish.sh
+```
+
+### 4. 查看启动
+```
+本地生成 log/agent_log.log文件，且打印 
+INFO:     Uvicorn running on http://0.0.0.0:8888 (Press CTRL+C to quit)
+```
+
+### 5. 集成到MCP客户端
+
+#### curl
+ ```shell
+http://0.0.0.0:8888/sse
+ ```
+
+#### Python模块方式
+```shell
+python -m minimax_qa_mcp.server
+```
+
+#### Claude Desktop 配置
+```json
+{
+    "agent_name": {
+      "command": "minimax-qa-mcp",
+      "args": ["--host", "0.0.0.0", "--port", "8888"]
+    }
+}
+```
+
+## API 说明
+
+### 天气服务(demo)
+- `get_alerts(state)`: 获取指定州的天气警报
+- `get_forecast(latitude, longitude)`: 获取指定位置的天气预报
+
+### 日志服务
+- `get_grafana_data(scene, psm, msg, from_time, to_time)`: 获取业务的grafana日志
+- `get_top_methods(scene, psm)`: 获取服务存在调用的接口列表
+- `get_http_data(scene, psm, api_path, from_time, to_time)`: 获取业务http接口流量日志
+- `query_code_segments(query, query_type, limit, exact, advanced, depth, direction, output)`: 查询代码片段
+- `gen_case(input_data, pwd)`: 生成link_case
+- `get_weaviate_info(input_data)`: 获取业务相关信息

mm_qa_mcp-0.2.0/minimax_qa_mcp/__init__.py

@@ -0,0 +1,14 @@
+"""
+coding:utf-8
+@Software: PyCharm
+@Time: 2025/5/9 18:17
+@Author: xingyun
+"""
+
+"""
+minimax_qa_mcp - QA agent service 集合
+"""
+
+__version__ = "0.1.0"
+__author__ = "xingyun"
+__all__ = []

mm_qa_mcp-0.2.0/minimax_qa_mcp/conf/__init__.py

@@ -0,0 +1,6 @@
+"""
+coding:utf-8
+@Software: PyCharm
+@Time: 2024/8/20 下午5:54
+@Author: xingyun
+"""

mm_qa_mcp-0.2.0/minimax_qa_mcp/conf/conf.ini

@@ -0,0 +1,86 @@
+[common]
+env = [test, prod]
+business = [xingye, talkie]
+grafana_url = https://mlogs.xaminim.com/api/v1/query
+swing_url = http://swing-babel-ali-prod.xaminim.com/
+
+[xingye_http_prod_business_info]
+grafana_cluster = ali-shanghai-prod-01
+grafana_name_space = kube-system
+
+[talkie_http_prod_business_info]
+grafana_cluster = ali-shanghai-prod-01
+grafana_name_space = kube-system
+
+[xingye_prod_business_info]
+grafana_cluster = ali-shanghai-prod-01
+grafana_name_space = xingye-prod
+
+[xingye_test_business_info]
+grafana_cluster = ali-shanghai-test-01
+grafana_name_space = xingye-test
+
+[talkie_prod_business_info]
+grafana_cluster = ali-virginia-test-01
+grafana_name_space = weaver
+
+[talkie_test_business_info]
+grafana_cluster = ali-virginia-prod-01
+grafana_name_space = weaver
+
+[hailuo_video_cn_pre_business_info]
+grafana_cluster = ali-wulan-prod-01
+grafana_name_space = inspo-pre
+
+[hailuo_video_cn_prod_business_info]
+grafana_cluster = ali-wulan-prod-01
+grafana_name_space = inspo-prod
+
+[hailuo_video_us_test_business_info]
+grafana_cluster = ali-virginia-prod-01
+grafana_name_space = moss-test
+
+[hailuo_video_us_prod_business_info]
+grafana_cluster = ali-virginia-prod-01
+grafana_name_space = moss-prod
+
+[open_platform_cn_prod_business_info]
+grafana_cluster = ali-wulan-prod-01
+grafana_name_space = algeng-prod
+
+[open_platform_us_prod_business_info]
+grafana_cluster = ali-virginia-prod-01
+grafana_name_space = algeng-prod
+
+[open_platform_cn_pre_business_info]
+grafana_cluster = ali-wulan-prod-01
+grafana_name_space = algeng-prod
+
+[open_platform_us_pre_business_info]
+grafana_cluster = ali-virginia-prod-01
+grafana_name_space = algeng-prod
+
+[weaviate_url]
+url_port = http://10.11.8.37:8080
+url = http://10.11.8.37
+port = 8080
+grpc_port = 50051
+
+
+[generator_case_conf]
+# 模型API URL，用于ModuleClient类调用
+module_api_url = http://swing-babel-ali-prod.xaminim.com/swing/api/get_module_result
+git_access_token = SrFXdGjtd3AtRVJCkpDn
+
+# 模型超时重视
+model_timeout = 200
+max_workers = 10
+
+# 各业务代码仓库地址
+xingye_git_url = https://gitlab.xaminim.com/qa/apicase_generate_tool.git
+xingye_branch = feat/link_case_demo
+
+# 各业务demo case路径
+xingye_link_case_demo_path = link_cases/xingye/comment/客态正常查看评论区/test_客态正常查看评论区.py
+xingye_pre_demo_case_path = link_cases/xingye/comment/客态正常查看评论区/pre_客态正常查看评论区.py
+

mm_qa_mcp-0.2.0/minimax_qa_mcp/server.py

@@ -0,0 +1,184 @@
+# 标准库导入
+import asyncio
+import sys
+import os
+
+# 将项目根目录添加到Python路径中
+current_dir = os.path.dirname(os.path.abspath(__file__))
+project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+sys.path.append(project_root)
+
+# MCP相关导入
+from mcp.server.fastmcp import FastMCP
+
+# 本地模块导入
+from minimax_qa_mcp.utils.logger import logger
+
+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 main, TYPE_API, TYPE_FUNC, TYPE_CODE, TYPE_ANY, TYPE_FUNC_DETAIL
+
+# Initialize FastMCP server
+mcp = FastMCP("mcp")
+
+
+@mcp.tool()
+async def get_grafana_data(scene: str, psm: str = "", msg: list = [], from_time: str = None,
+                           to_time: str = None) -> str:
+    """ 获取业务的grafana日志
+
+    Args:
+        scene: 枚举值，可选为[xingye_prod、xingye_test、talkie_prod、talkie_test、hailuo_video_cn_pre、hailuo_video_cn_prod、hailuo_video_us_test、hailuo_video_us_prod]其中xingye为星野业务、talkie为talkie业务、hailuo_video为海螺视频业务分国内(cn)和海外(us)，prod代表线上环境，test/pre代表线下环境
+        psm: 服务名称，可选
+        msg: 筛选关键字，可以是多个值的list，需要按照文本(比如trace_id)筛选日志时使用
+        from_time: 可选，但用户提到时间则必填，日志筛选时间区间-开始时间，时间是UTC+8标准格式，如：2025-03-17T15:22:42.885430+08:00
+        to_time: 可选，但用户提到时间则必填，日志筛选时间区间-结束时间，时间是UTC+8标准格式，如：2025-03-17T15:22:42.885430+08:00
+    """
+    # First get the forecast grid endpoint
+    resp = GetFromGrafana(scene, psm, from_time, to_time).post_grafana(msg)
+
+    return resp
+
+
+@mcp.tool()
+async def get_top_methods(scene: str, psm: str = "") -> list:
+    """ 获取服务存在调用的接口列表，以及接口基本说明
+
+    Args:
+        scene: 枚举值，选项列表为[xingye_rpc、talkie_rpc、xingye_http、talkie_http、hailuo_video_cn_http、hailuo_video_us_http]其中xingye为星野业务、talkie为talkie业务、hailuo_video为海螺视频业务分国内(cn)和海外(us)，默认http，用户显示要求时再拼接rpc
+        psm: 服务名称，可选，在用户显示指定时传递
+    """
+    # First get the forecast grid endpoint
+    resp = GetApiFromGrafana(scene, psm).get_need_method()
+
+    return resp
+
+
+@mcp.tool()
+async def get_http_data(scene: str, psm: str = "", api_path: str = "", from_time: str = None,
+                        to_time: str = None) -> str:
+    """ 获取业务http接口流量日志，包含req、resp等信息，会存为csv文件，返回文件访问地址
+
+    Args:
+        scene: 枚举值，可选为[xingye_http_prod、talkie_http_prod、hailuo_video_cn_pre、hailuo_video_cn_prod、hailuo_video_us_test、hailuo_video_us_prod]其中xingye为星野业务、talkie为talkie业务、hailuo_video为海螺视频业务分国内(cn)和海外(us)，prod代表线上环境，test/pre代表线下环境
+        psm: 需要查询的具体服务名称，可选参数
+        api_path: 需要查询的接口路径，必填参数
+        from_time: 日志筛选时间区间-开始时间，时间是UTC+8标准格式，如：2025-03-17T15:22:42.885430+08:00
+        to_time: 日志筛选时间区间-结束时间，时间是UTC+8标准格式，如：2025-03-17T15:22:42.885430+08:00
+    """
+
+    # 调用并行处理方法
+    results = CaseGrafanaService(scene, psm, from_time, to_time).process_qps_file(api_path)
+    return {"file_path": results}
+
+
+@mcp.tool()
+async def query_code_segments(query: str, query_type: str = None, limit: int = 10, exact: bool = False,
+                              advanced: bool = False,
+                              depth: int = 1, direction: str = "both", output: str = None) -> dict:
+    """ 查询代码片段，支持多种查询模式
+
+    Args:
+        query: 查询的内容，可以是API路径、函数名、包名、文件路径或其他代码内容
+        query_type: 查询类型，可选值：API(接口定义), FUNC(函数调用关系), CODE(代码变动的影响), FUNC_DETAIL(函数详情), ANY(所有类型，默认)
+            -- API：接口定义的说明
+            -- FUNC：函数之间的调用关系
+            -- CODE：代码变动带来的影响
+            -- FUNC_DETAIL：函数的具体实现
+            -- ANY：模糊匹配，所有可能的结果返回
+
+        limit: 返回结果的最大数量，默认10条
+        exact: 是否精确匹配，默认False表示模糊匹配
+        advanced: 使用高级代码影响分析，仅在查询类型为CODE时有效
+        depth: 调用链深度，默认为1层，仅在查询类型为FUNC时有效
+        direction: 调用链方向，可选值：caller(调用者), callee(被调用者), both(双向)，仅在查询类型为FUNC时有效
+        output: 输出文件路径，用于保存函数调用图谱，仅在查询类型为FUNC时有效
+
+    返回值说明:
+        - 当查询API时: 返回API的路径、入参、出参和实现逻辑
+        - 当查询函数时: 返回函数的调用链关系，谁调用了此函数，此函数调用了谁
+        - 当查询代码片段时: 返回代码影响的函数列表
+        - 当模糊查询时: 返回按类型分组的查询结果
+        - 当查询函数详情时(FUNC_DETAIL): 返回函数的详细实现代码、所在文件位置及调用关系
+    """
+    # 转换query_type参数
+    if query_type:
+        query_type = query_type.upper()
+        # 确保query_type是有效值
+        if query_type not in [TYPE_API, TYPE_FUNC, TYPE_CODE, TYPE_ANY, TYPE_FUNC_DETAIL]:
+            query_type = None
+
+    # 调用main函数处理查询
+    result = main(
+        input_content=query,
+        input_type=query_type,
+        limit=limit,
+        exact=exact,
+        depth=depth,
+        direction=direction,
+        advanced=advanced,
+        output=output
+    )
+
+    return result
+
+
+@mcp.tool()
+async def gen_case(input_data: dict, pwd: str) -> list:
+    """
+    link_case 生成
+    Args:
+        input_data: JSON格式的输入数据，包含Business、API和Case信息
+        pwd: 用户当前的目录地址
+    Return:
+        列表，模型返回的具体信息以及具体需要保存的路径地址，[{"pre_case_dir": "xxxxx", "pre_case_result", "case_dir": "xxx", "case_result": "xxx"}]
+        pre_case_dir:前置case需要保存的地址，你需要创建这个文件并写入：pre_case_result的内容
+        case_dir：真正case需要保存的地址，你需要创建这个文件并写入：case_result的内容
+    """
+    # 在线程池中执行阻塞操作
+    import concurrent.futures
+    loop = asyncio.get_running_loop()
+    with concurrent.futures.ThreadPoolExecutor() as pool:
+        generator = GeneratorCase(input_data, pwd)
+        logger.info(f"===== The input params is :{input_data}, pwd is :{pwd}")
+        result = await loop.run_in_executor(pool, generator.generator_case)
+    return result
+
+
+@mcp.tool()
+async def get_weaviate_info(input_data: str) -> dict:
+    """
+    检索知识库 获取业务相关信息
+    Args:
+        input_data: str，用户问题
+    Return:
+        业务信息
+    """
+    weaviate_client = GetWeaviateInfo(input_data)
+    result = weaviate_client.get_knowledge()
+    return result
+
+
+def main():
+    print("Starting Minimax QA MCP server")
+    """Run the Minimax QA MCP server"""
+    mcp.run()
+
+
+def run_server():
+    """命令行入口点函数，用于PyPI包安装后的命令行调用"""
+    # 确保当前工作目录在sys.path中
+    if os.getcwd() not in sys.path:
+        sys.path.insert(0, os.getcwd())
+    
+    # 输出启动信息
+    print("Starting Minimax QA MCP server from CLI")
+    
+    # 调用主函数
+    main()
+
+
+if __name__ == "__main__":
+    main()

mm_qa_mcp-0.2.0/minimax_qa_mcp/src/demo_langchain/__init__.py

@@ -0,0 +1,6 @@
+"""
+coding:utf-8
+@Software: PyCharm
+@Time: 2025/4/10 15:12
+@Author: xingyun
+"""

mm_qa_mcp-0.2.0/minimax_qa_mcp/src/demo_langchain/langchain_demo.py

@@ -0,0 +1 @@
+