aigroup-econ-mcp 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

aigroup_econ_mcp/__init__.py

@@ -10,7 +10,7 @@ AIGroup 计量经济学 MCP 服务
 - 模型诊断
 """
 
-__version__ = "0.1.0"
+__version__ = "0.1.1"
 __author__ = "AIGroup"
 __description__ = "专业计量经济学MCP工具 - 让大模型直接进行数据分析"
 

aigroup_econ_mcp/cli.py

@@ -2,21 +2,32 @@
 AIGroup 计量经济学 MCP 服务命令行入口
 """
 
+import sys
 import click
-import uvicorn
 from .server import create_mcp_server
 
 
 @click.command()
 @click.option('--port', default=8000, help='服务器端口')
 @click.option('--host', default='127.0.0.1', help='服务器地址')
-@click.option('--transport', default='streamable-http',
+@click.option('--transport', default='stdio',
               type=click.Choice(['stdio', 'streamable-http', 'sse']),
-              help='传输协议')
+              help='传输协议 (默认: stdio)')
 @click.option('--debug', is_flag=True, help='启用调试模式')
 @click.option('--mount-path', default=None, help='挂载路径')
-def main(port: int, host: str, transport: str, debug: bool, mount_path: str):
-    """启动aigroup-econ-mcp服务器"""
+@click.option('--version', is_flag=True, help='显示版本信息')
+def cli(port: int, host: str, transport: str, debug: bool, mount_path: str, version: bool):
+    """AIGroup 计量经济学 MCP 工具
+    
+    默认以stdio模式启动MCP服务器，适用于MCP客户端集成。
+    """
+    
+    # 处理版本标志
+    if version:
+        click.echo("aigroup-econ-mcp v0.1.0", err=True)
+        click.echo("Professional econometrics MCP tool", err=True)
+        click.echo("Author: AIGroup", err=True)
+        sys.exit(0)
 
     # 创建MCP服务器
     mcp_server = create_mcp_server()
@@ -24,29 +35,41 @@ def main(port: int, host: str, transport: str, debug: bool, mount_path: str):
     # 设置调试模式
     if debug:
         mcp_server.settings.debug = True
-        click.echo(f"[DEBUG] 调试模式已启用")
-
-    click.echo(f"[INFO] Starting aigroup-econ-mcp server")
-    click.echo(f"[INFO] Professional econometrics MCP tool for AI data analysis")
-    click.echo(f"[INFO] Transport protocol: {transport}")
-    click.echo(f"[INFO] Service address: http://{host}:{port}")
-    if mount_path:
-        click.echo(f"[INFO] Mount path: {mount_path}")
+        click.echo(f"[DEBUG] 调试模式已启用", err=True)
 
     # 根据传输协议启动服务器
     if transport == 'stdio':
-        # stdio模式直接运行
+        # stdio模式直接运行，不输出任何日志到stdout（MCP协议通信）
+        # 所有日志输出到stderr
+        if debug:
+            click.echo(f"[DEBUG] Starting in stdio mode", err=True)
         mcp_server.run(transport='stdio')
+        
     elif transport == 'streamable-http':
         # Streamable HTTP模式
+        click.echo(f"[INFO] Starting aigroup-econ-mcp server", err=True)
+        click.echo(f"[INFO] Professional econometrics MCP tool for AI data analysis", err=True)
+        click.echo(f"[INFO] Transport protocol: {transport}", err=True)
+        click.echo(f"[INFO] Service address: http://{host}:{port}", err=True)
+        if mount_path:
+            click.echo(f"[INFO] Mount path: {mount_path}", err=True)
+            
         mcp_server.run(
             transport='streamable-http',
             host=host,
             port=port,
             mount_path=mount_path or '/mcp'
         )
+        
     elif transport == 'sse':
         # SSE模式
+        click.echo(f"[INFO] Starting aigroup-econ-mcp server", err=True)
+        click.echo(f"[INFO] Professional econometrics MCP tool for AI data analysis", err=True)
+        click.echo(f"[INFO] Transport protocol: {transport}", err=True)
+        click.echo(f"[INFO] Service address: http://{host}:{port}", err=True)
+        if mount_path:
+            click.echo(f"[INFO] Mount path: {mount_path}", err=True)
+            
         mcp_server.run(
             transport='sse',
             host=host,
@@ -55,24 +78,5 @@ def main(port: int, host: str, transport: str, debug: bool, mount_path: str):
         )
 
 
-@click.command()
-def version():
-    """Show version information"""
-    click.echo("aigroup-econ-mcp v0.1.0")
-    click.echo("Professional econometrics MCP tool")
-    click.echo("Author: AIGroup")
-
-
-@click.group()
-def cli():
-    """AIGroup 计量经济学 MCP 工具"""
-    pass
-
-
-# 添加子命令
-cli.add_command(main)
-cli.add_command(version)
-
-
 if __name__ == "__main__":
     cli()

aigroup_econ_mcp/server.py

@@ -11,6 +11,7 @@ from dataclasses import dataclass
 import pandas as pd
 import numpy as np
 import statsmodels.api as sm
+from statsmodels.tsa import stattools
 from scipy import stats
 from pydantic import BaseModel, Field
 
@@ -320,7 +321,7 @@ async def hypothesis_testing(
 
         elif test_type == "adf":
             # ADF单位根检验
-            result = statsmodels.tsa.stattools.adfuller(data1)
+            result = stattools.adfuller(data1)
             test_result = HypothesisTestResult(
                 test_type="adf",
                 statistic=result[0],
@@ -370,11 +371,11 @@ async def time_series_analysis(
 
     try:
         # ADF单位根检验
-        adf_result = statsmodels.tsa.stattools.adfuller(data)
+        adf_result = stattools.adfuller(data)
 
         # 自相关和偏自相关函数
-        acf_values = statsmodels.tsa.stattools.acf(data, nlags=min(20, len(data)-1))
-        pacf_values = statsmodels.tsa.stattools.pacf(data, nlags=min(20, len(data)-1))
+        acf_values = stattools.acf(data, nlags=min(20, len(data)-1))
+        pacf_values = stattools.pacf(data, nlags=min(20, len(data)-1))
 
         result = TimeSeriesStatsResult(
             adf_statistic=adf_result[0],

{aigroup_econ_mcp-0.1.0.dist-info → aigroup_econ_mcp-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aigroup-econ-mcp
-Version: 0.1.0
+Version: 0.1.1
 Summary: 专业计量经济学MCP工具 - 让大模型直接进行数据分析
 Project-URL: Homepage, https://github.com/aigroup/aigroup-econ-mcp
 Project-URL: Repository, https://github.com/aigroup/aigroup-econ-mcp.git

aigroup_econ_mcp-0.1.1.dist-info/RECORD

@@ -0,0 +1,12 @@
+aigroup_econ_mcp/__init__.py,sha256=JX9x6D6f8r4vFb51i_kN9LmCp57GIpPG6MfiMEQ3bf0,472
+aigroup_econ_mcp/cli.py,sha256=QEQjK6Y7TpgLHpdd8QqQhCjATkJ3-BjYjiXaB4HbJMk,2972
+aigroup_econ_mcp/server.py,sha256=2g8qZBCu3Sed5Wlqp1IDqBBl8R_WA4H8nvLO5V8p7RM,14892
+aigroup_econ_mcp/tools/__init__.py,sha256=o_mAyxeUVxh-xBrggUzS5Yg0uO9FjKvroLXQhFO6ezg,141
+aigroup_econ_mcp/tools/regression.py,sha256=d-WiOSRSPIw5p7nFE6JE34ktQrBBXmZp6eQPfHXTFvA,6230
+aigroup_econ_mcp/tools/statistics.py,sha256=arw5ahGBH4d4pS-wRHh_5z35Yf4LMpiUblR_TLZSbZ0,3415
+aigroup_econ_mcp/tools/time_series.py,sha256=YFGBsKKA1a1FIH3x6gphw4bhVjgw8c-72ibJ4wOeDEM,8122
+aigroup_econ_mcp-0.1.1.dist-info/METADATA,sha256=Mph1KsWcJn39HcX37cBi5ovWmsz7-FH8ClFShDUsC98,5333
+aigroup_econ_mcp-0.1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+aigroup_econ_mcp-0.1.1.dist-info/entry_points.txt,sha256=j5ZJYOc4lAZV-X3XkAuGhzHtIRcJtZ6Gz8ZKPY_QTrM,62
+aigroup_econ_mcp-0.1.1.dist-info/licenses/LICENSE,sha256=x4onvmayh9xPtyeLWVwM9av5hlduofabbcq2z45BbPY,1063
+aigroup_econ_mcp-0.1.1.dist-info/RECORD,,

aigroup_econ_mcp-0.1.0.dist-info/RECORD

@@ -1,12 +0,0 @@
-aigroup_econ_mcp/__init__.py,sha256=z14iciYc9LvvPliqwqkOx4lOuAcJFQL9iIQeHo9JF5Y,472
-aigroup_econ_mcp/cli.py,sha256=3Cq2W9GMMcv3lsHwQkRcNJv5GbJo8pQZylopXyQPOF4,2193
-aigroup_econ_mcp/server.py,sha256=jBPJNUP5rAGsE0EuPBLglNaXt5eIsDjWzFMbwZSfMiQ,14918
-aigroup_econ_mcp/tools/__init__.py,sha256=o_mAyxeUVxh-xBrggUzS5Yg0uO9FjKvroLXQhFO6ezg,141
-aigroup_econ_mcp/tools/regression.py,sha256=d-WiOSRSPIw5p7nFE6JE34ktQrBBXmZp6eQPfHXTFvA,6230
-aigroup_econ_mcp/tools/statistics.py,sha256=arw5ahGBH4d4pS-wRHh_5z35Yf4LMpiUblR_TLZSbZ0,3415
-aigroup_econ_mcp/tools/time_series.py,sha256=YFGBsKKA1a1FIH3x6gphw4bhVjgw8c-72ibJ4wOeDEM,8122
-aigroup_econ_mcp-0.1.0.dist-info/METADATA,sha256=gSaDlcS73fcUcJJcX-AVlyZkfvC4XGG-v7a0W-HGbiw,5333
-aigroup_econ_mcp-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-aigroup_econ_mcp-0.1.0.dist-info/entry_points.txt,sha256=j5ZJYOc4lAZV-X3XkAuGhzHtIRcJtZ6Gz8ZKPY_QTrM,62
-aigroup_econ_mcp-0.1.0.dist-info/licenses/LICENSE,sha256=x4onvmayh9xPtyeLWVwM9av5hlduofabbcq2z45BbPY,1063
-aigroup_econ_mcp-0.1.0.dist-info/RECORD,,