Added more memrory methods

1 files changed, 25 insertions, 0 deletions

diff --git a/client.py b/client.py
index 0cbd91c..438f1fb 100644
--- a/client.py
+++ b/client.py
@@ -37,5 +37,30 @@ while True:
     if user_input.lower() in {"exit", "quit"}:
         break
 
+    match user_input:
+        case "/end":
+            agent.end_task()
+            print("[task ended]")
+            continue
+
+        case "/task":
+            print(f"\n[memory: {agent.working_memory}]\n")
+            continue
+
+        case "/pause":
+            agent.pause_task()
+            print("[task paused]")
+            continue
+
+        case "/resume":
+            agent.resume_task()
+            print("[task resumed]")
+            continue
+
+        case "/complete":
+            agent.complete_task()
+            print("[task completed]")
+            continue
+
     answer = agent.chat(user_input)
     print(f"Agent: {answer}")