ai-assistant / config /print_data.py
digitalai's picture
Update config/print_data.py
a9cbfc0 verified
raw
history blame contribute delete
615 Bytes
class ConversationProcessor:
def __init__(self, session_state):
self.session_state = session_state
def save_conversation_to_text(self, output_file):
with open(output_file, 'w', encoding='utf-8') as file:
file.write("Assistant: " + "\n")
for message in self.session_state['messages']:
role = message['role']
content = message['content']
if role == 'user':
file.write("User: " + content + "\n")
elif role == 'assistant':
file.write("Assistant: " + content + "\n")