OpenGradient
diff --git a/‎src/opengradient/client/llm.py‎
Lines changed: 2 additions & 0 deletions b/‎src/opengradient/client/llm.py‎
Lines changed: 2 additions & 0 deletions
diff --git a/‎uv.lock‎
Lines changed: 1 addition & 1 deletion b/‎uv.lock‎
Lines changed: 1 addition & 1 deletion
@@ -474,6 +474,8 @@ async def _chat_stream(self, params: _ChatParams, messages: List[Dict]) -> Async
                 exc,
             )
 
+        # Only reached if the first attempt failed before yielding any chunks.
+        # Re-resolve the TEE endpoint from the registry and retry once.
         await self._refresh_tee()
         headers = self._headers(params.x402_settlement_mode)
         async with self._http_client.stream(
Original file line number	Diff line number	Diff line change
`@@ -474,6 +474,8 @@ async def _chat_stream(self, params: _ChatParams, messages: List[Dict]) -> Async`
`474`	`474`	`exc,`
`475`	`475`	`)`
`476`	`476`
	`477`	`+ # Only reached if the first attempt failed before yielding any chunks.`
	`478`	`+ # Re-resolve the TEE endpoint from the registry and retry once.`
`477`	`479`	`await self._refresh_tee()`
`478`	`480`	`headers = self._headers(params.x402_settlement_mode)`
`479`	`481`	`async with self._http_client.stream(`