Simplify config example and log API bodies

This commit is contained in:
2026-06-08 13:26:22 +08:00
parent 5500947ddb
commit 64e3ea9631
4 changed files with 85 additions and 7 deletions

View File

@ -1,13 +1,17 @@
"""Standalone FastAPI server for Memory System API."""
from __future__ import annotations
from fastapi import FastAPI
import logging
from fastapi import FastAPI, Request, Response
from fastapi.middleware.cors import CORSMiddleware
from .api import router
from .config import Config, load_config, set_config
request_logger = logging.getLogger("memory_system_api.requests")
app = FastAPI(title="Memory System API", version="0.1.0")
app.add_middleware(
CORSMiddleware,
@ -16,6 +20,42 @@ app.add_middleware(
allow_methods=["*"],
allow_headers=["*"],
)
@app.middleware("http")
async def log_request_and_response(request: Request, call_next):
request_body = await request.body()
request_logger.info(
"request %s %s body=%s",
request.method,
_path_with_query(request),
_body_for_log(request_body),
)
async def receive():
return {"type": "http.request", "body": request_body, "more_body": False}
response = await call_next(Request(request.scope, receive))
response_body = b""
async for chunk in response.body_iterator:
response_body += chunk
request_logger.info(
"response %s %s status=%s body=%s",
request.method,
_path_with_query(request),
response.status_code,
_body_for_log(response_body),
)
return Response(
content=response_body,
status_code=response.status_code,
headers=dict(response.headers),
media_type=response.media_type,
background=response.background,
)
app.include_router(router)
@ -25,6 +65,17 @@ def create_app(config: Config | None = None) -> FastAPI:
return app
def _path_with_query(request: Request) -> str:
query = request.url.query
return f"{request.url.path}?{query}" if query else request.url.path
def _body_for_log(body: bytes) -> str:
if not body:
return ""
return body.decode("utf-8", errors="replace")
def main() -> None:
import argparse
import uvicorn
@ -41,6 +92,7 @@ def main() -> None:
if args.port:
config.server.port = args.port
set_config(config)
logging.basicConfig(level=config.logging.level.upper(), format=config.logging.format)
uvicorn.run(app, host=config.server.host, port=config.server.port, log_level=config.logging.level.lower())