package tech.easyflow.ai.dto;

import com.easyagents.rag.retrieval.RetrievalMode;
import tech.easyflow.ai.rag.KnowledgeRetrievalModes;

import java.math.BigInteger;

public class BotKnowledgeBindingRequest {

    private BigInteger knowledgeId;
    private String retrievalMode;

    public BigInteger getKnowledgeId() {
        return knowledgeId;
    }

    public void setKnowledgeId(BigInteger knowledgeId) {
        this.knowledgeId = knowledgeId;
    }

    public String getRetrievalMode() {
        return retrievalMode;
    }

    public void setRetrievalMode(String retrievalMode) {
        this.retrievalMode = retrievalMode;
    }

    public RetrievalMode resolveRetrievalMode() {
        return KnowledgeRetrievalModes.parse(retrievalMode);
    }
}
