dtlpymcp 0.1.9__tar.gz → 0.1.10__tar.gz

{dtlpymcp-0.1.9 → dtlpymcp-0.1.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dtlpymcp
-Version: 0.1.9
+Version: 0.1.10
 Summary: STDIO MCP proxy server for Dataloop platform.
 Author-email: Your Name <your.email@example.com>
 Classifier: Programming Language :: Python :: 3.10

dtlpymcp-0.1.10/dtlpymcp/__init__.py

@@ -0,0 +1,3 @@
+from .utils.dtlpy_context import DataloopContext, MCPSource
+
+__version__ = "0.1.10"

{dtlpymcp-0.1.9 → dtlpymcp-0.1.10}/dtlpymcp/__main__.py

@@ -24,11 +24,19 @@ def main():
         default=30.0,
         help="Timeout in seconds for Dataloop context initialization (default: 30.0)",
     )
+    start_parser.add_argument(
+        "--log-level",
+        "-l",
+        type=str,
+        default="INFO",
+        choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"],
+        help="Logging level (default: INFO)",
+    )
 
     args = parser.parse_args()
 
     if args.command == "start":
-        sys.exit(proxy_main(sources_file=args.sources_file, init_timeout=args.init_timeout))
+        sys.exit(proxy_main(sources_file=args.sources_file, init_timeout=args.init_timeout, log_level=args.log_level))
     else:
         parser.print_help()
         return 1

{dtlpymcp-0.1.9 → dtlpymcp-0.1.10}/dtlpymcp/proxy.py

@@ -2,6 +2,8 @@ from mcp.server.fastmcp import FastMCP, Context
 from mcp.server.fastmcp.tools.base import Tool, FuncMetadata
 from typing import Any, Optional, List
 import traceback
+from datetime import datetime
+from pathlib import Path
 import os
 import asyncio
 import logging
@@ -10,7 +12,48 @@ from mcp.server.fastmcp.utilities.func_metadata import ArgModelBase
 
 from dtlpymcp.utils.dtlpy_context import DataloopContext
 
-logger = logging.getLogger("dtlpymcp")
+
+def setup_logging(log_level: str = "INFO") -> logging.Logger:
+    """
+    Setup logging configuration with the specified log level.
+    
+    Args:
+        log_level: Logging level as string (DEBUG, INFO, WARNING, ERROR, CRITICAL)
+        
+    Returns:
+        logging.Logger: Configured logger instance
+    """
+    # Setup logging directory and file
+    log_dir = Path.home() / ".dataloop" / "mcplogs"
+    log_dir.mkdir(parents=True, exist_ok=True)
+    log_file = log_dir / f"{datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}.log"
+
+    # Remove any existing handlers
+    for handler in logging.root.handlers[:]:
+        logging.root.removeHandler(handler)
+
+    # File handler with timestamp
+    file_handler = logging.FileHandler(log_file, mode="a", encoding="utf-8")
+    file_handler.setFormatter(
+        logging.Formatter(fmt="%(asctime)s [%(levelname)s] %(name)s: %(message)s", datefmt="%Y-%m-%d %H:%M:%S")
+    )
+
+    # Console handler (default format)
+    console_handler = logging.StreamHandler()
+    console_handler.setFormatter(logging.Formatter(fmt="[%(levelname)s] %(name)s: %(message)s"))
+
+    # Convert string log level to logging constant
+    numeric_level = getattr(logging, log_level.upper(), logging.INFO)
+    
+    # Configure root logger
+    logging.basicConfig(level=numeric_level, handlers=[file_handler, console_handler])
+
+    # Get the main logger
+    logger = logging.getLogger("dtlpymcp")
+    logger.info(f"Logging configured with level: {log_level.upper()}")
+    logger.info(f"Log file: {log_file}")
+    
+    return logger
 
 
 def run_async(coro):
@@ -29,13 +72,13 @@ async def initialize_dataloop_context(sources_file: Optional[str] = None, init_t
     Initialize Dataloop context with timeout protection.
 
     Args:
-        dl_context: The DataloopContext instance to initialize
-        app: The FastMCP app instance to register tools with
+        sources_file: Path to sources file
         init_timeout: Timeout in seconds for initialization
 
     Returns:
         List[Tool]: List of tools
     """
+    logger = logging.getLogger("dtlpymcp")
     try:
         tools = []
         dl_context = DataloopContext(
@@ -101,8 +144,10 @@ def create_dataloop_mcp_server(sources_file: Optional[str] = None, init_timeout:
     return app
 
 
-def main(sources_file: Optional[str] = None, init_timeout: float = 30.0) -> int:
-
+def main(sources_file: Optional[str] = None, init_timeout: float = 30.0, log_level: str = "INFO") -> int:
+    # Setup logging with the specified level
+    logger = setup_logging(log_level)
+    
     logger.info("Starting Dataloop MCP server in stdio mode")
 
     # Validate environment variables

{dtlpymcp-0.1.9 → dtlpymcp-0.1.10}/dtlpymcp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dtlpymcp
-Version: 0.1.9
+Version: 0.1.10
 Summary: STDIO MCP proxy server for Dataloop platform.
 Author-email: Your Name <your.email@example.com>
 Classifier: Programming Language :: Python :: 3.10

{dtlpymcp-0.1.9 → dtlpymcp-0.1.10}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dtlpymcp"
-version = "0.1.9"
+version = "0.1.10"
 description = "STDIO MCP proxy server for Dataloop platform."
 authors = [
     { name = "Your Name", email = "your.email@example.com" }

dtlpymcp-0.1.9/dtlpymcp/__init__.py

@@ -1,34 +0,0 @@
-from .utils.dtlpy_context import DataloopContext, MCPSource
-import logging
-from datetime import datetime
-from pathlib import Path
-
-__version__ = "0.1.9"
-
-
-# Setup logging directory and file
-log_dir = Path.home() / ".dataloop" / "mcplogs"
-log_dir.mkdir(parents=True, exist_ok=True)
-log_file = log_dir / f"{datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}.log"
-
-# Remove any existing handlers
-for handler in logging.root.handlers[:]:
-    logging.root.removeHandler(handler)
-
-# File handler with timestamp
-file_handler = logging.FileHandler(log_file, mode="a", encoding="utf-8")
-file_handler.setFormatter(
-    logging.Formatter(fmt="%(asctime)s [%(levelname)s] %(name)s: %(message)s", datefmt="%Y-%m-%d %H:%M:%S")
-)
-
-# Console handler (default format)
-console_handler = logging.StreamHandler()   
-console_handler.setFormatter(logging.Formatter(fmt="[%(levelname)s] %(name)s: %(message)s"))
-
-# Configure root logger
-logging.basicConfig(level=logging.DEBUG, handlers=[file_handler, console_handler])
-
-# Get the main logger
-logger = logging.getLogger("dtlpymcp")
-logger.info(f"Logging configured with level: DEBUG")
-logger.info(f"Log file: {log_file}")