Update core/utils.py

core/utils.py

@@ -1,21 +1,24 @@
 # algoforge_prime/core/utils.py
-
-# This file can contain various helper functions used across the core modules.
-# For example, text parsing, sanitization, logging helpers, etc.
+import re
 
 def basic_text_cleanup(text: str) -> str:
-    """A very basic cleanup for LLM outputs."""
-    if not text:
+    """A basic cleanup for LLM outputs that might be wrapped in markdown code blocks."""
+    if not text or not isinstance(text, str):
         return ""
-    # Remove common code block markers if they are at the very start/end of the whole string
+    
     text = text.strip()
-    if text.startswith("```python"):
-        text = text[len("```python"):].strip()
-    elif text.startswith("```"):
-        text = text[3:].strip()
+    # Common markdown code block patterns
+    patterns = [
+        r"^```python\s*(.*?)\s*```$",  # ```python ... ```
+        r"^```\s*(.*?)\s*```$",        # ``` ... ```
+    ]
     
-    if text.endswith("```"):
-        text = text[:-3].strip()
-    return text
+    for pattern in patterns:
+        match = re.match(pattern, text, re.DOTALL | re.IGNORECASE)
+        if match:
+            return match.group(1).strip() # Return content within the markers
+            
+    return text # Return original if no common markers found
 
-# Add more utilities as needed.
+# Add more utilities as needed.
+print("DEBUG: core.utils - Module defined.")