Reduce log printing for expected exceptions

Author: AlonsoGuevara

graphrag/query/llm/text_utils.py

@@ -50,7 +50,7 @@ def chunk_text(
     yield from (token_encoder.decode(list(chunk)) for chunk in chunk_iterator)
 
 
-def try_parse_json_object(input: str) -> tuple[str, dict]:
+def try_parse_json_object(input: str, verbose: bool = True) -> tuple[str, dict]:
     """JSON cleaning and formatting utilities."""
     # Sometimes, the LLM returns a json string with some extra description, this function will clean it up.
 
@@ -59,7 +59,8 @@ def try_parse_json_object(input: str) -> tuple[str, dict]:
         # Try parse first
         result = json.loads(input)
     except json.JSONDecodeError:
-        log.info("Warning: Error decoding faulty json, attempting repair")
+        if verbose:
+            log.info("Warning: Error decoding faulty json, attempting repair")
 
     if result:
         return input, result
@@ -97,11 +98,13 @@ def try_parse_json_object(input: str) -> tuple[str, dict]:
         try:
             result = json.loads(input)
         except json.JSONDecodeError:
-            log.exception("error loading json, json=%s", input)
+            if verbose:
+                log.exception("error loading json, json=%s", input)
             return input, {}
         else:
             if not isinstance(result, dict):
-                log.exception("not expected dict type. type=%s:", type(result))
+                if verbose:
+                    log.exception("not expected dict type. type=%s:", type(result))
                 return input, {}
             return input, result
     else:

graphrag/query/structured_search/drift_search/action.py

@@ -7,6 +7,8 @@
 import logging
 from typing import Any
 
+from graphrag.query.llm.text_utils import try_parse_json_object
+
 log = logging.getLogger(__name__)
 
 
@@ -72,17 +74,12 @@ async def asearch(self, search_engine: Any, global_query: str, scorer: Any = Non
             drift_query=global_query, query=self.query
         )
 
-        try:
-            response = json.loads(search_result.response)
-        except json.JSONDecodeError:
-            error_message = "Failed to parse search response"
-            log.exception("%s: %s", error_message, search_result.response)
-            # Do not launch exception as it will roll up with other steps
-            # Instead return an empty response and let score -inf handle it
-            response = {}
+        # Do not launch exception as it will roll up with other steps
+        # Instead return an empty response and let score -inf handle it
+        _, response = try_parse_json_object(search_result.response, verbose=False)
 
         self.answer = response.pop("response", None)
-        self.score = response.pop("score", float("-inf"))
+        self.score = float(response.pop("score", "-inf"))
         self.metadata.update({"context_data": search_result.context_data})
 
         if self.answer is None: