Refactor

Author: coderphonui

README.md

@@ -1,9 +1,11 @@
 # LLM JSON auto repair
 
 A tiny library to repair JSON string output from LLM. It fixes most of the common issues from the LLM JSON output, eg:
+
 * Remove the ```json``` code block
 * Add missing commas
-* Add missing double quotes
+* Add missing double quotes when possible
+* Replace single quotes with double quotes
 * Escape special characters \t \n
 
 ## Usage
@@ -20,4 +22,6 @@ A tiny library to repair JSON string output from LLM. It fixes most of the commo
                 ```
                 """;
 String fixedJSON = jsonAutoRepairer.repair(originalJSON);
-```
+```
+
+It will automatically fix the JSON string and return the fixed JSON string if possible. In case, the JSON string cannot be fixed, it returns null

src/test/java/com/cdpn/jsonautorepair/JSONAutoRepairerTest.java

@@ -37,6 +37,8 @@ public void repair_should_cleanup_markdown_code_block_when_string_contain_json_c
 
     }
 
+
+
     @Test
     public void repair_should_return_original_string_when_the_string_is_a_valid_JSON() {
         String originalJSON = """
@@ -68,6 +70,7 @@ public void repair_should_not_wrap_string_in_brackets_when_the_string_is_a_JSON_
         assertEquals("[\"sea\",\"fish\"]", jsonAutoRepairer.repair("[\"sea\",\"fish\"]"));
     }
 
+
     @Test
     public void repair_should_add_quotes_around_unquoted_keys() {
         assertEquals("{\"name\":\"Alice\",\"age\":30}",
@@ -76,6 +79,23 @@ public void repair_should_add_quotes_around_unquoted_keys() {
                 jsonAutoRepairer.repair("{name: \"Alice\", age: 30 }"));
     }
 
+    @Test
+    public void repair_should_replace_single_quote_by_double_quote_around_key_and_value() {
+        assertEquals("{\"name\":\"Alice\"}",
+                jsonAutoRepairer.repair("{ 'name': 'Alice' }"));
+    }
+
+    @Test
+    public void repair_should_add_quote_to_key_and_value_when_possible() {
+        assertEquals("{\"name\":\"Alice\"}",
+                jsonAutoRepairer.repair("{ \"name\": Alice }"));
+
+        assertEquals("{\"name\":\"Alice\",\"sex\":\"female\"}",
+                jsonAutoRepairer.repair("{ \"name\": Alice, sex: female }"));
+
+    }
+
+
     @Test
     public void repair_should_escape_internal_quote() {
         String originalJSON = """