Simplify config example and log API bodies
This commit is contained in:
@ -36,6 +36,7 @@ class StorageConfig(BaseModel):
|
||||
|
||||
class LoggingConfig(BaseModel):
|
||||
level: str = "INFO"
|
||||
format: str = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
|
||||
|
||||
|
||||
class Config(BaseModel):
|
||||
|
||||
@ -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())
|
||||
|
||||
|
||||
Reference in New Issue
Block a user