cnhkmcp 1.1.5__tar.gz → 1.1.7__tar.gz

{cnhkmcp-1.1.5/cnhkmcp.egg-info → cnhkmcp-1.1.7}/PKG-INFO

@@ -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.5 → cnhkmcp-1.1.7}/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-1.1.5 → cnhkmcp-1.1.7}/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 → cnhkmcp-1.1.7/cnhkmcp.egg-info}/PKG-INFO

@@ -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.5 → cnhkmcp-1.1.7}/setup.py

@@ -1,64 +1,64 @@
-from setuptools import setup, find_packages
-import os
-
-# Read the README file
-def read_readme():
-    with open("README.md", "r", encoding="utf-8") as fh:
-        return fh.read()
-
-# Read requirements
-def read_requirements():
-    with open("requirements.txt", "r", encoding="utf-8") as fh:
-        return [line.strip() for line in fh if line.strip() and not line.startswith("#")]
-
-setup(
-    name="cnhkmcp",
-    version="1.1.5",
-    author="CNHK",
-    author_email="cnhk@example.com",
-    description="A comprehensive Model Context Protocol (MCP) server for quantitative trading platform integration",
-    long_description=read_readme(),
-    long_description_content_type="text/markdown",
-    url="https://github.com/cnhk/cnhkmcp",
-    packages=find_packages(),
-    classifiers=[
-        "Development Status :: 4 - Beta",
-        "Intended Audience :: Developers",
-        "Intended Audience :: Financial and Insurance Industry",
-        "License :: OSI Approved :: MIT License",
-        "Operating System :: OS Independent",
-        "Programming Language :: Python :: 3",
-        "Programming Language :: Python :: 3.8",
-        "Programming Language :: Python :: 3.9",
-        "Programming Language :: Python :: 3.10",
-        "Programming Language :: Python :: 3.11",
-        "Topic :: Office/Business :: Financial",
-        "Topic :: Software Development :: Libraries :: Python Modules",
-    ],
-    python_requires=">=3.8",
-    install_requires=read_requirements(),
-    extras_require={
-        "dev": [
-            "pytest>=6.0",
-            "pytest-asyncio>=0.18.0",
-            "black>=21.0",
-            "flake8>=3.8",
-        ],
-    },
-    entry_points={
-        "console_scripts": [
-            "cnhkmcp=cnhkmcp.untracked.platform_functions:main",
-        ],
-    },
-    include_package_data=True,
-    zip_safe=False,
-    package_data={
-        'cnhkmcp': ['untracked/*.py', 'untracked/*.json'],
-    },
-    keywords="mcp, quantitative, trading, api, client",
-    project_urls={
-        "Bug Reports": "https://github.com/cnhk/cnhkmcp/issues",
-        "Source": "https://github.com/cnhk/cnhkmcp",
-        "Documentation": "https://github.com/cnhk/cnhkmcp#readme",
-    },
+from setuptools import setup, find_packages
+import os
+
+# Read the README file
+def read_readme():
+    with open("README.md", "r", encoding="utf-8") as fh:
+        return fh.read()
+
+# Read requirements
+def read_requirements():
+    with open("requirements.txt", "r", encoding="utf-8") as fh:
+        return [line.strip() for line in fh if line.strip() and not line.startswith("#")]
+
+setup(
+    name="cnhkmcp",
+    version="1.1.7",
+    author="CNHK",
+    author_email="cnhk@example.com",
+    description="A comprehensive Model Context Protocol (MCP) server for quantitative trading platform integration",
+    long_description=read_readme(),
+    long_description_content_type="text/markdown",
+    url="https://github.com/cnhk/cnhkmcp",
+    packages=find_packages(),
+    classifiers=[
+        "Development Status :: 4 - Beta",
+        "Intended Audience :: Developers",
+        "Intended Audience :: Financial and Insurance Industry",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent",
+        "Programming Language :: Python :: 3",
+        "Programming Language :: Python :: 3.8",
+        "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
+        "Topic :: Office/Business :: Financial",
+        "Topic :: Software Development :: Libraries :: Python Modules",
+    ],
+    python_requires=">=3.8",
+    install_requires=read_requirements(),
+    extras_require={
+        "dev": [
+            "pytest>=6.0",
+            "pytest-asyncio>=0.18.0",
+            "black>=21.0",
+            "flake8>=3.8",
+        ],
+    },
+    entry_points={
+        "console_scripts": [
+            "cnhkmcp=cnhkmcp.untracked.platform_functions:main",
+        ],
+    },
+    include_package_data=True,
+    zip_safe=False,
+    package_data={
+        'cnhkmcp': ['untracked/*.py', 'untracked/*.json'],
+    },
+    keywords="mcp, quantitative, trading, api, client",
+    project_urls={
+        "Bug Reports": "https://github.com/cnhk/cnhkmcp/issues",
+        "Source": "https://github.com/cnhk/cnhkmcp",
+        "Documentation": "https://github.com/cnhk/cnhkmcp#readme",
+    },
 )