litellm/tests/image_gen_tests
user 4af58e1f97
[Security] Clear AWS Inspector CVE findings on Docker image
- Narrow /root/.cache COPY in Dockerfile to /root/.cache/prisma{,-python}
  only — drops ~660MB of uv build cache including a setuptools wheel
  that surfaced as CVE-2024-6345 / CVE-2025-47273 even though it was
  never on the runtime sys.path.
- DiskCache: switch to dc.JSONDisk to neutralize the pickle code path
  (CVE-2025-69872, no upstream fix). Values must be JSON-serializable;
  cleanup get_cache to skip the now-dead json.loads(dict) branch by
  guarding on isinstance(str).
- pyproject.toml: drop diskcache pin from [caching] extra (no fixed
  version exists). Stub kept so `pip install litellm[caching]` doesn't
  warn; users who want disk caching install diskcache themselves.
- Bump black 24.10.0 → 26.3.1 (CVE-2026-32274) + apply 296-file mechanical
  reformat. Black is dev-only (not in the runtime image), but bumping
  clears the manifest-scan finding.
- Refresh ui/litellm-dashboard/package-lock.json to pick up next 16.2.4
  (was 16.1.7, GHSA-q4gf-8mx6-v5v3), uuid 14.0.0, postcss 8.5.13.
- Refresh litellm-js/spend-logs/package-lock.json to pick up
  hono 4.12.16 (GHSA-458j-xx4x-4375).
- uv lock: gitpython 3.1.46 → 3.1.49 (clears two High GHSAs),
  langchain-text-splitters 1.1.1 → 1.1.2.
- Add tests/test_litellm/caching/test_disk_cache.py covering JSONDisk
  enforcement, dict/string round-trip, TTL, increment, delete/flush.

Net delta on combined trivy + grype scans: 17 findings → 4 (all
remaining 4 are Wolfi system python-3.13 CVEs marked WONTFIX upstream
in CPython 3.14; CVE-2026-3298 is Windows-unreachable on Linux).

Existing on-disk caches written by the previous pickle-format Disk
will silently miss after upgrade — diskcache is intended to be
ephemeral so impact is recreate-on-next-write.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
2026-05-05 20:21:15 +00:00
..
request_payloads test(image_gen): align Azure image gen fixture with body omitting model 2026-05-04 18:09:30 +05:30
base_image_generation_test.py fix(ci): remove unused is_expired variable (ruff F841) and handle ModelDeprecated in image gen test 2026-03-12 01:27:42 +00:00
conftest.py style: run black formatter on files from main merge 2026-04-17 13:02:59 -07:00
ishaan_github.png [Feat] Add Image Edits Support to LiteLLM (#11020) 2025-05-21 14:03:09 -07:00
litellm_site.png [Feat] Add Image Edits Support to LiteLLM (#11020) 2025-05-21 14:03:09 -07:00
test_bedrock_image_gen_unit_tests.py style: run black formatter on files from main merge 2026-04-17 13:02:59 -07:00
test_fal_ai_image_generation.py style: run black formatter on files from main merge 2026-04-17 13:02:59 -07:00
test_image.png [BETA] Add OpenAI /images/variations + Topaz API support (#7700) 2025-01-11 23:27:46 -08:00
test_image_edit.png test: cleanup tests 2026-03-30 16:20:01 -07:00
test_image_edits.py test(image_gen): expect no model in Azure image edit multipart (#26316) 2026-05-04 12:20:29 +05:30
test_image_generation.py [Security] Clear AWS Inspector CVE findings on Docker image 2026-05-05 20:21:15 +00:00
test_image_variation.py [Security] Clear AWS Inspector CVE findings on Docker image 2026-05-05 20:21:15 +00:00
test_xinference.py style: run black formatter on files from main merge 2026-04-17 13:02:59 -07:00
vertex_key.json test: update to new vertex ai keys 2026-03-28 20:19:05 -07:00