From 73acfc8b1f8c9690b37eb0e8824baa2b4cf2ad12 Mon Sep 17 00:00:00 2001
From: cuiweii <281024627@qq.com>
Date: Fri, 10 Jan 2025 19:16:30 +0800
Subject: [PATCH] Fix the JSON parsing exception returned by content security
 risk monitoring. #49

---
 .../model/zhipu/shared/ErrorResponse.java     | 16 ++++++++++++
 .../model/zhipu/shared/SensitiveFilter.java   | 25 +++++++++++++++++++
 2 files changed, 41 insertions(+)
 create mode 100644 models/langchain4j-community-zhipu-ai/src/main/java/dev/langchain4j/community/model/zhipu/shared/SensitiveFilter.java

diff --git a/models/langchain4j-community-zhipu-ai/src/main/java/dev/langchain4j/community/model/zhipu/shared/ErrorResponse.java b/models/langchain4j-community-zhipu-ai/src/main/java/dev/langchain4j/community/model/zhipu/shared/ErrorResponse.java
index d4c2f20..24ba629 100644
--- a/models/langchain4j-community-zhipu-ai/src/main/java/dev/langchain4j/community/model/zhipu/shared/ErrorResponse.java
+++ b/models/langchain4j-community-zhipu-ai/src/main/java/dev/langchain4j/community/model/zhipu/shared/ErrorResponse.java
@@ -1,8 +1,16 @@
 package dev.langchain4j.community.model.zhipu.shared;
 
+import static com.fasterxml.jackson.annotation.JsonInclude.Include.NON_NULL;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import com.fasterxml.jackson.annotation.JsonInclude;
+import java.util.List;
 import java.util.Map;
 
+@JsonInclude(NON_NULL)
+@JsonIgnoreProperties(ignoreUnknown = true)
 public class ErrorResponse {
+    private List<SensitiveFilter> contentFilter;
     private Map<String, String> error;
 
     public Map<String, String> getError() {
@@ -12,4 +20,12 @@ public Map<String, String> getError() {
     public void setError(Map<String, String> error) {
         this.error = error;
     }
+
+    public List<SensitiveFilter> getContentFilter() {
+        return contentFilter;
+    }
+
+    public void setContentFilter(List<SensitiveFilter> contentFilter) {
+        this.contentFilter = contentFilter;
+    }
 }
diff --git a/models/langchain4j-community-zhipu-ai/src/main/java/dev/langchain4j/community/model/zhipu/shared/SensitiveFilter.java b/models/langchain4j-community-zhipu-ai/src/main/java/dev/langchain4j/community/model/zhipu/shared/SensitiveFilter.java
new file mode 100644
index 0000000..e7b73d3
--- /dev/null
+++ b/models/langchain4j-community-zhipu-ai/src/main/java/dev/langchain4j/community/model/zhipu/shared/SensitiveFilter.java
@@ -0,0 +1,25 @@
+package dev.langchain4j.community.model.zhipu.shared;
+
+/**
+ * Illegal or harmful information: Severity level.
+ */
+public class SensitiveFilter {
+    private String role;
+    private Integer level;
+
+    public String getRole() {
+        return role;
+    }
+
+    public void setRole(final String role) {
+        this.role = role;
+    }
+
+    public Integer getLevel() {
+        return level;
+    }
+
+    public void setLevel(final Integer level) {
+        this.level = level;
+    }
+}