<dependencies>
      <!-- Spring AI Alibaba Agent Framework -->
      <dependency>
          <groupId>com.alibaba.cloud.ai</groupId>
          <artifactId>spring-ai-alibaba-agent-framework</artifactId>
          <version>1.1.2.0</version>
      </dependency>
      <!-- DashScope ChatModel 支持（如果使用其他模型，請(qǐng)?zhí)D(zhuǎn) Spring AI 文檔選擇對(duì)應(yīng)的 starter） -->
      <dependency>
          <groupId>com.alibaba.cloud.ai</groupId>
          <artifactId>spring-ai-alibaba-starter-dashscope</artifactId>
          <version>1.1.2.0</version>
      </dependency>
</dependencies>

@Test
void agentTest() throws GraphRunnerException {
    // 1. 初始化 DashScope API
    DashScopeApi dashScopeApi = DashScopeApi.builder()
            .apiKey(System.getenv("AliQwen_API"))
            .build();

    // 2. 創(chuàng)建 ChatModel
    ChatModel chatModel = DashScopeChatModel.builder()
            .dashScopeApi(dashScopeApi)
            .build();

    // 3. 定義天氣工具
    ToolCallback weatherTool = FunctionToolCallback.builder("get_weather", new WeatherTool())
            .description("獲取某個(gè)城市的天氣")
            .inputType(String.class)
            .build();

    // 4. 構(gòu)建React Agent
    ReactAgent agent = ReactAgent.builder()
            .name("weather_agent")
            .model(chatModel)
            .tools(weatherTool)
            .systemPrompt("你是一個(gè)非常有幫助的助手")
            .saver(new MemorySaver())
            .build();

    // 5. 調(diào)用Agent
    AssistantMessage response = agent.call("上海今天天氣怎么樣？");
    System.out.println(response.getText());
}

class WeatherTool implements BiFunction<String, ToolContext, String> {
    @Override
    public String apply(String city, ToolContext toolContext) {
        return city + "今天天氣非常好！";
    }
}

String SYSTEM_PROMPT = """
        你是一位擅長說**天氣冷笑話/諧音梗**的專業(yè)天氣預(yù)報(bào)員。
                    
        你可以使用兩個(gè)工具：
                    
        - **get_weather_for_location**：用于獲取指定地點(diǎn)的天氣
        - **get_user_location**：用于獲取用戶當(dāng)前所在位置
                    
        如果用戶詢問天氣，**必須先確認(rèn)地點(diǎn)**。
        如果從問題中能判斷出他們指的是**自己所在的地方**，
        就使用 **get_user_location** 工具獲取他們的位置。
        """;

ChatModel chatModel = DashScopeChatModel.builder()
        .dashScopeApi(dashScopeApi)
        .defaultOptions(DashScopeChatOptions.builder()
                .model(DashScopeChatModel.DEFAULT_MODEL_NAME)
                .temperature(0.5)      // 平衡創(chuàng)造性和準(zhǔn)確性
                .maxToken(1000)        // 控制響應(yīng)長度
                .build())
        .build();

// 天氣查詢工具
ToolCallback getWeatherTool = FunctionToolCallback
        .builder("getWeatherForLocation", new WeatherForLocationTool())
        .description("獲取一個(gè)給定城市的天氣")
        .inputType(String.class)
        .build();

// 用戶定位工具
ToolCallback getUserLocationTool = FunctionToolCallback
        .builder("getUserLocation", new UserLocationTool())
        .description("根據(jù)User Id獲取用戶位置")
        .inputType(String.class)
        .build();

ReactAgent agent = ReactAgent.builder()
        // ... 其他配置
        .outputType(ResponseFormat.class)  // 指定輸出格式
        .hooks(humanInTheLoopHook)
        .build();

RunnableConfig runnableConfig = RunnableConfig.builder()
        .threadId(Thread.currentThread().getId() + "")
        .build();

// 第一次調(diào)用
AssistantMessage response1 = agent.call("上海今天天氣怎么樣", runnableConfig);

// 第二次調(diào)用（保持上下文）
AssistantMessage response2 = agent.call("明天天氣怎么樣", runnableConfig);