memory is now in place

agent/_create.py

@@ -39,16 +39,23 @@ def agent(payload):
     workflow.add_conditional_edges("agent", should_continue, {"continue": "action", "end": END} )
     workflow.add_edge('action', 'agent')
 
-    from agent.memory import memory,ThreadStatus,threadID
-    app = workflow.compile(checkpointer=memory)
+    from agent.memory import Memory
+    memory = Memory(payload)
+
+    app = workflow.compile(checkpointer=memory.checkpoints)
 
 
     from agent.prompt import prompt
-    prompt=prompt[ThreadStatus]
-    prompt = prompt.format(input="hi! I'm bob")
+    prompt=prompt[memory.isNew]
+
+    input = payload.get("input") or "hi! I'm bob"
 
-    response = app.invoke(prompt, {"configurable": {"thread_id": threadID}})
+    prompt = prompt.format(input=input, thread_id=memory.thread_id)
+
+    response = app.invoke(prompt, {"configurable": {"thread_id": memory.thread_id}})
     print(response[-1].content)
 
+    print(payload)
+
     return response[-1].content
     

agent/memory.py

@@ -1,4 +1,42 @@
 from langgraph.checkpoint.sqlite import SqliteSaver
-memory = SqliteSaver.from_conn_string(":memory:")
-ThreadStatus=0
-threadID=2
+
+class extended_sqliteSaver(SqliteSaver):
+    def __init__(self, conn_string):
+        super().__init__(conn_string)
+        self.create_table("chat_history", {"thread_id": "TEXT", "message": "TEXT", "response": "TEXT"})
+
+    @classmethod
+    def from_conn_string(cls, conn_string: str) -> "SqliteSaver":
+        import sqlite3
+        return SqliteSaver(conn=sqlite3.connect(conn_string, check_same_thread=False))
+
+#memory = extended_sqliteSaver.from_conn_string("_chat_history.sqlite")
+
+class Memory: 
+
+    def __init__(self, payload):
+        self.checkpoints = extended_sqliteSaver.from_conn_string("_chat_history.sqlite")
+
+        if(payload.get("thread_id")):
+            self.thread_id = payload.get("thread_id")
+            self.isNew = 0
+        else:
+            self.thread_id = "1"
+            self.isNew = 1
+
+            try:
+
+                cursor = self.checkpoints.conn.cursor()
+
+                ## IF THERE ARE MORE THAN N THREADS, DELETE THE OLDEST ONES TO MAKE N THREADS
+                n=20
+                cursor.execute(f"DELETE FROM checkpoints WHERE thread_id NOT IN (SELECT thread_id FROM checkpoints ORDER BY thread_id DESC LIMIT {n})")
+        
+                ## GET THE HIGHEST THREAD VALUE AND ADD ONE TO CREATE THE NEW THREAD_ID
+                self.thread_id=cursor.execute("SELECT thread_id FROM checkpoints ORDER BY thread_id DESC LIMIT 1").fetchone()[0] or 0
+                self.thread_id=int(self.thread_id)+1
+            except:
+                self.thread_id=1
+
+            print("New thread created with id #", self.thread_id)
+

agent/prompt.py

@@ -2,7 +2,12 @@ from langchain_core.prompts import ChatPromptTemplate,SystemMessagePromptTemplat
 from agent.datastructures import parser
 
 prompt = {
-    0: # IF THE THREAD IS NEW, THE CHATBOT NEEDS TO BE PUMP-PROMPTED
+    0: # IF THREAD IS CONTINUING, WE CAN RELY ON THE ORIGINAL PROMPT 
+    ChatPromptTemplate.from_messages([
+        ("system", "The thread_id of this conversation is {thread_id}."),
+        ("human", "{input}")
+    ]),
+    1: # IF THE THREAD IS NEW, THE CHATBOT NEEDS TO BE PUMP-PROMPTED
     ChatPromptTemplate.from_messages([
         SystemMessagePromptTemplate.from_template("""                                               
 
@@ -12,13 +17,10 @@ prompt = {
 
             {response_format}
             
-            The thread_id of this conversation is 2.
-
             """
                                                   
         ).format(response_format=parser.get_format_instructions()),
+        ("system", "The thread_id of this conversation is {thread_id}."),
         ("human", "{input}")
-    ]),
-    1: # IF THREAD IS CONTINUING, WE CAN RELY ON THE ORIGINAL PROMPT 
-    ChatPromptTemplate.from_messages([("human", "{input}")])
+    ])
 }