package com.vincent.rsf.server.ai.service;

import com.fasterxml.jackson.databind.JsonNode;
import com.vincent.rsf.server.ai.dto.GatewayChatRequest;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;

@Service
public class AiTextCompletionService {

    @Resource
    private AiGatewayClient aiGatewayClient;

    /**
     * 用流式网关接口模拟一次“同步补全文本”调用。
     * 适合工具规划、结构化 JSON 选择这类短文本任务，内部仍然复用统一的流式网关能力。
     */
    public String complete(GatewayChatRequest request) throws Exception {
        final StringBuilder output = new StringBuilder();
        aiGatewayClient.stream(request, event -> handleEvent(event, output));
        return output.toString().trim();
    }

    /**
     * 只收集 delta 文本，并把 error / done 事件转换成同步调用语义。
     */
    private boolean handleEvent(JsonNode event, StringBuilder output) {
        String type = event.path("type").asText("");
        if ("delta".equals(type)) {
            output.append(event.path("content").asText(""));
            return true;
        }
        if ("error".equals(type)) {
            throw new IllegalStateException(event.path("message").asText("模型调用失败"));
        }
        return !"done".equals(type);
    }
}