cnhkmcp 1.1.8__py3-none-any.whl → 1.2.0__py3-none-any.whl

cnhkmcp/__init__.py

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

cnhkmcp/untracked/platform_functions.py

@@ -441,17 +441,40 @@ class BrainApiClient:
             self.log(f"Failed to get alpha PnL: {str(e)}", "ERROR")
             raise
     
-    async def get_user_alphas(self, stage: str = "IS", limit: int = 100, offset: int = 0) -> Dict[str, Any]:
-        """Get user's alphas."""
+    async def get_user_alphas(
+        self,
+        stage: str = "OS",
+        limit: int = 30,
+        offset: int = 0,
+        start_date: Optional[str] = None,
+        end_date: Optional[str] = None,
+        submission_start_date: Optional[str] = None,
+        submission_end_date: Optional[str] = None,
+        order: Optional[str] = None,
+        hidden: Optional[bool] = None,
+    ) -> Dict[str, Any]:
+        """Get user's alphas with advanced filtering."""
         await self.ensure_authenticated()
         
         try:
             params = {
-                'stage': stage,
-                'limit': limit,
-                'offset': offset
+                "stage": stage,
+                "limit": limit,
+                "offset": offset,
             }
-            
+            if start_date:
+                params["dateCreated>"] = start_date
+            if end_date:
+                params["dateCreated<"] = end_date
+            if submission_start_date:
+                params["dateSubmitted>"] = submission_start_date
+            if submission_end_date:
+                params["dateSubmitted<"] = submission_end_date
+            if order:
+                params["order"] = order
+            if hidden is not None:
+                params["hidden"] = str(hidden).lower()
+
             response = self.session.get(f"{self.base_url}/users/self/alphas", params=params)
             response.raise_for_status()
             return response.json()
@@ -1473,20 +1496,69 @@ async def get_alpha_pnl(alpha_id: str) -> Dict[str, Any]:
         return {"error": str(e)}
 
 @mcp.tool()
-async def get_user_alphas(stage: str = "IS", limit: int = 100, offset: int = 0) -> Dict[str, Any]:
+async def get_user_alphas(
+    stage: str = "IS",
+    limit: int = 30,
+    offset: int = 0,
+    start_date: Optional[str] = None,
+    end_date: Optional[str] = None,
+    submission_start_date: Optional[str] = None,
+    submission_end_date: Optional[str] = None,
+    order: Optional[str] = None,
+    hidden: Optional[bool] = None,
+) -> Dict[str, Any]:
     """
-    👤 Get user's alphas.
-    
+    👤 Get user's alphas with advanced filtering, pagination, and sorting.
+
+    This tool retrieves a list of your alphas, allowing for detailed filtering based on stage,
+    creation date, submission date, and visibility. It also supports pagination and custom sorting.
+
     Args:
-        stage: Alpha stage ("IS" for in-sample, "OS" for out-of-sample)
-        limit: Maximum number of alphas to return
-        offset: Offset for pagination, if the desired total number is larger than 100. then it is useful, every page, the load max is 100 alphas.
+        stage (str): The stage of the alphas to retrieve.
+            - "IS": In-Sample (alphas that have not been submitted).
+            - "OS": Out-of-Sample (alphas that have been submitted).
+            Defaults to "IS".
+        limit (int): The maximum number of alphas to return in a single request.
+            For example, `limit=50` will return at most 50 alphas. Defaults to 30.
+        offset (int): The number of alphas to skip from the beginning of the list.
+            Used for pagination. For example, `limit=50, offset=50` will retrieve alphas 51-100.
+            Defaults to 0.
+        start_date (Optional[str]): The earliest creation date for the alphas to be included.
+            Filters for alphas created on or after this date.
+            Example format: "2023-01-01T00:00:00Z".
+        end_date (Optional[str]): The latest creation date for the alphas to be included.
+            Filters for alphas created before this date.
+            Example format: "2023-12-31T23:59:59Z".
+        submission_start_date (Optional[str]): The earliest submission date for the alphas.
+            Only applies to "OS" alphas. Filters for alphas submitted on or after this date.
+            Example format: "2024-01-01T00:00:00Z".
+        submission_end_date (Optional[str]): The latest submission date for the alphas.
+            Only applies to "OS" alphas. Filters for alphas submitted before this date.
+            Example format: "2024-06-30T23:59:59Z".
+        order (Optional[str]): The sorting order for the returned alphas.
+            Prefix with a hyphen (-) for descending order.
+            Examples: "name" (sort by name ascending), "-dateSubmitted" (sort by submission date descending).
+        hidden (Optional[bool]): Filter alphas based on their visibility.
+            - `True`: Only return hidden alphas.
+            - `False`: Only return non-hidden alphas.
+            If not provided, both hidden and non-hidden alphas are returned.
 
     Returns:
-        User's alphas
+        Dict[str, Any]: A dictionary containing a list of alpha details under the 'results' key,
+        along with pagination information. If an error occurs, it returns a dictionary with an 'error' key.
     """
     try:
-        return await brain_client.get_user_alphas(stage, limit, offset)
+        return await brain_client.get_user_alphas(
+            stage=stage,
+            limit=limit,
+            offset=offset,
+            start_date=start_date,
+            end_date=end_date,
+            submission_start_date=submission_start_date,
+            submission_end_date=submission_end_date,
+            order=order,
+            hidden=hidden,
+        )
     except Exception as e:
         return {"error": str(e)}
 
@@ -1857,12 +1929,11 @@ async def get_pyramid_multipliers() -> Dict[str, Any]:
         return {"error": str(e)}
 
 @mcp.tool()
-async def get_pyramid_alphas(region: Optional[str] = None, delay: Optional[int] = None,
-                           category: Optional[str] = None, start_date: Optional[str] = None,
+async def get_pyramid_alphas(start_date: Optional[str] = None,
                            end_date: Optional[str] = None) -> Dict[str, Any]:
     """Get user's current alpha distribution across pyramid categories."""
     try:
-        return await brain_client.get_pyramid_alphas(region, delay, category, start_date, end_date)
+        return await brain_client.get_pyramid_alphas(start_date, end_date)
     except Exception as e:
         return {"error": str(e)}
 

{cnhkmcp-1.1.8.dist-info → cnhkmcp-1.2.0.dist-info}/METADATA

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

@@ -0,0 +1,10 @@
+cnhkmcp/__init__.py,sha256=cx523h6gaDOZpiANsRDTDKoTQ5oAD98trdlSeA-mNM8,2758
+cnhkmcp/untracked/forum_functions.py,sha256=78wzvN_UYWwbWU40q8_FJNSFPJnND6W9ZRey6MSSiEk,45516
+cnhkmcp/untracked/platform_functions.py,sha256=qdf8WAvOR0PELyaUY-UkQnigBklnMgz-ouS3zWM4IJk,80420
+cnhkmcp/untracked/user_config.json,sha256=_INn1X1qIsITrmEno-BRlQOAGm9wnNCw-6B333DEvnk,695
+cnhkmcp-1.2.0.dist-info/licenses/LICENSE,sha256=QLxO2eNMnJQEdI_R1UV2AOD-IvuA8zVrkHWA4D9gtoc,1081
+cnhkmcp-1.2.0.dist-info/METADATA,sha256=IQmOUwTVHkwwObLiv5JOKQ5KvUv5FNzKnIVKf0mxivs,5171
+cnhkmcp-1.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+cnhkmcp-1.2.0.dist-info/entry_points.txt,sha256=lTQieVyIvjhSMK4fT-XwnccY-JBC1H4vVQ3V9dDM-Pc,70
+cnhkmcp-1.2.0.dist-info/top_level.txt,sha256=x--ibUcSgOS9Z_RWK2Qc-vfs7DaXQN-WMaaxEETJ1Bw,8
+cnhkmcp-1.2.0.dist-info/RECORD,,

cnhkmcp-1.1.8.dist-info/RECORD

@@ -1,10 +0,0 @@
-cnhkmcp/__init__.py,sha256=TRhoxc6hfx6iGAFethhpcYiEi4C4og9J7nI1OzPb8gQ,2758
-cnhkmcp/untracked/forum_functions.py,sha256=78wzvN_UYWwbWU40q8_FJNSFPJnND6W9ZRey6MSSiEk,45516
-cnhkmcp/untracked/platform_functions.py,sha256=zV_2dT0p9QYK6yXD2TsKNBTWOrcDAPvvPs0aP-xtYIQ,77074
-cnhkmcp/untracked/user_config.json,sha256=_INn1X1qIsITrmEno-BRlQOAGm9wnNCw-6B333DEvnk,695
-cnhkmcp-1.1.8.dist-info/licenses/LICENSE,sha256=QLxO2eNMnJQEdI_R1UV2AOD-IvuA8zVrkHWA4D9gtoc,1081
-cnhkmcp-1.1.8.dist-info/METADATA,sha256=Aaq-8nx0k1evkJ9giAWtSUqs06jfK6qeThRPTuwjx_A,5171
-cnhkmcp-1.1.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-cnhkmcp-1.1.8.dist-info/entry_points.txt,sha256=lTQieVyIvjhSMK4fT-XwnccY-JBC1H4vVQ3V9dDM-Pc,70
-cnhkmcp-1.1.8.dist-info/top_level.txt,sha256=x--ibUcSgOS9Z_RWK2Qc-vfs7DaXQN-WMaaxEETJ1Bw,8
-cnhkmcp-1.1.8.dist-info/RECORD,,