Tecnhotron commited on
Commit
c444311
·
1 Parent(s): 48cea42
__pycache__/test_api.cpython-313-pytest-8.3.5.pyc ADDED
Binary file (16.2 kB). View file
 
api.py CHANGED
@@ -684,6 +684,16 @@ async def list_models():
684
  raise HTTPException(status_code=503, detail="Service unavailable: The backend driver is not ready.")
685
  except Exception as e:
686
  logger.error(f"An unexpected error occurred while fetching models: {e}", exc_info=True)
 
 
 
 
 
 
 
 
 
 
687
  raise HTTPException(status_code=500, detail=f"An unexpected error occurred while fetching models: {str(e)}")
688
 
689
  @app.post("/chat/completions", response_model=ChatCompletionResponse)
 
684
  raise HTTPException(status_code=503, detail="Service unavailable: The backend driver is not ready.")
685
  except Exception as e:
686
  logger.error(f"An unexpected error occurred while fetching models: {e}", exc_info=True)
687
+ try:
688
+ driver = driver_manager.get_driver()
689
+ logger.info("Attempting captcha solve after models fetch error...")
690
+ driver.uc_gui_click_captcha()
691
+ driver_manager._perform_sync_captcha_checks(driver)
692
+ png = driver.get_screenshot_as_png()
693
+ b64 = base64.b64encode(png).decode('utf-8')
694
+ logger.error(f"Screenshot base64 after captcha solve: {b64}")
695
+ except Exception as captcha_err:
696
+ logger.error(f"Failed captcha solve/screenshot in models endpoint: {captcha_err}", exc_info=True)
697
  raise HTTPException(status_code=500, detail=f"An unexpected error occurred while fetching models: {str(e)}")
698
 
699
  @app.post("/chat/completions", response_model=ChatCompletionResponse)