litellm/tests/test_litellm/llms/databricks
Mateo Wang a4a3348801
[internal copy of #28007] Fix/gcp model garden streaming (#28363)
* fix(vertex): stream Model Garden Gemma/Qwen responses correctly through /v1/messages

* test(vertex): cover _CombinedChunkSplitter defensive branches

* test(databricks): rename test file to avoid duplicate basename collision

* fix(databricks,anthropic): defensive token defaults; document single-mode splitter

Address greptile P2 concerns:
- databricks: default usage token fields to 0 when constructing
  ChatCompletionUsageBlock from a partially populated usage block — matches
  the defensive pattern used in ollama/vertex_ai/cohere/bedrock.
- _CombinedChunkSplitter: clarify in the docstring that an instance is
  single-mode (sync or async, not both), since the two iteration paths hold
  independent upstream iterator references.

Co-authored-by: Claude <claude@anthropic.com>

---------

Co-authored-by: Steven Kessler <9701252+stvnksslr@users.noreply.github.com>
Co-authored-by: Claude <claude@anthropic.com>
2026-06-10 12:31:00 -07:00
..
chat style: run black formatter on files from main merge 2026-04-17 13:02:59 -07:00
responses Address Greptile review: fix SDK auth fallback and remove unused imports 2026-02-18 15:57:50 +09:00
databricks_config.template.txt feat(databricks): Add enhanced authentication, security features, and custom user-agent support 2025-12-22 12:02:51 -08:00
test_databricks_common_utils.py
test_databricks_e2e.py feat(databricks): Add enhanced authentication, security features, and custom user-agent support 2025-12-22 12:02:51 -08:00
test_databricks_partner_integration.py style: run black formatter on files from main merge 2026-04-17 13:02:59 -07:00
test_databricks_pricing.py
test_databricks_streaming_utils.py [internal copy of #28007] Fix/gcp model garden streaming (#28363) 2026-06-10 12:31:00 -07:00