cnhkmcp 1.2.2__py3-none-any.whl → 1.2.4__py3-none-any.whl

cnhkmcp/__init__.py

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

cnhkmcp/untracked/platform_functions.py

@@ -26,6 +26,9 @@ from bs4 import BeautifulSoup
 from mcp.server.fastmcp import FastMCP
 from pydantic import BaseModel, Field, EmailStr
 
+from pathlib import Path
+
+
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
@@ -601,8 +604,8 @@ class BrainApiClient:
             self.log(f"Failed to get user profile: {str(e)}", "ERROR")
             raise
 
-    async def get_tutorials(self) -> Dict[str, Any]:
-        """Get available tutorials and learning materials."""
+    async def get_documentations(self) -> Dict[str, Any]:
+        """Get available documentations and learning materials."""
         await self.ensure_authenticated()
         
         try:
@@ -610,7 +613,7 @@ class BrainApiClient:
             response.raise_for_status()
             return response.json()
         except Exception as e:
-            self.log(f"Failed to get tutorials: {str(e)}", "ERROR")
+            self.log(f"Failed to get documentations: {str(e)}", "ERROR")
             raise
 
     # get_messages_summary function removed as requested
@@ -1189,8 +1192,8 @@ class BrainApiClient:
 
     # generate_alpha_links function removed as requested
 
-    async def get_tutorial_page(self, page_id: str) -> Dict[str, Any]:
-        """Retrieve detailed content of a specific tutorial page/article."""
+    async def get_documentation_page(self, page_id: str) -> Dict[str, Any]:
+        """Retrieve detailed content of a specific documentation page/article."""
         await self.ensure_authenticated()
         
         try:
@@ -1198,7 +1201,7 @@ class BrainApiClient:
             response.raise_for_status()
             return response.json()
         except Exception as e:
-            self.log(f"Failed to get tutorial page: {str(e)}", "ERROR")
+            self.log(f"Failed to get documentation page: {str(e)}", "ERROR")
             raise
 
     # Badge status function removed as requested
@@ -1213,39 +1216,44 @@ brain_client = BrainApiClient()
 CONFIG_FILE = "user_config.json"
 
 def _resolve_config_path(for_write: bool = False) -> str:
-    """Resolve the absolute path to the config file.
-
-    Resolution order:
-    1) BRAIN_CONFIG_PATH env var (explicit file path)
-    2) File in the same directory as this Python module
-    3) Current working directory (read fallback only)
+    """
+    Resolve the config file path with this priority:
+    1) BRAIN_CONFIG_PATH (file or directory)
+    2) Directory of running script when available, else current working directory
+    3) Current working directory
 
-    When for_write is True, prefers module directory (or env path) and will
-    return that even if it doesn't exist yet.
+    When for_write=True, returns the preferred path even if it doesn't exist yet.
     """
+    # 1) Explicit override via env var
+    env_path = os.environ.get("BRAIN_CONFIG_PATH")
+    if env_path:
+        p = Path(env_path).expanduser()
+        target = p / CONFIG_FILE if p.is_dir() else p
+        # For read, only if it exists; for write, allow regardless
+        if for_write or target.exists():
+            return str(target.resolve())
+
+    # 2) Script/module directory when available, else CWD (works in notebooks)
+    base_dir = Path.cwd()
     try:
-        # 1) Explicit override
-        env_path = os.environ.get("BRAIN_CONFIG_PATH")
-        if env_path:
-            return os.path.abspath(env_path)
-
-        # 2) Same directory as this .py module
-        module_dir = os.path.dirname(os.path.abspath(__file__))
-        module_path = os.path.join(module_dir, CONFIG_FILE)
-        if not for_write and os.path.exists(module_path):
-            return module_path
-
-        # 3) Fallback to CWD for backward compatibility (read-only preference)
-        cwd_path = os.path.abspath(CONFIG_FILE)
-        if not for_write and os.path.exists(cwd_path):
-            return cwd_path
-
-        # For writes (or when nothing exists), prefer the module directory
-        return module_path
-    except Exception as e:
-        # As a last resort, use the bare filename in CWD
-        logger.error(f"Failed to resolve config path, defaulting to CWD: {e}")
-        return os.path.abspath(CONFIG_FILE)
+        # __file__ is not defined in notebooks; this will fail there and keep CWD
+        script_dir = Path(__file__).resolve().parent  # type: ignore[name-defined]
+        base_dir = script_dir
+    except Exception:
+        # Fall back to current working directory for notebooks/REPL
+        pass
+
+    module_path = base_dir / CONFIG_FILE
+    if not for_write and module_path.exists():
+        return str(module_path.resolve())
+
+    # 3) Fallback to CWD for backward compatibility
+    cwd_path = Path.cwd() / CONFIG_FILE
+    if not for_write and cwd_path.exists():
+        return str(cwd_path.resolve())
+
+    # For writes (or when nothing exists), prefer the module/base directory
+    return str(module_path.resolve())
 
 def load_config() -> Dict[str, Any]:
     """Load configuration from file with robust path resolution.
@@ -1262,19 +1270,6 @@ def load_config() -> Dict[str, Any]:
             logger.error(f"Failed to load config from '{path}': {e}")
     return {}
 
-def load_brain_credentials() -> Optional[tuple]:
-    """Load credentials from .brain_credentials file in home directory."""
-    try:
-        from os.path import expanduser
-        credentials_file = expanduser('~/.brain_credentials')
-        if os.path.exists(credentials_file):
-            with open(credentials_file, 'r') as f:
-                credentials = json.load(f)
-                if isinstance(credentials, list) and len(credentials) == 2:
-                    return tuple(credentials)
-    except Exception as e:
-        logger.error(f"Failed to load .brain_credentials: {e}")
-    return None
 
 def save_config(config: Dict[str, Any]):
     """Save configuration to file using the resolved config path.
@@ -1293,7 +1288,7 @@ def save_config(config: Dict[str, Any]):
 # MCP Tools
 
 @mcp.tool()
-async def authenticate(email: str = "", password: str = "") -> Dict[str, Any]:
+async def authenticate(email: Optional[str] = "", password: Optional[str] = "") -> Dict[str, Any]:
     """
     🔐 Authenticate with WorldQuant BRAIN platform.
     
@@ -1307,20 +1302,12 @@ async def authenticate(email: str = "", password: str = "") -> Dict[str, Any]:
         Authentication result with user info and permissions
     """
     try:
-        # First try to load from .brain_credentials file
-        brain_creds = load_brain_credentials()
-        if brain_creds and not email and not password:
-            email, password = brain_creds
-            print(f"📧 Using credentials from .brain_credentials file: {email}")
-        
-        # If still no credentials, try config file
-        if not email or not password:
-            config = load_config()
-            if 'credentials' in config:
-                if not email:
-                    email = config['credentials'].get('email', '')
-                if not password:
-                    password = config['credentials'].get('password', '')
+        config = load_config()
+        if 'credentials' in config:
+            if not email:
+                email = config['credentials'].get('email', '')
+            if not password:
+                password = config['credentials'].get('password', '')
         
         if not email or not password:
             return {"error": "Email and password required. Either provide them as arguments, configure them in user_config.json, or create a .brain_credentials file in your home directory with format: [\"email\", \"password\"]"}
@@ -1756,15 +1743,15 @@ async def get_user_profile(user_id: str = "self") -> Dict[str, Any]:
         return {"error": str(e)}
 
 @mcp.tool()
-async def get_tutorials() -> Dict[str, Any]:
+async def get_documentations() -> Dict[str, Any]:
     """
-    📚 Get available tutorials and learning materials.
+    📚 Get available documentations and learning materials.
     
     Returns:
-        List of tutorials
+        List of documentations
     """
     try:
-        return await brain_client.get_tutorials()
+        return await brain_client.get_documentations()
     except Exception as e:
         return {"error": str(e)}
 
@@ -2029,10 +2016,10 @@ async def expand_nested_data(data: List[Dict[str, Any]], preserve_original: bool
 # generate_alpha_links MCP tool removed as requested
 
 @mcp.tool()
-async def get_tutorial_page(page_id: str) -> Dict[str, Any]:
-    """Retrieve detailed content of a specific tutorial page/article."""
+async def get_documentation_page(page_id: str) -> Dict[str, Any]:
+    """Retrieve detailed content of a specific documentation page/article."""
     try:
-        return await brain_client.get_tutorial_page(page_id)
+        return await brain_client.get_documentation_page(page_id)
     except Exception as e:
         return {"error": str(e)}
 

{cnhkmcp-1.2.2.dist-info → cnhkmcp-1.2.4.dist-info}/METADATA

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

@@ -0,0 +1,10 @@
+cnhkmcp/__init__.py,sha256=-laBHbRG7AtBQp4pu6I9nSFtNBoBp_p35pQzNcf2Jkw,2758
+cnhkmcp/untracked/forum_functions.py,sha256=78wzvN_UYWwbWU40q8_FJNSFPJnND6W9ZRey6MSSiEk,45516
+cnhkmcp/untracked/platform_functions.py,sha256=o0l1MNcUUmvCRgjhxbn0QaxdZSLyV_C1DI-2DaXx9Eg,81739
+cnhkmcp/untracked/user_config.json,sha256=_INn1X1qIsITrmEno-BRlQOAGm9wnNCw-6B333DEvnk,695
+cnhkmcp-1.2.4.dist-info/licenses/LICENSE,sha256=QLxO2eNMnJQEdI_R1UV2AOD-IvuA8zVrkHWA4D9gtoc,1081
+cnhkmcp-1.2.4.dist-info/METADATA,sha256=a86cCJaC48vkhIHafwnyDZYliEfK6Ht6qGZdO7qCsy8,5171
+cnhkmcp-1.2.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+cnhkmcp-1.2.4.dist-info/entry_points.txt,sha256=lTQieVyIvjhSMK4fT-XwnccY-JBC1H4vVQ3V9dDM-Pc,70
+cnhkmcp-1.2.4.dist-info/top_level.txt,sha256=x--ibUcSgOS9Z_RWK2Qc-vfs7DaXQN-WMaaxEETJ1Bw,8
+cnhkmcp-1.2.4.dist-info/RECORD,,

cnhkmcp-1.2.2.dist-info/RECORD

@@ -1,10 +0,0 @@
-cnhkmcp/__init__.py,sha256=jLSpv4B0gds0czwoFi1w5-qZBViB9f8yUZN4_9U1yf8,2758
-cnhkmcp/untracked/forum_functions.py,sha256=78wzvN_UYWwbWU40q8_FJNSFPJnND6W9ZRey6MSSiEk,45516
-cnhkmcp/untracked/platform_functions.py,sha256=7Vnul6prlHwCNB0L0X_sLwR_-C53y1sKDhAKF6KOE1A,82452
-cnhkmcp/untracked/user_config.json,sha256=_INn1X1qIsITrmEno-BRlQOAGm9wnNCw-6B333DEvnk,695
-cnhkmcp-1.2.2.dist-info/licenses/LICENSE,sha256=QLxO2eNMnJQEdI_R1UV2AOD-IvuA8zVrkHWA4D9gtoc,1081
-cnhkmcp-1.2.2.dist-info/METADATA,sha256=xAuZdgGb0hWe_FmVuTT0JTVwjb3tstKBcyerAlr8rMI,5171
-cnhkmcp-1.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-cnhkmcp-1.2.2.dist-info/entry_points.txt,sha256=lTQieVyIvjhSMK4fT-XwnccY-JBC1H4vVQ3V9dDM-Pc,70
-cnhkmcp-1.2.2.dist-info/top_level.txt,sha256=x--ibUcSgOS9Z_RWK2Qc-vfs7DaXQN-WMaaxEETJ1Bw,8
-cnhkmcp-1.2.2.dist-info/RECORD,,