test

Author: cosminacho

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

@@ -1,10 +1,27 @@
+import logging
+from typing import Any
+
 from langchain.agents import create_agent
 from langchain_community.tools import DuckDuckGoSearchResults
 from langgraph.graph import END, START, MessagesState, StateGraph
 from pydantic import BaseModel
+from uipath_langchain_client.clients.normalized.chat_models import UiPathChat as _NormalizedChat
 
 from uipath_langchain.chat import UiPathChat
 
+logger = logging.getLogger(__name__)
+
+# Monkey-patch to log raw response on failure
+_original_postprocess = _NormalizedChat._postprocess_response
+
+def _debug_postprocess(self: Any, response: dict[str, Any]) -> Any:
+    if "choices" not in response:
+        logger.error("RAW RESPONSE (no 'choices' key): %s", response)
+        print(f"DEBUG RAW RESPONSE: {response}")
+    return _original_postprocess(self, response)
+
+_NormalizedChat._postprocess_response = _debug_postprocess
+
 # Configuration constants