Update llm_utils.py

llm_utils.py

@@ -14,7 +14,22 @@ def load_theorem_db(path="theorems.yaml"):
         raise ValueError("Invalid theorems.yaml structure")
 
     return data
+def load_theorem_context(yaml_path="theorems.yaml"):
+    with open(yaml_path, 'r') as f:
+        data = yaml.safe_load(f)
+
+    context_lines = []
+    for idx, th in enumerate(data.get('theorems', []), 1):
+        context_lines.append(
+            f"**Theorem {idx}: {th['name']}**\n"
+            f"- **Statement**: {th['statement']}\n"
+            f"- **Tags**: {', '.join(th['tags'])}\n"
+            f"- **When to Use**: {th['when_to_use']}\n"
+            f"- **Short Explanation**: {th['short_explanation']}\n"
+        )
+        context_lines.append('---')
 
+    return "\n".join(context_lines)
 
 def generate_context(theorems, equation_type):
     context = f"# Theorem Reference Context for Solving {equation_type.capitalize()} Equations\n\n"