mm-qa-mcp 0.2.0__py3-none-any.whl

minimax_qa_mcp/utils/logger.py

@@ -0,0 +1,38 @@
+import logging
+import os
+from logging.handlers import RotatingFileHandler
+
+# 创建日志目录（如果不存在）
+log_dir = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), 'logs')
+os.makedirs(log_dir, exist_ok=True)
+log_file = os.path.join(log_dir, 'agent_service.log')
+
+# 配置日志格式
+log_format = '[%(asctime)s]-[%(process)d]-[%(levelname)s]- %(message)s'
+formatter = logging.Formatter(log_format)
+
+# 创建logger
+logger = logging.getLogger("agent_service")
+logger.setLevel(logging.INFO)
+
+# 清除已有的处理器，避免重复
+if logger.handlers:
+    logger.handlers.clear()
+
+# 添加控制台处理器
+console_handler = logging.StreamHandler()
+console_handler.setFormatter(formatter)
+logger.addHandler(console_handler)
+
+# 添加文件处理器（使用RotatingFileHandler支持日志轮转）
+file_handler = RotatingFileHandler(
+    log_file,
+    maxBytes=10*1024*1024,  # 10MB
+    backupCount=5,
+    encoding='utf-8'
+)
+file_handler.setFormatter(formatter)
+logger.addHandler(file_handler)
+
+# 避免日志传播到根logger
+logger.propagate = False

minimax_qa_mcp/utils/utils.py

@@ -0,0 +1,246 @@
+"""
+coding:utf-8
+@Software: PyCharm
+@Time: 2024/8/21 上午11:25
+@Author: xingyun
+"""
+import configparser
+import json
+import os
+import subprocess
+import sys
+import pytz
+import importlib.resources
+import pkgutil
+
+from datetime import datetime
+
+
+class Utils:
+    @staticmethod
+    def get_env_info():
+        if 'bedrock' in os.environ:
+            print(os.environ['bedrock'])
+            if 'env' in json.loads(os.environ['bedrock']):
+                return json.loads(os.environ['bedrock'])['env'], json.loads(os.environ['bedrock'])['business']
+        return Utils.get_conf("common", "env"), Utils.get_conf("common", "business")
+
+    @staticmethod
+    def get_conf(section, key):
+        if os.environ.get(key, None):
+            return os.environ[key]
+        else:
+            file = Utils.get_conf_abspath()
+            config = configparser.ConfigParser()
+            config.read(file)
+            try:
+                return config[section][key]
+            except KeyError:
+                raise KeyError(f"conf.ini, section:{section},key:{key} not fould, please check!")
+
+    @staticmethod
+    def set_conf(section, key, value):
+        """
+        设置或更新配置文件中的指定项。
+        :param section: 配置文件中的 section 名称
+        :param key: 配置项名称
+        :param value: 配置值
+        """
+        file = Utils.get_conf_abspath()
+        config = configparser.ConfigParser()
+
+        # 读取现有配置文件（如果存在）
+        if os.path.exists(file):
+            config.read(file)
+
+        # 如果 section 不存在，创建新的 section
+        if not config.has_section(section):
+            config.add_section(section)
+
+        # 设置或更新 key 的值
+        config.set(section, key, value)
+        # 写回配置文件
+        with open(file, "w") as configfile:
+            config.write(configfile)
+
+    @staticmethod
+    def get_conf_abspath():
+        # 首先尝试从包资源中读取配置
+        try:
+            # 检查配置是否作为包资源存在
+            if pkgutil.find_loader("minimax_qa_mcp.conf"):
+                conf_content = pkgutil.get_data("minimax_qa_mcp.conf", "conf.ini")
+                if conf_content:
+                    # 创建临时文件并写入配置内容
+                    temp_dir = os.path.join(os.path.expanduser("~"), ".minimax_qa_mcp")
+                    os.makedirs(temp_dir, exist_ok=True)
+                    temp_file = os.path.join(temp_dir, "conf.ini")
+                    
+                    # 如果文件不存在或需要更新，则写入内容
+                    if not os.path.exists(temp_file):
+                        with open(temp_file, "wb") as f:
+                            f.write(conf_content)
+                    
+                    return temp_file
+        except Exception as e:
+            print(f"从包资源读取配置失败: {e}")
+        
+        # 如果从包资源读取失败，尝试通过路径查找
+        runner_abspath_dirname = os.path.dirname(os.path.abspath(sys.argv[-1]))
+        now_abspath_dirname = runner_abspath_dirname
+        for i in range(10):
+            for root, dirs, files in os.walk(now_abspath_dirname):
+                if "conf" in dirs:
+                    conf_path = os.path.join(now_abspath_dirname, "conf", "conf.ini")
+                    if os.path.exists(conf_path):
+                        return conf_path
+            now_abspath_dirname = os.path.dirname(now_abspath_dirname)
+        
+        # 最后，尝试从当前工作目录查找
+        local_conf = os.path.join(os.getcwd(), "conf", "conf.ini") 
+        if os.path.exists(local_conf):
+            return local_conf
+            
+        raise Exception("not found /conf/conf.ini")
+
+    @staticmethod
+    def get_link_conf_abspath():
+        # 首先尝试从包资源中读取配置
+        try:
+            # 检查配置是否作为包资源存在
+            if pkgutil.find_loader("minimax_qa_mcp.conf"):
+                conf_content = pkgutil.get_data("minimax_qa_mcp.conf", "link_api.txt")
+                if conf_content:
+                    # 创建临时文件并写入配置内容
+                    temp_dir = os.path.join(os.path.expanduser("~"), ".minimax_qa_mcp")
+                    os.makedirs(temp_dir, exist_ok=True)
+                    temp_file = os.path.join(temp_dir, "link_api.txt")
+                    
+                    # 如果文件不存在或需要更新，则写入内容
+                    if not os.path.exists(temp_file):
+                        with open(temp_file, "wb") as f:
+                            f.write(conf_content)
+                    
+                    return temp_file
+        except Exception as e:
+            print(f"从包资源读取配置失败: {e}")
+        
+        # 如果从包资源读取失败，尝试通过路径查找
+        runner_abspath_dirname = os.path.dirname(os.path.abspath(sys.argv[-1]))
+        now_abspath_dirname = runner_abspath_dirname
+        for i in range(10):
+            for root, dirs, files in os.walk(now_abspath_dirname):
+                if "conf" in dirs:
+                    link_path = os.path.join(now_abspath_dirname, "conf", "link_api.txt")
+                    if os.path.exists(link_path):
+                        return link_path
+            now_abspath_dirname = os.path.dirname(now_abspath_dirname)
+        
+        # 最后，尝试从当前工作目录查找
+        local_conf = os.path.join(os.getcwd(), "conf", "link_api.txt")
+        if os.path.exists(local_conf):
+            return local_conf
+        
+        return None
+
+    @staticmethod
+    def get_json_data(path):
+        with open(path, 'r') as f:
+            load_dict = json.load(f)
+        return load_dict
+
+    @staticmethod
+    def write_json(json_data, json_path):
+        with open(json_path, "w") as fp:
+            fp.write(json.dumps(json_data, indent=4))
+
+    @staticmethod
+    def replace_bool_values(data):
+        if isinstance(data, dict):
+            return {k: Utils.replace_bool_values(v) for k, v in data.items()}
+        elif isinstance(data, list):
+            return [Utils.replace_bool_values(item) for item in data]
+        elif data is False:
+            return False
+        elif data is True:
+            return True
+        else:
+            return data
+
+    @staticmethod
+    def get_report_path() -> str:
+        if 'lib' in os.getcwd():
+            root_path = os.path.dirname(os.path.abspath(os.getcwd()))
+        # 命令行处理
+        else:
+            root_path = os.getcwd()
+        report_path = root_path + '/allure/'
+
+        return report_path
+
+    @staticmethod
+    def time_formatted() -> str:
+        # 获取当前日期和时间
+        current_date = datetime.now()
+        # 获取上海时区
+        shanghai_tz = pytz.timezone('Asia/Shanghai')
+        # 将当前日期和时间转换为上海时区
+        shanghai_time = current_date.astimezone(shanghai_tz)
+        # 格式化日期为 YYYYMMDD 格式
+        formatted_date = shanghai_time.strftime('%Y%m%d')
+        return formatted_date
+
+    @staticmethod
+    def time_formatted_shanghai() -> str:
+        # 获取当前日期和时间
+        current_date = datetime.now()
+        # 获取上海时区
+        shanghai_tz = pytz.timezone('Asia/Shanghai')
+        # 将当前日期和时间转换为上海时区
+        shanghai_time = current_date.astimezone(shanghai_tz)
+        # 格式化日期为 YYYYMMDD 格式
+        formatted_date = shanghai_time.strftime('%Y%m%d%H%M%S')
+        return formatted_date
+
+    @staticmethod
+    def run_cmd(cmd):
+        """运行cmd 命令"""
+
+        # 运行CMD命令并获取输出
+        result = subprocess.run(cmd, shell=True, capture_output=True, text=True)
+
+        # 打印命令的输出
+        print("Output:", result.stdout)
+        print("Error:", result.stderr)
+
+
+if __name__ == '__main__':
+    utils = Utils()
+    req = {
+        "cid": 180114436932074,
+        "user_id": 160633434722592,
+        "npc_id": 174392236024298,
+        "mid": 180115467682152,
+        "suggest_count": 0,
+        "scene": 1,
+        "use_safe_model": False,  # 这里是 False 而不是 false
+        "user_msg_text": "（语气平静）多谢。",
+        "base_req": {
+            "AppID": 600,
+            "os": 2,
+            "sys_language": "zh",
+            "ip_region": "cn",
+            "channel": "xy_XIAOMI",
+            "bus_data": {},
+            "Extra": {
+                "UserID": "160633434722592",
+                "DeviceID": "160634651599227",
+                "VersionCode": 1350003
+            }
+        }
+    }
+
+    # 调用函数进行替换
+    req = utils.replace_bool_values(req)
+
+    print(req)

mm_qa_mcp-0.2.0.dist-info/METADATA

@@ -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.dist-info/RECORD

@@ -0,0 +1,28 @@
+minimax_qa_mcp/__init__.py,sha256=7r3jCBYTrmA8VWKtzFA-WSHnlRND2xg9oaZDFV3ozWA,188
+minimax_qa_mcp/server.py,sha256=jgozKCjL-YiA0Sfje4jnCse9tLzRny-oMwccvgWpomE,8178
+minimax_qa_mcp/conf/__init__.py,sha256=s_q9K05OfxIzPXLshF3qPedjADeGzjlfU4w_6PiR62g,85
+minimax_qa_mcp/conf/conf.ini,sha256=ny4CtXIib0ggTxI7wmT4v3dZcdNadN7EcUFcyeREaqo,2393
+minimax_qa_mcp/src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+minimax_qa_mcp/src/demo_langchain/__init__.py,sha256=PZDCBXwlgCJ-UMfOikaTGBrLcOtpJVbuPBzM6RdiLlg,80
+minimax_qa_mcp/src/demo_langchain/langchain_demo.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
+minimax_qa_mcp/src/gateway_case/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+minimax_qa_mcp/src/gateway_case/get_case.py,sha256=djn77_zn_252Mhmk_qn_9k_uz4NAjadq2bEnKyPi7iw,27116
+minimax_qa_mcp/src/generator_case/__init__.py,sha256=Sy4ncJFUvzzFAlwlJAC7cQ2YdD4JFXpmovG3jVVboHc,79
+minimax_qa_mcp/src/generator_case/generator_case.py,sha256=AO6K-LUrpboW54gDZrx4_mpF3jkhmucrPs2_oCXbnnc,50074
+minimax_qa_mcp/src/generator_case/generator_case_langchain.py,sha256=i0OnLBFEWtQymBVH28iILR1XqfKWXhgpYZzewUNSzl0,43256
+minimax_qa_mcp/src/get_weaviate_info/__init__.py,sha256=sMKXpobyDsZBnRFR7ozqPfpJEN1x3cIBM7mTTPrv1PE,80
+minimax_qa_mcp/src/get_weaviate_info/get_weaviate_info.py,sha256=kKjfqDZ0Ka13YuAAT5ru8nB1Twm-JkeJnv1a80lHk8Q,10390
+minimax_qa_mcp/src/grafana/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+minimax_qa_mcp/src/grafana/service.py,sha256=PwgbB99kgwPhy7g1zafkdzCHqwXW8zfG53mlZfAMYqA,4255
+minimax_qa_mcp/src/query_segments/__init__.py,sha256=nNfm8AmQKuQMPPVMtimtgiz_f_yYKY70JoUCa-s7TJ0,80
+minimax_qa_mcp/src/query_segments/query_segments.py,sha256=ZtFEjR9npd1zoCQOVkukvUFXmCub6MaJ_GTy0abLQes,93237
+minimax_qa_mcp/src/xmind2markdown/__init__.py,sha256=IqnCkza9xw5r-0mEetNQoQrkoX-E8yaMmkcgi4MOoPk,80
+minimax_qa_mcp/src/xmind2markdown/xmind_to_markdown.py,sha256=86WZ4aTf1Y4jDLsoBaJZySOeU0G__j_8ZBAtQqccdV4,41903
+minimax_qa_mcp/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+minimax_qa_mcp/utils/logger.py,sha256=R1SyLTra9ngjSu4vKJ21krgl83Or33Q0dRz6-K_OYS8,1089
+minimax_qa_mcp/utils/utils.py,sha256=Xb9d6lz4p3wVDbLKrctfG3v5_Lx1ge6vFMmtAqZ9qF4,8703
+mm_qa_mcp-0.2.0.dist-info/METADATA,sha256=ubINaJAX3AZQRdshpFS6j8VBJO5GiS46IUEe9ngjsjM,4405
+mm_qa_mcp-0.2.0.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+mm_qa_mcp-0.2.0.dist-info/entry_points.txt,sha256=iP2Pl72UfrJCDakHv_Zxg6ci6gtvZipc8cN1SvqueJk,63
+mm_qa_mcp-0.2.0.dist-info/top_level.txt,sha256=cXfV8AjI2zW5yK2aw4Z2TMV9oRPx6eU04K9ZxwqAtxw,15
+mm_qa_mcp-0.2.0.dist-info/RECORD,,

mm_qa_mcp-0.2.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.3.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

mm_qa_mcp-0.2.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+mm_qa_mcp = minimax_qa_mcp.server:run_server

mm_qa_mcp-0.2.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+minimax_qa_mcp