store execution results or error messages in conversation history to allow for easier back-reference, e.g. when asking 'Please fix the code'

Author: dasmy

gpt_code_ui/webapp/main.py

@@ -83,6 +83,12 @@ def append(self, role: str, content: str):
     def upload_file(self, filename: str, file_info: str = None):
         self.append("user", f"In the following, I will refer to the file {filename}.\n{file_info}")
 
+    def add_execution_result(self, result: str):
+        self.append("user", f"Executing this code yielded the following output:\n{result}")
+
+    def add_error(self, message: str):
+        self.append("user", f"Executing this code lead to an error.\nThe error message reads:\n{message}")
+
     def __call__(self):
         return self._buffer
 
@@ -213,6 +219,15 @@ def proxy_kernel_manager(path):
     else:
         resp = requests.get(f'http://localhost:{KERNEL_APP_PORT}/{path}')
 
+    # store execution results in conversation history to allow back-references by the user
+    for res in json.loads(resp.content).get('results', []):
+        if res['type'] == "message":
+            chat_history.add_execution_result(res['value'])
+        elif res['type'] == "message_error":
+            chat_history.add_error(res['value'])
+
+        print(res)
+
     excluded_headers = ['content-encoding',
                         'content-length', 'transfer-encoding', 'connection']
     headers = [(name, value) for (name, value) in resp.raw.headers.items()