cnhkmcp 1.2.5__tar.gz → 1.2.6__tar.gz

{cnhkmcp-1.2.5/cnhkmcp.egg-info → cnhkmcp-1.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cnhkmcp
-Version: 1.2.5
+Version: 1.2.6
 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.5 → cnhkmcp-1.2.6}/cnhkmcp/__init__.py

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

{cnhkmcp-1.2.5 → cnhkmcp-1.2.6}/cnhkmcp/untracked/platform_functions.py

@@ -2428,6 +2428,83 @@ async def _wait_for_multisimulation_completion(location: str, expected_children:
     except Exception as e:
         return {"error": f"Error waiting for multisimulation completion: {str(e)}"}
 
+@mcp.tool()
+async def get_daily_and_quarterly_payment(email: str = "", password: str = "") -> Dict[str, Any]:
+    """
+    Get daily and quarterly payment information from WorldQuant BRAIN platform.
+    
+    This function retrieves both base payments (daily alpha performance payments) and 
+    other payments (competition rewards, quarterly payments, referrals, etc.).
+    
+    Args:
+        email: Your BRAIN platform email address (optional if in config)
+        password: Your BRAIN platform password (optional if in config)
+    
+    Returns:
+        Dictionary containing base payment and other payment data with summaries and detailed records
+    """
+    try:
+        # Initialize client
+        brain_client = BrainApiClient()
+        
+        # Authenticate if credentials provided
+        if email and password:
+            auth_result = await brain_client.authenticate(email, password)
+            if auth_result.get('status') != 'authenticated':
+                return {"error": f"Authentication failed: {auth_result.get('message', 'Unknown error')}"}
+        else:
+            # Try to use existing session or config
+            config = await manage_config("get")
+            if not config.get('is_authenticated'):
+                return {"error": "Not authenticated. Please provide email and password or authenticate first."}
+        
+        # Get base payment data
+        base_payment_response = brain_client.session.get(
+            'https://api.worldquantbrain.com/users/self/activities/base-payment'
+        )
+        
+        if base_payment_response.status_code != 200:
+            return {"error": f"Failed to get base payment data: {base_payment_response.status_code}"}
+        
+        base_payment_data = base_payment_response.json()
+        
+        # Get other payment data
+        other_payment_response = brain_client.session.get(
+            'https://api.worldquantbrain.com/users/self/activities/other-payment'
+        )
+        
+        if other_payment_response.status_code != 200:
+            return {"error": f"Failed to get other payment data: {other_payment_response.status_code}"}
+        
+        other_payment_data = other_payment_response.json()
+        
+        # Return comprehensive payment information
+        return {
+            "success": True,
+            "base_payment": {
+                "summary": {
+                    "yesterday": base_payment_data.get("yesterday"),
+                    "current_quarter": base_payment_data.get("current"),
+                    "previous_quarter": base_payment_data.get("previous"),
+                    "year_to_date": base_payment_data.get("ytd"),
+                    "total_all_time": base_payment_data.get("total"),
+                    "currency": base_payment_data.get("currency")
+                },
+                "daily_records": base_payment_data.get("records", {}).get("records", []),
+                "schema": base_payment_data.get("records", {}).get("schema")
+            },
+            "other_payment": {
+                "total_all_time": other_payment_data.get("total"),
+                "special_payments": other_payment_data.get("records", {}).get("records", []),
+                "schema": other_payment_data.get("records", {}).get("schema"),
+                "currency": other_payment_data.get("currency")
+            },
+            "timestamp": datetime.now().isoformat()
+        }
+        
+    except Exception as e:
+        return {"error": f"Error retrieving payment information: {str(e)}"}
+
 if __name__ == "__main__":
     print("🧠 WorldQuant BRAIN MCP Server Starting...", file=sys.stderr)
     mcp.run()

{cnhkmcp-1.2.5 → cnhkmcp-1.2.6/cnhkmcp.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cnhkmcp
-Version: 1.2.5
+Version: 1.2.6
 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.5 → cnhkmcp-1.2.6}/setup.py

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