config to hide internal types

Author: Jiarui Guo

sdk/translation/azure-ai-translation-text/customization/src/main/java/TextTranslationCustomization.java

@@ -48,5 +48,17 @@ public void customize(LibraryCustomization customization, Logger logger) {
                 }
             )
         );
+
+        logger.info("Customizing the client classes - public protocol methods");
+        customization.getClass("com.azure.ai.translation.text", "TextTranslationClient").customizeAst(ast -> 
+            ast.getClassByName("TextTranslationClient").ifPresent(clazz -> {
+                clazz.getMethodsByName("translateWithResponse").forEach(method -> method.setPublic(true));
+                clazz.getMethodsByName("transliterateWithResponse").forEach(method -> method.setPublic(true));
+            }));
+        customization.getClass("com.azure.ai.translation.text", "TextTranslationAsyncClient").customizeAst(ast -> 
+            ast.getClassByName("TextTranslationAsyncClient").ifPresent(clazz -> {
+                clazz.getMethodsByName("translateWithResponse").forEach(method -> method.setPublic(true));
+                clazz.getMethodsByName("transliterateWithResponse").forEach(method -> method.setPublic(true));
+            }));
     }
 }

sdk/translation/azure-ai-translation-text/src/main/java/com/azure/ai/translation/text/TextTranslationAsyncClient.java

@@ -4,16 +4,16 @@
 package com.azure.ai.translation.text;
 
 import com.azure.ai.translation.text.implementation.TextTranslationClientImpl;
+import com.azure.ai.translation.text.implementation.models.InputTextItem;
+import com.azure.ai.translation.text.implementation.models.TranslateBody;
+import com.azure.ai.translation.text.implementation.models.TranslationResult;
+import com.azure.ai.translation.text.implementation.models.TransliterateBody;
+import com.azure.ai.translation.text.implementation.models.TransliterateResult;
 import com.azure.ai.translation.text.models.GetSupportedLanguagesResult;
-import com.azure.ai.translation.text.models.InputTextItem;
 import com.azure.ai.translation.text.models.LanguageScope;
 import com.azure.ai.translation.text.models.TranslateInputItem;
-import com.azure.ai.translation.text.models.TranslateInputs;
 import com.azure.ai.translation.text.models.TranslatedTextItem;
-import com.azure.ai.translation.text.models.TranslationResult;
 import com.azure.ai.translation.text.models.TranslationTarget;
-import com.azure.ai.translation.text.models.TransliterateInputs;
-import com.azure.ai.translation.text.models.TransliterateResult;
 import com.azure.ai.translation.text.models.TransliteratedText;
 import com.azure.core.annotation.Generated;
 import com.azure.core.annotation.ReturnType;
@@ -309,27 +309,63 @@ public Mono<Response<BinaryData>> transliterateWithResponse(String language, Str
      * @param fromScript Specifies the script used by the input text. Look up supported languages using the
      * transliteration scope,
      * to find input scripts available for the selected language.
-     * @param toScript Specifies the output script. Look up supported languages using the transliteration
-     * scope, to find output
+     * @param toScript Specifies the output script. Look up supported languages using the transliteration scope, to find
+     * output
      * scripts available for the selected combination of input language and input script.
-     * @param inputs Defines the text inputs to transliterate.
+     * @param transliterateInput Defines the content of the request.
+     * @param clientTraceId A client-generated GUID to uniquely identify the request.
      * @throws IllegalArgumentException thrown if parameters fail the validation.
      * @throws HttpResponseException thrown if the request is rejected by server.
      * @throws ClientAuthenticationException thrown if the request is rejected by server on status code 401.
      * @throws ResourceNotFoundException thrown if the request is rejected by server on status code 404.
      * @throws ResourceModifiedException thrown if the request is rejected by server on status code 409.
      * @throws RuntimeException all other wrapped checked exceptions if the request fails to be sent.
-     * @return the response body on successful completion of {@link Mono}.
+     * @return response for the transliteration API on successful completion of {@link Mono}.
+     */
+    @Generated
+    @ServiceMethod(returns = ReturnType.SINGLE)
+    Mono<TransliterateResult> transliterate(String language, String fromScript, String toScript,
+        TransliterateBody transliterateInput, String clientTraceId) {
+        // Generated convenience method for transliterateWithResponse
+        RequestOptions requestOptions = new RequestOptions();
+        if (clientTraceId != null) {
+            requestOptions.setHeader(HttpHeaderName.fromString("X-ClientTraceId"), clientTraceId);
+        }
+        return transliterateWithResponse(language, fromScript, toScript, BinaryData.fromObject(transliterateInput),
+            requestOptions).flatMap(FluxUtil::toMono)
+                .map(protocolMethodData -> protocolMethodData.toObject(TransliterateResult.class));
+    }
+
+    /**
+     * Transliterate Text.
+     *
+     * @param language Specifies the language of the text to convert from one script to another.
+     * Possible languages are listed in the transliteration scope obtained by querying the service
+     * for its supported languages.
+     * @param fromScript Specifies the script used by the input text. Look up supported languages using the
+     * transliteration scope,
+     * to find input scripts available for the selected language.
+     * @param toScript Specifies the output script. Look up supported languages using the transliteration scope, to find
+     * output
+     * scripts available for the selected combination of input language and input script.
+     * @param transliterateInput Defines the content of the request.
+     * @throws IllegalArgumentException thrown if parameters fail the validation.
+     * @throws HttpResponseException thrown if the request is rejected by server.
+     * @throws ClientAuthenticationException thrown if the request is rejected by server on status code 401.
+     * @throws ResourceNotFoundException thrown if the request is rejected by server on status code 404.
+     * @throws ResourceModifiedException thrown if the request is rejected by server on status code 409.
+     * @throws RuntimeException all other wrapped checked exceptions if the request fails to be sent.
+     * @return response for the transliteration API on successful completion of {@link Mono}.
      */
-    private Mono<List<TransliteratedText>> transliterateInner(String language, String fromScript, String toScript,
-        List<InputTextItem> inputs) {
+    @Generated
+    @ServiceMethod(returns = ReturnType.SINGLE)
+    Mono<TransliterateResult> transliterate(String language, String fromScript, String toScript,
+        TransliterateBody transliterateInput) {
+        // Generated convenience method for transliterateWithResponse
         RequestOptions requestOptions = new RequestOptions();
-        TransliterateInputs transliterateBody = new TransliterateInputs(inputs);
-        return transliterateWithResponse(language, fromScript, toScript, BinaryData.fromObject(transliterateBody),
-            requestOptions).flatMap(FluxUtil::toMono).map(protocolMethodData -> {
-                TransliterateResult result = protocolMethodData.toObject(TransliterateResult.class);
-                return result.getValue();
-            });
+        return transliterateWithResponse(language, fromScript, toScript, BinaryData.fromObject(transliterateInput),
+            requestOptions).flatMap(FluxUtil::toMono)
+                .map(protocolMethodData -> protocolMethodData.toObject(TransliterateResult.class));
     }
 
     /**
@@ -359,7 +395,8 @@ private Mono<List<TransliteratedText>> transliterateInner(String language, Strin
     @ServiceMethod(returns = ReturnType.SINGLE)
     public Mono<List<TransliteratedText>> transliterate(String language, String fromScript, String toScript,
         List<String> inputs) {
-        return transliterateInner(language, fromScript, toScript, convertTextsToInputTextItems(inputs));
+        TransliterateBody transliterateBody = new TransliterateBody(convertTextsToInputTextItems(inputs));
+        return transliterate(language, fromScript, toScript, transliterateBody).map(result -> result.getValue());
     }
 
     /**
@@ -482,6 +519,52 @@ public Mono<Response<BinaryData>> translateWithResponse(BinaryData translateInpu
         return this.serviceClient.translateWithResponseAsync(translateInput, requestOptions);
     }
 
+    /**
+     * Translate Text.
+     *
+     * @param translateInput Defines the content of the request.
+     * @param clientTraceId A client-generated GUID to uniquely identify the request.
+     * @throws IllegalArgumentException thrown if parameters fail the validation.
+     * @throws HttpResponseException thrown if the request is rejected by server.
+     * @throws ClientAuthenticationException thrown if the request is rejected by server on status code 401.
+     * @throws ResourceNotFoundException thrown if the request is rejected by server on status code 404.
+     * @throws ResourceModifiedException thrown if the request is rejected by server on status code 409.
+     * @throws RuntimeException all other wrapped checked exceptions if the request fails to be sent.
+     * @return response for the translation API on successful completion of {@link Mono}.
+     */
+    @Generated
+    @ServiceMethod(returns = ReturnType.SINGLE)
+    Mono<TranslationResult> translate(TranslateBody translateInput, String clientTraceId) {
+        // Generated convenience method for translateWithResponse
+        RequestOptions requestOptions = new RequestOptions();
+        if (clientTraceId != null) {
+            requestOptions.setHeader(HttpHeaderName.fromString("X-ClientTraceId"), clientTraceId);
+        }
+        return translateWithResponse(BinaryData.fromObject(translateInput), requestOptions).flatMap(FluxUtil::toMono)
+            .map(protocolMethodData -> protocolMethodData.toObject(TranslationResult.class));
+    }
+
+    /**
+     * Translate Text.
+     *
+     * @param translateInput Defines the content of the request.
+     * @throws IllegalArgumentException thrown if parameters fail the validation.
+     * @throws HttpResponseException thrown if the request is rejected by server.
+     * @throws ClientAuthenticationException thrown if the request is rejected by server on status code 401.
+     * @throws ResourceNotFoundException thrown if the request is rejected by server on status code 404.
+     * @throws ResourceModifiedException thrown if the request is rejected by server on status code 409.
+     * @throws RuntimeException all other wrapped checked exceptions if the request fails to be sent.
+     * @return response for the translation API on successful completion of {@link Mono}.
+     */
+    @Generated
+    @ServiceMethod(returns = ReturnType.SINGLE)
+    Mono<TranslationResult> translate(TranslateBody translateInput) {
+        // Generated convenience method for translateWithResponse
+        RequestOptions requestOptions = new RequestOptions();
+        return translateWithResponse(BinaryData.fromObject(translateInput), requestOptions).flatMap(FluxUtil::toMono)
+            .map(protocolMethodData -> protocolMethodData.toObject(TranslationResult.class));
+    }
+
     /**
      * Translate Text.
      *
@@ -496,14 +579,8 @@ public Mono<Response<BinaryData>> translateWithResponse(BinaryData translateInpu
      */
     @ServiceMethod(returns = ReturnType.SINGLE)
     public Mono<List<TranslatedTextItem>> translate(List<TranslateInputItem> inputs) {
-        // Generated convenience method for translateWithResponse
-        RequestOptions requestOptions = new RequestOptions();
-        TranslateInputs translateBody = new TranslateInputs(inputs);
-        return translateWithResponse(BinaryData.fromObject(translateBody), requestOptions).flatMap(FluxUtil::toMono)
-            .map(protocolMethodData -> {
-                TranslationResult result = protocolMethodData.toObject(TranslationResult.class);
-                return result.getValue();
-            });
+        TranslateBody translateBody = new TranslateBody(inputs);
+        return translate(translateBody).map(result -> result.getValue());
     }
 
     /**