ai-docs-gen 0.1.5__tar.gz → 0.1.8__tar.gz

{ai_docs_gen-0.1.5 → ai_docs_gen-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-docs-gen
-Version: 0.1.5
+Version: 0.1.8
 Summary: CLI-инструмент для генерации технической документации по коду и конфигурациям
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown

{ai_docs_gen-0.1.5 → ai_docs_gen-0.1.8}/ai_docs/generator.py

@@ -1,6 +1,7 @@
 import json
 from datetime import datetime
 import os
+import re
 import shutil
 import subprocess
 from pathlib import Path
@@ -942,6 +943,7 @@ def generate_docs(
         if not mkdocs_bin:
             raise RuntimeError("mkdocs is not installed or not on PATH")
         subprocess.check_call([mkdocs_bin, "build", "-f", "mkdocs.yml"], cwd=output_root)
+        _postprocess_mermaid_html(output_root / "ai_docs_site")
         print("[ai-docs] mkdocs: done")
 
     # Save cache
@@ -957,3 +959,33 @@ def generate_docs(
         print("[ai-docs] errors summary:")
         for item in errors:
             print(f"[ai-docs] error: {item}")
+
+
+def _postprocess_mermaid_html(site_dir: Path) -> None:
+    if not site_dir.exists():
+        return
+
+    mermaid_div_re = re.compile(r'(<div class="mermaid">)(.*?)(</div>)', re.DOTALL)
+    bracket_re = re.compile(r"\\[([^\\]]*)\\]")
+
+    for path in site_dir.rglob("*.html"):
+        text = path.read_text(encoding="utf-8")
+        if 'class="mermaid"' not in text:
+            continue
+
+        def _fix_mermaid(match: re.Match) -> str:
+            head, body, tail = match.groups()
+            body = body.replace("&gt;", ">")
+
+            def _fix_brackets(bmatch: re.Match) -> str:
+                inner = bmatch.group(1).replace("(", " ").replace(")", "")
+                inner = re.sub(r"[ \\t]+", " ", inner)
+                return f"[{inner}]"
+
+            body = bracket_re.sub(_fix_brackets, body)
+            body = re.sub(r"[ \\t]+", " ", body)
+            return f"{head}{body}{tail}"
+
+        updated = mermaid_div_re.sub(_fix_mermaid, text)
+        if updated != text:
+            path.write_text(updated, encoding="utf-8")

{ai_docs_gen-0.1.5 → ai_docs_gen-0.1.8}/ai_docs/tokenizer.py

@@ -16,7 +16,7 @@ def get_encoding(model: str):
         return tiktoken.encoding_for_model(model)
     except KeyError:
         try:
-            return tiktoken.get_encoding("o200k_base")
+            return tiktoken.get_encoding("cl100k_base")
         except Exception:
             return _ByteEncoding()
 

{ai_docs_gen-0.1.5 → ai_docs_gen-0.1.8}/ai_docs_gen.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-docs-gen
-Version: 0.1.5
+Version: 0.1.8
 Summary: CLI-инструмент для генерации технической документации по коду и конфигурациям
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown

{ai_docs_gen-0.1.5 → ai_docs_gen-0.1.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ai-docs-gen"
-version = "0.1.5"
+version = "0.1.8"
 description = "CLI-инструмент для генерации технической документации по коду и конфигурациям"
 readme = "README.md"
 requires-python = ">=3.8"