chore: shorten code

geprog · Sep 8, 2023 · 38ece57 · 38ece57
1 parent 3163a46
commit 38ece57
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/ai/question.py b/ai/question.py
@@ -30,10 +30,8 @@ def _get_chat(chat_id: str)->str:
                 memory_key="chat_history", return_messages=True, output_key="answer"
             ),
             "lastQuestionTime": time.time(),
-        })   
-
-    chatMemories[chat_id]["lastQuestion"] = time.time()
-
+        }).set('lastQuestionTime',time.time())   
+
     return chatMemories.get(chat_id,{}).get("memory")