langchain-ai/streamlit-agent
1
0
Fork 0
mirror of https://github.com/langchain-ai/streamlit-agent.git synced 2026-07-01 09:25:05 -04:00
Code Issues 8 Packages Projects Releases Wiki Activity

Simplify basic_streaming a bit

Browse Source
This commit is contained in:
Joshua Carroll
2023-07-03 10:49:55 -07:00
parent 0c6a1cc872
commit cffdb1f7e0
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
streamlit_agent/basic_streaming.py
+1 -3
View File
@@ -32,9 +32,7 @@ if prompt := st.chat_input():
st.stop()
with st.chat_message("assistant"):
container = st.empty()
stream_handler = StreamHandler(container)
stream_handler = StreamHandler(st.empty())
llm = ChatOpenAI(openai_api_key=openai_api_key, streaming=True, callbacks=[stream_handler])
response = llm(st.session_state.messages)
st.session_state.messages.append(ChatMessage(role="assistant", content=response.content))
container.markdown(response.content)