Added tags to the trace for the front end flows

pages/010_LLM_Architectures.py

@@ -4,7 +4,7 @@ import streamlit as st
 from time import time
 
 from src.st_helpers import st_setup
-from src.common import img_dir, escape_dollars
+from src.common import img_dir, escape_dollars, test_group_tag
 from src.architectures import *
 
 
@@ -39,10 +39,11 @@ def show_side_by_side() -> None:
             # Now dispatch the messages per architecture
             for i, a in enumerate(Architecture.architectures):
                 request = ArchitectureRequest(query=prompt)
+                group_tag = test_group_tag()
                 with arch_cols[i]:
                     with st.spinner('Architecture processing request'):
                         start = time()
-                        a(request)
+                        a(request, trace_tags=["UI", "SideBySideCompare", group_tag])
                         elapsed_in_s = (int((time() - start) * 10))/10  # round to 1dp in seconds
                         st.write('##### Timing')
                         st.write(f'Request took **{elapsed_in_s}s**')
@@ -98,7 +99,7 @@ def show_architecture(architecture: str) -> None:
                 with st.chat_message("user"):
                     st.write(prompt)
                 request = ArchitectureRequest(query=prompt)
-                trace = arch(request)
+                trace = arch(request, trace_tags=["UI", "SingleArchTest"])
                 with st.chat_message("assistant"):
                     st.write(escape_dollars(request.response))
                 with trace_col: