Fix extracting inferred queries from chat history db

Inferred queries is stored with underscore in db but aliased with - in memory.

This conversation.messages logic was broken, so inferred queries field
of chat message history was getting ignored.

This change fixes that issue and improve previous image generation
description for better context for subsequent image generation attempts.
This commit is contained in:
Debanjum
2025-08-25 00:28:36 -07:00
parent 892e4d4077
commit ff004d31ef

View File

@@ -636,9 +636,9 @@ class Conversation(DbBaseModel):
for msg in self.conversation_log.get("chat", []):
try:
# Clean up inferred queries if they contain None
if msg.get("intent") and msg["intent"].get("inferred-queries"):
if msg.get("intent") and msg["intent"].get("inferred_queries"):
msg["intent"]["inferred-queries"] = [
q for q in msg["intent"]["inferred-queries"] if q is not None and isinstance(q, str)
q for q in msg["intent"]["inferred_queries"] if q is not None and isinstance(q, str)
]
msg["message"] = str(msg.get("message", ""))
validated_messages.append(ChatMessageModel.model_validate(msg))