style: apply ruff fixes
This commit is contained in:
@@ -14,12 +14,12 @@ from fastapi.responses import Response
|
|||||||
from fastapi.staticfiles import StaticFiles
|
from fastapi.staticfiles import StaticFiles
|
||||||
from metrics import observe_request, prometheus_metrics
|
from metrics import observe_request, prometheus_metrics
|
||||||
from middleware import CorrelationIdMiddleware
|
from middleware import CorrelationIdMiddleware
|
||||||
|
from routes.ask import router as ask_router
|
||||||
from routes.config import router as config_router
|
from routes.config import router as config_router
|
||||||
from routes.events import router as events_router
|
from routes.events import router as events_router
|
||||||
from routes.fetch import router as fetch_router
|
from routes.fetch import router as fetch_router
|
||||||
from routes.fetch import run_fetch
|
from routes.fetch import run_fetch
|
||||||
from routes.health import router as health_router
|
from routes.health import router as health_router
|
||||||
from routes.ask import router as ask_router
|
|
||||||
from routes.rules import router as rules_router
|
from routes.rules import router as rules_router
|
||||||
from routes.webhooks import router as webhooks_router
|
from routes.webhooks import router as webhooks_router
|
||||||
|
|
||||||
|
|||||||
@@ -46,7 +46,9 @@ def _extract_time_range(question: str) -> tuple[str | None, str | None]:
|
|||||||
start = now.replace(hour=0, minute=0, second=0, microsecond=0)
|
start = now.replace(hour=0, minute=0, second=0, microsecond=0)
|
||||||
else:
|
else:
|
||||||
num = int(m.group(1))
|
num = int(m.group(1))
|
||||||
delta = {"days": timedelta(days=num), "hours": timedelta(hours=num), "minutes": timedelta(minutes=num)}[unit]
|
delta = {"days": timedelta(days=num), "hours": timedelta(hours=num), "minutes": timedelta(minutes=num)}[
|
||||||
|
unit
|
||||||
|
]
|
||||||
start = now - delta
|
start = now - delta
|
||||||
return start.isoformat().replace("+00:00", "Z"), now.isoformat().replace("+00:00", "Z")
|
return start.isoformat().replace("+00:00", "Z"), now.isoformat().replace("+00:00", "Z")
|
||||||
|
|
||||||
@@ -238,11 +240,7 @@ async def ask_question(body: AskRequest, user: dict = Depends(require_auth)):
|
|||||||
query = _build_event_query(entity, start, end)
|
query = _build_event_query(entity, start, end)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
cursor = (
|
cursor = events_collection.find(query).sort([("timestamp", -1)]).limit(LLM_MAX_EVENTS)
|
||||||
events_collection.find(query)
|
|
||||||
.sort([("timestamp", -1)])
|
|
||||||
.limit(LLM_MAX_EVENTS)
|
|
||||||
)
|
|
||||||
events = list(cursor)
|
events = list(cursor)
|
||||||
except Exception as exc:
|
except Exception as exc:
|
||||||
logger.error("Failed to query events for ask", error=str(exc))
|
logger.error("Failed to query events for ask", error=str(exc))
|
||||||
|
|||||||
@@ -1,11 +1,7 @@
|
|||||||
from datetime import UTC, datetime, timedelta
|
from datetime import UTC, datetime, timedelta
|
||||||
|
|
||||||
import pytest
|
|
||||||
from fastapi.testclient import TestClient
|
|
||||||
|
|
||||||
from routes.ask import _build_event_query, _extract_entity, _extract_time_range
|
from routes.ask import _build_event_query, _extract_entity, _extract_time_range
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
# ---------------------------------------------------------------------------
|
||||||
# Unit tests: time-range extraction
|
# Unit tests: time-range extraction
|
||||||
# ---------------------------------------------------------------------------
|
# ---------------------------------------------------------------------------
|
||||||
@@ -44,7 +40,7 @@ class TestExtractTimeRange:
|
|||||||
def test_today(self):
|
def test_today(self):
|
||||||
start, end = _extract_time_range("What happened today?")
|
start, end = _extract_time_range("What happened today?")
|
||||||
start_dt = datetime.fromisoformat(start.replace("Z", "+00:00"))
|
start_dt = datetime.fromisoformat(start.replace("Z", "+00:00"))
|
||||||
end_dt = datetime.fromisoformat(end.replace("Z", "+00:00"))
|
# end_dt is not needed for this assertion
|
||||||
# Should be from midnight today to now
|
# Should be from midnight today to now
|
||||||
assert start_dt.hour == 0
|
assert start_dt.hour == 0
|
||||||
assert start_dt.minute == 0
|
assert start_dt.minute == 0
|
||||||
|
|||||||
Reference in New Issue
Block a user