cnhkmcp 1.2.0__tar.gz → 1.2.2__tar.gz

{cnhkmcp-1.2.0/cnhkmcp.egg-info → cnhkmcp-1.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cnhkmcp
-Version: 1.2.0
+Version: 1.2.2
 Summary: A comprehensive Model Context Protocol (MCP) server for quantitative trading platform integration
 Home-page: https://github.com/cnhk/cnhkmcp
 Author: CNHK

{cnhkmcp-1.2.0 → cnhkmcp-1.2.2}/cnhkmcp/__init__.py

@@ -50,7 +50,7 @@ from .untracked.forum_functions import (
     read_full_forum_post
 )
 
-__version__ = "1.2.0"
+__version__ = "1.2.2"
 __author__ = "CNHK"
 __email__ = "cnhk@example.com"
 

{cnhkmcp-1.2.0 → cnhkmcp-1.2.2}/cnhkmcp/untracked/platform_functions.py

@@ -389,7 +389,7 @@ class BrainApiClient:
             response = self.session.get(f"{self.base_url}/data-sets", params=params)
             response.raise_for_status()
             response = response.json()
-            response['extraNote'] = "if your returned result is 0, you may want to check your parameter by using get_instrument_options tool to got correct parameter"
+            response['extraNote'] = "if your returned result is 0, you may want to check your parameter by using get_platform_setting_options tool to got correct parameter"
             return response
         except Exception as e:
             self.log(f"Failed to get datasets: {str(e)}", "ERROR")
@@ -423,7 +423,7 @@ class BrainApiClient:
             response = self.session.get(f"{self.base_url}/data-fields", params=params)
             response.raise_for_status()
             response = response.json()
-            response['extraNote'] = "if your returned result is 0, you may want to check your parameter by using get_instrument_options tool to got correct parameter"
+            response['extraNote'] = "if your returned result is 0, you may want to check your parameter by using get_platform_setting_options tool to got correct parameter"
             return response
         except Exception as e:
             self.log(f"Failed to get datafields: {str(e)}", "ERROR")
@@ -436,7 +436,15 @@ class BrainApiClient:
         try:
             response = self.session.get(f"{self.base_url}/alphas/{alpha_id}/recordsets/pnl")
             response.raise_for_status()
-            return response.json()
+            # Some alphas may return 204 No Content or an empty body
+            text = (response.text or "").strip()
+            if not text:
+                return {}
+            try:
+                return response.json()
+            except Exception as parse_err:
+                self.log(f"PnL JSON parse failed for {alpha_id}: {parse_err}", "WARNING")
+                return {}
         except Exception as e:
             self.log(f"Failed to get alpha PnL: {str(e)}", "ERROR")
             raise
@@ -786,9 +794,11 @@ class BrainApiClient:
         await self.ensure_authenticated()
         
         try:
-            response = self.session.get(f"{self.base_url}/alphas/{alpha_id}/recordsets/production-correlation")
+            response = self.session.get(f"{self.base_url}/alphas/{alpha_id}/correlations/prod")
             response.raise_for_status()
-            return response.json()
+            if response.text:
+                return response.json()
+            return {}  # Return empty dict for empty response
         except Exception as e:
             self.log(f"Failed to get production correlation: {str(e)}", "ERROR")
             raise
@@ -798,9 +808,11 @@ class BrainApiClient:
         await self.ensure_authenticated()
         
         try:
-            response = self.session.get(f"{self.base_url}/alphas/{alpha_id}/recordsets/self-correlation")
+            response = self.session.get(f"{self.base_url}/alphas/{alpha_id}/correlations/self")
             response.raise_for_status()
-            return response.json()
+            if response.text:
+                return response.json()
+            return {}  # Return empty dict for empty response
         except Exception as e:
             self.log(f"Failed to get self correlation: {str(e)}", "ERROR")
             raise
@@ -1050,7 +1062,7 @@ class BrainApiClient:
             self.log(f"Failed to get competition agreement: {str(e)}", "ERROR")
             raise
 
-    async def get_instrument_options(self) -> Dict[str, Any]:
+    async def get_platform_setting_options(self) -> Dict[str, Any]:
         """Get available instrument types, regions, delays, and universes."""
         await self.ensure_authenticated()
         
@@ -1200,14 +1212,54 @@ brain_client = BrainApiClient()
 # Configuration management
 CONFIG_FILE = "user_config.json"
 
+def _resolve_config_path(for_write: bool = False) -> str:
+    """Resolve the absolute path to the config file.
+
+    Resolution order:
+    1) BRAIN_CONFIG_PATH env var (explicit file path)
+    2) File in the same directory as this Python module
+    3) Current working directory (read fallback only)
+
+    When for_write is True, prefers module directory (or env path) and will
+    return that even if it doesn't exist yet.
+    """
+    try:
+        # 1) Explicit override
+        env_path = os.environ.get("BRAIN_CONFIG_PATH")
+        if env_path:
+            return os.path.abspath(env_path)
+
+        # 2) Same directory as this .py module
+        module_dir = os.path.dirname(os.path.abspath(__file__))
+        module_path = os.path.join(module_dir, CONFIG_FILE)
+        if not for_write and os.path.exists(module_path):
+            return module_path
+
+        # 3) Fallback to CWD for backward compatibility (read-only preference)
+        cwd_path = os.path.abspath(CONFIG_FILE)
+        if not for_write and os.path.exists(cwd_path):
+            return cwd_path
+
+        # For writes (or when nothing exists), prefer the module directory
+        return module_path
+    except Exception as e:
+        # As a last resort, use the bare filename in CWD
+        logger.error(f"Failed to resolve config path, defaulting to CWD: {e}")
+        return os.path.abspath(CONFIG_FILE)
+
 def load_config() -> Dict[str, Any]:
-    """Load configuration from file."""
-    if os.path.exists(CONFIG_FILE):
+    """Load configuration from file with robust path resolution.
+
+    Looks for the config in this order: BRAIN_CONFIG_PATH -> module directory -> CWD.
+    Returns an empty dict when not found or on error.
+    """
+    path = _resolve_config_path(for_write=False)
+    if os.path.exists(path):
         try:
-            with open(CONFIG_FILE, 'r') as f:
+            with open(path, 'r', encoding='utf-8') as f:
                 return json.load(f)
         except Exception as e:
-            logger.error(f"Failed to load config: {e}")
+            logger.error(f"Failed to load config from '{path}': {e}")
     return {}
 
 def load_brain_credentials() -> Optional[tuple]:
@@ -1225,10 +1277,16 @@ def load_brain_credentials() -> Optional[tuple]:
     return None
 
 def save_config(config: Dict[str, Any]):
-    """Save configuration to file."""
+    """Save configuration to file using the resolved config path.
+
+    Uses BRAIN_CONFIG_PATH if set; otherwise writes next to this module.
+    Ensures the target directory exists.
+    """
     try:
-        with open(CONFIG_FILE, 'w') as f:
-            json.dump(config, f, indent=2)
+        path = _resolve_config_path(for_write=True)
+        os.makedirs(os.path.dirname(path), exist_ok=True)
+        with open(path, 'w', encoding='utf-8') as f:
+            json.dump(config, f, indent=2, ensure_ascii=False)
     except Exception as e:
         logger.error(f"Failed to save config: {e}")
 
@@ -1636,35 +1694,7 @@ async def save_simulation_data(simulation_id: str, filename: str) -> Dict[str, A
     except Exception as e:
         return {"error": str(e)}
 
-@mcp.tool()
-async def analyze_alpha_performance(alpha_id: str) -> Dict[str, Any]:
-    """
-    📊 Comprehensive alpha performance analysis.
-    
-    Args:
-        alpha_id: The alpha ID to analyze
-    
-    Returns:
-        Comprehensive performance analysis
-    """
-    try:
-        # Get alpha details
-        alpha_details = await brain_client.get_alpha_details(alpha_id)
-        
-        # Get PnL data
-        pnl_data = await brain_client.get_alpha_pnl(alpha_id)
-        
-        # Get yearly stats if available
-        yearly_stats = await brain_client.get_alpha_yearly_stats(alpha_id)
-        
-        return {
-            "alpha_details": alpha_details,
-            "pnl_data": pnl_data,
-            "yearly_stats": yearly_stats,
-            "analysis_timestamp": datetime.now().isoformat()
-        }
-    except Exception as e:
-        return {"error": str(e)}
+
 
 @mcp.tool()
 async def get_operators() -> Dict[str, Any]:
@@ -1962,10 +1992,18 @@ async def get_competition_agreement(competition_id: str) -> Dict[str, Any]:
         return {"error": str(e)}
 
 @mcp.tool()
-async def get_instrument_options() -> Dict[str, Any]:
-    """Get available instrument types, regions, delays, and universes."""
+async def get_platform_setting_options() -> Dict[str, Any]:
+    """Discover valid simulation setting options (instrument types, regions, delays, universes, neutralization).
+
+    Use this when a simulation request might contain an invalid/mismatched setting. If an AI or user supplies
+    incorrect parameters (e.g., wrong region for an instrument type), call this tool to retrieve the authoritative
+    option sets and correct the inputs before proceeding.
+
+    Returns:
+        A structured list of valid combinations and choice lists to validate or fix simulation settings.
+    """
     try:
-        return await brain_client.get_instrument_options()
+        return await brain_client.get_platform_setting_options()
     except Exception as e:
         return {"error": str(e)}
 

{cnhkmcp-1.2.0 → cnhkmcp-1.2.2/cnhkmcp.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cnhkmcp
-Version: 1.2.0
+Version: 1.2.2
 Summary: A comprehensive Model Context Protocol (MCP) server for quantitative trading platform integration
 Home-page: https://github.com/cnhk/cnhkmcp
 Author: CNHK

{cnhkmcp-1.2.0 → cnhkmcp-1.2.2}/setup.py

@@ -13,7 +13,7 @@ def read_requirements():
 
 setup(
     name="cnhkmcp",
-    version="1.2.0",
+    version="1.2.2",
     author="CNHK",
     author_email="cnhk@example.com",
     description="A comprehensive Model Context Protocol (MCP) server for quantitative trading platform integration",