aigroup-econ-mcp 0.4.2__py3-none-any.whl → 1.4.3__py3-none-any.whl

tools/time_series_panel_data_tools.py

@@ -0,0 +1,65 @@
+"""
+时间序列和面板数据工具注册
+将时间序列和面板数据模型注册为MCP工具
+"""
+
+from typing import List, Optional
+from .time_series_panel_data_adapter import TimeSeriesPanelDataAdapter
+
+
+def register_time_series_panel_data_tools():
+    """
+    注册时间序列和面板数据工具
+    """
+    # 这个函数可以用于初始化时注册工具
+    pass
+
+
+# 工具函数别名，便于直接调用
+arima_model = TimeSeriesPanelDataAdapter.arima_model
+exponential_smoothing_model = TimeSeriesPanelDataAdapter.exponential_smoothing_model
+garch_model = TimeSeriesPanelDataAdapter.garch_model
+unit_root_tests = TimeSeriesPanelDataAdapter.unit_root_tests
+var_svar_model = TimeSeriesPanelDataAdapter.var_svar_model
+cointegration_analysis = TimeSeriesPanelDataAdapter.cointegration_analysis
+dynamic_panel_model = TimeSeriesPanelDataAdapter.dynamic_panel_model
+
+
+# 工具列表
+TIME_SERIES_PANEL_DATA_TOOLS = [
+    {
+        "name": "arima_model",
+        "function": arima_model,
+        "description": "ARIMA模型，用于时间序列建模和预测"
+    },
+    {
+        "name": "exponential_smoothing_model",
+        "function": exponential_smoothing_model,
+        "description": "指数平滑模型，用于时间序列建模和预测"
+    },
+    {
+        "name": "garch_model",
+        "function": garch_model,
+        "description": "GARCH模型，用于波动率建模"
+    },
+    {
+        "name": "unit_root_tests",
+        "function": unit_root_tests,
+        "description": "单位根检验，用于检验时间序列的平稳性"
+    },
+    {
+        "name": "var_svar_model",
+        "function": var_svar_model,
+        "description": "向量自回归(VAR)和结构向量自回归(SVAR)模型"
+    },
+    {
+        "name": "cointegration_analysis",
+        "function": cointegration_analysis,
+        "description": "协整分析，包括Engle-Granger检验、Johansen检验和VECM模型"
+    },
+    {
+        "name": "dynamic_panel_model",
+        "function": dynamic_panel_model,
+        "description": "动态面板模型，包括差分GMM和系统GMM"
+    }
+]

aigroup_econ_mcp/__init__.py

@@ -1,19 +0,0 @@
-"""
-AIGroup 计量经济学 MCP 服务
-专业计量经济学MCP工具 - 让大模型直接进行数据分析
-
-提供：
-- 描述性统计分析
-- OLS回归分析
-- 时间序列分析
-- 假设检验
-- 模型诊断
-"""
-
-__version__ = "0.4.2"
-__author__ = "AIGroup"
-__description__ = "专业计量经济学MCP工具 - 让大模型直接进行数据分析"
-
-from .server import create_mcp_server
-
-__all__ = ["create_mcp_server", "__version__", "__author__", "__description__"]

aigroup_econ_mcp/cli.py

@@ -1,82 +0,0 @@
-"""
-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='stdio',
-              type=click.Choice(['stdio', 'streamable-http', 'sse']),
-              help='传输协议 (默认: stdio)')
-@click.option('--debug', is_flag=True, help='启用调试模式')
-@click.option('--mount-path', default=None, help='挂载路径')
-@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:
-        from . import __version__
-        click.echo(f"aigroup-econ-mcp v{__version__}", 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()
-
-    # 设置调试模式
-    if debug:
-        mcp_server.settings.debug = True
-        click.echo(f"[DEBUG] 调试模式已启用", err=True)
-
-    # 根据传输协议启动服务器
-    if transport == 'stdio':
-        # stdio模式直接运行，不输出任何日志到stdout（MCP协议通信）
-        # 所有日志输出到stderr
-        from . import __version__
-        click.echo(f"[INFO] aigroup-econ-mcp v{__version__} starting...", err=True)
-        click.echo(f"[INFO] Transport: stdio (MCP protocol)", err=True)
-        if debug:
-            click.echo(f"[DEBUG] Debug mode enabled", err=True)
-        click.echo(f"[INFO] Server ready. Waiting for MCP client connection...", err=True)
-        mcp_server.run(transport='stdio')
-        
-    elif transport == 'streamable-http':
-        # Streamable HTTP模式 - 使用uvicorn手动启动
-        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)
-        
-        # 获取Starlette应用并启动uvicorn服务器
-        app = mcp_server.streamable_http_app()
-        uvicorn.run(app, host=host, port=port, log_level="info")
-        
-    elif transport == 'sse':
-        # SSE模式 - 使用uvicorn手动启动
-        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)
-        
-        # 获取Starlette应用并启动uvicorn服务器
-        app = mcp_server.sse_app()
-        uvicorn.run(app, host=host, port=port, log_level="info")
-
-
-if __name__ == "__main__":
-    cli()