litellm/tests/test_litellm/llms
Felipe Garé a696ffe4a6
Litellm gemini batch (#14733)
* feat: add Vertex AI support for file content retrieval

- Extended `custom_llm_provider` to include "vertex_ai" in `afile_content` function.
- Implemented file content retrieval logic for Vertex AI in `VertexAIFilesHandler`.
- Added helper method to extract bucket and object from URL-encoded file_id.
- Created comprehensive unit and integration tests for Vertex AI file handling.
- Updated transformation logic to ensure compatibility with Vertex AI file responses.

* fix: update Vertex AI file transformation logic

- Modified the transformation logic in `VertexAIFilesConfig` to return a newline-separated JSON string for batch JSONL files instead of a array if JSON strings.

* fix: enhance Vertex AI output handling in transformation logic

- Updated the transformation logic in `VertexAIBatchTransformation` to utilize the new `OutputInfo` TypedDict for retrieving the GCS output directory.
- Added `OutputInfo` class to type definitions for better structure and clarity in Vertex AI responses.
2025-09-19 15:22:52 -07:00
..
anthropic test: update tests 2025-09-16 22:30:40 -07:00
azure [Feat]Add cancel endpoint support for openai and azure (#14561) 2025-09-15 07:08:56 -07:00
azure_ai/chat
baseten/chat
bedrock Merge branch 'litellm_contributor_prs_09_18_2025_p2' into fix/issue-14685-bedrock-titan-v2-encoding-format 2025-09-18 17:54:33 -07:00
bytez/chat
chat
cohere/chat
cometapi/chat
compactifai Fix CompactifAI provider tests and implementation 2025-09-15 22:03:42 +02:00
custom_httpx Fix linting 2025-09-07 18:04:00 -07:00
dashscope [Fixes] Using Qwen API Tiered Pricing (#14479) 2025-09-11 20:07:41 -07:00
databricks Merge branch 'main' into feature/databricks-function-call-missing-pass-description 2025-09-09 22:42:32 -07:00
datarobot
deepgram
deepinfra fix deepinfra test 2025-09-10 19:39:23 -07:00
featherless_ai/chat
fireworks_ai/chat
gemini Fix gemini cli error (#14417) 2025-09-12 11:56:51 -07:00
github_copilot
heroku
hosted_vllm
huggingface
jina_ai/embedding
litellm_proxy/chat
llamafile/chat
lm_studio fix(lm_studio): resolve illegal Bearer header value issue 2025-09-12 22:41:30 +02:00
meta_llama
mistral
moonshot
nebius
novita/chat
nscale/chat
oci/chat Merge pull request #14092 from gotsysdba/main 2025-09-09 22:39:16 -07:00
ollama test: add unit tests for ollama chat request transformation 2025-09-12 17:59:59 -07:00
openai [Bug Fix] x-litellm-tags not routing with Responses API (#14289) 2025-09-05 09:40:37 -07:00
openrouter/chat
ovhcloud feat: Add OVHCloud AI Endpoints as a provider 2025-09-12 13:37:03 +02:00
perplexity
pg_vector/vector_stores
recraft
sagemaker
sambanova
vercel_ai_gateway
vertex_ai Litellm gemini batch (#14733) 2025-09-19 15:22:52 -07:00
volcengine add more test 2025-09-16 12:11:10 +08:00
watsonx
xai