Fixed new/save/execute_plan

ai-ops-cli.py

@@ -76,7 +76,7 @@ def new_session(self):
         response.raise_for_status()
 
         self.current_session = {'sid': response.json()['sid'], 'name': session_name}
-        self.chat()
+        self.chat(print_name=True)
 
     def save_session(self):
         """Save the current session"""
@@ -87,7 +87,6 @@ def save_session(self):
             self.console.print(f'[!] Failed: {response.status_code}')
         else:
             self.console.print(f'[+] Saved')
-        self.chat()
 
     def rename_session(self):
         """Renames the current session"""
@@ -102,7 +101,7 @@ def rename_session(self):
         )
         response.raise_for_status()
         self.current_session['name'] = session_name
-        self.chat()
+        self.chat(print_name=True)
 
     def delete_session(self):
         """Deletes the current session"""
@@ -199,6 +198,8 @@ def execute_plan(self):
             for task_str in resp.iter_content():
                 if task_str:
                     self.console.print(task_str.decode(), end='')
+            print()
+        self.chat()
 
     def list_plans(self):
         """Retrieve the plans in the current session and

src/api.py

@@ -127,6 +127,7 @@ def new_session(name: str):
     else:
         new_id = max(sorted(sessions.keys())) + 1
     agent.new_session(new_id)
+    agent.get_session(new_id).name = name
 
     return {'sid': new_id}