github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-05 20:58:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: chat message histrory mypy fix (#18250)

Browse Source 
Description: Fixed type: ignore's for mypy for
chat_message_histories(streamlit)
Adresses #17048 

Planning to add more based on reviews
This commit is contained in:
Yudhajit Sinha 2024-03-02 01:47:18 +05:30 committed by GitHub
parent b9416dc96a
commit e2b901c35b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
libs/community/langchain_community/chat_message_histories/streamlit.py
View File

@ -22,17 +22,26 @@ class StreamlitChatMessageHistory(BaseChatMessageHistory):
if key not in st.session_state: if key not in st.session_state:
st.session_state[key] = [] st.session_state[key] = []
self._messages = st.session_state[key] self._key = key
@property @property
def messages(self) -> List[BaseMessage]: # type: ignore def messages(self) -> List[BaseMessage]:
"""Retrieve the current list of messages""" """Retrieve the current list of messages"""
return self._messages import streamlit as st
return st.session_state[self._key]
@messages.setter
def messages(self, value: List[BaseMessage]) -> None:
"""Set the messages list with a new value"""
import streamlit as st
st.session_state[self._key] = value
def add_message(self, message: BaseMessage) -> None: def add_message(self, message: BaseMessage) -> None:
"""Add a message to the session memory""" """Add a message to the session memory"""
self._messages.append(message) self.messages.append(message)
def clear(self) -> None: def clear(self) -> None:
"""Clear session memory""" """Clear session memory"""
self._messages.clear() self.messages.clear()