aigroup-econ-mcp 0.3.2__py3-none-any.whl → 0.3.4__py3-none-any.whl

aigroup_econ_mcp/__init__.py

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

aigroup_econ_mcp/server.py

@@ -2203,7 +2203,7 @@ async def var_forecast(
                     text=f"VAR模型预测结果：\n"
                          f"预测步数: {steps}\n"
                          f"模型滞后阶数: {result['model_order']}\n"
-                         f"AIC = {result['model_aic']:.2f}, BIC = {result['model_bic']:.2f}\n"
+                         f"AIC = {result.get('model_aic', 0):.2f}, BIC = {result.get('model_bic', 0):.2f}\n"
                          f"各变量预测值已生成"
                 )
             ],
@@ -2621,7 +2621,7 @@ async def random_forest_regression_analysis(
                          f"平均绝对误差 = {result.mae:.4f}\n"
                          f"树的数量 = {result.n_estimators}\n"
                          f"最大深度 = {result.max_depth if result.max_depth != -1 else '无限制'}\n"
-                         f"袋外得分 = {result.oob_score:.4f if result.oob_score else 'N/A'}\n\n"
+                         f"袋外得分 = {result.oob_score:.4f if result.oob_score is not None else 'N/A'}\n\n"
                          f"特征重要性：\n" + "\n".join([
                              f"  {feature}: {importance:.4f}" 
                              for feature, importance in result.feature_importance.items()

aigroup_econ_mcp/tools/time_series.py

@@ -262,9 +262,9 @@ def garch_model(
         if not data:
             raise ValueError("Data cannot be empty")
         
-        # Reduced data length requirement from 50 to 30 observations
-        if len(data) < 30:
-            raise ValueError(f"GARCH model requires at least 30 observations, currently have {len(data)}")
+        # Reduced data length requirement from 50 to 20 observations
+        if len(data) < 20:
+            raise ValueError(f"GARCH模型至少需要20个观测点，当前只有{len(data)}个观测点")
         
         # Convert to return series (if data is not returns)
         series = pd.Series(data)
@@ -417,15 +417,33 @@ def impulse_response_analysis(
         df = pd.DataFrame(data)
         
         # Check data length
-        if len(df) < max_lags + 10:
-            raise ValueError(f"Data length ({len(df)}) insufficient, need at least {max_lags + 10} observations")
+        min_obs = max(max_lags + 10, 20)  # 确保足够的数据点
+        if len(df) < min_obs:
+            raise ValueError(f"数据长度({len(df)})不足，需要至少{min_obs}个观测点")
+        
+        # 数据平稳性检查
+        from statsmodels.tsa.stattools import adfuller
+        stationary_vars = []
+        for col in df.columns:
+            adf_result = adfuller(df[col].dropna())
+            if adf_result[1] < 0.05:  # p值 < 0.05 表示平稳
+                stationary_vars.append(col)
+        
+        if len(stationary_vars) < len(df.columns):
+            print(f"警告: 变量 {set(df.columns) - set(stationary_vars)} 可能非平稳，建议进行差分处理")
         
         # Fit VAR model
         model = VAR(df)
         
-        # Select optimal lag order
-        lag_order = model.select_order(maxlags=max_lags)
-        best_lag = lag_order.aic
+        # Select optimal lag order with error handling
+        try:
+            lag_order = model.select_order(maxlags=max_lags)
+            best_lag = lag_order.aic
+            if best_lag is None or best_lag == 0:
+                best_lag = 1  # 默认滞后阶数
+        except Exception as e:
+            print(f"滞后阶数选择失败，使用默认滞后阶数1: {e}")
+            best_lag = 1
         
         # Fit model with optimal lag
         fitted_model = model.fit(best_lag)
@@ -462,15 +480,8 @@ def impulse_response_analysis(
                 "model_order": best_lag
             }
         
-        return {
-            "impulse_responses": impulse_responses,
-            "orthogonalized": irf.orth_irfs.tolist() if hasattr(irf, 'orth_irfs') else None,
-            "cumulative_effects": irf.cum_effects.tolist() if hasattr(irf, 'cum_effects') else None,
-            "model_order": best_lag
-        }
-        
     except Exception as e:
-        raise ValueError(f"Impulse response analysis failed: {str(e)}")
+        raise ValueError(f"脉冲响应分析失败: {str(e)}")
 
 
 def variance_decomposition(
@@ -484,15 +495,33 @@ def variance_decomposition(
         df = pd.DataFrame(data)
         
         # Check data length
-        if len(df) < max_lags + 10:
-            raise ValueError(f"Data length ({len(df)}) insufficient, need at least {max_lags + 10} observations")
+        min_obs = max(max_lags + 10, 20)  # 确保足够的数据点
+        if len(df) < min_obs:
+            raise ValueError(f"数据长度({len(df)})不足，需要至少{min_obs}个观测点")
+        
+        # 数据平稳性检查
+        from statsmodels.tsa.stattools import adfuller
+        stationary_vars = []
+        for col in df.columns:
+            adf_result = adfuller(df[col].dropna())
+            if adf_result[1] < 0.05:  # p值 < 0.05 表示平稳
+                stationary_vars.append(col)
+        
+        if len(stationary_vars) < len(df.columns):
+            print(f"警告: 变量 {set(df.columns) - set(stationary_vars)} 可能非平稳，建议进行差分处理")
         
         # Fit VAR model
         model = VAR(df)
         
-        # Select optimal lag order
-        lag_order = model.select_order(maxlags=max_lags)
-        best_lag = lag_order.aic
+        # Select optimal lag order with error handling
+        try:
+            lag_order = model.select_order(maxlags=max_lags)
+            best_lag = lag_order.aic
+            if best_lag is None or best_lag == 0:
+                best_lag = 1  # 默认滞后阶数
+        except Exception as e:
+            print(f"滞后阶数选择失败，使用默认滞后阶数1: {e}")
+            best_lag = 1
         
         # Fit model with optimal lag
         fitted_model = model.fit(best_lag)
@@ -525,7 +554,7 @@ def variance_decomposition(
         }
         
     except Exception as e:
-        raise ValueError(f"Variance decomposition failed: {str(e)}")
+        raise ValueError(f"方差分解失败: {str(e)}")
 
 
 def vecm_model(
@@ -549,47 +578,66 @@ def vecm_model(
     try:
         # Data validation
         if not data:
-            raise ValueError("Data cannot be empty")
+            raise ValueError("数据不能为空")
         
         if len(data) < 2:
-            raise ValueError("VECM model requires at least 2 variables")
+            raise ValueError("VECM模型至少需要2个变量")
         
         # Convert to DataFrame
         df = pd.DataFrame(data)
         
         # Check data length
-        if len(df) < max_lags + 10:
-            raise ValueError(f"Data length ({len(df)}) insufficient, need at least {max_lags + 10} observations")
+        min_obs = max(max_lags + 10, 30)  # 确保足够的数据点
+        if len(df) < min_obs:
+            raise ValueError(f"数据长度({len(df)})不足，需要至少{min_obs}个观测点")
         
-        # Simplified implementation:
-        # Simplified implementation: use VAR model as base
-        # In practice, should use specialized VECM implementation
+        # 数据平稳性检查
+        from statsmodels.tsa.stattools import adfuller
+        stationary_vars = []
+        for col in df.columns:
+            adf_result = adfuller(df[col].dropna())
+            if adf_result[1] < 0.05:  # p值 < 0.05 表示平稳
+                stationary_vars.append(col)
+        
+        if len(stationary_vars) < len(df.columns):
+            print(f"警告: 变量 {set(df.columns) - set(stationary_vars)} 可能非平稳，建议进行差分处理")
+        
+        # 简化实现：使用VAR模型作为基础
+        # 在实际应用中，应该使用专门的VECM实现
         
         # Fit VAR model
         model = VAR(df)
-        lag_order = model.select_order(maxlags=max_lags)
-        best_lag = lag_order.aic
+        
+        # Select optimal lag order with error handling
+        try:
+            lag_order = model.select_order(maxlags=max_lags)
+            best_lag = lag_order.aic
+            if best_lag is None or best_lag == 0:
+                best_lag = 1  # 默认滞后阶数
+        except Exception as e:
+            print(f"滞后阶数选择失败，使用默认滞后阶数1: {e}")
+            best_lag = 1
         
         fitted_model = model.fit(best_lag)
         
-        # Build coefficients
+        # Build coefficients with proper error handling
         coefficients = {}
         for i, col in enumerate(df.columns):
             coefficients[col] = {}
             # Add constant term
-            coefficients[col]['const'] = 0.0  # Simplified implementation
+            coefficients[col]['const'] = 0.0  # 简化实现
             # Add error correction term
-            coefficients[col]['ecm'] = -0.1  # Simplified implementation
+            coefficients[col]['ecm'] = -0.1  # 简化实现
         
         # Build error correction terms
         error_correction = {}
         for col in df.columns:
-            error_correction[col] = -0.1  # Simplified implementation
+            error_correction[col] = -0.1  # 简化实现
         
-        # Build cointegration vectors
+        # Build cointegration vectors with proper rank handling
         cointegration_vectors = []
-        for i in range(coint_rank):
-            vector = [1.0] + [-0.5] * (len(df.columns) - 1)  # Simplified implementation
+        for i in range(min(coint_rank, len(df.columns))):  # 确保秩不超过变量数量
+            vector = [1.0] + [-0.5] * (len(df.columns) - 1)  # 简化实现
             cointegration_vectors.append(vector)
         
         return VECMModelResult(
@@ -603,7 +651,7 @@ def vecm_model(
         )
         
     except Exception as e:
-        raise ValueError(f"VECM model fitting failed: {str(e)}")
+        raise ValueError(f"VECM模型拟合失败: {str(e)}")
 
 
 def forecast_var(

{aigroup_econ_mcp-0.3.2.dist-info → aigroup_econ_mcp-0.3.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aigroup-econ-mcp
-Version: 0.3.2
+Version: 0.3.4
 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.3.2.dist-info → aigroup_econ_mcp-0.3.4.dist-info}/RECORD

@@ -1,7 +1,7 @@
-aigroup_econ_mcp/__init__.py,sha256=h59QFKzpqwLe9sM2qFg36ELx8GLqtZvDMbraakPCAVw,490
+aigroup_econ_mcp/__init__.py,sha256=1JAE9wCfHjjYgRBmpjztBnEv_PRYxy_ikQDOjOpjXg4,490
 aigroup_econ_mcp/cli.py,sha256=oAYGd-BqTzvwx-sqcJsLiK2V8GieE90c68mGMtEoYjI,3378
 aigroup_econ_mcp/config.py,sha256=ab5X4-H8isIe2nma0c0AOqlyYgwhf5kfe9Zx5XRrzIo,18876
-aigroup_econ_mcp/server.py,sha256=GjNzsc0Pj-0E-e6JWsPEcDKobr4oLQeWsblTQjJi2s8,106680
+aigroup_econ_mcp/server.py,sha256=Oipgc7SCxz_FwB3a6EX_gR1HxR7E1jw8r1uUoAb4Yj8,106706
 aigroup_econ_mcp/tools/__init__.py,sha256=gJCT-Tzx5cPnVhV68GRffModLCY5DdyETvK_UBZg7J0,325
 aigroup_econ_mcp/tools/base.py,sha256=CwZFtvagcv732OAyCecEfwj8vekrOHSKjPXwrWamW2g,8163
 aigroup_econ_mcp/tools/cache.py,sha256=Urv2zuycp5dS7Qh-XQWEMrwszq9RZ-il8cz_-WniGgc,15311
@@ -11,10 +11,10 @@ aigroup_econ_mcp/tools/optimized_example.py,sha256=tZVQ2jTzHY_zixTynm4Sq8gj5hz6e
 aigroup_econ_mcp/tools/panel_data.py,sha256=288CZoUN40lEApoYbTXkyoenV-eWFDbyPeV46NhU3-4,18782
 aigroup_econ_mcp/tools/regression.py,sha256=uMGRGUQo4mU1sb8fwpP2FpkCqt_e9AtqEtUpInACtJo,6443
 aigroup_econ_mcp/tools/statistics.py,sha256=2cHgNSUXwPYPLxntVOEOL8yF-x92mrgjK-R8kkxDihg,4239
-aigroup_econ_mcp/tools/time_series.py,sha256=clBJVbL8TT05nq6BBfU1nt9YO4vJFbCgL0aU7x_5Sgk,23008
+aigroup_econ_mcp/tools/time_series.py,sha256=5Zy-CPH9HD5u268U3BkFK-DWq7akiEbMZBnrmNfjSK8,25187
 aigroup_econ_mcp/tools/validation.py,sha256=F7LHwog5xtFIMjD9D48kd8jAF5MsZb7wjdrgaOg8EKo,16657
-aigroup_econ_mcp-0.3.2.dist-info/METADATA,sha256=Vyd1AGr0EPm2LFWW0e1tHEi9WLkK5GmVdDlI2Kq3cmU,11159
-aigroup_econ_mcp-0.3.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-aigroup_econ_mcp-0.3.2.dist-info/entry_points.txt,sha256=j5ZJYOc4lAZV-X3XkAuGhzHtIRcJtZ6Gz8ZKPY_QTrM,62
-aigroup_econ_mcp-0.3.2.dist-info/licenses/LICENSE,sha256=DoyCJUWlDzKbqc5KRbFpsGYLwLh-XJRHKQDoITjb1yc,1083
-aigroup_econ_mcp-0.3.2.dist-info/RECORD,,
+aigroup_econ_mcp-0.3.4.dist-info/METADATA,sha256=ZGPwVCzOI3ijbPKaV_71iXt8frf9ZjWyxcPDz2elk9s,11159
+aigroup_econ_mcp-0.3.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+aigroup_econ_mcp-0.3.4.dist-info/entry_points.txt,sha256=j5ZJYOc4lAZV-X3XkAuGhzHtIRcJtZ6Gz8ZKPY_QTrM,62
+aigroup_econ_mcp-0.3.4.dist-info/licenses/LICENSE,sha256=DoyCJUWlDzKbqc5KRbFpsGYLwLh-XJRHKQDoITjb1yc,1083
+aigroup_econ_mcp-0.3.4.dist-info/RECORD,,