Update qlib/mcp_output/mcp_plugin/mcp_service.py

qlib/mcp_output/mcp_plugin/mcp_service.py

@@ -37,7 +37,7 @@ def initialize_exchange(
     min_cost: float = 5.0,
     limit_threshold: Union[Tuple[str, str], float, None] = None,
     deal_price: Union[str, Tuple[str, str], List[str]] = None,
-    **kwargs: Any,
+    extra_kwargs: Dict[str, Any] = None,
 ) -> dict:
     """
     Initialize and return an Exchange object for backtesting or trading simulations.
@@ -54,12 +54,16 @@ def initialize_exchange(
         min_cost (float): Minimum transaction cost.
         limit_threshold (Union[Tuple[str, str], float, None]): Price movement limits.
         deal_price (Union[str, Tuple[str, str], List[str]]): Price configuration.
-        kwargs (Any): Additional keyword arguments.
+        extra_kwargs (Dict[str, Any]): Additional keyword arguments as a dictionary.
 
     Returns:
         dict: A dictionary containing success, result, or error fields.
     """
     try:
+        # Handle extra_kwargs
+        if extra_kwargs is None:
+            extra_kwargs = {}
+        
         exchange_obj = get_exchange(
             exchange=exchange,
             freq=freq,
@@ -72,9 +76,9 @@ def initialize_exchange(
             min_cost=min_cost,
             limit_threshold=limit_threshold,
             deal_price=deal_price,
-            **kwargs,
+            **extra_kwargs,
         )
-        return {"success": True, "result": exchange_obj}
+        return {"success": True, "result": str(exchange_obj)}
     except Exception as e:
         return {"success": False, "error": str(e)}
 
@@ -108,7 +112,7 @@ def create_account(
             account=account,
             pos_type=pos_type,
         )
-        return {"success": True, "result": account_obj}
+        return {"success": True, "result": str(account_obj)}
     except Exception as e:
         return {"success": False, "error": str(e)}
 
@@ -151,7 +155,7 @@ def initialize_strategy_executor(
             exchange_kwargs=exchange_kwargs,
             pos_type=pos_type,
         )
-        return {"success": True, "result": strategy_executor}
+        return {"success": True, "result": str(strategy_executor)}
     except Exception as e:
         return {"success": False, "error": str(e)}
 
@@ -194,7 +198,13 @@ def run_backtest(
             exchange_kwargs=exchange_kwargs,
             pos_type=pos_type,
         )
-        return {"success": True, "result": {"portfolio_metrics": portfolio_metrics, "trading_indicators": trading_indicators}}
+        return {
+            "success": True, 
+            "result": {
+                "portfolio_metrics": portfolio_metrics.to_dict() if hasattr(portfolio_metrics, 'to_dict') else str(portfolio_metrics),
+                "trading_indicators": trading_indicators.to_dict() if hasattr(trading_indicators, 'to_dict') else str(trading_indicators)
+            }
+        }
     except Exception as e:
         return {"success": False, "error": str(e)}
 
@@ -241,7 +251,7 @@ def collect_trade_data(
             return_value=return_value,
         )
         data = list(data_generator)
-        return {"success": True, "result": data}
+        return {"success": True, "result": [str(item) for item in data]}
     except Exception as e:
         return {"success": False, "error": str(e)}
 
@@ -259,7 +269,7 @@ def format_trade_decisions(decisions: List[Any]) -> dict:
     """
     try:
         formatted_decisions = format_decisions(decisions)
-        return {"success": True, "result": formatted_decisions}
+        return {"success": True, "result": str(formatted_decisions)}
     except Exception as e:
         return {"success": False, "error": str(e)}