(airavata) branch master updated: Supporting streamed outputs in kernel execution

[dimuthuupe]

This is an automated email from the ASF dual-hosted git repository.

dimuthuupe pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/airavata.git


The following commit(s) were added to refs/heads/master by this push:
     new ee3920e7a3 Supporting streamed outputs in kernel execution
ee3920e7a3 is described below

commit ee3920e7a3fa3d7b355eace3c1e860f671469c94
Author: Dimuthu Wannipurage <dimut...@gatech.edu>
AuthorDate: Thu Jul 18 04:13:32 2024 -0400

    Supporting streamed outputs in kernel execution
---
 modules/agent-framework/airavata-agent/jupyter/kernel.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/modules/agent-framework/airavata-agent/jupyter/kernel.py 
b/modules/agent-framework/airavata-agent/jupyter/kernel.py
index 53fe3b4cea..83886a1296 100644
--- a/modules/agent-framework/airavata-agent/jupyter/kernel.py
+++ b/modules/agent-framework/airavata-agent/jupyter/kernel.py
@@ -40,18 +40,28 @@ def execute():
     kc.execute(code)
 
     # Wait for the result and display it
+
+    execution_noticed = False
+    content_text = ""
     while True:
         try:
             msg = kc.get_iopub_msg(timeout=1)
-            #print(msg)
             content = msg["content"]
+            parent_header = msg["parent_header"]
 
             # When a message with the text stream comes and it's the result of 
our execution
+            if msg["msg_type"] == "execute_input":
+                execution_noticed = True
             if msg["msg_type"] == "stream" and content["name"] == "stdout":
                 print(content["text"])
-                return jsonify({'result': content["text"]}), 200
+                content_text = content_text + content["text"]
             if msg["msg_type"] == "error":
                 return jsonify({'error': content}), 200
+            if msg["msg_type"] == "status" and execution_noticed:
+                if content["execution_state"] and content["execution_state"] 
== "idle":
+                    if parent_header and parent_header["msg_type"]:
+                        if parent_header["msg_type"] == "execute_request": ## 
This is a result without stdout like a = 12
+                            return jsonify({'result': content_text}), 200
         except KeyboardInterrupt:
             print("Interrupted by user.")
             return jsonify({'error': "Intterrupted by user"}), 500