数据导入分析接口调整
This commit is contained in:
@ -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
|
||||
|
||||
Reference in New Issue
Block a user