fix(jsonl-parser): safe web_search result_count + slug Optional typing (#29)

Author: clean6378-max-it

tests/test_jsonl_parser.py

@@ -171,6 +171,13 @@ def test_web_search_result(self):
         assert r["query"] == "q"
         assert r["result_count"] == 1
 
+    def test_web_search_results_none_or_non_sized_yields_zero_count(self):
+        r = _parse_tool_result({"query": "q", "results": None})
+        assert r["result_type"] == "web_search"
+        assert r["result_count"] == 0
+        r2 = _parse_tool_result({"query": "q", "results": "not-a-list"})
+        assert r2["result_count"] == 0
+
     def test_web_fetch_result(self):
         r = _parse_tool_result({"url": "https://x", "code": 200, "durationMs": 40})
         assert r["result_type"] == "web_fetch"

utils/jsonl_parser.py

@@ -426,7 +426,11 @@ def _tool_result_build_web_search(tr: dict, base: dict) -> dict:
     result = dict(base)
     result["result_type"] = "web_search"
     result["query"] = tr.get("query", "")
-    result["result_count"] = len(tr.get("results", []))
+    raw_results = tr.get("results")
+    if isinstance(raw_results, (list, tuple, set, dict)):
+        result["result_count"] = len(raw_results)
+    else:
+        result["result_count"] = 0
     result["duration_seconds"] = tr.get("durationSeconds")
     return result
 
@@ -553,7 +557,7 @@ def _tool_result_build_plan(tr: dict, base: dict) -> dict:
 )
 
 
-def _parse_tool_result(tool_result, slug: str = None) -> dict | None:
+def _parse_tool_result(tool_result, slug: str | None = None) -> dict | None:
     """Figure out what kind of tool result this is (bash, file edit, glob, etc.)
     by looking at which keys are present, since the JSONL doesn't always tag them.