cnhkmcp 1.2.5__py3-none-any.whl → 1.2.7__py3-none-any.whl

cnhkmcp/__init__.py

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

cnhkmcp/untracked/platform_functions.py

@@ -810,7 +810,7 @@ class BrainApiClient:
             self.log(f"Forum search failed: {str(e)}", "ERROR")
             return {"error": str(e)}
 
-    async def get_forum_post(self, email: str, password: str, article_id: str, 
+    async def read_forum_post(self, email: str, password: str, article_id: str, 
                            headless: bool = False) -> Dict[str, Any]:
         """Get forum post."""
         try:
@@ -2030,7 +2030,7 @@ async def search_forum_posts(search_query: str, email: str = "", password: str =
         return {"error": str(e)}
 
 @mcp.tool()
-async def get_forum_post(article_id: str, email: str = "", password: str = "", 
+async def read_forum_post(article_id: str, email: str = "", password: str = "", 
                         headless: bool = False) -> Dict[str, Any]:
     """
     📄 Get a specific forum post by article ID.
@@ -2428,6 +2428,81 @@ 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:
+        # 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()
+    mcp.run()
+    

{cnhkmcp-1.2.5.dist-info → cnhkmcp-1.2.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cnhkmcp
-Version: 1.2.5
+Version: 1.2.7
 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.7.dist-info/RECORD

@@ -0,0 +1,10 @@
+cnhkmcp/__init__.py,sha256=VI3A7Rs-oPVrJX-aR6KauxyjODzST-gEyIdqFQc9WP0,2758
+cnhkmcp/untracked/forum_functions.py,sha256=QW-CplAsqDkw-Wcwq-1tuZBq48dEO-vXZ8xw7X65EuE,42303
+cnhkmcp/untracked/platform_functions.py,sha256=lJ_ENgn7SU94MoDr0Vta9MwcJS34SE7-4M_j8FMTk88,105451
+cnhkmcp/untracked/user_config.json,sha256=_INn1X1qIsITrmEno-BRlQOAGm9wnNCw-6B333DEvnk,695
+cnhkmcp-1.2.7.dist-info/licenses/LICENSE,sha256=QLxO2eNMnJQEdI_R1UV2AOD-IvuA8zVrkHWA4D9gtoc,1081
+cnhkmcp-1.2.7.dist-info/METADATA,sha256=Yr5UmRTdagf8iX_Xn1p4TId8ApYQ4hhjWHL0Aq-rlvA,5171
+cnhkmcp-1.2.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+cnhkmcp-1.2.7.dist-info/entry_points.txt,sha256=lTQieVyIvjhSMK4fT-XwnccY-JBC1H4vVQ3V9dDM-Pc,70
+cnhkmcp-1.2.7.dist-info/top_level.txt,sha256=x--ibUcSgOS9Z_RWK2Qc-vfs7DaXQN-WMaaxEETJ1Bw,8
+cnhkmcp-1.2.7.dist-info/RECORD,,

cnhkmcp-1.2.5.dist-info/RECORD

@@ -1,10 +0,0 @@
-cnhkmcp/__init__.py,sha256=9lcmFTzQJ3bfvmROp2IucO7n4RWArWsslGxIzyQWvZU,2758
-cnhkmcp/untracked/forum_functions.py,sha256=QW-CplAsqDkw-Wcwq-1tuZBq48dEO-vXZ8xw7X65EuE,42303
-cnhkmcp/untracked/platform_functions.py,sha256=a_WZR9sXAGa_7obv0CXY1-xViEZr4DWwGN8g3_W1IQ0,102025
-cnhkmcp/untracked/user_config.json,sha256=_INn1X1qIsITrmEno-BRlQOAGm9wnNCw-6B333DEvnk,695
-cnhkmcp-1.2.5.dist-info/licenses/LICENSE,sha256=QLxO2eNMnJQEdI_R1UV2AOD-IvuA8zVrkHWA4D9gtoc,1081
-cnhkmcp-1.2.5.dist-info/METADATA,sha256=bo9Zt5xdJD3GgUiC7-GZJVMt36zagZvIrPCC6Fk3ba8,5171
-cnhkmcp-1.2.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-cnhkmcp-1.2.5.dist-info/entry_points.txt,sha256=lTQieVyIvjhSMK4fT-XwnccY-JBC1H4vVQ3V9dDM-Pc,70
-cnhkmcp-1.2.5.dist-info/top_level.txt,sha256=x--ibUcSgOS9Z_RWK2Qc-vfs7DaXQN-WMaaxEETJ1Bw,8
-cnhkmcp-1.2.5.dist-info/RECORD,,