Log ChatMessageModel validation errors during conversation save

This commit is contained in:
Debanjum
2025-06-19 16:45:39 -07:00
parent 494e7b3856
commit 22d71cab44

View File

@@ -445,20 +445,26 @@ async def save_to_conversation_log(
if generated_mermaidjs_diagram: if generated_mermaidjs_diagram:
khoj_message_metadata["mermaidjsDiagram"] = generated_mermaidjs_diagram khoj_message_metadata["mermaidjsDiagram"] = generated_mermaidjs_diagram
updated_conversation = message_to_log( try:
user_message=q, updated_conversation = message_to_log(
chat_response=chat_response, user_message=q,
user_message_metadata=user_message_metadata, chat_response=chat_response,
khoj_message_metadata=khoj_message_metadata, user_message_metadata=user_message_metadata,
chat_history=chat_history, khoj_message_metadata=khoj_message_metadata,
) chat_history=chat_history,
await ConversationAdapters.save_conversation( )
user, except ValidationError as e:
updated_conversation, updated_conversation = None
client_application=client_application, logger.error(f"Error constructing chat history: {e}")
conversation_id=conversation_id,
user_message=q, if updated_conversation:
) await ConversationAdapters.save_conversation(
user,
updated_conversation,
client_application=client_application,
conversation_id=conversation_id,
user_message=q,
)
if is_promptrace_enabled(): if is_promptrace_enabled():
merge_message_into_conversation_trace(q, chat_response, tracer) merge_message_into_conversation_trace(q, chat_response, tracer)