test

Author: cosminacho

testcases/company-research-agent/src/graph.py

@@ -1,7 +1,5 @@
 import json
 import logging
-import sys
-from pathlib import Path
 from typing import Any
 
 from langchain.agents import create_agent
@@ -14,16 +12,17 @@
 
 logger = logging.getLogger(__name__)
 
-# Monkey-patch to dump raw response to file on failure
+# Monkey-patch to surface raw response on failure
 _original_postprocess = _NormalizedChat._postprocess_response
-_debug_file = Path(__file__).resolve().parent.parent / "debug_response.json"
 
 def _debug_postprocess(self: Any, response: dict[str, Any]) -> Any:
     if "choices" not in response:
         dump = json.dumps(response, indent=2, default=str)
-        _debug_file.write_text(dump)
-        sys.stderr.write(f"\n\n===== DEBUG RAW RESPONSE (no 'choices' key) =====\n{dump}\n=====\n\n")
-        sys.stderr.flush()
+        raise ValueError(
+            f"LLM response missing 'choices' key. "
+            f"Got keys: {list(response.keys())}. "
+            f"Full response: {dump}"
+        )
     return _original_postprocess(self, response)
 
 _NormalizedChat._postprocess_response = _debug_postprocess