diff --git a/src/backend-java-quarkus/src/main/java/ai/azure/openai/rag/workshop/backend/rest/ChatMessage.java b/src/backend-java-quarkus/src/main/java/ai/azure/openai/rag/workshop/backend/rest/ChatMessage.java
index 3b15f03..7bce637 100644
--- a/src/backend-java-quarkus/src/main/java/ai/azure/openai/rag/workshop/backend/rest/ChatMessage.java
+++ b/src/backend-java-quarkus/src/main/java/ai/azure/openai/rag/workshop/backend/rest/ChatMessage.java
@@ -3,5 +3,6 @@
 public class ChatMessage {
 
   public String content;
+  public String role;
 
 }
diff --git a/src/backend-java-quarkus/src/main/java/ai/azure/openai/rag/workshop/backend/rest/ChatResponse.java b/src/backend-java-quarkus/src/main/java/ai/azure/openai/rag/workshop/backend/rest/ChatResponse.java
index 7cec2c1..48d1c21 100644
--- a/src/backend-java-quarkus/src/main/java/ai/azure/openai/rag/workshop/backend/rest/ChatResponse.java
+++ b/src/backend-java-quarkus/src/main/java/ai/azure/openai/rag/workshop/backend/rest/ChatResponse.java
@@ -14,6 +14,7 @@ public static ChatResponse fromMessage(String message) {
     choice.index = 0;
     choice.message = new ai.azure.openai.rag.workshop.backend.rest.ChatMessage();
     choice.message.content = message;
+    choice.message.role = "assistant";
     chatResponse.choices.add(choice);
     return chatResponse;
   }