Upd user_id

helpers/diagram.py

@@ -107,13 +107,13 @@ async def generate_mermaid_diagram(
         if retry < max_retries:
             return await generate_mermaid_diagram(
                 instructions, detailed_analysis, gemini_rotator, nvidia_rotator,
-                render_error="Diagram did not include recognizable Mermaid diagram keyword.", retry=retry+1
+                render_error="Diagram did not include recognizable Mermaid diagram keyword.", retry=retry+1, user_id=user_id
             )
 
     return diagram
 
 
-async def _render_mermaid_with_retry(mermaid_text: str, max_retries: int = 3) -> bytes:
+async def _render_mermaid_with_retry(mermaid_text: str, max_retries: int = 3, user_id: str = "") -> bytes:
     """
     Render mermaid code to PNG with retry logic and AI-powered error correction.
     """

routes/reports.py

@@ -891,7 +891,7 @@ Create a comprehensive, authoritative report with proper hierarchical structure
         report = ensure_hierarchical_structure(report)
         
         # Fix heading numbering using AI
-        report = await fix_heading_numbering(report, nvidia_rotator)
+        report = await fix_heading_numbering(report, nvidia_rotator, user_id)
 
         # Optionally enrich with Mermaid diagrams when useful
         try:
@@ -1218,7 +1218,7 @@ def remove_meta_commentary(text: str) -> str:
     return text
 
 
-async def fix_heading_numbering(report: str, nvidia_rotator) -> str:
+async def fix_heading_numbering(report: str, nvidia_rotator, user_id: str = "") -> str:
     """
     Extract headings from the report, use AI to re-number them properly, then apply the fixes.
     """

utils/service/pdf.py

@@ -70,7 +70,7 @@ async def _parse_markdown_content(content: str, heading1_style, heading2_style,
                         mermaid_code = '\n'.join(code_lines)
                         # Use retry logic from diagram.py
                         from helpers.diagram import _render_mermaid_with_retry
-                        img_bytes = await _render_mermaid_with_retry(mermaid_code)
+                        img_bytes = await _render_mermaid_with_retry(mermaid_code, user_id=user_id)
                         
                         if img_bytes and len(img_bytes) > 0:
                             import io