langroid 0.16.4__tar.gz → 0.16.5__tar.gz

{langroid-0.16.4 → langroid-0.16.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langroid
-Version: 0.16.4
+Version: 0.16.5
 Summary: Harness LLMs with Multi-Agent Programming
 License: MIT
 Author: Prasad Chalasani
@@ -55,7 +55,7 @@ Requires-Dist: grpcio (>=1.62.1,<2.0.0)
 Requires-Dist: halo (>=0.0.31,<0.0.32)
 Requires-Dist: huggingface-hub (>=0.21.2,<0.22.0) ; extra == "hf-transformers" or extra == "all" or extra == "transformers"
 Requires-Dist: jinja2 (>=3.1.2,<4.0.0)
-Requires-Dist: json-repair (>=0.27.0,<0.28.0)
+Requires-Dist: json-repair (>=0.29.0,<0.30.0)
 Requires-Dist: lancedb (>=0.8.2,<0.9.0) ; extra == "vecdbs" or extra == "lancedb"
 Requires-Dist: litellm (>=1.30.1,<2.0.0) ; extra == "all" or extra == "litellm"
 Requires-Dist: lxml (>=4.9.3,<5.0.0)

{langroid-0.16.4 → langroid-0.16.5}/langroid/parsing/parse_json.py

@@ -60,13 +60,11 @@ def parse_imperfect_json(json_string: str) -> Union[Dict[str, Any], List[Any]]:
     except (ValueError, SyntaxError):
         pass
 
-    # If ast.literal_eval fails or returns non-dict/list, try json.loads
-    try:
-        json_string = repair_json(json_string)
-        result = json.loads(json_string)
-        if isinstance(result, (dict, list)):
-            return result
-    except json.JSONDecodeError:
+    # If ast.literal_eval fails or returns non-dict/list, try repair_json
+    json_repaired_obj = repair_json(json_string, return_objects=True)
+    if isinstance(json_repaired_obj, (dict, list)):
+        return json_repaired_obj
+    else:
         try:
             # fallback on yaml
             yaml_result = yaml.safe_load(json_string)
@@ -86,9 +84,14 @@ def try_repair_json_yaml(s: str) -> str | None:
     NOTE - replacing \n with space will result in format loss,
     which may matter in generated code (e.g. python, toml, etc)
     """
-    s_repaired = repair_json(s)
-    if s_repaired != "":
-        return s_repaired  # type: ignore
+    s_repaired_obj = repair_json(s, return_objects=True)
+    if isinstance(s_repaired_obj, list):
+        if len(s_repaired_obj) > 0:
+            s_repaired_obj = s_repaired_obj[0]
+        else:
+            s_repaired_obj = None
+    if s_repaired_obj is not None:
+        return json.dumps(s_repaired_obj)  # type: ignore
     else:
         try:
             yaml_result = yaml.safe_load(s)

{langroid-0.16.4 → langroid-0.16.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langroid"
-version = "0.16.4"
+version = "0.16.5"
 description = "Harness LLMs with Multi-Agent Programming"
 authors = ["Prasad Chalasani <pchalasani@gmail.com>"]
 readme = "README.md"
@@ -87,7 +87,7 @@ nest-asyncio = "^1.6.0"
 async-generator = "^1.10"
 
 python-magic = "^0.4.27"
-json-repair = "^0.27.0"
+json-repair = "^0.29.0"
 cerebras-cloud-sdk = "^1.1.0"