fix(api): cap memory message limit and top_n at REST_API_MAX_PAGE_SIZE (#15376)

## Related issues

Closes #15375

### What problem does this PR solve?

`GET /api/v1/messages` and `GET /api/v1/messages/search` accepted
unbounded `limit` / `top_n` query parameters while other REST list
endpoints enforce `REST_API_MAX_PAGE_SIZE` (100) via
`validate_rest_api_page_size()`. Oversized values can trigger expensive
memory index queries and large result sets (DoS risk).

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

### Changes

| File | Change |
|------|--------|
| `api/apps/restful_apis/memory_api.py` | Cap `limit` and `top_n` with
`validate_rest_api_page_size`; return argument error when exceeded |
|
`test/testcases/test_web_api/test_message_app/test_message_routes_unit.py`
| Regression tests for oversized `limit` / `top_n` |

### Test plan

- [x] Unit tests added
- [ ] `pytest
test/testcases/test_web_api/test_message_app/test_message_routes_unit.py`

Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
jony376
2026-07-05 20:20:13 -07:00
committed by GitHub
parent 534a3a7faa
commit aaade3530e
2 changed files with 41 additions and 2 deletions

View File

@@ -260,7 +260,10 @@ async def search_message():
query = args.get("query")
similarity_threshold = float(args.get("similarity_threshold", 0.2))
keywords_similarity_weight = float(args.get("keywords_similarity_weight", 0.7))
top_n = int(args.get("top_n", 5))
try:
top_n = validate_rest_api_page_size(int(args.get("top_n", 5)))
except ValueError as exc:
return get_error_argument_result(str(exc))
agent_id = args.get("agent_id", "")
session_id = args.get("session_id", "")
user_id = args.get("user_id", "")
@@ -280,7 +283,10 @@ async def get_messages():
memory_ids = memory_ids[0].split(",")
agent_id = args.get("agent_id", "")
session_id = args.get("session_id", "")
limit = int(args.get("limit", 10))
try:
limit = validate_rest_api_page_size(int(args.get("limit", 10)))
except ValueError as exc:
return get_error_argument_result(str(exc))
if not memory_ids:
return get_error_argument_result("memory_ids is required.")
try: