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 add testing and remove redundant function 2025-08-21 11:25:26 -07:00
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 feat: add CometAPI provider support with chat completions and streaming (#13458) 2025-08-11 18:06:37 -07:00
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 Updated URL handling for DataRobot provider base 2025-08-21 19:42:46 -06:00
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 Merge branch 'main' into feat/github-copilot-thinking-reasoning-support 2025-08-27 22:09:15 -07:00
heroku adds tool calling test 2025-08-06 14:23:55 -06:00
hosted_vllm [Feat] Add reasoning_effort param for hosted_vllm provider (#13620) 2025-08-14 10:10:30 -07:00
huggingface
jina_ai/embedding feat(JinaAI): support multimodal embedding models (#13181) 2025-08-05 19:21:56 -07:00
litellm_proxy/chat LiteLLM SDK <-> Proxy: support user param + Prisma - remove use_prisma_migrate flag - redundant as this is now default (#13555) 2025-08-12 22:03:39 -07:00
llamafile/chat
lm_studio fix(lm_studio): resolve illegal Bearer header value issue 2025-09-12 22:41:30 +02:00
meta_llama
mistral Merge branch 'main' into litellm_dev_08_15_2025_p1 2025-08-23 10:51:56 -07:00
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 add Perplexity citation annotations support (#13225) 2025-08-02 08:47:35 -07:00
pg_vector/vector_stores
recraft
sagemaker
sambanova Feat/sambanova embeddings (#13308) 2025-08-12 17:15:26 -07:00
vercel_ai_gateway Convert integration tests to mocks 2025-07-31 23:57:53 -07:00
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 fix xai cost calculation (#14127) 2025-09-01 11:31:56 -07:00