cnhkmcp 1.3.7__tar.gz → 1.4.0__tar.gz

{cnhkmcp-1.3.7/cnhkmcp.egg-info → cnhkmcp-1.4.0}/PKG-INFO

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

{cnhkmcp-1.3.7 → cnhkmcp-1.4.0}/cnhkmcp/__init__.py

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

{cnhkmcp-1.3.7 → cnhkmcp-1.4.0}/cnhkmcp/untracked/APP/hkSimulator/autosimulator.py

@@ -1,4 +1,36 @@
-# autosimulator.py
+import sys
+import subprocess
+# --- Dependency Check & Auto-Install ---
+required_imports = [
+    ("getpass", None),
+    ("json", None),
+    ("logging", None),
+    ("os", None),
+    ("threading", None),
+    ("time", None),
+    ("functools", None),
+    ("multiprocessing", None),
+    ("pathlib", None),
+    ("typing", None),
+    ("urllib.parse", None),
+    ("pandas", "pandas"),
+    ("requests", "requests"),
+    ("tqdm", "tqdm"),
+    ("pandas.io.formats.style", "pandas"),
+]
+for mod, pipname in required_imports:
+    try:
+        if "." in mod:
+            __import__(mod.split(".")[0])
+        else:
+            __import__(mod)
+    except ImportError:
+        if pipname:
+            print(f"Installing missing package: {pipname}")
+            subprocess.check_call([sys.executable, "-m", "pip", "install", pipname])
+        else:
+            print(f"Module {mod} is a built-in or not installable via pip.")
+# --- Script Description ---
 """
 Autosimulator for WorldQuant BRAIN platform
 - Timestamped logger

{cnhkmcp-1.3.7 → cnhkmcp-1.4.0}/cnhkmcp/untracked/APP/static/script.js

@@ -1615,10 +1615,9 @@ window.onclick = function(event) {
 // Apply template variables
 function applyTemplate() {
     const variableInput = document.getElementById('variableInput');
-    const variables = variableInput.value
-        .split(',')
-        .map(v => v.trim())
-        .filter(v => v !== '');
+    
+    // Special handling for bucket() functions to avoid splitting on commas inside them
+    const variables = parseVariablesWithBucketSupport(variableInput.value);
     
     if (variables.length === 0) {
         alert('Please enter at least one variable');
@@ -1646,6 +1645,109 @@ function applyTemplate() {
     errorsDiv.innerHTML = `<div class="success-message">✓ Template <${currentTemplate}/> configured as ${currentConfigType} with ${variables.length} variable${variables.length > 1 ? 's' : ''}</div>`;
 }
 
+// Parse variables with special support for bucket() functions
+function parseVariablesWithBucketSupport(input) {
+    const variables = [];
+    let currentVariable = '';
+    let i = 0;
+    
+    while (i < input.length) {
+        const char = input[i];
+        
+        // Check if we're starting a bucket function
+        if (char === 'b' && i + 7 <= input.length && input.substring(i, i + 7) === 'bucket(') {
+            // Add any previous variable before the bucket function
+            const trimmed = currentVariable.trim();
+            if (trimmed !== '') {
+                variables.push(trimmed);
+            }
+            
+            // Find the complete bucket function
+            const bucketFunction = extractBucketFunction(input, i);
+            if (bucketFunction) {
+                // Add the complete bucket function
+                variables.push(bucketFunction.function);
+                currentVariable = '';
+                i = bucketFunction.endIndex + 1; // Move past the bucket function
+                continue;
+            } else {
+                currentVariable += char;
+            }
+        }
+        
+        // Regular comma handling
+        if (char === ',') {
+            const trimmed = currentVariable.trim();
+            if (trimmed !== '') {
+                variables.push(trimmed);
+            }
+            currentVariable = '';
+        } else {
+            currentVariable += char;
+        }
+        
+        i++;
+    }
+    
+    // Add the last variable if there is one
+    const trimmed = currentVariable.trim();
+    if (trimmed !== '') {
+        variables.push(trimmed);
+    }
+    
+    return variables;
+}
+
+// Extract complete bucket function including all nested content
+function extractBucketFunction(input, startIndex) {
+    let parenthesesCount = 0;
+    let inQuotes = false;
+    let quoteChar = null;
+    let i = startIndex;
+    
+    // Find the start of bucket(
+    if (input.substring(i, i + 7) !== 'bucket(') {
+        return null;
+    }
+    
+    parenthesesCount = 1;
+    i += 7; // Move past "bucket("
+    
+    while (i < input.length) {
+        const char = input[i];
+        
+        // Handle quotes
+        if ((char === '"' || char === "'") && !inQuotes) {
+            inQuotes = true;
+            quoteChar = char;
+        } else if (char === quoteChar && inQuotes) {
+            inQuotes = false;
+            quoteChar = null;
+        }
+        
+        // Only count parentheses when not inside quotes
+        if (!inQuotes) {
+            if (char === '(') {
+                parenthesesCount++;
+            } else if (char === ')') {
+                parenthesesCount--;
+                if (parenthesesCount === 0) {
+                    // Found the end of bucket function
+                    const functionText = input.substring(startIndex, i + 1);
+                    return {
+                        function: functionText,
+                        endIndex: i
+                    };
+                }
+            }
+        }
+        
+        i++;
+    }
+    
+    return null; // No matching closing parenthesis found
+}
+
 // Clear editor
 function clearEditor() {
     const editor = document.getElementById('expressionEditor');

{cnhkmcp-1.3.7 → cnhkmcp-1.4.0}/cnhkmcp/untracked/platform_functions.py

@@ -13,6 +13,7 @@ from dataclasses import dataclass, asdict
 from datetime import datetime, timedelta
 import os
 import sys
+import math
 from time import sleep
 
 import requests
@@ -371,6 +372,34 @@ class BrainApiClient:
         except Exception as e:
             self.log(f"Failed to get alpha details: {str(e)}", "ERROR")
             raise
+
+    def _is_atom(self, detail: Optional[Dict[str, Any]]) -> bool:
+        """Match atom detection used in extract_regular_alphas.py:
+        - Primary signal: 'classifications' entries containing 'SINGLE_DATA_SET'
+        - Fallbacks: tags list contains 'atom' or classification id/name contains 'ATOM'
+        """
+        if not detail or not isinstance(detail, dict):
+            return False
+
+        classifications = detail.get('classifications') or []
+        for c in classifications:
+            cid = (c.get('id') or c.get('name') or '')
+            if isinstance(cid, str) and 'SINGLE_DATA_SET' in cid:
+                return True
+
+        # Fallbacks
+        tags = detail.get('tags') or []
+        if isinstance(tags, list):
+            for t in tags:
+                if isinstance(t, str) and t.strip().lower() == 'atom':
+                    return True
+
+        for c in classifications:
+            cid = (c.get('id') or c.get('name') or '')
+            if isinstance(cid, str) and 'ATOM' in cid.upper():
+                return True
+
+        return False
     
     async def get_datasets(self, instrument_type: str = "EQUITY", region: str = "USA", 
                           delay: int = 1, universe: str = "TOP3000", theme: str = "false", search: Optional[str] = None) -> Dict[str, Any]:
@@ -1165,6 +1194,121 @@ class BrainApiClient:
             self.log(f"Failed to get pyramid multipliers: {str(e)}", "ERROR")
             raise
 
+    async def get_diversity_score(self, start_date: str, end_date: str) -> Dict[str, Any]:
+        """Compute diversity score for regular alphas in a date range.
+
+        Description:
+        This diversity score measures three key aspects of work output: the proportion of works
+        with the "Atom" tag (S_A), atom porportion, the breadth of pyramids covered (S_P), and how evenly works
+        are distributed across those pyramids (S_H). Calculated as their product, it rewards
+        strong performance across all three dimensions—encouraging more Atom-tagged works,
+        wider pyramid coverage, and balanced distribution—with weaknesses in any area lowering
+        the total score significantly.
+
+        Inputs (hints for AI callers):
+        - start_date (str): ISO UTC start datetime, e.g. '2025-08-14T00:00:00Z'
+        - end_date (str): ISO UTC end datetime, e.g. '2025-08-18T23:59:59Z'
+        - Note: this tool always uses 'OS' (submission dates) to define the window; callers do not need to supply a stage.
+                - Note: P_max (total number of possible pyramids) is derived from the platform
+                    pyramid-multipliers endpoint and not supplied by callers.
+
+        Returns (compact JSON): {
+            'diversity_score': float,
+            'N': int,  # total regular alphas in window
+            'A': int,  # number of Atom-tagged works (is_single_data_set)
+            'P': int,  # pyramid coverage count in the sample
+            'P_max': int, # used max for normalization
+            'S_A': float, 'S_P': float, 'S_H': float,
+            'per_pyramid_counts': {pyramid_name: count}
+        }
+        """
+        # Fetch user alphas (always use OS / submission dates per product policy)
+        await self.ensure_authenticated()
+        alphas_resp = await self.get_user_alphas(stage='OS', limit=500, submission_start_date=start_date, submission_end_date=end_date)
+
+        if not isinstance(alphas_resp, dict) or 'results' not in alphas_resp:
+            return {'error': 'Unexpected response from get_user_alphas', 'raw': alphas_resp}
+
+        alphas = alphas_resp['results']
+        regular = [a for a in alphas if a.get('type') == 'REGULAR']
+
+        # Fetch details for each regular alpha
+        pyramid_list = []
+        atom_count = 0
+        per_pyramid = {}
+        for a in regular:
+            try:
+                detail = await self.get_alpha_details(a.get('id'))
+            except Exception:
+                continue
+
+            is_atom = self._is_atom(detail)
+            if is_atom:
+                atom_count += 1
+
+            # Extract pyramids
+            ps = []
+            if isinstance(detail.get('pyramids'), list):
+                ps = [p.get('name') for p in detail.get('pyramids') if p.get('name')]
+            else:
+                pt = detail.get('pyramidThemes') or {}
+                pss = pt.get('pyramids') if isinstance(pt, dict) else None
+                if pss and isinstance(pss, list):
+                    ps = [p.get('name') for p in pss if p.get('name')]
+
+            for p in ps:
+                pyramid_list.append(p)
+                per_pyramid[p] = per_pyramid.get(p, 0) + 1
+
+        N = len(regular)
+        A = atom_count
+        P = len(per_pyramid)
+
+        # Determine P_max similarly to the script: use pyramid multipliers if available
+        P_max = None
+        try:
+            pm = await self.get_pyramid_multipliers()
+            if isinstance(pm, dict) and 'pyramids' in pm:
+                pyramids_list = pm.get('pyramids') or []
+                P_max = len(pyramids_list)
+        except Exception:
+            P_max = None
+
+        if not P_max or P_max <= 0:
+            P_max = max(P, 1)
+
+        # Component scores
+        S_A = (A / N) if N > 0 else 0.0
+        S_P = (P / P_max) if P_max > 0 else 0.0
+
+        # Entropy
+        S_H = 0.0
+        if P <= 1 or not per_pyramid:
+            S_H = 0.0
+        else:
+            total_occ = sum(per_pyramid.values())
+            H = 0.0
+            for cnt in per_pyramid.values():
+                q = cnt / total_occ if total_occ > 0 else 0
+                if q > 0:
+                    H -= q * math.log2(q)
+            max_H = math.log2(P) if P > 0 else 1
+            S_H = (H / max_H) if max_H > 0 else 0.0
+
+        diversity_score = S_A * S_P * S_H
+
+        return {
+            'diversity_score': diversity_score,
+            'N': N,
+            'A': A,
+            'P': P,
+            'P_max': P_max,
+            'S_A': S_A,
+            'S_P': S_P,
+            'S_H': S_H,
+            'per_pyramid_counts': per_pyramid
+        }
+
     async def get_pyramid_alphas(self, start_date: Optional[str] = None,
                                end_date: Optional[str] = None) -> Dict[str, Any]:
         """Get user's current alpha distribution across pyramid categories."""
@@ -1514,6 +1658,25 @@ async def authenticate(email: Optional[str] = "", password: Optional[str] = "")
     except Exception as e:
         return {"error": str(e)}
 
+
+@mcp.tool()
+async def get_diversity_score(start_date: str, end_date: str) -> Dict[str, Any]:
+    """Compute and return the diversity score for REGULAR alphas in a submission-date window.
+
+    This MCP tool wraps BrainApiClient.get_diversity_score and always uses submission dates (OS).
+
+    Inputs:
+        - start_date: ISO UTC start datetime (e.g. '2025-08-14T00:00:00Z')
+        - end_date: ISO UTC end datetime (e.g. '2025-08-18T23:59:59Z')
+        - p_max: optional integer total number of pyramid categories for normalization
+
+    Returns: compact JSON with diversity_score, N, A, P, P_max, S_A, S_P, S_H, per_pyramid_counts
+    """
+    try:
+        return await brain_client.get_diversity_score(start_date=start_date, end_date=end_date)
+    except Exception as e:
+        return {"error": str(e)}
+
 @mcp.tool()
 async def manage_config(action: str = "get", settings: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
     """
@@ -2284,7 +2447,7 @@ async def read_specific_documentation(page_id: str) -> Dict[str, Any]:
 # Badge status MCP tool removed as requested
 
 @mcp.tool()
-async def create_multi_regularAlpha_simulation(
+async def create_multiSim(
     alpha_expressions: List[str],
     instrument_type: str = "EQUITY",
     region: str = "USA",
@@ -2574,10 +2737,10 @@ async def get_daily_and_quarterly_payment(email: str = "", password: str = "") -
 
 
 
-# New MCP tool: get_error_message_fromAlphaLocation
+# New MCP tool: get_SimError_detail
 from typing import Sequence
 @mcp.tool()
-async def get_error_message_fromAlphaLocation(locations: Sequence[str]) -> dict:
+async def get_SimError_detail(locations: Sequence[str]) -> dict:
     """
     Fetch and parse error/status from multiple simulation locations (URLs).
     Args:
@@ -2616,6 +2779,6 @@ async def get_error_message_fromAlphaLocation(locations: Sequence[str]) -> dict:
     return {"results": results}
 
 if __name__ == "__main__":
-    print("🧠 WorldQuant BRAIN MCP Server Starting...", file=sys.stderr)
+    print("WorldQuant BRAIN MCP Server Starting...", file=sys.stderr)
     mcp.run()
     

{cnhkmcp-1.3.7 → cnhkmcp-1.4.0}/cnhkmcp/untracked//347/244/272/344/276/213/345/267/245/344/275/234/346/265/201_daily_report_workflow.md

@@ -34,9 +34,11 @@
   2. **社区动态**：从消息中提取社区相关信息，如研究论文或热门话题。
   3. **排行榜变化**：记录用户位置变化。
      - 使用工具：`mcp_brain-api_get_leaderboard`（设置 `user_id` 为用户 ID，如 "CQ89422"）。
+   4. **多样性分数**：收集用户最近一个季度的多样性分数，提出具体建议。
 - **使用的 MCP 工具**：
   - `mcp_brain-api_get_messages`：获取平台公告和社区动态。
   - `mcp_brain-api_get_leaderboard`：获取用户排行榜统计。
+  - `mcp_brain-api_get_diversity_scores`：获取用户多样性分数。
 
 ### 3. 比赛参与与进度
 - **步骤**：

{cnhkmcp-1.3.7 → cnhkmcp-1.4.0/cnhkmcp.egg-info}/PKG-INFO

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

{cnhkmcp-1.3.7 → cnhkmcp-1.4.0}/cnhkmcp.egg-info/SOURCES.txt

@@ -70,4 +70,5 @@ cnhkmcp/untracked/APP/templates/idea_house.html
 cnhkmcp/untracked/APP/templates/index.html
 cnhkmcp/untracked/APP/templates/inspiration_house.html
 cnhkmcp/untracked/APP/templates/paper_analysis.html
-cnhkmcp/untracked/APP/templates/simulator.html
+cnhkmcp/untracked/APP/templates/simulator.html
+cnhkmcp/untracked/__pycache__/forum_functions.cpython-313.pyc

{cnhkmcp-1.3.7 → cnhkmcp-1.4.0}/setup.py

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