新手也能懂！手把手教你搭建一个会说双关语的天气预报 AI Agent

<!-- 如果用阿里云DashScope -->
<dependency>
    <groupId>com.alibaba.cloud.ai</groupId>
    <artifactId>spring-ai-alibaba-starter-dashscope</artifactId>
    <version>1.1.0.0-RC2</version>
</dependency>

<!-- 如果用OpenAI -->
<dependency>
    <groupId>org.springframework.ai</groupId>
    <artifactId>spring-ai-starter-model-openai</artifactId>
    <version>1.1.0</version>
</dependency>

String SYSTEM_PROMPT = """
    你是一位资深气象专家,特别喜欢用双关语说话。
    
    你可以使用两个工具:
    • get_weather_for_location: 查询指定城市的天气
    • get_user_location: 获取用户所在位置
    
    当用户问天气时,一定要先确认位置。
    如果用户说"这里"、"我这边"等模糊词,
    就用get_user_location工具查询用户位置。
    """;

public class WeatherForLocationTool implements BiFunction<String, ToolContext, String> {
    @Override
    public String apply(
        @ToolParam(description = "城市名称") String city,
        ToolContext toolContext) {
        // 这里可以接入真实的天气API
        // 现在先返回模拟数据
        return city + "今天阳光明媚,气温25度!";
    }
}

public class UserLocationTool implements BiFunction<String, ToolContext, String> {
    @Override
    public String apply(
        @ToolParam(description = "用户查询") String query,
        ToolContext toolContext) {
        // 从上下文中获取用户ID
        String userId = "";
        if (toolContext != null && toolContext.getContext() != null) {
            RunnableConfig config = (RunnableConfig) 
                toolContext.getContext().get(AGENT_CONFIG_CONTEXT_KEY);
            Optional<Object> userIdObj = config.metadata("user_id");
            if (userIdObj.isPresent()) {
                userId = (String) userIdObj.get();
            }
        }
        
        // 根据用户ID返回位置(实际应用中从数据库查询)
        return "1".equals(userId) ? "深圳" : "北京";
    }
}

ToolCallback weatherTool = FunctionToolCallback
    .builder("getWeatherForLocation", new WeatherForLocationTool())
    .description("查询指定城市的天气情况")
    .inputType(String.class)
    .build();

ToolCallback locationTool = FunctionToolCallback
    .builder("getUserLocation", new UserLocationTool())
    .description("根据用户ID获取用户位置")
    .inputType(String.class)
    .build();

// 以阿里云DashScope为例
DashScopeApi dashScopeApi = DashScopeApi.builder()
    .apiKey(System.getenv("AI_DASHSCOPE_API_KEY"))  // 从环境变量读取
    .build();

ChatModel chatModel = DashScopeChatModel.builder()
    .dashScopeApi(dashScopeApi)
    .defaultOptions(DashScopeChatOptions.builder()
        .withModel(DashScopeChatModel.DEFAULT_MODEL_NAME)
        .withTemperature(0.5)  // 控制回答的创造性,0-1之间
        .withMaxToken(1000)    // 限制回答长度
        .build())
    .build();

public class ResponseFormat {
    // 必须有的双关语回复
    private String punnyResponse;
    
    // 天气详情(如果有的话)
    private String weatherConditions;
    
    // Getters和Setters
    public String getPunnyResponse() {
        return punnyResponse;
    }
    
    public void setPunnyResponse(String punnyResponse) {
        this.punnyResponse = punnyResponse;
    }
    
    public String getWeatherConditions() {
        return weatherConditions;
    }
    
    public void setWeatherConditions(String weatherConditions) {
        this.weatherConditions = weatherConditions;
    }
}

// 创建Agent时加入记忆组件
ReactAgent agent = ReactAgent.builder()
    .name("weather_agent")
    .saver(new MemorySaver())  // 内存存储,简单快速
    .build();

// 用同一个threadId,Agent就能记住之前的对话
String threadId = "user_123_chat";
RunnableConfig config = RunnableConfig.builder()
    .threadId(threadId)
    .build();

// 第一次问
agent.call("深圳今天天气怎么样?", config);

// 第二次问(注意用同一个threadId)
agent.call("明天呢?", config);  
// Agent知道你问的是深圳明天的天气!

// 创建你的智能天气助手
ReactAgent agent = ReactAgent.builder()
    .name("weather_pun_agent")              // 给Agent起个名字
    .model(chatModel)                       // 配置的AI模型
    .systemPrompt(SYSTEM_PROMPT)            // Agent人设
    .tools(locationTool, weatherTool)       // 提供的工具
    .outputType(ResponseFormat.class)       // 输出格式
    .saver(new MemorySaver())               // 记忆组件
    .build();

// 开始对话!
String threadId = "chat_001";
RunnableConfig config = RunnableConfig.builder()
    .threadId(threadId)
    .addMetadata("user_id", "1")  // 传入用户ID
    .build();

// 第一轮对话
AssistantMessage response = agent.call("外面天气怎么样?", config);
System.out.println(response.getText());
// 可能输出: "深圳今天'晴'有独钟!阳光明媚25度,
//          简直是'天气晴郎'的好时'光'啊!"

// 继续对话
response = agent.call("谢谢!", config);
System.out.println(response.getText());
// 可能输出: "不'客气',能为您'预'报天气是我的'荣'幸!"

String customSchema = """
    请按以下JSON格式输出:
    {
        "title": "标题",
        "content": "内容",
        "style": "风格"
    }
    """;

ReactAgent agent = ReactAgent.builder()
    .outputSchema(customSchema)
    .build();

Optional<OverAllState> result = agent.invoke("帮我查天气");

if (result.isPresent()) {
    OverAllState state = result.get();
    // 查看所有消息历史
    List<Message> messages = state.value("messages", new ArrayList<>());
    // 调试神器!
    System.out.println("Agent内部状态: " + state);
}

ModelCallLimitHook hook = ModelCallLimitHook.builder()
    .runLimit(5)  // 最多调5次模型
    .exitBehavior(ModelCallLimitHook.ExitBehavior.ERROR)
    .build();

ReactAgent agent = ReactAgent.builder()
    .hooks(hook)
    .build();

Hook humanApproval = HumanInTheLoopHook.builder()
    .approvalOn("getWeatherTool", 
        ToolConfig.builder()
            .description("是否允许查询天气?")
            .build())
    .build();

ReactAgent agent = ReactAgent.builder()
    .hooks(humanApproval)
    .build();