Add logic for prepare and execute calls of API Chat feature

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

@@ -508,7 +508,7 @@ public final class APIConstants {
     public static final String API_CHAT_ENDPOINT = API_CHAT + "Endpoint";
     public static final String API_CHAT_HEALTH_RESOURCE = "/health"; // "/api-chat/health"
     public static final String API_CHAT_PREPARE_RESOURCE = "/prepare"; // "/api-chat/prepare"
-    public static final String API_CHAT_EXECUTE_RESOURCE = "/execute"; // "/api-chat/execute"
+    public static final String API_CHAT_EXECUTE_RESOURCE = "/chat"; // "/api-chat/chat"
 
     //documentation rxt
 

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

@@ -10392,6 +10392,22 @@ public static boolean isApiChatEnabled() {
         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;
+    }
+
     /**
      * This method is used for AI Service health check purposes. This will be utilized by API-Chat feature and
      * Marketplace-Assistant feature

components/apimgt/org.wso2.carbon.apimgt.rest.api.common/src/main/resources/devportal-api.yaml

@@ -5419,6 +5419,14 @@ components:
             properties too will affect the maximum password length allowed and an
             intersection of all conditions will be considered finally to validate
             the password.
+        isApiChatEnabled:
+          type: boolean
+          description: Specifies whether API Chat feature is enabled.
+          default: true
+        isAIFeatureAuthTokenProvided:
+          type: boolean
+          description: Checks if the auth token is provided for AI service usage.
+          default: false
     ApplicationAttribute:
       title: Application attributes
       type: object
@@ -5912,9 +5920,7 @@ components:
         path:
           type: string
           description: Path of the Http resource
-        queryParameters:
-          type: object
-        pathParameters:
+        parameters:
           type: object
         requestBody:
           type: object
@@ -5944,6 +5950,10 @@ components:
               type: integer
               description: HTTP status code of the response
               example: 201
+            path:
+              type: string
+              description: HTTP path url with encoded parameters
+              example: "/order/123"
             headers:
               type: object
               description: Response headers

components/apimgt/org.wso2.carbon.apimgt.rest.api.store.v1/src/gen/java/org/wso2/carbon/apimgt/rest/api/store/v1/dto/ApiChatExecuteRequestResponseDTO.java

@@ -21,6 +21,7 @@
 public class ApiChatExecuteRequestResponseDTO   {
 
     private Integer code = null;
+    private String path = null;
     private Object headers = null;
     private Object body = null;
 
@@ -42,6 +43,24 @@ public void setCode(Integer code) {
     this.code = code;
   }
 
+  /**
+   * HTTP path url with encoded parameters
+   **/
+  public ApiChatExecuteRequestResponseDTO path(String path) {
+    this.path = path;
+    return this;
+  }
+
+
+  @ApiModelProperty(example = "/order/123", value = "HTTP path url with encoded parameters")
+  @JsonProperty("path")
+  public String getPath() {
+    return path;
+  }
+  public void setPath(String path) {
+    this.path = path;
+  }
+
   /**
    * Response headers
    **/
@@ -91,13 +110,14 @@ public boolean equals(java.lang.Object o) {
     }
     ApiChatExecuteRequestResponseDTO apiChatExecuteRequestResponse = (ApiChatExecuteRequestResponseDTO) o;
     return Objects.equals(code, apiChatExecuteRequestResponse.code) &&
+        Objects.equals(path, apiChatExecuteRequestResponse.path) &&
         Objects.equals(headers, apiChatExecuteRequestResponse.headers) &&
         Objects.equals(body, apiChatExecuteRequestResponse.body);
   }
 
   @Override
   public int hashCode() {
-    return Objects.hash(code, headers, body);
+    return Objects.hash(code, path, headers, body);
   }
 
   @Override
@@ -106,6 +126,7 @@ public String toString() {
     sb.append("class ApiChatExecuteRequestResponseDTO {\n");
 
     sb.append("    code: ").append(toIndentedString(code)).append("\n");
+    sb.append("    path: ").append(toIndentedString(path)).append("\n");
     sb.append("    headers: ").append(toIndentedString(headers)).append("\n");
     sb.append("    body: ").append(toIndentedString(body)).append("\n");
     sb.append("}");

components/apimgt/org.wso2.carbon.apimgt.rest.api.store.v1/src/gen/java/org/wso2/carbon/apimgt/rest/api/store/v1/dto/HttpToolDTO.java

@@ -60,8 +60,7 @@ public static MethodEnum fromValue(String v) {
     }
     private MethodEnum method = null;
     private String path = null;
-    private Object queryParameters = null;
-    private Object pathParameters = null;
+    private Object parameters = null;
     private Object requestBody = null;
 
   /**
@@ -142,38 +141,20 @@ public void setPath(String path) {
 
   /**
    **/
-  public HttpToolDTO queryParameters(Object queryParameters) {
-    this.queryParameters = queryParameters;
+  public HttpToolDTO parameters(Object parameters) {
+    this.parameters = parameters;
     return this;
   }
 
 
   @ApiModelProperty(value = "")
       @Valid
-  @JsonProperty("queryParameters")
-  public Object getQueryParameters() {
-    return queryParameters;
+  @JsonProperty("parameters")
+  public Object getParameters() {
+    return parameters;
   }
-  public void setQueryParameters(Object queryParameters) {
-    this.queryParameters = queryParameters;
-  }
-
-  /**
-   **/
-  public HttpToolDTO pathParameters(Object pathParameters) {
-    this.pathParameters = pathParameters;
-    return this;
-  }
-
-
-  @ApiModelProperty(value = "")
-      @Valid
-  @JsonProperty("pathParameters")
-  public Object getPathParameters() {
-    return pathParameters;
-  }
-  public void setPathParameters(Object pathParameters) {
-    this.pathParameters = pathParameters;
+  public void setParameters(Object parameters) {
+    this.parameters = parameters;
   }
 
   /**
@@ -208,14 +189,13 @@ public boolean equals(java.lang.Object o) {
         Objects.equals(description, httpTool.description) &&
         Objects.equals(method, httpTool.method) &&
         Objects.equals(path, httpTool.path) &&
-        Objects.equals(queryParameters, httpTool.queryParameters) &&
-        Objects.equals(pathParameters, httpTool.pathParameters) &&
+        Objects.equals(parameters, httpTool.parameters) &&
         Objects.equals(requestBody, httpTool.requestBody);
   }
 
   @Override
   public int hashCode() {
-    return Objects.hash(name, description, method, path, queryParameters, pathParameters, requestBody);
+    return Objects.hash(name, description, method, path, parameters, requestBody);
   }
 
   @Override
@@ -227,8 +207,7 @@ public String toString() {
     sb.append("    description: ").append(toIndentedString(description)).append("\n");
     sb.append("    method: ").append(toIndentedString(method)).append("\n");
     sb.append("    path: ").append(toIndentedString(path)).append("\n");
-    sb.append("    queryParameters: ").append(toIndentedString(queryParameters)).append("\n");
-    sb.append("    pathParameters: ").append(toIndentedString(pathParameters)).append("\n");
+    sb.append("    parameters: ").append(toIndentedString(parameters)).append("\n");
     sb.append("    requestBody: ").append(toIndentedString(requestBody)).append("\n");
     sb.append("}");
     return sb.toString();

components/apimgt/org.wso2.carbon.apimgt.rest.api.store.v1/src/gen/java/org/wso2/carbon/apimgt/rest/api/store/v1/dto/SettingsDTO.java

@@ -39,6 +39,8 @@ public class SettingsDTO   {
     private String passwordPolicyPattern = null;
     private Integer passwordPolicyMinLength = null;
     private Integer passwordPolicyMaxLength = null;
+    private Boolean isApiChatEnabled = true;
+    private Boolean isAIFeatureAuthTokenProvided = false;
 
   /**
    **/
@@ -235,7 +237,7 @@ public SettingsDTO isJWTEnabledForLoginTokens(Boolean isJWTEnabledForLoginTokens
     return this;
   }
 
-  
+
   @ApiModelProperty(value = "")
   @JsonProperty("IsJWTEnabledForLoginTokens")
   public Boolean isIsJWTEnabledForLoginTokens() {
@@ -317,6 +319,42 @@ public void setPasswordPolicyMaxLength(Integer passwordPolicyMaxLength) {
     this.passwordPolicyMaxLength = passwordPolicyMaxLength;
   }
 
+  /**
+   * Specifies whether API Chat feature is enabled.
+   **/
+  public SettingsDTO isApiChatEnabled(Boolean isApiChatEnabled) {
+    this.isApiChatEnabled = isApiChatEnabled;
+    return this;
+  }
+
+
+  @ApiModelProperty(value = "Specifies whether API Chat feature is enabled.")
+  @JsonProperty("isApiChatEnabled")
+  public Boolean isIsApiChatEnabled() {
+    return isApiChatEnabled;
+  }
+  public void setIsApiChatEnabled(Boolean isApiChatEnabled) {
+    this.isApiChatEnabled = isApiChatEnabled;
+  }
+
+  /**
+   * Checks if the auth token is provided for AI service usage.
+   **/
+  public SettingsDTO isAIFeatureAuthTokenProvided(Boolean isAIFeatureAuthTokenProvided) {
+    this.isAIFeatureAuthTokenProvided = isAIFeatureAuthTokenProvided;
+    return this;
+  }
+
+
+  @ApiModelProperty(value = "Checks if the auth token is provided for AI service usage.")
+  @JsonProperty("isAIFeatureAuthTokenProvided")
+  public Boolean isIsAIFeatureAuthTokenProvided() {
+    return isAIFeatureAuthTokenProvided;
+  }
+  public void setIsAIFeatureAuthTokenProvided(Boolean isAIFeatureAuthTokenProvided) {
+    this.isAIFeatureAuthTokenProvided = isAIFeatureAuthTokenProvided;
+  }
+
 
   @Override
   public boolean equals(java.lang.Object o) {
@@ -342,12 +380,14 @@ public boolean equals(java.lang.Object o) {
         Objects.equals(userStorePasswordPattern, settings.userStorePasswordPattern) &&
         Objects.equals(passwordPolicyPattern, settings.passwordPolicyPattern) &&
         Objects.equals(passwordPolicyMinLength, settings.passwordPolicyMinLength) &&
-        Objects.equals(passwordPolicyMaxLength, settings.passwordPolicyMaxLength);
+        Objects.equals(passwordPolicyMaxLength, settings.passwordPolicyMaxLength) &&
+        Objects.equals(isApiChatEnabled, settings.isApiChatEnabled) &&
+        Objects.equals(isAIFeatureAuthTokenProvided, settings.isAIFeatureAuthTokenProvided);
   }
 
   @Override
   public int hashCode() {
-    return Objects.hash(grantTypes, scopes, applicationSharingEnabled, mapExistingAuthApps, apiGatewayEndpoint, monetizationEnabled, recommendationEnabled, isUnlimitedTierPaid, identityProvider, isAnonymousModeEnabled, isPasswordChangeEnabled, isJWTEnabledForLoginTokens, userStorePasswordPattern, passwordPolicyPattern, passwordPolicyMinLength, passwordPolicyMaxLength);
+    return Objects.hash(grantTypes, scopes, applicationSharingEnabled, mapExistingAuthApps, apiGatewayEndpoint, monetizationEnabled, recommendationEnabled, isUnlimitedTierPaid, identityProvider, isAnonymousModeEnabled, isPasswordChangeEnabled, userStorePasswordPattern, passwordPolicyPattern, passwordPolicyMinLength, passwordPolicyMaxLength);
   }
 
   @Override
@@ -371,6 +411,8 @@ public String toString() {
     sb.append("    passwordPolicyPattern: ").append(toIndentedString(passwordPolicyPattern)).append("\n");
     sb.append("    passwordPolicyMinLength: ").append(toIndentedString(passwordPolicyMinLength)).append("\n");
     sb.append("    passwordPolicyMaxLength: ").append(toIndentedString(passwordPolicyMaxLength)).append("\n");
+    sb.append("    isApiChatEnabled: ").append(toIndentedString(isApiChatEnabled)).append("\n");
+    sb.append("    isAIFeatureAuthTokenProvided: ").append(toIndentedString(isAIFeatureAuthTokenProvided)).append("\n");
     sb.append("}");
     return sb.toString();
   }