From 330febaa1a8510be1b2eaa60185d8d5000fd5d6c Mon Sep 17 00:00:00 2001 From: Debanjum Singh Solanky Date: Fri, 13 Jan 2023 21:22:40 -0300 Subject: [PATCH] Update conversation logs from /beta/summary API endpoint too --- src/routers/api_beta.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/routers/api_beta.py b/src/routers/api_beta.py index 8170999a..021e7e9a 100644 --- a/src/routers/api_beta.py +++ b/src/routers/api_beta.py @@ -47,6 +47,10 @@ def summarize_beta(q: str): model = state.processor_config.conversation.model api_key = state.processor_config.conversation.openai_api_key + # Load Conversation History + chat_session = state.processor_config.conversation.chat_session + meta_log = state.processor_config.conversation.meta_log + # Converse with OpenAI GPT result_list = search(q, n=1, t=SearchType.Org, r=True) collated_result = "\n".join([item.entry for item in result_list]) @@ -58,6 +62,10 @@ def summarize_beta(q: str): gpt_response = str(e) status = 'error' + # Update Conversation History + state.processor_config.conversation.chat_session = message_to_prompt(q, chat_session, gpt_message=gpt_response) + state.processor_config.conversation.meta_log['chat'] = message_to_log(q, gpt_response, conversation_log=meta_log.get('chat', [])) + return {'status': status, 'response': gpt_response}