Retrieve response content.
(
self,
check_client_disconnected: Callable,
prompt_length: int = 0,
timeout: Optional[float] = None,
)
| 257 | await self.page.wait_for_load_state("domcontentloaded", timeout=30000) |
| 258 | |
| 259 | async def get_response( |
| 260 | self, |
| 261 | check_client_disconnected: Callable, |
| 262 | prompt_length: int = 0, |
| 263 | timeout: Optional[float] = None, |
| 264 | ) -> str: |
| 265 | """Retrieve response content.""" |
| 266 | submit_btn = self.page.locator(SUBMIT_BUTTON_SELECTOR) |
| 267 | edit_btn = self.page.locator(EDIT_MESSAGE_BUTTON_SELECTOR) |
| 268 | input_field = self.page.locator(PROMPT_TEXTAREA_SELECTOR) |
| 269 | await _wait_for_response_completion( |
| 270 | self.page, |
| 271 | input_field, |
| 272 | submit_btn, |
| 273 | edit_btn, |
| 274 | self.req_id, |
| 275 | check_client_disconnected, |
| 276 | None, |
| 277 | prompt_length=prompt_length, |
| 278 | timeout=timeout, |
| 279 | ) |
| 280 | content = await _get_final_response_content( |
| 281 | self.page, self.req_id, check_client_disconnected |
| 282 | ) |
| 283 | if not content or not content.strip(): |
| 284 | verified = await self.verify_response_integrity(check_client_disconnected) |
| 285 | return verified.get("content", "") |
| 286 | return content |
| 287 | |
| 288 | async def verify_response_integrity( |
| 289 | self, check_client_disconnected: Callable, trigger_reason: str = "" |