Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

feat: exception handling middleware for sandbox_configs + identities + tools (#5143)

Browse Source
This commit is contained in:
Sarah Wooders
2025-10-06 23:52:05 -07:00
committed by Caren Thomas
parent 307c85ca9a
commit 324933edd3
6 changed files with 307 additions and 503 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/mcp_tests/test_mcp_schema_validation.py
View File

@@ -163,12 +163,14 @@ async def test_add_mcp_tool_rejects_invalid_schemas():
# Should raise HTTPException for invalid schema
headers = HeaderParams(actor_id="test_user")
with pytest.raises(HTTPException) as exc_info:
from letta.errors import LettaInvalidMCPSchemaError
with pytest.raises(LettaInvalidMCPSchemaError) as exc_info:
await add_mcp_tool(mcp_server_name="test_server", mcp_tool_name="test_tool", server=mock_server, headers=headers)
assert exc_info.value.status_code == 400
assert "invalid schema" in exc_info.value.detail["message"].lower()
assert exc_info.value.detail["health_status"] == SchemaHealth.INVALID.value
assert "invalid schema" in exc_info.value.message.lower()
assert exc_info.value.details["mcp_tool_name"] == "test_tool"
assert exc_info.value.details["reasons"] == ["Missing 'type' at root level"]
def test_mcp_schema_healing_for_optional_fields():