Merge pull request #3 from jonathan-buttner/ml-fix-unified-switch-case-usage

fixing switch with class issue

Author: maxhniebergall

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/action/TransportInferenceAction.java

@@ -12,7 +12,6 @@
 import org.elasticsearch.action.support.ActionFilters;
 import org.elasticsearch.inference.InferenceService;
 import org.elasticsearch.inference.InferenceServiceRegistry;
-import org.elasticsearch.inference.InferenceServiceResults;
 import org.elasticsearch.inference.Model;
 import org.elasticsearch.inference.UnparsedModel;
 import org.elasticsearch.injection.guice.Inject;

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/external/request/openai/OpenAiUnifiedChatCompletionRequestEntity.java

@@ -66,18 +66,21 @@ public XContentBuilder toXContent(XContentBuilder builder, Params params) throws
             for (UnifiedCompletionRequest.Message message : unifiedRequest.messages()) {
                 builder.startObject();
                 {
-                    switch (message.content()) {
-                        case UnifiedCompletionRequest.ContentString contentString -> builder.field(CONTENT_FIELD, contentString.content());
-                        case UnifiedCompletionRequest.ContentObjects contentObjects -> {
-                            builder.startArray(CONTENT_FIELD);
-                            for (UnifiedCompletionRequest.ContentObject contentObject : contentObjects.contentObjects()) {
-                                builder.startObject();
-                                builder.field(TEXT_FIELD, contentObject.text());
-                                builder.field(TYPE_FIELD, contentObject.type());
-                                builder.endObject();
-                            }
-                            builder.endArray();
+                    if (message.content() instanceof UnifiedCompletionRequest.ContentString contentString) {
+                        builder.field(CONTENT_FIELD, contentString.content());
+                    } else if (message.content() instanceof UnifiedCompletionRequest.ContentObjects contentObjects) {
+                        builder.startArray(CONTENT_FIELD);
+                        for (UnifiedCompletionRequest.ContentObject contentObject : contentObjects.contentObjects()) {
+                            builder.startObject();
+                            builder.field(TEXT_FIELD, contentObject.text());
+                            builder.field(TYPE_FIELD, contentObject.type());
+                            builder.endObject();
                         }
+                        builder.endArray();
+                    } else {
+                        throw new IllegalArgumentException(
+                            Strings.format("Unsupported message.content class received: %s", message.content().getClass().getSimpleName())
+                        );
                     }
 
                     builder.field(ROLE_FIELD, message.role());