chore/ 删除无用文件

This commit is contained in:
Kevin
2026-03-19 14:36:14 +08:00
parent 2f60858c9c
commit c6e07ce5ca
135 changed files with 2111 additions and 4510 deletions

View File

@@ -7,7 +7,14 @@ import tempfile
logger = get_logger(__name__)
_DEFAULT_CACHE_DIR = os.path.normpath(
os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "..", "..", "models", "whisper")
os.path.join(
os.path.dirname(os.path.abspath(__file__)),
"..",
"..",
"..",
"models",
"whisper",
)
)
@@ -36,6 +43,7 @@ class WhisperASRProvider:
if device == "auto":
try:
import torch # type: ignore[import-untyped]
device = "cuda" if torch.cuda.is_available() else "cpu"
except ImportError:
device = "cpu"

View File

@@ -48,7 +48,11 @@ class LiblibImageGenerator:
try:
job = {"job_id": task_id}
result = self._provider.poll_until_complete(job, self._poll_interval, 1)
status = TaskStatus.COMPLETED if result.get("status") == "completed" else TaskStatus.PROCESSING
status = (
TaskStatus.COMPLETED
if result.get("status") == "completed"
else TaskStatus.PROCESSING
)
return ImageResult(
status=status,
task_id=task_id,

View File

@@ -2,6 +2,7 @@
Liblib 图生 SDK 封装,位于 adapters 层;实现细节不暴露给 feature。
Feature 通过 port ImageGenerator 使用,本模块仅被 app.adapters.image_gen.liblib 使用。
"""
import base64
import hmac
import logging
@@ -61,8 +62,12 @@ class LiblibImageProvider:
self.http_client = http_client or httpx.Client(timeout=120)
self.access_key = access_key or (settings.liblib_access_key or "")
self.secret_key = secret_key or (settings.liblib_secret_key or "")
self.base_url = (base_url or settings.liblib_base_url or "https://openapi.liblibai.cloud").rstrip("/")
self.template_uuid = template_uuid or (settings.liblib_template_uuid or DEFAULT_LIBLIB_TEMPLATE_UUID)
self.base_url = (
base_url or settings.liblib_base_url or "https://openapi.liblibai.cloud"
).rstrip("/")
self.template_uuid = template_uuid or (
settings.liblib_template_uuid or DEFAULT_LIBLIB_TEMPLATE_UUID
)
self.allowed_download_hosts = _build_allowed_download_hosts(
self.base_url,
allowed_download_hosts=allowed_download_hosts,
@@ -133,7 +138,9 @@ class LiblibImageProvider:
response.raise_for_status()
data = response.json()
if data.get("code") != 0:
raise RuntimeError(f"Liblib status query failed: {data.get('msg', data)}")
raise RuntimeError(
f"Liblib status query failed: {data.get('msg', data)}"
)
result = data.get("data", {})
status = result.get("generateStatus")
if status == 5:
@@ -144,17 +151,28 @@ class LiblibImageProvider:
"image_url": images[0]["imageUrl"],
"job_id": job["job_id"],
}
raise RuntimeError(f"Liblib returned success but no images for {job['job_id']}")
raise RuntimeError(
f"Liblib returned success but no images for {job['job_id']}"
)
if status == 6:
raise RuntimeError(f"Liblib generation failed: {result.get('generateMsg', 'unknown')}")
raise RuntimeError(
f"Liblib generation failed: {result.get('generateMsg', 'unknown')}"
)
if status == 7:
raise TimeoutError(f"Liblib returned undocumented status 7 for {job['job_id']}")
raise TimeoutError(
f"Liblib returned undocumented status 7 for {job['job_id']}"
)
logger.debug(
"Liblib poll attempt %d/%d, status=%s, job=%s",
attempt + 1, max_attempts, status, job["job_id"],
attempt + 1,
max_attempts,
status,
job["job_id"],
)
time.sleep(poll_interval_seconds)
raise TimeoutError(f"Liblib image generation timed out after {max_attempts} attempts for {job['job_id']}")
raise TimeoutError(
f"Liblib image generation timed out after {max_attempts} attempts for {job['job_id']}"
)
def download_image(self, job: dict) -> bytes:
image_url = job["image_url"]
@@ -183,9 +201,13 @@ class _LiblibAuthRedactionFilter(logging.Filter):
record.msg = _redact_sensitive_query_values(record.msg)
if record.args:
if isinstance(record.args, dict):
record.args = {k: _redact_sensitive_query_values(v) for k, v in record.args.items()}
record.args = {
k: _redact_sensitive_query_values(v) for k, v in record.args.items()
}
else:
record.args = tuple(_redact_sensitive_query_values(v) for v in record.args)
record.args = tuple(
_redact_sensitive_query_values(v) for v in record.args
)
return True
@@ -196,7 +218,13 @@ def _redact_sensitive_query_values(value):
def _install_http_log_redaction() -> None:
for logger_name in ("httpx", "httpcore", "httpcore.connection", "httpcore.http11", "httpcore.proxy"):
for logger_name in (
"httpx",
"httpcore",
"httpcore.connection",
"httpcore.http11",
"httpcore.proxy",
):
target_logger = get_logger(logger_name)
if getattr(target_logger, "_liblib_auth_redaction_installed", False):
continue
@@ -219,7 +247,10 @@ def _build_allowed_download_hosts(
default_hosts: set[str] = set()
if base_hostname:
default_hosts.add(base_hostname)
if base_hostname.endswith(".liblibai.cloud") or base_hostname == "liblibai.cloud":
if (
base_hostname.endswith(".liblibai.cloud")
or base_hostname == "liblibai.cloud"
):
default_hosts.add("liblibai.cloud")
default_hosts.add("liblib.cloud")
return tuple(sorted(default_hosts.union(configured_hosts)))
@@ -230,7 +261,9 @@ def _validate_download_url(image_url: str, allowed_hosts: tuple[str, ...]) -> No
hostname = (parsed.hostname or "").lower()
if parsed.scheme != "https" or not hostname:
raise ValueError(f"Unsupported image download URL: {image_url}")
if not any(_hostname_matches(hostname, allowed_host) for allowed_host in allowed_hosts):
if not any(
_hostname_matches(hostname, allowed_host) for allowed_host in allowed_hosts
):
raise ValueError(f"Image download host is not allowed: {hostname}")

View File

@@ -75,7 +75,13 @@ class DeepSeekLLMProvider:
def _to_langchain_messages(messages: list[dict]) -> list:
from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
mapping = {"system": SystemMessage, "human": HumanMessage, "user": HumanMessage, "ai": AIMessage, "assistant": AIMessage}
mapping = {
"system": SystemMessage,
"human": HumanMessage,
"user": HumanMessage,
"ai": AIMessage,
"assistant": AIMessage,
}
result = []
for msg in messages:
cls = mapping.get(msg.get("role", ""), HumanMessage)

View File

@@ -63,7 +63,11 @@ class TencentSmsSender:
error_code = status.Code if status else "UNKNOWN"
if "TemplateParamSetNotMatchApprovedTemplate" in error_code:
continue
logger.error("SMS send failed: %s - %s", error_code, status.Message if status else "")
logger.error(
"SMS send failed: %s - %s",
error_code,
status.Message if status else "",
)
return False
except TencentCloudSDKException as e:

View File

@@ -19,7 +19,9 @@ class TencentCosStorage:
token: str = "",
):
self._bucket = bucket
self._base_url = (base_url or f"https://{bucket}.cos.{region}.myqcloud.com").rstrip("/")
self._base_url = (
base_url or f"https://{bucket}.cos.{region}.myqcloud.com"
).rstrip("/")
config = CosConfig(
Region=region,
SecretId=secret_id,

View File

@@ -134,9 +134,7 @@ class TencentTTSProvider:
results: list[bytes] = []
for chunk in chunks:
audio = await asyncio.to_thread(
self._synthesize_sync, chunk, voice_type
)
audio = await asyncio.to_thread(self._synthesize_sync, chunk, voice_type)
if not audio:
return b""
results.append(audio)