From d77b933c3f5d623d743e19fb059e523700171c3a Mon Sep 17 00:00:00 2001
From: Stefan Bratanov <stefan.bratanov93@gmail.com>
Date: Tue, 13 Feb 2024 10:41:16 +0200
Subject: [PATCH] Add helper initializers for content parts

---
 .../stefanbratanov/jvm/openai/ChatMessage.java       | 12 ++++++++++++
 .../stefanbratanov/jvm/openai/TestDataUtil.java      |  9 ++++-----
 2 files changed, 16 insertions(+), 5 deletions(-)

diff --git a/src/main/java/io/github/stefanbratanov/jvm/openai/ChatMessage.java b/src/main/java/io/github/stefanbratanov/jvm/openai/ChatMessage.java
index c7defa9..035b2db 100644
--- a/src/main/java/io/github/stefanbratanov/jvm/openai/ChatMessage.java
+++ b/src/main/java/io/github/stefanbratanov/jvm/openai/ChatMessage.java
@@ -59,6 +59,18 @@ public String type() {
 
           public record ImageUrl(String url, Optional<String> detail) {}
         }
+
+        static TextContentPart textContentPart(String text) {
+          return new TextContentPart(text);
+        }
+
+        static ImageContentPart imageContentPart(String url) {
+          return new ImageContentPart(new ImageContentPart.ImageUrl(url, Optional.empty()));
+        }
+
+        static ImageContentPart imageContentPart(String url, String detail) {
+          return new ImageContentPart(new ImageContentPart.ImageUrl(url, Optional.of(detail)));
+        }
       }
     }
   }
diff --git a/src/test/java/io/github/stefanbratanov/jvm/openai/TestDataUtil.java b/src/test/java/io/github/stefanbratanov/jvm/openai/TestDataUtil.java
index 33a7196..2f9a67f 100644
--- a/src/test/java/io/github/stefanbratanov/jvm/openai/TestDataUtil.java
+++ b/src/test/java/io/github/stefanbratanov/jvm/openai/TestDataUtil.java
@@ -1,7 +1,6 @@
 package io.github.stefanbratanov.jvm.openai;
 
 import io.github.stefanbratanov.jvm.openai.ChatMessage.UserMessage.UserMessageWithContentParts.ContentPart;
-import io.github.stefanbratanov.jvm.openai.ChatMessage.UserMessage.UserMessageWithContentParts.ContentPart.ImageContentPart;
 import io.github.stefanbratanov.jvm.openai.CreateChatCompletionRequest.ResponseFormat;
 import io.github.stefanbratanov.jvm.openai.ThreadMessage.Content.ImageFileContent;
 import io.github.stefanbratanov.jvm.openai.ThreadMessage.Content.TextContent;
@@ -438,10 +437,10 @@ private ChatMessage randomChatMessage() {
                 randomInt(1, 5),
                 () ->
                     oneOf(
-                        new ContentPart.TextContentPart(randomString(15)),
-                        new ImageContentPart(
-                            new ImageContentPart.ImageUrl(
-                                randomString(7), Optional.of(oneOf("auto", "low", "high"))))),
+                        ContentPart.textContentPart(randomString(15)),
+                        ContentPart.imageContentPart(randomString(7)),
+                        ContentPart.imageContentPart(
+                            randomString(7), oneOf("auto", "low", "high"))),
                 ContentPart[]::new)),
         ChatMessage.assistantMessage(randomString(10)),
         ChatMessage.assistantMessage(