Implemented token authentication for chat and generate api payload resources

Author: KamalikaSene

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

@@ -1,7 +1,7 @@
 /*
- *  Copyright (c) 2021, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
+ *  Copyright (c) 2025, WSO2 LLC. (http://www.wso2.org) All Rights Reserved.
  *
- *  WSO2 Inc. licenses this file to you under the Apache License,
+ *  WSO2 LLC. licenses this file to you under the Apache License,
  *  Version 2.0 (the "License"); you may not use this file except
  *  in compliance with the License.
  *  You may obtain a copy of the License at
@@ -557,7 +557,6 @@ public static class AI {
         public static final String DESIGN_ASSISTANT_TOKEN_ENDPOINT = "TokenEndpoint";
         public static final String DESIGN_ASSISTANT_CHAT_RESOURCE = "ChatResource";
         public static final String DESIGN_ASSISTANT_GEN_API_PAYLOAD_RESOURCE = "GenApiPayloadResource";
-        public static final String DESIGN_ASSISTANT_CONFIGURATION = "AiConfiguration";
 
         private AI() {
 

components/apimgt/org.wso2.carbon.apimgt.rest.api.publisher.v1/src/main/java/org/wso2/carbon/apimgt/rest/api/publisher/v1/impl/DesignAssistantApiServiceImpl.java

@@ -17,6 +17,8 @@
  */
 package org.wso2.carbon.apimgt.rest.api.publisher.v1.impl;
 
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.JsonMappingException;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.google.gson.Gson;
 import org.apache.commons.lang3.StringUtils;
@@ -52,7 +54,6 @@
 public class DesignAssistantApiServiceImpl implements DesignAssistantApiService {
 
     private static final Log log = LogFactory.getLog(DesignAssistantApiServiceImpl.class);
-
     private static DesignAssistantConfigurationDTO configDto;
 
     @Override
@@ -84,12 +85,13 @@ public Response designAssistantApiPayloadGen(DesignAssistantGenAPIPayloadDTO des
                 if (configDto.isKeyProvided()) {
                     response = APIUtil.invokeAIService(configDto.getEndpoint(), configDto.getTokenEndpoint(),
                             configDto.getKey(), configDto.getGenApiPayloadResource(), payload.toString(), null);
+ 
                 } else {
                     response = APIUtil.invokeAIService(configDto.getEndpoint(), null,
                             configDto.getAccessToken(), configDto.getGenApiPayloadResource(), payload.toString(), null);
+
                 }
 
-                ObjectMapper objectMapper = new ObjectMapper();
                 DesignAssistantAPIPayloadResponseDTO responseDTO = new DesignAssistantAPIPayloadResponseDTO();
                 responseDTO.setGeneratedPayload(response);
 
@@ -122,35 +124,34 @@ public Response designAssistantChat(DesignAssistantChatQueryDTO designAssistantC
         }
         try {
             if (configDto.isKeyProvided() || configDto.isAuthTokenProvided()) {
-
-                boolean isChatQueryEmpty = StringUtils.isEmpty(designAssistantChatQueryDTO.getSessionId());
+                String sessionId = designAssistantChatQueryDTO.getSessionId();
+                boolean isChatQueryEmpty = StringUtils.isEmpty(sessionId);
                 if (isChatQueryEmpty) {
                     String errorMessage = "Payload is badly formatted. Expected to have 'sessionId'";
                     RestApiUtil.handleBadRequest(errorMessage, log);
                     return null;
                 }
 
                 JSONObject payload = new JSONObject();
-                String text = new Gson().toJson(designAssistantChatQueryDTO.getText());
-                String sessionId = new Gson().toJson(designAssistantChatQueryDTO.getSessionId());
 
-                payload.put(APIConstants.QUERY, designAssistantChatQueryDTO.getSessionId());
-                payload.put(APIConstants.SESSIONID, designAssistantChatQueryDTO.getSessionId());
+                payload.put(APIConstants.TEXT, designAssistantChatQueryDTO.getText());
+                payload.put(APIConstants.SESSIONID, sessionId);
 
                 String response;
                 if (configDto.isKeyProvided()) {
                     response = APIUtil.invokeAIService(configDto.getEndpoint(), configDto.getTokenEndpoint(),
                             configDto.getKey(), configDto.getChatResource(), payload.toString(), null);
+    
                 } else {
                     response = APIUtil.invokeAIService(configDto.getEndpoint(), null,
                             configDto.getAccessToken(), configDto.getChatResource(), payload.toString(), null);
+
                 }
-              
+
                 ObjectMapper objectMapper = new ObjectMapper();
                 DesignAssistantChatResponseDTO responseDTO = objectMapper.readValue(response, DesignAssistantChatResponseDTO.class);
 
                 return Response.ok(responseDTO).build();
-                
             }
         } catch (APIManagementException e) {
             if (RestApiUtil.isDueToAIServiceNotAccessible(e)) {
@@ -162,7 +163,9 @@ public Response designAssistantChat(DesignAssistantChatQueryDTO designAssistantC
                         "Assistant service";
                 RestApiUtil.handleInternalServerError(errorMessage, e, log);
             }
+        } catch (JsonProcessingException e) {
+            throw new RuntimeException(e);
         }
         return null;
     }
-}
+}