Precommit fixes (#355)
* precommit fixes * Fix error handling for large messages in get_velociraptor_artifact_recommendation_response
taylor_socfortress committed
Dec 5, 2024 at 15:54 UTC
9bfd1048f92a31b16accc7bce8101ad94f50209f
2 files changed
+6
-3
backend/app/threat_intel/routes/socfortress.py
+5
-2
@@ -1,19 +1,21 @@
1
+from datetime import datetime
2
+
3
from fastapi import APIRouter
4
from fastapi import Depends
5
from fastapi import HTTPException
6
from fastapi import Security
7
from loguru import logger
8
from sqlalchemy.ext.asyncio import AsyncSession
7
-from datetime import datetime
9
+
10
from app.agents.services.status import get_agent_os_by_id
11
from app.auth.utils import AuthHandler
12
from app.connectors.velociraptor.services.artifacts import get_artifacts
13
from app.db.db_session import get_db
14
from app.incidents.schema.db_operations import CommentCreate
13
-from app.incidents.services.db_operations import create_comment
15
from app.incidents.schema.incident_alert import CreateAlertRequest
16
from app.incidents.schema.incident_alert import CreateAlertRequestRoute
17
from app.incidents.schema.incident_alert import GenericAlertModel
18
+from app.incidents.services.db_operations import create_comment
19
from app.incidents.services.incident_alert import get_single_alert_details
20
from app.middleware.license import get_license
21
from app.middleware.license import is_feature_enabled
@@ -200,6 +202,7 @@ async def process_name_intel_socfortress(
202
)
203
return socfortress_lookup
204
205
+
206
async def current_time():
207
return datetime.now().strftime("%Y-%m-%d %H:%M:%S")
208
backend/app/threat_intel/services/socfortress.py
+1
-1
@@ -444,7 +444,7 @@ async def get_velociraptor_artifact_recommendation_response(
444
status_code=429,
445
detail="Message is too large. Please try again with a smaller message.",
446
)
447
- elif "too large" in response_data.get("message", ""):
447
+ elif "too large" in response_data.get("detail", ""):
448
raise HTTPException(
449
status_code=429,
450
detail="Message is too large. Please try again with a smaller message.",