[BUG] Use ZhipuAiChatModel to input sensitive words

[yongxingMa]

Describe the bug

Use ZhipuAiChatModel to input some sensitive words，error message not returned properly，Leads to a direct interruption of the programme, I would like to be able to return it normally via Response .

The error message is as follows：
status code: 400; body: {"error":{"code":"1301","message":"系统检测到输入或生成内容可能包含不安全或敏感内容，请您避免输入易产生敏感内容的提示语，感谢您的配合。"}}

Response<AiMessage> resMessage = zhipuAiChatModel.generate(asList(systemMessage,userMessage));

[dosubot]

To address the issue where ZhipuAiChatModel encounters sensitive words and returns a status code 400, leading to a program interruption, you can modify the error handling to manage this scenario gracefully. Specifically, you can adjust the onFailure method within your EventSourceListener to check for a status code of 400. When this error is detected, you can then log the error and create a custom AiMessage indicating the issue, which allows the program to continue running by returning a response rather than crashing.

Here's a snippet to guide you through handling a 400 error:

@Override
public void onFailure(EventSource eventSource, Throwable t, okhttp3.Response response) {
    if (response != null && response.code() == 400) {
        try {
            String responseBody = response.body().string();
            log.error("Encountered sensitive words: {}", responseBody);
            AiMessage errorMessage = AiMessage.from("Encountered sensitive words. Please adjust your query.");
            Response<AiMessage> errorResponse = Response.from(errorMessage, null, FinishReason.ERROR);
            handler.onComplete(errorResponse);
        } catch (IOException e) {
            log.error("Error handling sensitive words response", e);
            handler.onError(new RuntimeException("Failed to handle sensitive words error.", e));
        }
    } else if (t != null) {
        handler.onError(t);
    } else {
        handler.onError(new RuntimeException(String.format("Unexpected error: status code: %s; body: %s", response.code(), response.body())));
    }
}

This approach ensures that when sensitive words are detected, resulting in a 400 error, your application can handle the situation more elegantly by logging the error and notifying the user through a custom message, thus avoiding an abrupt termination of the program.

Details

Sources

• langchain4j-zhipu-ai/src/main/java/dev/langchain4j/model/zhipu/ZhipuAiClient.java
• langchain4j-core/src/main/java/dev/langchain4j/data/message/AiMessage.java
• langchain4j-zhipu-ai/src/main/java/dev/langchain4j/model/zhipu/ZhipuAiChatModel.java
• langchain4j-core/src/main/java/dev/langchain4j/model/output/Response.java
• langchain4j-qianfan/src/main/java/dev/langchain4j/model/qianfan/client/StreamingRequestExecutor.java

About Dosu
This response is meant to be useful and save you time. It is not meant to be a precise solution, but rather a starting point for your own research.

Help me be more useful! Please leave a 👍 if this is helpful and 👎 if it is irrelevant.

To continue the conversation, mention @dosu.

[langchain4j]

cc @1402564807