数据导入分析接口调整

This commit is contained in:
zhaoawd
2025-10-30 22:38:05 +08:00
parent 39911d78ab
commit 455b884551
6 changed files with 141 additions and 29 deletions

View File

@ -1,6 +1,7 @@
from __future__ import annotations
import logging
import os
from typing import Any, Dict, List
import httpx
@ -13,6 +14,23 @@ from app.providers.base import LLMProviderClient
logger = logging.getLogger(__name__)
def _resolve_timeout_seconds() -> float:
raw = os.getenv("DEEPSEEK_TIMEOUT_SECONDS")
if raw is None:
return 60.0
try:
return float(raw)
except ValueError:
logger.warning(
"Invalid value for DEEPSEEK_TIMEOUT_SECONDS=%r, falling back to 60 seconds",
raw,
)
return 60.0
DEEPSEEK_TIMEOUT_SECONDS = _resolve_timeout_seconds()
class DeepSeekProvider(LLMProviderClient):
name = LLMProvider.DEEPSEEK.value
api_key_env = "DEEPSEEK_API_KEY"
@ -40,9 +58,12 @@ class DeepSeekProvider(LLMProviderClient):
"Authorization": f"Bearer {self.api_key}",
"Content-Type": "application/json",
}
timeout = httpx.Timeout(DEEPSEEK_TIMEOUT_SECONDS)
try:
response = await client.post(self.base_url, json=payload, headers=headers)
response = await client.post(
self.base_url, json=payload, headers=headers, timeout=timeout
)
response.raise_for_status()
except httpx.HTTPStatusError as exc:
status_code = exc.response.status_code