package tech.easyflow.agent.service;

import com.mybatisflex.core.service.IService;
import tech.easyflow.agent.entity.AgentKnowledgeBinding;

import java.math.BigInteger;
import java.util.List;

/**
 * Agent 知识库绑定服务。
 */
public interface AgentKnowledgeBindingService extends IService<AgentKnowledgeBinding> {

    /**
     * 替换 Agent 知识库绑定。
     *
     * @param agentId Agent ID
     * @param bindings 新绑定列表
     * @return 保存后的绑定列表
     */
    List<AgentKnowledgeBinding> replaceBindings(BigInteger agentId, List<AgentKnowledgeBinding> bindings);

    /**
     * 查询 Agent 启用知识库绑定。
     *
     * @param agentId Agent ID
     * @return 启用绑定列表
     */
    List<AgentKnowledgeBinding> listEnabled(BigInteger agentId);
}
