Implement simple "Agent Think Frame" , Tracy can do teach summary now.

src/aios_kernel/chatsession.py

@@ -50,7 +50,9 @@ class ChatSessionDB:
                     SessionID TEXT PRIMARY KEY,
                     SessionOwner TEXT,
                     SessionTopic TEXT,
-                    StartTime TEXT
+                    StartTime TEXT,
+                    SummarizePos INTEGER,
+                    Summary TEXT
                 );
             """)
 
@@ -92,8 +94,8 @@ class ChatSessionDB:
         try:
             conn = self._get_conn()
             conn.execute("""
-                INSERT INTO ChatSessions (SessionID, SessionOwner,SessionTopic, StartTime)
-                VALUES (?,?, ?, ?)
+                INSERT INTO ChatSessions (SessionID, SessionOwner,SessionTopic, StartTime,SummarizePos,Summary)
+                VALUES (?,?, ?, ?,0,"")
             """, (session_id, session_owner,session_topic, start_time))
             conn.commit()
             return 0  # return 0 if successful
@@ -159,16 +161,17 @@ class ChatSessionDB:
         chatsession = c.fetchone()
         return chatsession
 
-    def get_chatsessions(self, limit, offset):
+    def list_chatsessions(self, owner_id, limit, offset):
         """ retrieve sessions with pagination """
         try:
             conn = self._get_conn()
             cursor = conn.cursor()
             cursor.execute("""
-                SELECT * FROM ChatSessions
+                SELECT SessionID FROM ChatSessions
+                WHERE SessionOwner = ?           
                 ORDER BY StartTime DESC
-                LIMIT ? OFFSET ?
-            """, (limit, offset))
+                LIMIT ? OFFSET ? 
+            """, (owner_id,limit, offset))
             results = cursor.fetchall()
             #self.close()
             return results  # return 0 and the result if successful
@@ -184,6 +187,25 @@ class ChatSessionDB:
         message = c.fetchone()
         return message
 
+    # read message from begin->now
+    def read_message(self,session_id,limit,offset):
+        try:
+            conn = self._get_conn()
+            cursor = conn.cursor()
+            cursor.execute("""
+                SELECT MessageID, SessionID, MsgType, PrevMsgID, SenderID, ReceiverID, Timestamp, Topic,Mentions,ContentMIME,Content,ActionName,ActionParams,ActionResult,DoneTime,Status FROM Messages
+                WHERE SessionID = ?
+                ORDER BY Timestamp 
+                LIMIT ? OFFSET ?
+            """, (session_id, limit, offset))
+            results = cursor.fetchall()
+            #self.close()
+            return results  # return 0 and the result if successful
+        except Error as e:
+            logging.error("Error occurred while getting messages: %s", e)
+            return -1, None  # return -1 and None if an error occurs
+        
+    # read message from  now->beign    
     def get_messages(self, session_id, limit, offset):
         """ retrieve messages of a session with pagination """
         try:
@@ -217,6 +239,20 @@ class ChatSessionDB:
             logging.error("Error occurred while updating message status: %s", e)
             return -1  # return -1 if an error occurs
 
+    def update_session_summary(self, session_id, summarize_pos, summary):
+        """ update the summary of a session """
+        try:
+            conn = self._get_conn()
+            conn.execute("""
+                UPDATE ChatSessions
+                SET SummarizePos = ?, Summary = ?
+                WHERE SessionID = ?
+            """, (summarize_pos, summary, session_id))
+            conn.commit()
+            return 0  # return 0 if successful
+        except Error as e:
+            logging.error("Error occurred while updating session summary: %s", e)
+            return -1
 
 # chat session store the chat history between owner and agent
 # chat session might be large, so can read / write at stream mode.
@@ -232,7 +268,7 @@ class AIChatSession:
     #    #result = AIChatSession()
 
     @classmethod
-    def get_session(cls,owner_id:str,session_topic:str,db_path:str,auto_create = True) -> str:
+    def get_session(cls,owner_id:str,session_topic:str,db_path:str,auto_create = True) -> 'AIChatSession':
         db = cls._dbs.get(db_path)
         if db is None:
             db = ChatSessionDB(db_path)
@@ -248,8 +284,42 @@ class AIChatSession:
         else:
             result = AIChatSession(owner_id,session[0],db)
             result.topic = session_topic
+            result.summarize_pos = session[4]
+            result.summary = session[5]
 
         return result
+    
+    @classmethod
+    def get_session_by_id(cls,session_id:str,db_path:str)->'AIChatSession':
+        db = cls._dbs.get(db_path)
+        if db is None:
+            db = ChatSessionDB(db_path)
+            cls._dbs[db_path] = db
+
+        result = None
+        session = db.get_chatsession_by_id(session_id)
+        if session is None:
+            return None
+        else:
+            result = AIChatSession(session[1],session[0],db)
+            result.topic = session[2]
+            result.summarize_pos = session[4]
+            result.summary = session[5]
+
+        return result        
+
+    @classmethod
+    def list_session(cls,owner_id:str,db_path:str) -> list[str]:
+        db = cls._dbs.get(db_path)
+        if db is None:
+            db = ChatSessionDB(db_path)
+            cls._dbs[db_path] = db
+
+        result = db.list_chatsessions(owner_id,16,0)
+        result_ids = []
+        for r in result:
+            result_ids.append(r[0])
+        return result_ids    
 
 
     def __init__(self,owner_id:str, session_id:str, db:ChatSessionDB) -> None:
@@ -259,12 +329,18 @@ class AIChatSession:
 
         self.topic : str = None
         self.start_time : str = None
+        self.summarize_pos : int = 0
+        self.summary = None
 
     def get_owner_id(self) -> str:
         return self.owner_id
 
-    def read_history(self, number:int=10,offset=0) -> [AgentMsg]:
-        msgs = self.db.get_messages(self.session_id, number, offset)
+    def read_history(self, number:int=10,offset=0,order="revers") -> [AgentMsg]:
+        if order == "revers":
+            msgs = self.db.get_messages(self.session_id, number, offset)
+        else:
+            msgs = self.db.read_message(self.session_id, number, offset)
+            
         result = []
         for msg in msgs:
             agent_msg = AgentMsg()
@@ -294,6 +370,12 @@ class AIChatSession:
         msg.session_id = self.session_id
         self.db.insert_message(msg)
 
+
+    def update_think_progress(self,progress:int,new_summary:str) -> None:
+        self.db.update_session_summary(self.session_id,progress,new_summary)
+        self.summarize_pos = progress
+        self.summary = new_summary
+
     #def attach_event_handler(self,handler) -> None:
     #    """chat session changed event handler"""
     #    pass