From 0a4896d754855711984e52570f0d33c76d9732fa Mon Sep 17 00:00:00 2001
From: Ayush Morbar <108860844+ayushmorbar@users.noreply.github.com>
Date: Sun, 3 Mar 2024 06:06:28 +0000
Subject: [PATCH] Add search functionality to magic_commands.py

---
 neogpt/utils/magic_commands.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/neogpt/utils/magic_commands.py b/neogpt/utils/magic_commands.py
index 09de959..7357e6c 100644
--- a/neogpt/utils/magic_commands.py
+++ b/neogpt/utils/magic_commands.py
@@ -210,6 +210,25 @@ def magic_commands(user_input, chain):
         cprint("📄 /export - Export the current chat memory to the settings/settings.yaml file")
 
         return True
+    
+    # If the user inputs '/search [keyword]', search the chat history for the keyword
+    elif user_input.startswith("/search"):
+        # Extract the keyword from the command
+        keyword = user_input.split(" ", 1)[1].strip().replace("'", "").replace('"', "")
+        # Create a flag to indicate if any messages are found
+        found = False
+        for message in chain.combine_documents_chain.memory.chat_memory.messages:
+            if keyword in message.content:
+                if isinstance(message, HumanMessage):
+                    cprint(
+                        f"   [bright_yellow]{get_username()} [/bright_yellow]{message.content}"
+                    )
+                else:
+                    cprint(f"   [bright_blue]NeoGPT: [/bright_blue]{message.content}")
+                found = True
+        if not found:
+            cprint(f"No messages found with the keyword '{keyword}'.")
+        return True
 
     # If the command is not recognized, print an error message
     else: