klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-07 13:23:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Keep search results passed to GPT as context in conversation logs

Browse Source 
This will be useful to
1. Show source references used to arrive at answer
2. Carry out multi-turn conversations
This commit is contained in:
Debanjum Singh Solanky
2023-03-05 15:00:20 -06:00
parent 7cad1c9428
commit 45f461d175
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/khoj/processor/conversation/gpt.py
View File

@@ -8,6 +8,7 @@ import openai
# Internal Packages # Internal Packages
from khoj.utils.constants import empty_escape_sequences from khoj.utils.constants import empty_escape_sequences
from khoj.utils.helpers import merge_dicts
def answer(text, user_query, model, api_key=None, temperature=0.5, max_tokens=500): def answer(text, user_query, model, api_key=None, temperature=0.5, max_tokens=500):
@@ -192,7 +193,7 @@ def message_to_log(user_message, gpt_message, user_message_metadata={}, conversa
current_dt = datetime.now().strftime("%Y-%m-%d %H:%M:%S") current_dt = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
# Create json log from Human's message # Create json log from Human's message
human_log = user_message_metadata or default_user_message_metadata human_log = merge_dicts(user_message_metadata, default_user_message_metadata)
human_log["message"] = user_message human_log["message"] = user_message
human_log["by"] = "you" human_log["by"] = "you"
human_log["created"] = current_dt human_log["created"] = current_dt

2
src/khoj/routers/api_beta.py
View File

@@ -111,7 +111,7 @@ def chat(q: Optional[str] = None):
# Update Conversation History # Update Conversation History
state.processor_config.conversation.chat_session = message_to_prompt(q, chat_session, gpt_message=gpt_response) 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( state.processor_config.conversation.meta_log["chat"] = message_to_log(
q, gpt_response, conversation_log=meta_log.get("chat", []) q, gpt_response, user_message_metadata={"context": collated_result}, conversation_log=meta_log.get("chat", [])
) )
return {"status": status, "response": gpt_response} return {"status": status, "response": gpt_response}