package com.ard.agent.service;

import com.ard.agent.tool.KnowledgeTools;
import com.ard.agent.tool.MonitorTools;
import com.ard.common.security.utils.SecurityUtils;
import jakarta.annotation.Resource;
import org.springframework.ai.chat.client.ChatClient;
import org.springframework.ai.chat.client.advisor.MessageChatMemoryAdvisor;
import org.springframework.ai.chat.model.ChatResponse;
import org.springframework.stereotype.Service;
import reactor.core.publisher.Flux;

@Service
public class AIChatService {

    @Resource
    private ChatClient chatClient;
    @Resource
    private MonitorTools monitorTools;  // 注入工具类
    @Resource
    private KnowledgeTools knowledgeTools;
    private static final String SYSTEM_PROMPT = """
        你是摄像头管理助手。请严格遵守以下规则：
        
        1. **工具优先原则**：当用户问题涉及摄像机的实时状态、修改操作、查询具体设备信息时，
           必须调用对应的工具（getCameraDetail/searchCameras/updateCameraInfo），
           绝对不要使用自己的记忆或上下文中的静态文本来回答。
        
        2. **知识库使用场景**：只有当用户询问操作指南、故障排查手册、产品说明等静态知识时，
           才使用知识库中的内容。
        
        3. **冲突处理**：如果知识库中提到了某类操作，但用户要求执行具体操作，
           必须以工具调用的结果为准。
        
        ========== 输出格式 ==========
        使用 Markdown 格式输出：
        - ## 标题
        - - 列表项
        - **加粗** 强调
        - 表格用于数据展示
        """;
    /**
     * 流式对话
     */
    public Flux<String> streamChat(String userMessage) {
        Long userId = SecurityUtils.getUserId();
        return chatClient.prompt()
                .system(SYSTEM_PROMPT)
                .user(userMessage)
                .advisors(advisor -> advisor.param("chat_memory_conversation_id", userId))
                .tools(monitorTools, knowledgeTools)
                .stream()
                .content();
    }
}