最佳实践

import com.coze.loop.client.CozeLoopClient;
import com.coze.loop.client.CozeLoopClientBuilder;
import com.coze.loop.entity.ExecuteParam;
import com.coze.loop.entity.ExecuteResult;
import com.coze.loop.spec.tracespec.ModelChoice;
import com.coze.loop.spec.tracespec.ModelInput;
import com.coze.loop.spec.tracespec.ModelMessage;
import com.coze.loop.spec.tracespec.ModelOutput;
import com.coze.loop.trace.CozeLoopSpan;

import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

public class CozeLoopBestPractice {

    public static void main(String[] args) {
        // 在非 Spring Boot 环境中，通过 Builder 手动构建客户端。
        // SDK 会自动从环境变量 COZELOOP_WORKSPACE_ID 和 COZELOOP_API_TOKEN 读取配置。
        try (CozeLoopClient client = new CozeLoopClientBuilder()
                .serviceName("java-best-practice-example")
                .build()) {

            // 1. 创建 Root Span，追踪整个业务流程
            try (CozeLoopSpan rootSpan = client.startSpan("travel_plan_generation", "custom")) {
                rootSpan.setTag("user_id", "user_67890");
                rootSpan.setTag("request_source", "mobile_app");

                // 2. 在业务逻辑中创建 Model Span，用于追踪模型调用
                callLLM(client, "为我推荐一个上海的周末两日游路线");

                // 3. (可选) 执行 Prompt
                executePrompt(client, "your_prompt_key", "你好");

            } finally {
                // 在短生命周期的应用（如脚本）结束前，强制刷新以确保数据上报
                 client.flush();
            }

        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 模拟一次大模型调用，并上报一个 Model Span。
     */
    private static void callLLM(CozeLoopClient client, String userInput) {
        // 使用 try-with-resources 自动管理 Span 的生命周期
        try (CozeLoopSpan modelSpan = client.startSpan("llm_call_doubao", "model")) {
            // 设置模型提供商和名称
            modelSpan.setModelProvider("volcengine");
            modelSpan.setModelName("doubao-pro-32k");

            // 准备模型输入
            ModelInput input = new ModelInput();
            input.setMessages(Collections.singletonList(new ModelMessage("user", userInput)));
            modelSpan.setInput(input);

            // ----- 在此处执行你的实际 LLM 调用逻辑 -----
            // String llmResponse = yourLLMApi.chat(userInput);
            String llmResponse = "好的，这是为您规划的上海两日游路线：第一天...";
            int inputTokens = 25;
            int outputTokens = 150;
            // -----------------------------------------

            // 记录模型输出和 Token 消耗
            ModelOutput output = new ModelOutput();
            output.setChoices(
                    Collections.singletonList(
                            new ModelChoice(new ModelMessage("assistant", llmResponse), "stop", 0L)));
            modelSpan.setOutput(output);
            modelSpan.setInputTokens(inputTokens);
            modelSpan.setOutputTokens(outputTokens);
        }
    }

    /**
     * (可选) 执行在 CozeLoop 平台上定义的 Prompt。
     */
    private static void executePrompt(CozeLoopClient client, String promptKey, String query) {
         if (promptKey == null || promptKey.equals("your_prompt_key")) {
             System.out.println("\n提示：请替换 `your_prompt_key` 为你在平台创建的实际 Prompt Key 来测试执行功能。");
             return;
         }
        try (CozeLoopSpan executeSpan = client.startSpan("execute_prompt_example", "custom")) {
            Map<String, Object> variables = new HashMap<>();
            variables.put("query", query); // "query" 是你在 Prompt 中定义的变量名

            ExecuteParam param = ExecuteParam.builder()
                    .promptKey(promptKey)
                    .variableVals(variables)
                    .build();

            // 同步执行 Prompt
            ExecuteResult result = client.execute(param);

            executeSpan.setTag("prompt_key", promptKey);
            executeSpan.setInput(variables);
            if (result.getMessage() != null) {
                executeSpan.setOutput(result.getMessage().getContent());
            }
            if (result.getUsage() != null) {
                executeSpan.setTag("total_tokens", result.getUsage().getTotalTokens());
            }
        }
    }
}