Add devportal rest api changes for API Chat feature

components/apimgt/org.wso2.carbon.apimgt.api/src/main/java/org/wso2/carbon/apimgt/api/APIConsumer.java

@@ -799,6 +799,28 @@ Set<SubscribedAPI> getPaginatedSubscribedAPIsByApplication(Application applicati
      */
     Tier getThrottlePolicyByName(String name, int policyType, String organization) throws APIManagementException;
 
+    /**
+     * Returns the API Chat execute call response as a string
+     *
+     * @param apiChatRequestId Request UUID
+     * @param requestPayload Request payload to be used for the AI service execute call
+     * @return execution response as a string
+     * @throws APIManagementException if execute call failed
+     */
+    String invokeApiChatExecute(String apiChatRequestId, String requestPayload) throws APIManagementException;
+
+    /**
+     * Returns the API Chat prepare call response as a string
+     *
+     * @param apiId            ID of the API
+     * @param apiChatRequestId Request UUID
+     * @param organization     Identifier of an organization
+     * @return prepare response
+     * @throws APIManagementException if prepare call failed
+     */
+    String invokeApiChatPrepare(String apiId, String apiChatRequestId, String organization)
+            throws APIManagementException;
+
     /**
      * This method used to retrieve key manager configurations for tenant
      * @param organization organization of the key manager

components/apimgt/org.wso2.carbon.apimgt.api/src/main/java/org/wso2/carbon/apimgt/api/ExceptionCodes.java

@@ -548,7 +548,11 @@
     KEY_MANAGER_RESTRICTED_FOR_USER(902013, "Unauthorized Access to Key Manager", 403, "Key Manager is Restricted for this user"),
     // Admin portal get apis and api provider change related errors
     CHANGE_API_PROVIDER_FAILED(903011, "Error while changing the API provider", 500, "Error while changing the API provider in the registry or DB"),
-    GET_SEARCH_APIS_IN_ADMIN_FAILED(903012, "Error while getting the apis", 500, "Error while getting/searching the apis from registry");
+    GET_SEARCH_APIS_IN_ADMIN_FAILED(903012, "Error while getting the apis", 500, "Error while getting/searching the apis from registry"),
+
+    // AI service invocation related exceptions
+    AI_SERVICE_INVALID_RESPONSE(903100, "Invalid response from AI service", 500, "Error while invoking AI service. %s", false),
+    AI_SERVICE_INVALID_ACCESS_TOKEN(900906, "Invalid access token provided for AI service", 401, "Invalid access token provided for AI service");
 
     private final long errorCode;
     private final String errorMessage;

components/apimgt/org.wso2.carbon.apimgt.impl/src/main/java/org/wso2/carbon/apimgt/impl/APIConstants.java

@@ -501,6 +501,14 @@ public final class APIConstants {
     public static final String DIGEST = "x5t#S256";
     public static final String CNF = "cnf";
 
+    // Constants related to AI features: API chat and Marketplace Assistant
+    public static final String API_CHAT =  "APIChat.";
+    public static final String API_CHAT_ENABLED = API_CHAT + "Enabled";
+    public static final String API_CHAT_AUTH_TOKEN = API_CHAT + "AuthToken";
+    public static final String API_CHAT_ENDPOINT = API_CHAT + "Endpoint";
+    public static final String API_CHAT_PREPARE_RESOURCE = "/prepare"; // "/api-chat/prepare"
+    public static final String API_CHAT_EXECUTE_RESOURCE = "/chat"; // "/api-chat/chat"
+
     //documentation rxt
 
     public static final String DOC_NAME = "overview_name";

components/apimgt/org.wso2.carbon.apimgt.impl/src/main/java/org/wso2/carbon/apimgt/impl/APIConsumerImpl.java

@@ -3308,6 +3308,23 @@
         return APIUtil.removeXMediationScriptsFromSwagger(definition);
     }
 
+    @Override
+    public String invokeApiChatExecute(String apiChatRequestId, String requestPayload) throws APIManagementException {
+        return APIUtil.invokeAIService(APIConstants.API_CHAT_ENDPOINT,
+                APIConstants.API_CHAT_AUTH_TOKEN, APIConstants.API_CHAT_EXECUTE_RESOURCE, requestPayload,
+                apiChatRequestId);
+    }
+
+    @Override
+    public String invokeApiChatPrepare(String apiId, String apiChatRequestId, String organization)
+            throws APIManagementException {
+        String swaggerDefinition = getOpenAPIDefinition(apiId, organization);
+        String payload = "{\"openapi\": " + swaggerDefinition + "}";
+        String prepareResponse = APIUtil.invokeAIService(APIConstants.API_CHAT_ENDPOINT,
+                APIConstants.API_CHAT_AUTH_TOKEN, APIConstants.API_CHAT_PREPARE_RESOURCE, payload, apiChatRequestId);
+        return prepareResponse;
+    }
+
     @Override
     public String getOpenAPIDefinitionForEnvironment(API api, String environmentName)
             throws APIManagementException {

components/apimgt/org.wso2.carbon.apimgt.impl/src/main/java/org/wso2/carbon/apimgt/impl/utils/APIUtil.java

@@ -10371,4 +10371,76 @@
         }
         return scopesStringBuilder.toString().trim();
     }
+
+    /**
+     * Check whether API Chat feature is enabled
+     *
+     * @return returns true if API Chat feature is enabled, false if disabled.
+     */
+    public static boolean isApiChatEnabled() {
+
+        APIManagerConfiguration config = ServiceReferenceHolder.getInstance().
+                getAPIManagerConfigurationService().getAPIManagerConfiguration();
+        String isApiChatEnabled = config.getFirstProperty(APIConstants.API_CHAT_ENABLED);
+        if (isApiChatEnabled == null) {
+            return false;
+        }
+
+        return Boolean.parseBoolean(isApiChatEnabled);
+    }
+
+    /**
+     * Checks whether an auth token is provided for AI features to use. This token is utilized for authentication and
+     * throttling purposes.
+     *
+     * @return returns true if a valid auth token is found, false otherwise.
+     */
+    public static boolean isAuthTokenProvidedForAIFeatures() {
+        APIManagerConfiguration config = ServiceReferenceHolder.getInstance().
+                getAPIManagerConfigurationService().getAPIManagerConfiguration();
+        String authToken = config.getFirstProperty(APIConstants.API_CHAT_AUTH_TOKEN);
+        if (authToken == null || authToken.equals("")) {
+            return false;
+        }
+        return true;
+    }
+
+    public static String invokeAIService(String endpointConfigName, String authTokenConfigName,
+            String resource, String payload, String requestId) throws  APIManagementException {
+
+        APIManagerConfiguration config = ServiceReferenceHolder.getInstance().
+                getAPIManagerConfigurationService().getAPIManagerConfiguration();
+        String endpoint = config.getFirstProperty(endpointConfigName);
+        String authToken = config.getFirstProperty(authTokenConfigName);
+        try {
+            HttpPost preparePost = new HttpPost(endpoint + resource);
+            preparePost.setHeader(HttpHeaders.AUTHORIZATION, APIConstants.AUTHORIZATION_BEARER + authToken);
+            preparePost.setHeader(HttpHeaders.CONTENT_TYPE, APIConstants.APPLICATION_JSON_MEDIA_TYPE);
+            preparePost.setHeader("x-request-id", requestId);
+            StringEntity requestEntity = new StringEntity(payload, ContentType.APPLICATION_JSON);
+            preparePost.setEntity(requestEntity);
+
+            URL url = new URL(endpoint);
+            int port = url.getPort();
+            String protocol = url.getProtocol();
+            HttpClient httpClient = APIUtil.getHttpClient(port, protocol);
+
+            CloseableHttpResponse response = executeHTTPRequest(preparePost, httpClient);
+            int statusCode = response.getStatusLine().getStatusCode();
+            String responseStr = EntityUtils.toString(response.getEntity());
+            if (statusCode == HttpStatus.SC_CREATED) {
+                return responseStr;
+            } else if (statusCode == HttpStatus.SC_UNAUTHORIZED) {
+                throw new APIManagementException("Unexpected response detected from the AI service." + responseStr,
+                        ExceptionCodes.AI_SERVICE_INVALID_ACCESS_TOKEN);
+            } else {
+                throw new APIManagementException("Unexpected response detected from the AI service." + responseStr,
+                        ExceptionCodes.AI_SERVICE_INVALID_RESPONSE);
+            }
+        } catch (MalformedURLException e) {
+            throw new APIManagementException("Invalid/malformed URL encountered. URL: " + endpoint, e);
+        } catch (APIManagementException | IOException e) {
+            throw new APIManagementException("Error encountered while connecting to service", e);
+        }
+    }
 }