Fixed a display issue in the markdown with dollar signs in prices

pages/010_LLM_Architectures.py

@@ -2,7 +2,7 @@ import pandas as pd
 import streamlit as st
 
 from src.st_helpers import st_setup
-from src.common import img_dir
+from src.common import img_dir, escape_dollars
 from src.architectures import *
 
 
@@ -51,7 +51,7 @@ def show_architecture(architecture: str) -> None:
                 request = ArchitectureRequest(query=prompt)
                 trace = arch(request)
                 with st.chat_message("assistant"):
-                    st.write(request.response)
+                    st.write(escape_dollars(request.response))
                 with trace_col:
                     st.write("#### Architecture Trace")
                     st.markdown(trace.as_markdown())

src/architectures.py

@@ -14,7 +14,7 @@ from time import time
 from typing import List, Optional
 from better_profanity import profanity
 
-from src.common import config_dir, data_dir, hf_api_token
+from src.common import config_dir, data_dir, hf_api_token, escape_dollars
 from src.models import HFLlamaChatModel
 
 
@@ -59,7 +59,7 @@ class ArchitectureRequest:
         md += "\n- **Response evolution**"
         for r in self._response:
             md += f"\n  - {r}"
-        return md
+        return escape_dollars(md)
 
 
 class ArchitectureTraceOutcome(Enum):
@@ -112,7 +112,7 @@ class ArchitectureTraceStep:
         elif self.outcome == ArchitectureTraceOutcome.EXCEPTION:
             outcome = f"Exception ({self._exception})"
         md += f"  - **Outcome**: {outcome}"
-        return md
+        return escape_dollars(md)
 
 
 class ArchitectureTrace:

src/common.py

@@ -30,3 +30,13 @@ def join_items_comma_and(items: List[str]) -> str:
     if string_count == 1:
         return items[0]
     return f"{', '.join(items[:-1])} and {items[-1]}"
+
+
+def escape_dollars(text: str) -> str:
+    """
+    Convenience function to escape dollar signs for prices in markdown
+    """
+    if text is None:
+        return text
+    return text.replace("$", "\\$")
+