Log Brave search responses

This commit is contained in:
2026-05-19 21:49:31 +02:00
parent 5ed1104c12
commit cc783ab8ed
2 changed files with 104 additions and 1 deletions

View File

@@ -1,6 +1,7 @@
from __future__ import annotations
from dataclasses import dataclass, field
import logging
import re
from typing import Any
@@ -8,8 +9,11 @@ import httpx
from app.config import AIWebSearchConfig
log = logging.getLogger(__name__)
SAFESEARCH_VALUES = {"off", "moderate", "strict"}
FRESHNESS_RE = re.compile(r"^(pd|pw|pm|py|\d{4}-\d{2}-\d{2}to\d{4}-\d{2}-\d{2})$")
MAX_LOG_BODY_CHARS = 1000
@dataclass(slots=True)
@@ -231,16 +235,59 @@ class BraveWebSearchClient:
raise ValueError(f"Brave web search request failed: {exc}") from exc
if response.status_code >= 400:
_log_error_response(response)
raise ValueError(_format_error_response(response))
try:
payload = response.json()
except ValueError as exc:
_log_error_response(response)
raise ValueError("Brave web search returned invalid JSON") from exc
if not isinstance(payload, dict):
_log_response_summary(response, query.query, results_count=0)
raise ValueError("Brave web search returned an unexpected response")
return parse_response(query.query, payload)
parsed = parse_response(query.query, payload)
_log_response_summary(response, query.query, results_count=len(parsed.results))
return parsed
def _response_log_body(response: httpx.Response) -> str:
body = response.text.replace("\n", "\\n").replace("\r", "\\r")
if len(body) > MAX_LOG_BODY_CHARS:
return body[:MAX_LOG_BODY_CHARS] + "...(truncated)"
return body
def _response_url(response: httpx.Response) -> str:
try:
return str(response.request.url)
except RuntimeError:
return str(response.url)
def _log_response_summary(
response: httpx.Response,
query: str,
*,
results_count: int,
) -> None:
log.info(
"Brave web search response status=%s query=%r results=%s url=%s",
response.status_code,
query,
results_count,
_response_url(response),
)
def _log_error_response(response: httpx.Response) -> None:
log.warning(
"Brave web search error status=%s url=%s body=%s",
response.status_code,
_response_url(response),
_response_log_body(response),
)
def _format_error_response(response: httpx.Response) -> str: