update godot scenese

Author: hazelnutcloud

godot/examples/simple/message.gd

@@ -0,0 +1,17 @@
+class_name Message
+extends Node
+
+@onready var text_container = %Text
+@onready var icon = %Panel
+@export_enum("user", "assistant") var sender: String
+
+var completion_id: int = -1
+var pending: bool = false
+var errored: bool = false
+
+func set_text(new_text: String):
+	text_container.text = new_text
+	
+func append_text(new_text: String):
+	text_container.text += new_text
+

godot/examples/simple/message.tscn

@@ -1,10 +1,37 @@
-[gd_scene format=3 uid="uid://t862t0v8ht2q"]
+[gd_scene load_steps=5 format=3 uid="uid://t862t0v8ht2q"]
 
-[node name="RichTextLabel" type="RichTextLabel"]
-offset_right = 40.0
-offset_bottom = 40.0
+[ext_resource type="Script" path="res://examples/simple/message.gd" id="1_pko33"]
+[ext_resource type="Texture2D" uid="uid://dplw232htshgc" path="res://addons/godot-llama-cpp/assets/godot-llama-cpp-1024x1024.svg" id="2_dvc7y"]
+
+[sub_resource type="StyleBoxTexture" id="StyleBoxTexture_t8bgj"]
+texture = ExtResource("2_dvc7y")
+
+[sub_resource type="Theme" id="Theme_bw3pb"]
+Panel/styles/panel = SubResource("StyleBoxTexture_t8bgj")
+
+[node name="RichTextLabel" type="HBoxContainer"]
+anchors_preset = 15
+anchor_right = 1.0
+anchor_bottom = 1.0
+grow_horizontal = 2
+grow_vertical = 2
+size_flags_horizontal = 3
+theme_override_constants/separation = 20
+script = ExtResource("1_pko33")
+sender = "assistant"
+
+[node name="Panel" type="Panel" parent="."]
+unique_name_in_owner = true
+custom_minimum_size = Vector2(80, 80)
+layout_mode = 2
+size_flags_vertical = 0
+theme = SubResource("Theme_bw3pb")
+
+[node name="Text" type="RichTextLabel" parent="."]
+unique_name_in_owner = true
+layout_mode = 2
 size_flags_horizontal = 3
 focus_mode = 2
+text = "..."
 fit_content = true
-scroll_active = false
 selection_enabled = true

godot/examples/simple/simple.gd

@@ -1,22 +1,40 @@
 extends Node
 
+const message = preload("res://examples/simple/message.tscn")
+
 @onready var messages_container = %MessagesContainer
 @onready var llama_context = %LlamaContext
 
-var message = preload("res://examples/simple/message.tscn")
-
 func _on_text_edit_submit(input: String) -> void:
 	handle_input(input)
 
 func handle_input(input: String) -> void:
-	var new_message = message.instantiate()
-	new_message.text = input
-	messages_container.add_child(new_message)
+	var completion_id = llama_context.request_completion(input)
+	
+	var user_message: Message = message.instantiate()
+	messages_container.add_child(user_message)
+	user_message.set_text(input)
+	user_message.sender = "user"
+	user_message.completion_id = completion_id
 
-	var id = llama_context.request_completion(input)
-	print("request id: ", id)
+	var ai_message: Message = message.instantiate()
+	messages_container.add_child(ai_message)
+	ai_message.sender = "assistant"
+	ai_message.completion_id = completion_id
+	ai_message.pending = true
 
 
 
 func _on_llama_context_completion_generated(chunk: Dictionary) -> void:
-	print("new chunk: ", chunk)
+	var completion_id = chunk.id
+	for message: Message in messages_container.get_children():
+		if message.completion_id != completion_id or message.sender != "assistant":
+			continue
+		if chunk.has("error"):
+			message.errored = true
+		elif chunk.has("text"):
+			if message.pending:
+				message.pending = false
+				message.set_text(chunk["text"])
+			else:
+				message.append_text(chunk["text"])

godot/examples/simple/simple.tscn

@@ -40,9 +40,12 @@ unique_name_in_owner = true
 layout_mode = 2
 size_flags_horizontal = 3
 size_flags_vertical = 3
+theme_override_constants/separation = 30
 
 [node name="RichTextLabel2" parent="Panel/MarginContainer/VBoxContainer/ScrollContainer/MessagesContainer" instance=ExtResource("2_7iip7")]
 layout_mode = 2
+
+[node name="Text" parent="Panel/MarginContainer/VBoxContainer/ScrollContainer/MessagesContainer/RichTextLabel2" index="1"]
 text = "How can I help you?"
 
 [node name="HBoxContainer" type="HBoxContainer" parent="Panel/MarginContainer/VBoxContainer"]
@@ -69,3 +72,5 @@ unique_name_in_owner = true
 [connection signal="submit" from="Panel/MarginContainer/VBoxContainer/HBoxContainer/TextEdit" to="." method="_on_text_edit_submit"]
 [connection signal="pressed" from="Panel/MarginContainer/VBoxContainer/HBoxContainer/Button" to="Panel/MarginContainer/VBoxContainer/HBoxContainer/TextEdit" method="_on_button_pressed"]
 [connection signal="completion_generated" from="LlamaContext" to="." method="_on_llama_context_completion_generated"]
+
+[editable path="Panel/MarginContainer/VBoxContainer/ScrollContainer/MessagesContainer/RichTextLabel2"]