39 lines
1.4 KiB
Python
39 lines
1.4 KiB
Python
"""Code-fence helpers shared by the stream consumer and the gateway final send."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import re
|
|
|
|
|
|
def escape_code_fences_for_display(text: str) -> str:
|
|
"""Replace each ``` with \\`\\`\\` so text can be wrapped in an outer ``` block.
|
|
|
|
Reasoning content that quotes code would otherwise break the outer fence.
|
|
"""
|
|
return text.replace("```", "\\`\\`\\`") if isinstance(text, str) else text
|
|
|
|
|
|
def ensure_closed_code_fences(text: str) -> str:
|
|
"""Append a closing ``` and/or ` if the text has orphaned code markers.
|
|
|
|
Output truncated mid-code-block (finish_reason="length") would otherwise render
|
|
everything after the orphan as one code block / inline span; a spurious close is
|
|
far less harmful. Odd ``` count → fence on its own line; then, with complete
|
|
```…``` regions stripped, odd ` count → a backtick.
|
|
"""
|
|
if not isinstance(text, str) or not text:
|
|
return text
|
|
|
|
if text.count("```") % 2 == 1:
|
|
text = text.rstrip("\n") + "\n```"
|
|
|
|
# Strip complete fenced regions (and any trailing unclosed ``` that leaks
|
|
# through) so their internal backticks don't pollute the standalone count.
|
|
without_fences = re.sub(r"```.*?```", "", text, flags=re.DOTALL)
|
|
without_fences = re.sub(r"```[^`]*$", "", without_fences)
|
|
|
|
if without_fences.count("`") % 2 == 1:
|
|
text = text + "`"
|
|
|
|
return text
|