Spaces:
Paused
Paused
File size: 1,601 Bytes
7861a83 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
import logging
from urllib.parse import urlparse
def is_generate_content_endpoint(url):
"""
Check if the URL is a GenerateContent endpoint
"""
return 'GenerateContent' in url
def parse_proxy_url(proxy_url):
"""
Parse a proxy URL into its components
Returns:
tuple: (scheme, host, port, username, password)
"""
if not proxy_url:
return None, None, None, None, None
parsed = urlparse(proxy_url)
scheme = parsed.scheme
host = parsed.hostname
port = parsed.port
username = parsed.username
password = parsed.password
return scheme, host, port, username, password
def setup_logger(name, log_file=None, level=logging.INFO):
"""
Set up a logger with the specified name and configuration
Args:
name (str): Logger name
log_file (str, optional): Path to log file
level (int, optional): Logging level
Returns:
logging.Logger: Configured logger
"""
logger = logging.getLogger(name)
logger.setLevel(level)
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
# Add console handler
console_handler = logging.StreamHandler()
console_handler.setFormatter(formatter)
logger.addHandler(console_handler)
# Add file handler if specified
if log_file:
file_handler = logging.FileHandler(log_file)
file_handler.setFormatter(formatter)
logger.addHandler(file_handler)
return logger
|