@RestController
@RequestMapping("/helloworld")
public class HelloworldController {

    private final ChatClient chatClient;

    public HelloworldController(ChatClient.Builder chatClientBuilder) {
        this.chatClient = chatClientBuilder
            .defaultSystem("You are a helpful AI assistant.")
            .build();
    }

    @GetMapping("/simple/chat")
    public Map<String, Object> simpleChat() {
        String response = chatClient.prompt()
            .user("Hello! Can you introduce yourself?")
            .call()
            .content();

        return Map.of(
            "response", response,
            "timestamp", Instant.now().toString()
        );
    }
}

@GetMapping("/stream/chat")
public Flux<String> streamChat() {
    return chatClient.prompt()
        .user("Please tell me a story")
        .stream()
        .content();
}

@Component
public class CustomChatAdvisor implements RequestResponseAdvisor {

    @Override
    public AdvisorResponse adviseResponse(AdvisorRequest request,
                                        AdvisorResponse response) {
        // 添加自定义响应处理
        String content = response.response().getResult().getOutput().getContent();

        // 记录对话日志
        logConversation(request, content);

        return response;
    }

    private void logConversation(AdvisorRequest request, String content) {
        System.out.println("User: " + request.userText());
        System.out.println("AI: " + content);
    }
}

@Component
public class InMemoryChatMemory implements ChatMemory {

    private final Map<String, List<Message>> conversationHistory = new ConcurrentHashMap<>();

    @Override
    public void add(String conversationId, Message message) {
        conversationHistory.computeIfAbsent(conversationId, k -> new ArrayList<>())
                         .add(message);
    }

    @Override
    public List<Message> get(String conversationId, int lastN) {
        List<Message> history = conversationHistory.getOrDefault(conversationId, Collections.emptyList());
        int size = history.size();
        return history.subList(Math.max(0, size - lastN), size);
    }
}

@SpringBootTest
@AutoConfigureTestDatabase
class HelloworldControllerTest {

    @Autowired
    private TestRestTemplate restTemplate;

    @MockBean
    private ChatClient chatClient;

    @Test
    void testSimpleChat() {
        // 模拟 ChatClient 响应
        when(chatClient.prompt().user(anyString()).call().content())
            .thenReturn("Hello! I'm an AI assistant.");

        ResponseEntity<String> response = restTemplate.getForEntity(
            "/helloworld/simple/chat", String.class);

        assertEquals(HttpStatus.OK, response.getStatusCode());
        assertTrue(response.getBody().contains("AI assistant"));
    }
}

@ControllerAdvice
public class AIExceptionHandler {

    @ExceptionHandler(ChatModelException.class)
    public ResponseEntity<String> handleChatModelException(ChatModelException e) {
        log.error("AI model error", e);
        return ResponseEntity.status(HttpStatus.SERVICE_UNAVAILABLE)
            .body("AI service temporarily unavailable");
    }
}

@Configuration
public class ChatConfig {

    @Bean
    @Primary
    public ChatClient.Builder chatClientBuilder(ChatModel chatModel,
                                              ChatMemory chatMemory) {
        return ChatClient.builder(chatModel)
            .defaultSystem("You are a helpful assistant.")
            .defaultAdvisors(new MessageChatMemoryAdvisor(chatMemory))
            .defaultOptions(DashScopeChatOptions.builder()
                .withTemperature(0.7)
                .withMaxTokens(2000)
                .build());
    }
}

@Component
public class SecurityAdvisor implements RequestResponseAdvisor {

    private static final List<String> BLOCKED_KEYWORDS =
        Arrays.asList("password", "secret", "token");

    @Override
    public AdvisorRequest adviseRequest(AdvisorRequest request) {
        String userInput = request.userText();

        for (String keyword : BLOCKED_KEYWORDS) {
            if (userInput.toLowerCase().contains(keyword)) {
                throw new SecurityException("Input contains sensitive information");
            }
        }

        return request;
    }
}

@RestController
@RequestMapping("/multimodal")
public class MultimodalController {

    @PostMapping("/analyze")
    public String analyzeImage(@RequestParam("file") MultipartFile file) {
        // 实现图像分析功能
        return chatClient.prompt()
            .user(user -> user.text("Describe this image").media(MimeTypeUtils.IMAGE_JPEG, file.getResource()))
            .call()
            .content();
    }
}

@Component
public class WeatherService {

    @JsonTypeInfo(use = JsonTypeInfo.Id.NONE)
    @JsonClassDescription("Get the current weather in a given location")
    public record WeatherRequest(
        @JsonProperty(required = true, value = "location")
        @JsonPropertyDescription("The city and state, e.g. San Francisco, CA")
        String location) {
    }

    @Function("getCurrentWeather")
    public String getCurrentWeather(WeatherRequest request) {
        // 调用天气 API
        return String.format("The weather in %s is 72°F and sunny.", request.location());
    }
}