sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2025-02-17 16:14:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Update conversation logs from /beta/summary API endpoint too

Browse source
This commit is contained in:
Debanjum Singh Solanky 2023-01-13 21:22:40 -03:00
parent cb6f0b53c9
commit 330febaa1a
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/routers/api_beta.py
View file

@ -47,6 +47,10 @@ def summarize_beta(q: str):
model = state.processor_config.conversation.model model = state.processor_config.conversation.model
api_key = state.processor_config.conversation.openai_api_key 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 # Converse with OpenAI GPT
result_list = search(q, n=1, t=SearchType.Org, r=True) result_list = search(q, n=1, t=SearchType.Org, r=True)
collated_result = "\n".join([item.entry for item in result_list]) collated_result = "\n".join([item.entry for item in result_list])
@ -58,6 +62,10 @@ def summarize_beta(q: str):
gpt_response = str(e) gpt_response = str(e)
status = 'error' 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} return {'status': status, 'response': gpt_response}