1
1
/*
2
- * (C) Copyright IBM Corp. 2020, 2021 .
2
+ * (C) Copyright IBM Corp. 2022 .
3
3
*
4
4
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
5
5
* the License. You may obtain a copy of the License at
@@ -119,12 +119,6 @@ public void testMessageInputStateless() throws Throwable {
119
119
.location (new java .util .ArrayList <Long >(java .util .Arrays .asList (Long .valueOf ("26" ))))
120
120
.value ("testString" )
121
121
.confidence (Double .valueOf ("72.5" ))
122
- .metadata (
123
- new java .util .HashMap <String , Object >() {
124
- {
125
- put ("foo" , "testString" );
126
- }
127
- })
128
122
.groups (
129
123
new java .util .ArrayList <CaptureGroup >(java .util .Arrays .asList (captureGroupModel )))
130
124
.interpretation (runtimeEntityInterpretationModel )
@@ -139,13 +133,6 @@ public void testMessageInputStateless() throws Throwable {
139
133
new java .util .ArrayList <Long >(java .util .Arrays .asList (Long .valueOf ("26" ))));
140
134
assertEquals (runtimeEntityModel .value (), "testString" );
141
135
assertEquals (runtimeEntityModel .confidence (), Double .valueOf ("72.5" ));
142
- assertEquals (
143
- runtimeEntityModel .metadata (),
144
- new java .util .HashMap <String , Object >() {
145
- {
146
- put ("foo" , "testString" );
147
- }
148
- });
149
136
assertEquals (
150
137
runtimeEntityModel .groups (),
151
138
new java .util .ArrayList <CaptureGroup >(java .util .Arrays .asList (captureGroupModel )));
@@ -156,6 +143,11 @@ public void testMessageInputStateless() throws Throwable {
156
143
java .util .Arrays .asList (runtimeEntityAlternativeModel )));
157
144
assertEquals (runtimeEntityModel .role (), runtimeEntityRoleModel );
158
145
146
+ MessageInputAttachment messageInputAttachmentModel =
147
+ new MessageInputAttachment .Builder ().url ("testString" ).mediaType ("testString" ).build ();
148
+ assertEquals (messageInputAttachmentModel .url (), "testString" );
149
+ assertEquals (messageInputAttachmentModel .mediaType (), "testString" );
150
+
159
151
MessageInputOptionsSpelling messageInputOptionsSpellingModel =
160
152
new MessageInputOptionsSpelling .Builder ().suggestions (true ).autoCorrect (true ).build ();
161
153
assertEquals (messageInputOptionsSpellingModel .suggestions (), Boolean .valueOf (true ));
@@ -182,6 +174,9 @@ public void testMessageInputStateless() throws Throwable {
182
174
.entities (
183
175
new java .util .ArrayList <RuntimeEntity >(java .util .Arrays .asList (runtimeEntityModel )))
184
176
.suggestionId ("testString" )
177
+ .attachments (
178
+ new java .util .ArrayList <MessageInputAttachment >(
179
+ java .util .Arrays .asList (messageInputAttachmentModel )))
185
180
.options (messageInputOptionsStatelessModel )
186
181
.build ();
187
182
assertEquals (messageInputStatelessModel .messageType (), "text" );
@@ -193,6 +188,10 @@ public void testMessageInputStateless() throws Throwable {
193
188
messageInputStatelessModel .entities (),
194
189
new java .util .ArrayList <RuntimeEntity >(java .util .Arrays .asList (runtimeEntityModel )));
195
190
assertEquals (messageInputStatelessModel .suggestionId (), "testString" );
191
+ assertEquals (
192
+ messageInputStatelessModel .attachments (),
193
+ new java .util .ArrayList <MessageInputAttachment >(
194
+ java .util .Arrays .asList (messageInputAttachmentModel )));
196
195
assertEquals (messageInputStatelessModel .options (), messageInputOptionsStatelessModel );
197
196
198
197
String json = TestUtilities .serialize (messageInputStatelessModel );
0 commit comments