Class LLM_Wrapper

LLMwrapper.py:78–1781

    return re.sub(r'<think>.*?</think>', '', text, flags=re.DOTALL)

class LLM_Wrapper:
    @staticmethod
    def _download_image_to_base64(img_url: str, timeout: int = 10) -> str:
        """
        Download an image from URL and convert it to base64.
        
        Args:
            img_url (str): The URL of the image to download
            timeout (int): Request timeout in seconds
            
        Returns:
            str: Base64 encoded image data
            
        Raises:
            Exception: If image download or conversion fails
        """
        try:
            headers = {
                'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
            }
            
            response = requests.get(img_url, timeout=timeout, headers=headers)
            response.raise_for_status()
            
            # Convert to base64
            img_base64 = base64.b64encode(response.content).decode('utf-8')
            return img_base64
            
        except Exception as e:
            raise Exception(f"Failed to download and convert image from {img_url}: {str(e)}")

    @staticmethod
    def _send_api_key_usage(
        api_key: str,
        model_name: str,
        source_name: str,
        prompt_tokens: int,
        completion_tokens: int,
        create_time: datetime,
        finish_time: datetime,
        execution_time: float,
        status: bool,
        request_id: str,
        remark: str = "",
    ) -> bool:
        """Send API key usage information to the API key manager."""
        try:
            usage_data = {
                "request_id": request_id,
                "api_key": api_key,
                "model_name": model_name,
                "source_name": source_name,
                "prompt_tokens": prompt_tokens,
                "completion_tokens": completion_tokens,
                "create_time": create_time.isoformat(),
                "finish_time": finish_time.isoformat(),
                "execution_time": execution_time,
                "status": status,