sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-27 17:35:07 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Json dump contents in prompt tracer to make structure discernable

Browse source
This commit is contained in:
Debanjum 2024-10-31 18:04:43 -07:00
parent 5b15176e20
commit 89597aefe9
1 changed files with 5 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/khoj/processor/conversation/utils.py
View file

@ -1,4 +1,5 @@
import base64 import base64
import json
import logging import logging
import math import math
import mimetypes import mimetypes
@ -501,15 +502,12 @@ def commit_conversation_trace(
Returns the path to the repository. Returns the path to the repository.
""" """
# Serialize session, system message and response to yaml # Serialize session, system message and response to yaml
system_message_yaml = yaml.dump(system_message, allow_unicode=True, sort_keys=False, default_flow_style=False) system_message_yaml = json.dumps(system_message, ensure_ascii=False, sort_keys=False)
response_yaml = yaml.dump(response, allow_unicode=True, sort_keys=False, default_flow_style=False) response_yaml = json.dumps(response, ensure_ascii=False, sort_keys=False)
formatted_session = [{"role": message.role, "content": message.content} for message in session] formatted_session = [{"role": message.role, "content": message.content} for message in session]
session_yaml = yaml.dump(formatted_session, allow_unicode=True, sort_keys=False, default_flow_style=False) session_yaml = json.dumps(formatted_session, ensure_ascii=False, sort_keys=False)
query = ( query = (
yaml.dump(session[-1].content, allow_unicode=True, sort_keys=False, default_flow_style=False) json.dumps(session[-1].content, ensure_ascii=False, sort_keys=False).strip().removeprefix("'").removesuffix("'")
.strip()
.removeprefix("'")
.removesuffix("'")
) # Extract serialized query from chat session ) # Extract serialized query from chat session
# Extract chat metadata for session # Extract chat metadata for session