55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
|
import asyncio
|
|||
|
|
import httpx
|
|||
|
|
from typing import Optional
|
|||
|
|
|
|||
|
|
from app.config import settings
|
|||
|
|
from app.utils.logger import logger
|
|||
|
|
|
|||
|
|
|
|||
|
|
class AIAnalyzer:
|
|||
|
|
def __init__(self):
|
|||
|
|
self.endpoint = settings.ai_model.endpoint
|
|||
|
|
self.api_key = settings.ai_model.api_key
|
|||
|
|
|
|||
|
|
async def analyze_alert(
|
|||
|
|
self,
|
|||
|
|
alert_id: int,
|
|||
|
|
snapshot_url: str,
|
|||
|
|
alert_info: dict,
|
|||
|
|
) -> Optional[dict]:
|
|||
|
|
if not self.endpoint:
|
|||
|
|
logger.debug("AI模型端点未配置，跳过分析")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
async with httpx.AsyncClient(timeout=30.0) as client:
|
|||
|
|
response = await client.post(
|
|||
|
|
f"{self.endpoint}/analyze",
|
|||
|
|
json={
|
|||
|
|
"alert_id": alert_id,
|
|||
|
|
"snapshot_url": snapshot_url,
|
|||
|
|
"alert_info": alert_info,
|
|||
|
|
},
|
|||
|
|
headers={"Authorization": f"Bearer {self.api_key}"} if self.api_key else {},
|
|||
|
|
)
|
|||
|
|
response.raise_for_status()
|
|||
|
|
result = response.json()
|
|||
|
|
logger.info(f"AI分析完成: alert_id={alert_id}")
|
|||
|
|
return result
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"AI分析失败: alert_id={alert_id}, error={e}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
|
|||
|
|
ai_analyzer = AIAnalyzer()
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def trigger_async_analysis(alert_id: int, snapshot_url: str, alert_info: dict):
|
|||
|
|
asyncio.create_task(
|
|||
|
|
ai_analyzer.analyze_alert(alert_id, snapshot_url, alert_info)
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_ai_analyzer() -> AIAnalyzer:
|
|||
|
|
return ai_analyzer
|