tree-sitter-analyzer 0.8.1__py3-none-any.whl → 0.8.3__py3-none-any.whl

tree_sitter_analyzer/mcp/tools/analyze_scale_tool_cli_compatible.py

@@ -99,14 +99,12 @@ class AnalyzeScaleToolCLICompatible:
         logger.info(f"Analyzing code scale for {file_path} (language: {language})")
 
         try:
-            start_time = time.time()
-
-            # Use AdvancedAnalyzer for comprehensive analysis
+            # Use higher precision timer and measure ONLY the engine call
+            start_time = time.perf_counter()
             analysis_result = await self.analysis_engine.analyze_file(file_path)
+            analysis_time_ms = round((time.perf_counter() - start_time) * 1000, 2)
 
             # Handle potential None result (for testing purposes with mocked engine)
-            # This can only happen in tests where the engine is mocked to return None
-            # Use cast to tell MyPy this is possible in testing scenarios
             if cast(Any, analysis_result) is None:
                 return {
                     "file_path": file_path,
@@ -119,13 +117,10 @@ class AnalyzeScaleToolCLICompatible:
                         "fields": 0,
                         "annotations": 0,
                     },
-                    "analysis_time_ms": round((time.time() - start_time) * 1000, 2),
+                    "analysis_time_ms": analysis_time_ms,
                     "error_message": f"Failed to analyze file: {file_path}",
                 }
 
-            # Calculate analysis time
-            analysis_time_ms = round((time.time() - start_time) * 1000, 2)
-
             # Build CLI-compatible result structure (exact match with CLI --advanced --statistics)
             result = {
                 "file_path": file_path,

tree_sitter_analyzer/mcp/utils/error_handler.py

@@ -491,6 +491,24 @@ def handle_mcp_errors(
         async def async_wrapper(*args: Any, **kwargs: Any) -> Any:
             try:
                 return await func(*args, **kwargs)
+            except RuntimeError as e:
+                # Handle initialization errors specifically
+                if "not fully initialized" in str(e):
+                    logger.warning(f"Request received before initialization complete: {operation}")
+                    raise MCPError(
+                        "Server is still initializing. Please wait a moment and try again.",
+                        category=ErrorCategory.CONFIGURATION,
+                        severity=ErrorSeverity.LOW
+                    ) from e
+                # Handle other runtime errors normally
+                error_handler = get_error_handler()
+                context = {
+                    "function": func.__name__,
+                    "args": str(args)[:200],  # Limit length
+                    "kwargs": str(kwargs)[:200],
+                }
+                error_info = error_handler.handle_error(e, context, operation)
+                raise
             except Exception as e:
                 error_handler = get_error_handler()
                 context = {