Merge pull request #551 from Pingdred/why_in_wm

Added `why` in the working memory

core/cat/looking_glass/cheshire_cat.py

@@ -478,10 +478,7 @@ def __call__(self, user_message_json):
         log.info("cat_message:")
         log.info(cat_message)
 
-        # update conversation history
         user_message = user_working_memory["user_message_json"]["text"]
-        user_working_memory.update_conversation_history(who="Human", message=user_message)
-        user_working_memory.update_conversation_history(who="AI", message=cat_message["output"])
 
         # store user message in episodic memory
         # TODO: vectorize and store also conversation chunks
@@ -513,6 +510,10 @@ def __call__(self, user_message_json):
 
         final_output = self.mad_hatter.execute_hook("before_cat_sends_message", final_output)
 
+        # update conversation history
+        user_working_memory.update_conversation_history(who="Human", message=user_message)
+        user_working_memory.update_conversation_history(who="AI", message=final_output["content"], why=final_output["why"])
+
         return final_output
 
 

core/cat/memory/working_memory.py

@@ -27,7 +27,7 @@ def get_user_id(self):
 
         return self["user_message_json"]["user_id"]
 
-    def update_conversation_history(self, who, message):
+    def update_conversation_history(self, who, message, why={}):
         """Update the conversation history.
 
         The methods append to the history key the last three conversation turns.
@@ -41,7 +41,7 @@ def update_conversation_history(self, who, message):
         
         """
         # append latest message in conversation
-        self["history"].append({"who": who, "message": message})
+        self["history"].append({"who": who, "message": message, "why": why})
 
         # do not allow more than k messages in convo history (+2 which are the current turn)
         # TODO: allow infinite history, but only insert in prompts the last k messages

core/cat/routes/websocket.py

@@ -1,7 +1,7 @@
 import traceback
 import asyncio
 from cat.looking_glass.cheshire_cat import CheshireCat
-from typing import Dict, Optional
+from typing import Dict
 from fastapi import APIRouter, WebSocketDisconnect, WebSocket
 from cat.log import log
 from fastapi.concurrency import run_in_threadpool
@@ -88,8 +88,9 @@ async def websocket_endpoint(websocket: WebSocket, user_id: str = "user"):
     # Retrieve the `ccat` instance from the application's state.
     ccat = websocket.app.state.ccat
 
+    # Skip the coroutine if the same user is already connected via WebSocket.
     if user_id in manager.active_connections:
-        # Skip the coroutine if the same user is already connected via WebSocket.
+        log.error(f"A websocket connection with ID '{user_id}' has already been opened.")
         return
 
     # Add the new WebSocket connection to the manager.