mirror of
https://github.com/tiennm99/litellm.git
synced 2026-08-03 10:23:28 +00:00
michelligabrieleandGitHub
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
Testing for litellm/
This directory 1:1 maps the the litellm/ directory, and can only contain mocked tests.
The point of this is to:
- Increase test coverage of
litellm/ - Make it easy for contributors to add tests for the
litellm/package and easily run tests without needing LLM API keys.
File name conventions
litellm/proxy/test_caching_routes.pymaps tolitellm/proxy/caching_routes.pytest_<filename>.pymaps tolitellm/<filename>.py