fix: fixed with rendering messages

KenyonY · Jul 20, 2024 · 25c3b89 · 25c3b89
1 parent 94ca947
commit 25c3b89
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/openai_forward/__main__.py b/openai_forward/__main__.py
@@ -120,6 +120,8 @@ def _start_streamlit(self, port):
                 'true',
                 '--theme.base',
                 'light',
+                '--browser.gatherUsageStats',
+                'false',
             ]
         )
 

diff --git a/openai_forward/webui/chat.py b/openai_forward/webui/chat.py
@@ -9,7 +9,8 @@ def render_chat_log_message(msg: Dict):
     if msg.get("user_role"):
         with st.chat_message(name="human"):
             messages = msg.pop('messages')
-            st.write({msg_item['role']: msg_item['content'] for msg_item in messages})
+            for msg_item in messages:
+                st.write(f"`{msg_item['role']}`: {msg_item['content']}")
             st.write(msg)
     elif msg.get("assistant_role"):
         with st.chat_message(name="ai"):

diff --git a/openai_forward/webui/run.py b/openai_forward/webui/run.py
@@ -441,7 +441,6 @@ def display_other_configuration():
             uid, msg = q.get()
             uid: bytes
             print(f"{uid=}")
-            time.sleep(0.01)
             item = orjson.loads(msg)
             if uid.startswith(b"0"):
                 item['user_role'] = True