mirror of
https://github.com/tiennm99/litellm.git
synced 2026-06-18 17:28:19 +00:00
michelligabriele
c47f77a348
* fix(agentcore): handle JSON responses from agents using sync return BedrockAgentCoreApp agents that use synchronous `return` (instead of async `yield`) respond with Content-Type: application/json instead of text/event-stream. The streaming parser only handles SSE format, silently discarding the JSON body and returning empty content to the client. This adds Content-Type detection in both sync and async streaming wrappers — when application/json is received, the response is parsed and converted to a single-chunk stream. Also extends _parse_json_response with a fallback chain supporting multiple agent response schemas (standard AgentCore, Strands framework, plain string, raw JSON fallback). * fix(agentcore): add dict-type guard to _parse_json_response Prevent AttributeError when json.loads() returns a non-dict (e.g. JSON array or primitive) by adding an isinstance check at the top of _parse_json_response. Non-dict values fall back to raw JSON string content. * fix(agentcore): handle malformed JSON and split streaming chunks - Wrap json.loads() in try/except in both sync and async streaming wrappers so malformed JSON bodies raise a structured BedrockError instead of a raw JSONDecodeError - Split the JSON-fallback streaming path into two chunks (content chunk with finish_reason=None, then stop sentinel with empty delta) to match the SSE path convention * fix(agentcore): catch IO errors in streaming JSON path + async error test - Broaden except clause to catch both json.JSONDecodeError and IO-level exceptions (httpx.ReadError, etc.) from response.read()/aread(), so all failures surface as structured BedrockError - Add async malformed-JSON test to mirror the sync test coverage
In total litellm runs 1000+ tests
[02/20/2025] Update:
To make it easier to contribute and map what behavior is tested,
we've started mapping the litellm directory in tests/test_litellm
This folder can only run mock tests.