From ff004d31efbcca2e6392b4753b30cc1de08b8f3a Mon Sep 17 00:00:00 2001 From: Debanjum Date: Mon, 25 Aug 2025 00:28:36 -0700 Subject: [PATCH] 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. --- src/khoj/database/models/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/khoj/database/models/__init__.py b/src/khoj/database/models/__init__.py index e0874f59..535fa7e7 100644 --- a/src/khoj/database/models/__init__.py +++ b/src/khoj/database/models/__init__.py @@ -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))