ilotoki_thu
/
line_bot
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

improved logging

This commit is contained in:
Ian Griffin 2023-08-05 10:17:43 +07:00
parent 1d23682b94
commit 8896ced97f
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
linebot_app.py
View File

@ -36,6 +36,16 @@ from linebot.v3.messaging import (
app = Flask(__name__) app = Flask(__name__)
# logging
log_file_name = "./linebot.message.log"
# initiate log file
log_file = open(log_file_name, "w")
log_file.write("")
log_file.close()
# User Sessions # User Sessions
user_sessions = {} user_sessions = {}
@ -94,12 +104,13 @@ def callback():
@handler.add(MessageEvent, message=TextMessageContent) @handler.add(MessageEvent, message=TextMessageContent)
def message_text(event): def message_text(event):
log_file = open(log_file_name, "a")
log_file.write("user_id: " + event.source.user_id + ", time: " + event.timestamp + ", message:" + event.message.text)
# User Session # User Session
# create session if none exist # create session if none exist
if event.source.user_id not in user_sessions.keys(): if event.source.user_id not in user_sessions.keys():
user_sessions[event.source.user_id] = ConversationalRetrievalChain.from_llm(llm,retriever=retriever,memory=ConversationBufferMemory(memory_key="chat_history", return_messages=True)) user_sessions[event.source.user_id] = ConversationalRetrievalChain.from_llm(llm,retriever=retriever,memory=ConversationBufferMemory(memory_key="chat_history", return_messages=True))
print(event.source.user_id, ":", event.message.text)
# unique_docs = retriever_from_llm.get_relevant_documents(query=event.message.text) # unique_docs = retriever_from_llm.get_relevant_documents(query=event.message.text)
with ApiClient(configuration) as api_client: with ApiClient(configuration) as api_client:
try: try:
@ -114,7 +125,8 @@ def message_text(event):
messages=[TextMessage(text=answer)] messages=[TextMessage(text=answer)]
)) ))
print(answer) log_file.write("bot_answer: " + answer + "\n")
log_file.close()
# with ApiClient(configuration) as api_client: # with ApiClient(configuration) as api_client:
# line_bot_api = MessagingApi(api_client) # line_bot_api = MessagingApi(api_client)
@ -135,4 +147,4 @@ def message_text(event):
if __name__ == "__main__": if __name__ == "__main__":
# start web server # start web server
app.run(port=srv_secrets.srv_port) app.run(port=srv_secrets.srv_port)