prerna.engine.impl.model.responses.InstructModelEngineResponse Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of semoss Show documentation
SEMOSS
The newest version!
package prerna.engine.impl.model.responses;

import java.util.Map;
import java.util.List;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

public class InstructModelEngineResponse extends AbstractModelEngineResponse>> {
	private static final Logger classLogger = LogManager.getLogger(InstructModelEngineResponse.class);


    private static final long serialVersionUID = 1L;
    
	public static final String MESSAGE_ID = "messageId";
	public static final String ROOM_ID = "roomId";

	private String messageId;
	private String roomId;

    public InstructModelEngineResponse(List> response, Integer numberOfTokensInPrompt, Integer numberOfTokensInResponse) {
        super(response, numberOfTokensInPrompt, numberOfTokensInResponse);
    }
    
	public void setMessageId(String messageId) {
		this.messageId = messageId;
	}
	
	public String getMessageId() {
		return this.messageId;
	}
	
	public void setRoomId(String roomId) {
		this.roomId = roomId;
	}
	
	public String getRoomId() {
		return this.roomId;
	}

    @Override
    public Map toMap() {
        Map responseMap = super.toMap();
    	responseMap.put(MESSAGE_ID, this.messageId);
    	responseMap.put(ROOM_ID, this.roomId);
        return responseMap;
    }

    public static InstructModelEngineResponse fromMap(Map modelResponse) {
        Object responseObject = modelResponse.get(RESPONSE);
        List> responseList = null;

        if (responseObject instanceof List) {
            responseList = (List>) responseObject;
        } else {
            throw new IllegalArgumentException("Invalid response type: " + responseObject.getClass());
        }

        Integer tokensInPrompt = getTokens(modelResponse.get(NUMBER_OF_TOKENS_IN_PROMPT));
        Integer tokensInResponse = getTokens(modelResponse.get(NUMBER_OF_TOKENS_IN_RESPONSE));

        return new InstructModelEngineResponse(responseList, tokensInPrompt, tokensInResponse);
    }

    @SuppressWarnings("unchecked")
    public static InstructModelEngineResponse fromObject(Object responseObject) {
        if (responseObject instanceof Map) {
            Map modelResponse = (Map) responseObject;
            return fromMap(modelResponse);
        } else {
        	classLogger.error("responseObject : {}", responseObject);
            throw new IllegalArgumentException("Expected a Map but got: " + responseObject.getClass());
        }
    }
}