fix: resolve event loop conflicts and mermaid diagram rendering

Signed-off-by: gopal-raj-suresh <gopal.raj.dummugudupu@cloud2labs.com>

Author: gopal-raj-suresh

sample_solutions/Docugen-Microagents/api/agents/mermaid_agent.py

@@ -54,6 +54,29 @@
 4. VALIDATE with validate_mermaid_syntax()
 5. Fix errors if validation fails
 
+**FORBIDDEN Patterns (will cause syntax errors):**
+- ❌ NO self-referencing edges: Service -->|X| Service
+- ❌ NO curly braces in labels: -->|GET {id}|
+- ❌ NO endpoint paths in labels: -->|POST /upload|
+- ❌ NO colons in labels: -->|Method: POST|
+- ✅ GOOD: -->|Upload File|, -->|Process Request|
+
+**Required Architecture Pattern:**
+```
+User → Frontend → Backend Service(s) → Database/Storage
+```
+Show the FLOW of data, NOT the API endpoints.
+
+**CRITICAL - Final Response Format:**
+Your FINAL response MUST contain ONLY the mermaid diagram code wrapped in triple backticks.
+Do NOT add any explanatory text before or after the diagram.
+Format your final response EXACTLY like this:
+
+```mermaid
+graph TD
+    [your diagram here]
+```
+
 **Limit:** 12 tool calls."""
 
 

sample_solutions/Docugen-Microagents/api/models/log_manager.py

@@ -4,6 +4,7 @@
 """
 
 import asyncio
+import threading
 import logging
 from collections import defaultdict
 from typing import Dict, List, Optional
@@ -29,7 +30,7 @@ def __init__(self, max_logs_per_job: int = 1000):
         self.max_logs_per_job = max_logs_per_job
         self._logs: Dict[str, List[AgentLog]] = defaultdict(list)
         self._subscribers: Dict[str, List[asyncio.Queue]] = defaultdict(list)
-        self._lock = asyncio.Lock()
+        self._lock = threading.Lock()
 
     async def add_log(self, log: AgentLog) -> None:
         """
@@ -38,7 +39,7 @@ async def add_log(self, log: AgentLog) -> None:
         Args:
             log: Agent log entry
         """
-        async with self._lock:
+        with self._lock:
             # Add to storage
             job_logs = self._logs[log.job_id]
             job_logs.append(log)
@@ -51,10 +52,10 @@ async def add_log(self, log: AgentLog) -> None:
             dead_queues = []
             for queue in self._subscribers[log.job_id]:
                 try:
-                    await asyncio.wait_for(queue.put(log), timeout=1.0)
-                except (asyncio.TimeoutError, asyncio.QueueFull):
+                    queue.put_nowait(log)
+                except asyncio.QueueFull:
                     dead_queues.append(queue)
-                    logger.warning(f"Queue full or timeout for job {log.job_id}")
+                    logger.warning(f"Queue full for job {log.job_id}")
 
             # Clean up dead queues
             for queue in dead_queues:
@@ -153,13 +154,13 @@ async def subscribe(self, job_id: str) -> asyncio.Queue:
             Queue that will receive new logs
         """
         queue: asyncio.Queue = asyncio.Queue(maxsize=100)
-        async with self._lock:
+        with self._lock:
             self._subscribers[job_id].append(queue)
 
             # Send existing logs immediately
             for log in self._logs.get(job_id, []):
                 try:
-                    await queue.put(log)
+                    queue.put_nowait(log)
                 except asyncio.QueueFull:
                     logger.warning(f"Initial logs queue full for job {job_id}")
                     break
@@ -174,7 +175,7 @@ async def unsubscribe(self, job_id: str, queue: asyncio.Queue) -> None:
             job_id: Job identifier
             queue: Queue to remove
         """
-        async with self._lock:
+        with self._lock:
             if queue in self._subscribers[job_id]:
                 self._subscribers[job_id].remove(queue)
 

sample_solutions/Docugen-Microagents/api/services/llm_service.py

@@ -30,9 +30,8 @@ def get_llm(model_name: Optional[str] = None, temperature: float = 0.7) -> ChatO
 
     logger.info(f"Initializing LLM with model: {model_name}")
 
-    # Create httpx client with configurable SSL verification
+    # Create httpx async client with configurable SSL verification
     import httpx
-    http_client = httpx.Client(verify=settings.VERIFY_SSL)
     async_http_client = httpx.AsyncClient(verify=settings.VERIFY_SSL)
 
     return ChatOpenAI(
@@ -41,6 +40,5 @@ def get_llm(model_name: Optional[str] = None, temperature: float = 0.7) -> ChatO
         openai_api_key=settings.INFERENCE_API_TOKEN,
         openai_api_base=f"{settings.INFERENCE_API_ENDPOINT}/v1",
         max_tokens=settings.AGENT_MAX_TOKENS,
-        http_client=http_client,
         http_async_client=async_http_client
     )

sample_solutions/Docugen-Microagents/api/tools/repo_tools.py

@@ -1028,7 +1028,7 @@ def validate_mermaid_syntax_tool(mermaid_code: str) -> str:
             # Check 3: Edge labels must not contain unsafe characters
             # Pattern: |label text|
             edge_label_pattern = r'\|([^|]+)\|'
-            unsafe_edge_chars = ['/', '(', ')', ':', '`', '"']
+            unsafe_edge_chars = ['/', '(', ')', ':', '`', '"', '{', '}']
             for line in lines[1:]:
                 if not line.strip():
                     continue
@@ -1046,6 +1046,18 @@ def validate_mermaid_syntax_tool(mermaid_code: str) -> str:
                             )
                             break
 
+            # Check 4: No self-referencing edges (Node --> Node)
+            self_ref_pattern = r'([A-Za-z][A-Za-z0-9_]*)\s*-->\s*\1(?:\s|$|\|)'
+            for line in lines[1:]:
+                if not line.strip():
+                    continue
+                matches = re.findall(self_ref_pattern, line)
+                for node_id in matches:
+                    errors.append(
+                        f"Self-referencing edge detected: {node_id} --> {node_id}. "
+                        f"Show data flow between different components instead."
+                    )
+
         # Check sequenceDiagram syntax
         if code.startswith("sequenceDiagram"):
             if "->" not in code and "->>" not in code:

sample_solutions/Docugen-Microagents/ui/src/components/Mermaid.jsx

@@ -1,17 +1,18 @@
 import { useEffect, useRef } from 'react'
 import mermaid from 'mermaid'
 
-// Initialize mermaid with default config
-mermaid.initialize({
-  startOnLoad: false,
-  theme: 'default',
-  securityLevel: 'strict',
-  fontFamily: 'ui-sans-serif, system-ui, sans-serif'
-})
-
 function Mermaid({ chart }) {
   const containerRef = useRef(null)
-  const idRef = useRef(`mermaid-${Math.random().toString(36).substr(2, 9)}`)
+
+  useEffect(() => {
+    // Initialize mermaid with config
+    mermaid.initialize({
+      startOnLoad: false,
+      theme: 'default',
+      securityLevel: 'strict',
+      fontFamily: 'ui-sans-serif, system-ui, sans-serif'
+    })
+  }, [])
 
   useEffect(() => {
     if (containerRef.current && chart) {
@@ -20,20 +21,24 @@ function Mermaid({ chart }) {
           // Clear previous content
           containerRef.current.innerHTML = ''
 
+          // Generate new ID for each render to avoid conflicts
+          const newId = `mermaid-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`
+
           // Render the mermaid chart
-          const { svg } = await mermaid.render(idRef.current, chart)
+          const { svg } = await mermaid.render(newId, chart)
           containerRef.current.innerHTML = svg
         } catch (error) {
           console.error('Mermaid rendering error:', error)
-          containerRef.current.innerHTML = `<pre style="color: red;">Error rendering diagram: ${error.message}</pre>`
+          console.error('Chart content:', chart)
+          containerRef.current.innerHTML = `<pre style="color: red; padding: 1rem; background: #fee; border-radius: 8px;">Error rendering diagram: ${error.message}</pre>`
         }
       }
 
       renderDiagram()
     }
   }, [chart])
 
-  return <div ref={containerRef} />
+  return <div ref={containerRef} style={{ minHeight: '200px' }} />
 }
 
 export default Mermaid