fix code block copy and insert

TerminalFi · Jul 2, 2024 · a8ebab8 · a8ebab8
1 parent 7431723
commit a8ebab8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/plugin/ui/chat.py b/plugin/ui/chat.py
@@ -252,7 +252,6 @@ def process_code_block(reply, inside_code_block, code_block_index):
                 + "\n\n"
                 + code_block_lines[0]
             )
-            print(reply)
             return reply, code_block_lines
 
         transformed_conversation = []
@@ -278,8 +277,9 @@ def process_code_block(reply, inside_code_block, code_block_index):
                         self.conversation_manager.insert_code_block_index(
                             code_block_index, "".join(current_entry["code_block"])
                         )
+                        current_entry["code_block"] = []
                     else:
-                        current_entry["code_block"].extend(reply)
+                        current_entry["code_block"].append(reply)
                 current_entry["messages"].append(reply)
             else:
                 if current_entry: