Browse Source

feat(glmocr_vl_adapter): 添加连通性测试以验证与 GLM-OCR API 的连接

zhch158_admin 1 week ago
parent
commit
59f2fdd74c
1 changed files with 10 additions and 0 deletions
  1. 10 0
      ocr_tools/universal_doc_parser/models/adapters/glmocr_vl_adapter.py

+ 10 - 0
ocr_tools/universal_doc_parser/models/adapters/glmocr_vl_adapter.py

@@ -121,6 +121,16 @@ class GLMOCRVLRecognizer(BaseVLRecognizer):
                     'Authorization': f'Bearer {self.api_key}'
                 })
             
+            # 需要向MinerU-VL,进行连通测试
+            try:
+                test_response = self.session.get(self.api_url, timeout=(self.connect_timeout, self.http_timeout), verify=self.verify_ssl)
+                if test_response.status_code == 200:
+                    logger.debug(f"Successfully connected to GLM-OCR API at {self.api_url}")
+                else:
+                    logger.warning(f"Received unexpected status code {test_response.status_code} from GLM-OCR API: {test_response.text}")
+            except requests.exceptions.RequestException as e:
+                logger.error(f"Failed to connect to GLM-OCR API at {self.api_url}: {e}")
+                raise
             logger.success(f"✅ GLM-OCR VL recognizer initialized: {self.api_url}")
             
         except Exception as e: