cnhkmcp 1.1.5__py3-none-any.whl → 1.1.7__py3-none-any.whl

cnhkmcp/__init__.py

@@ -1,125 +1,125 @@
-"""
-CNHK MCP Server - A comprehensive Model Context Protocol server for quantitative trading platform integration.
-"""
-
-from .untracked.platform_functions import (
-    BrainApiClient,
-    authenticate,
-    create_simulation,
-    get_simulation_status,
-    wait_for_simulation,
-    get_alpha_details,
-    get_datasets,
-    get_datafields,
-    get_alpha_pnl,
-    get_user_alphas,
-    submit_alpha,
-    get_events,
-    get_leaderboard,
-    batch_process_alphas,
-    analyze_alpha_performance,
-    get_operators,
-    run_selection,
-    get_user_profile,
-    get_tutorials,
-    get_messages_summary,
-    get_messages,
-    get_alpha_record_sets,
-    check_production_correlation,
-    check_self_correlation,
-    get_submission_check,
-    set_alpha_properties,
-    get_user_activities,
-    get_pyramid_multipliers,
-    get_pyramid_alphas,
-    get_user_competitions,
-    get_competition_details,
-    get_competition_agreement,
-    get_instrument_options,
-    performance_comparison,
-    combine_test_results,
-    generate_alpha_links,
-    get_tutorial_page,
-    get_tutorial_badge_status
-)
-
-from .untracked.forum_functions import (
-    ForumClient,
-    get_glossary_terms,
-    search_forum_posts,
-    read_full_forum_post
-)
-
-__version__ = "1.1.5"
-__author__ = "CNHK"
-__email__ = "cnhk@example.com"
-
-__all__ = [
-    # Main API client
-    "BrainApiClient",
-    
-    # Authentication
-    "authenticate",
-    
-    # Simulation management
-    "create_simulation",
-    "get_simulation_status",
-    "wait_for_simulation",
-    
-    # Alpha management
-    "get_alpha_details",
-    "submit_alpha",
-    "get_user_alphas",
-    "get_alpha_pnl",
-    "get_alpha_record_sets",
-    "set_alpha_properties",
-    "check_production_correlation",
-    "check_self_correlation",
-    "get_submission_check",
-    
-    # Data access
-    "get_datasets",
-    "get_datafields",
-    "get_operators",
-    "run_selection",
-    "get_instrument_options",
-    
-    # Performance analysis
-    "analyze_alpha_performance",
-    "performance_comparison",
-    "combine_test_results",
-    
-    # User management
-    "get_user_profile",
-    "get_user_activities",
-    "get_user_competitions",
-    
-    # Competition and events
-    "get_events",
-    "get_leaderboard",
-    "get_competition_details",
-    "get_competition_agreement",
-    
-    # Pyramid system
-    "get_pyramid_multipliers",
-    "get_pyramid_alphas",
-    
-    # Tutorials and documentation
-    "get_tutorials",
-    "get_tutorial_page",
-    "get_tutorial_badge_status",
-    
-    # Messages
-    "get_messages_summary",
-    "get_messages",
-    
-    # Utilities
-    "batch_process_alphas",
-    "generate_alpha_links",
-    
-    # Forum functionality
-    "ForumClient",
-    "get_glossary_terms",
-    "search_forum_posts",
-    "read_full_forum_post"
+"""
+CNHK MCP Server - A comprehensive Model Context Protocol server for quantitative trading platform integration.
+"""
+
+from .untracked.platform_functions import (
+    BrainApiClient,
+    authenticate,
+    create_simulation,
+    get_simulation_status,
+    wait_for_simulation,
+    get_alpha_details,
+    get_datasets,
+    get_datafields,
+    get_alpha_pnl,
+    get_user_alphas,
+    submit_alpha,
+    get_events,
+    get_leaderboard,
+    batch_process_alphas,
+    analyze_alpha_performance,
+    get_operators,
+    run_selection,
+    get_user_profile,
+    get_tutorials,
+    get_messages_summary,
+    get_messages,
+    get_alpha_record_sets,
+    check_production_correlation,
+    check_self_correlation,
+    get_submission_check,
+    set_alpha_properties,
+    get_user_activities,
+    get_pyramid_multipliers,
+    get_pyramid_alphas,
+    get_user_competitions,
+    get_competition_details,
+    get_competition_agreement,
+    get_instrument_options,
+    performance_comparison,
+    combine_test_results,
+    generate_alpha_links,
+    get_tutorial_page,
+    get_tutorial_badge_status
+)
+
+from .untracked.forum_functions import (
+    ForumClient,
+    get_glossary_terms,
+    search_forum_posts,
+    read_full_forum_post
+)
+
+__version__ = "1.1.7"
+__author__ = "CNHK"
+__email__ = "cnhk@example.com"
+
+__all__ = [
+    # Main API client
+    "BrainApiClient",
+    
+    # Authentication
+    "authenticate",
+    
+    # Simulation management
+    "create_simulation",
+    "get_simulation_status",
+    "wait_for_simulation",
+    
+    # Alpha management
+    "get_alpha_details",
+    "submit_alpha",
+    "get_user_alphas",
+    "get_alpha_pnl",
+    "get_alpha_record_sets",
+    "set_alpha_properties",
+    "check_production_correlation",
+    "check_self_correlation",
+    "get_submission_check",
+    
+    # Data access
+    "get_datasets",
+    "get_datafields",
+    "get_operators",
+    "run_selection",
+    "get_instrument_options",
+    
+    # Performance analysis
+    "analyze_alpha_performance",
+    "performance_comparison",
+    "combine_test_results",
+    
+    # User management
+    "get_user_profile",
+    "get_user_activities",
+    "get_user_competitions",
+    
+    # Competition and events
+    "get_events",
+    "get_leaderboard",
+    "get_competition_details",
+    "get_competition_agreement",
+    
+    # Pyramid system
+    "get_pyramid_multipliers",
+    "get_pyramid_alphas",
+    
+    # Tutorials and documentation
+    "get_tutorials",
+    "get_tutorial_page",
+    "get_tutorial_badge_status",
+    
+    # Messages
+    "get_messages_summary",
+    "get_messages",
+    
+    # Utilities
+    "batch_process_alphas",
+    "generate_alpha_links",
+    
+    # Forum functionality
+    "ForumClient",
+    "get_glossary_terms",
+    "search_forum_posts",
+    "read_full_forum_post"
 ] 

cnhkmcp/untracked/platform_functions.py

@@ -81,8 +81,8 @@ class BrainApiClient:
         print(f"[{level}] {message}", file=sys.stderr)
     
     async def authenticate(self, email: str, password: str) -> Dict[str, Any]:
-        """Authenticate with WorldQuant BRAIN platform using Basic Authentication."""
-        self.log("🔐 Starting Basic Authentication process...", "INFO")
+        """Authenticate with WorldQuant BRAIN platform with biometric support."""
+        self.log("🔐 Starting Authentication process...", "INFO")
         
         try:
             # Store credentials for potential re-authentication
@@ -106,7 +106,7 @@ class BrainApiClient:
             
             # Check for successful authentication (status code 201)
             if response.status_code == 201:
-                self.log("✅ Basic Authentication successful", "SUCCESS")
+                self.log("✅ Authentication successful", "SUCCESS")
                 
                 # Check if JWT token was automatically stored by session
                 jwt_token = self.session.cookies.get('t')
@@ -120,10 +120,26 @@ class BrainApiClient:
                     'user': {'email': email},
                     'status': 'authenticated',
                     'permissions': ['read', 'write'],
-                    'message': 'Basic Authentication successful',
+                    'message': 'Authentication successful',
                     'status_code': response.status_code,
                     'has_jwt': jwt_token is not None
                 }
+            
+            # Check if biometric authentication is required (401 with persona)
+            elif response.status_code == 401:
+                www_auth = response.headers.get("WWW-Authenticate")
+                location = response.headers.get("Location")
+                
+                if www_auth == "persona" and location:
+                    self.log("🔴 Biometric authentication required", "INFO")
+                    
+                    # Handle biometric authentication
+                    from urllib.parse import urljoin
+                    biometric_url = urljoin(response.url, location)
+                    
+                    return await self._handle_biometric_auth(biometric_url, email)
+                else:
+                    raise Exception("Incorrect email or password")
             else:
                 raise Exception(f"Authentication failed with status code: {response.status_code}")
                     
@@ -134,6 +150,97 @@ class BrainApiClient:
             self.log(f"❌ Authentication failed: {str(e)}", "ERROR")
             raise
     
+    async def _handle_biometric_auth(self, biometric_url: str, email: str) -> Dict[str, Any]:
+        """Handle biometric authentication using browser automation."""
+        self.log("🌐 Starting biometric authentication...", "INFO")
+        
+        try:
+            # Import selenium for browser automation
+            from selenium import webdriver
+            from selenium.webdriver.chrome.options import Options
+            import time
+            
+            # Setup Chrome options
+            options = Options()
+            options.add_argument('--no-sandbox')
+            options.add_argument('--disable-dev-shm-usage')
+            
+            driver = None
+            try:
+                # Open browser with timeout
+                driver = webdriver.Chrome(options=options)
+                # Set a short timeout so it doesn't wait forever
+                driver.set_page_load_timeout(80)  # Only wait 5 seconds
+                
+                self.log("🌐 Opening browser for biometric authentication...", "INFO")
+                
+                # Try to open the URL but handle timeout
+                try:
+                    driver.get(biometric_url)
+                    self.log("✅ Browser page loaded successfully", "SUCCESS")
+                except Exception as timeout_error:
+                    self.log(f"⚠️ Page load timeout (expected): {str(timeout_error)[:50]}...", "WARNING")
+                    self.log("✅ Browser window is open for biometric authentication", "INFO")
+                
+                # Print instructions
+                print("\n" + "="*60, file=sys.stderr)
+                print("🔒 BIOMETRIC AUTHENTICATION REQUIRED", file=sys.stderr)
+                print("="*60, file=sys.stderr)
+                print("🌐 Browser window is open with biometric authentication page", file=sys.stderr)
+                print("🔧 Complete the biometric authentication in the browser", file=sys.stderr)
+                print("⏳ The system will automatically check when you're done...", file=sys.stderr)
+                print("="*60, file=sys.stderr)
+                
+                # Keep checking until authentication is complete
+                max_attempts = 60  # 5 minutes maximum (60 * 5 seconds)
+                attempt = 0
+                
+                while attempt < max_attempts:
+                    time.sleep(5)  # Check every 5 seconds
+                    attempt += 1
+                    
+                    # Check if authentication completed
+                    check_response = self.session.post(biometric_url)
+                    self.log(f"🔄 Checking authentication status (attempt {attempt}/{max_attempts}): {check_response.status_code}", "INFO")
+                    
+                    if check_response.status_code == 201:
+                        self.log("✅ Biometric authentication successful!", "SUCCESS")
+                        
+                        # Close browser
+                        driver.quit()
+                        
+                        # Check JWT token
+                        jwt_token = self.session.cookies.get('t')
+                        if jwt_token:
+                            self.log("✅ JWT token received", "SUCCESS")
+                        
+                        # Return success response
+                        return {
+                            'user': {'email': email},
+                            'status': 'authenticated',
+                            'permissions': ['read', 'write'],
+                            'message': 'Biometric authentication successful',
+                            'status_code': check_response.status_code,
+                            'has_jwt': jwt_token is not None
+                        }
+                
+                # If we get here, authentication timed out
+                if driver:
+                    driver.quit()
+                raise Exception("Biometric authentication timed out")
+                
+            except Exception as driver_error:
+                if driver:
+                    try:
+                        driver.quit()
+                    except:
+                        pass
+                raise Exception(f"Browser automation error: {driver_error}")
+                
+        except Exception as e:
+            self.log(f"❌ Biometric authentication failed: {str(e)}", "ERROR")
+            raise
+    
     async def is_authenticated(self) -> bool:
         """Check if currently authenticated using JWT token."""
         try:
@@ -263,7 +370,7 @@ class BrainApiClient:
             raise
     
     async def get_datasets(self, instrument_type: str = "EQUITY", region: str = "USA", 
-                          delay: int = 1, universe: str = "TOP3000", theme: str = "false") -> Dict[str, Any]:
+                          delay: int = 1, universe: str = "TOP3000", theme: str = "false", search: Optional[str] = None) -> Dict[str, Any]:
         """Get available datasets."""
         await self.ensure_authenticated()
         
@@ -276,9 +383,14 @@ class BrainApiClient:
                 'theme': theme
             }
             
+            if search:
+                params['search'] = search
+            
             response = self.session.get(f"{self.base_url}/data-sets", params=params)
             response.raise_for_status()
-            return response.json()
+            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"
+            return response
         except Exception as e:
             self.log(f"Failed to get datasets: {str(e)}", "ERROR")
             raise
@@ -310,7 +422,9 @@ class BrainApiClient:
             
             response = self.session.get(f"{self.base_url}/data-fields", params=params)
             response.raise_for_status()
-            return response.json()
+            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"
+            return response
         except Exception as e:
             self.log(f"Failed to get datafields: {str(e)}", "ERROR")
             raise
@@ -827,26 +941,43 @@ class BrainApiClient:
             self.log(f"Failed to get pyramid multipliers: {str(e)}", "ERROR")
             raise
 
-    async def get_pyramid_alphas(self, region: Optional[str] = None, delay: Optional[int] = None,
-                               category: Optional[str] = None, start_date: Optional[str] = None,
+    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."""
         await self.ensure_authenticated()
         
         try:
             params = {}
-            if region:
-                params['region'] = region
-            if delay is not None:
-                params['delay'] = delay
-            if category:
-                params['category'] = category
             if start_date:
                 params['startDate'] = start_date
             if end_date:
                 params['endDate'] = end_date
             
-            response = self.session.get(f"{self.base_url}/pyramid/alphas", params=params)
+            # Try the user-specific activities endpoint first (like pyramid-multipliers)
+            response = self.session.get(f"{self.base_url}/users/self/activities/pyramid-alphas", params=params)
+            
+            # If that fails, try alternative endpoints
+            if response.status_code == 404:
+                # Try alternative endpoint structure
+                response = self.session.get(f"{self.base_url}/users/self/pyramid/alphas", params=params)
+                
+                if response.status_code == 404:
+                    # Try yet another alternative
+                    response = self.session.get(f"{self.base_url}/activities/pyramid-alphas", params=params)
+                    
+                    if response.status_code == 404:
+                        # Return an informative error with what we tried
+                        return {
+                            "error": "Pyramid alphas endpoint not found",
+                            "tried_endpoints": [
+                                "/users/self/activities/pyramid-alphas",
+                                "/users/self/pyramid/alphas", 
+                                "/activities/pyramid-alphas",
+                                "/pyramid/alphas"
+                            ],
+                            "suggestion": "This endpoint may not be available in the current API version"
+                        }
+            
             response.raise_for_status()
             return response.json()
         except Exception as e:

{cnhkmcp-1.1.5.dist-info → cnhkmcp-1.1.7.dist-info}/METADATA

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

@@ -0,0 +1,10 @@
+cnhkmcp/__init__.py,sha256=zA4Je1BkR8_9NorDUU5n_QEozwqBquxVGsUrGk3yHjc,2758
+cnhkmcp/untracked/forum_functions.py,sha256=78wzvN_UYWwbWU40q8_FJNSFPJnND6W9ZRey6MSSiEk,45516
+cnhkmcp/untracked/platform_functions.py,sha256=sDW-tOhWUT41YuO-A3ugimkrNEj0ShMEUYp4EFVQpeA,77441
+cnhkmcp/untracked/user_config.json,sha256=_INn1X1qIsITrmEno-BRlQOAGm9wnNCw-6B333DEvnk,695
+cnhkmcp-1.1.7.dist-info/licenses/LICENSE,sha256=QLxO2eNMnJQEdI_R1UV2AOD-IvuA8zVrkHWA4D9gtoc,1081
+cnhkmcp-1.1.7.dist-info/METADATA,sha256=sNnIQ4NvGvk6qScb5HBmfhUiP9-OiDG1hsGQLn-RazA,5171
+cnhkmcp-1.1.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+cnhkmcp-1.1.7.dist-info/entry_points.txt,sha256=lTQieVyIvjhSMK4fT-XwnccY-JBC1H4vVQ3V9dDM-Pc,70
+cnhkmcp-1.1.7.dist-info/top_level.txt,sha256=x--ibUcSgOS9Z_RWK2Qc-vfs7DaXQN-WMaaxEETJ1Bw,8
+cnhkmcp-1.1.7.dist-info/RECORD,,

cnhkmcp-1.1.5.dist-info/RECORD

@@ -1,10 +0,0 @@
-cnhkmcp/__init__.py,sha256=tguzH5AFA9YrG97csGDkzjyEixyN4p5H3-lIlQ58uCY,2882
-cnhkmcp/untracked/forum_functions.py,sha256=78wzvN_UYWwbWU40q8_FJNSFPJnND6W9ZRey6MSSiEk,45516
-cnhkmcp/untracked/platform_functions.py,sha256=LuZt6GidH8Qy27DtcUgBCgED3iZf5WF-Hq_3xm-HOjM,70748
-cnhkmcp/untracked/user_config.json,sha256=_INn1X1qIsITrmEno-BRlQOAGm9wnNCw-6B333DEvnk,695
-cnhkmcp-1.1.5.dist-info/licenses/LICENSE,sha256=QLxO2eNMnJQEdI_R1UV2AOD-IvuA8zVrkHWA4D9gtoc,1081
-cnhkmcp-1.1.5.dist-info/METADATA,sha256=_bTEFa_BKyb2lbedU-JNhCU6oKsx1bH2xXmHgnBziKI,5171
-cnhkmcp-1.1.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-cnhkmcp-1.1.5.dist-info/entry_points.txt,sha256=lTQieVyIvjhSMK4fT-XwnccY-JBC1H4vVQ3V9dDM-Pc,70
-cnhkmcp-1.1.5.dist-info/top_level.txt,sha256=x--ibUcSgOS9Z_RWK2Qc-vfs7DaXQN-WMaaxEETJ1Bw,8
-cnhkmcp-1.1.5.dist-info/RECORD,,