fix: missing role in api response

Azure-Samples · Apr 22, 2024 · 4b10b9f · 4b10b9f
1 parent b468e25
commit 4b10b9f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/...end-java-quarkus/src/main/java/ai/azure/openai/rag/workshop/backend/rest/ChatMessage.java b/...end-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/...nd-java-quarkus/src/main/java/ai/azure/openai/rag/workshop/backend/rest/ChatResponse.java b/...nd-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;
   }