From 95a4233087bb034c69fe767294fe70c55da9496a Mon Sep 17 00:00:00 2001 From: Matthew Farrellee Date: Thu, 5 Sep 2024 07:31:47 -0400 Subject: [PATCH] add warning for debugging purposes --- libs/ai-endpoints/langchain_nvidia_ai_endpoints/_common.py | 1 + 1 file changed, 1 insertion(+) diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_common.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_common.py index 0d0893f7..55aa88ba 100644 --- a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_common.py +++ b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_common.py @@ -383,6 +383,7 @@ def _wait(self, response: Response, session: requests.Session) -> Response: "NVCF-REQID" in response.headers ), "Received 202 response with no request id to follow" request_id = response.headers.get("NVCF-REQID") + warnings.warn(f"Polling for response: {request_id}") # todo: remove payload = { "url": self.polling_url_tmpl.format(request_id=request_id), "headers": self.headers_tmpl["call"],