Enable ChatCompletions to handle ToolCall streaming

pom.xml

@@ -6,7 +6,7 @@
 
   <groupId>io.github.sashirestela</groupId>
   <artifactId>simple-openai</artifactId>
-  <version>2.3.1</version>
+  <version>2.3.2</version>
   <packaging>jar</packaging>
 
   <name>simple-openai</name>

src/main/java/io/github/sashirestela/openai/domain/chat/Choice.java

@@ -6,10 +6,12 @@
 import io.github.sashirestela.openai.domain.chat.message.ChatMsgResponse;
 import lombok.Getter;
 import lombok.NoArgsConstructor;
+import lombok.Setter;
 import lombok.ToString;
 
 @NoArgsConstructor
 @Getter
+@Setter
 @ToString
 @JsonNaming(PropertyNamingStrategies.SnakeCaseStrategy.class)
 public class Choice {

src/main/java/io/github/sashirestela/openai/domain/chat/message/ChatMsg.java

@@ -10,4 +10,8 @@ public Role getRole() {
         return role;
     }
 
+    public void setRole(Role role) {
+        this.role = role;
+    }
+
 }

src/main/java/io/github/sashirestela/openai/domain/chat/message/ChatMsgResponse.java

@@ -5,11 +5,13 @@
 import io.github.sashirestela.openai.domain.chat.tool.ChatToolCall;
 import lombok.Getter;
 import lombok.NoArgsConstructor;
+import lombok.Setter;
 
 import java.util.List;
 
 @NoArgsConstructor
 @Getter
+@Setter
 @JsonNaming(PropertyNamingStrategies.SnakeCaseStrategy.class)
 public class ChatMsgResponse extends ChatMsg {
 

src/main/java/io/github/sashirestela/openai/domain/chat/tool/ChatFunctionCall.java

@@ -4,10 +4,12 @@
 import lombok.AllArgsConstructor;
 import lombok.Getter;
 import lombok.NoArgsConstructor;
+import lombok.Setter;
 
 @NoArgsConstructor
 @AllArgsConstructor
 @Getter
+@Setter
 public class ChatFunctionCall {
 
     @Required

src/main/java/io/github/sashirestela/openai/domain/chat/tool/ChatToolCall.java

@@ -4,12 +4,16 @@
 import lombok.AllArgsConstructor;
 import lombok.Getter;
 import lombok.NoArgsConstructor;
+import lombok.Setter;
 
 @NoArgsConstructor
 @AllArgsConstructor
 @Getter
+@Setter
 public class ChatToolCall {
 
+    private Integer index;
+
     @Required
     private String id;
 

src/test/java/io/github/sashirestela/openai/domain/chat/ChatDomainTest.java

@@ -158,6 +158,7 @@ void testChatCompletionsCreateWithFunctionAnswer() throws IOException {
                 .message(new ChatMsgSystem("You are an expert in Mathematics"))
                 .message(new ChatMsgUser("What is the product of 123 and 456?"))
                 .message(new ChatMsgAssistant(null, List.of(new ChatToolCall(
+                        0,
                         "call_tAoX6VHyjQVLnM9CZvEsTEwW",
                         ChatToolType.FUNCTION,
                         new ChatFunctionCall("product", "{\"multiplicand\":123,\"multiplier\":456}")))))