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 Litellm dev 06 18 2025 p1 (#11872) 2025-06-18 21:24:36 -07:00
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 Integration: Bytez as a model provider (#12121) 2025-07-12 10:50:39 -07:00
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 [Feat] Add Eleven Labs - Speech To Text Support on LiteLLM (#12119) 2025-06-27 17:50:49 -07:00
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 refactor: cleanup huggingface rerank transformation 2025-06-06 10:30:44 -07:00
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 test llamafile 2025-07-03 22:35:22 -07:00
lm_studio fix(lm_studio): resolve illegal Bearer header value issue 2025-09-12 22:41:30 +02:00
meta_llama [Bug Fix] Fix Error code: 307 for LlamaAPI Streaming Chat (#11946) 2025-06-30 16:52:42 -07:00
mistral Merge branch 'main' into litellm_dev_08_15_2025_p1 2025-08-23 10:51:56 -07:00
moonshot [Feat] New LLM API Integration - Add Moonshot API (Kimi) (#12551) (#12592) 2025-07-14 15:23:34 -07:00
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 Openrouter - filter out cache_control flag for non-anthropic models (allows usage with claude code) (#12850) 2025-07-21 22:15:48 -07:00
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 [Bug Fix] QA - Use PG Vector Vector Store with LiteLLM (#12716) 2025-07-18 08:41:18 -07:00
recraft [Feat] Add Recraft API - Image Edits Support (#12874) 2025-07-22 15:03:08 -07:00
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 Fixing watsonx error: 'model_id' or 'model' cannot be specified in the request body for models in a deployment space (#11854) 2025-06-23 10:14:10 -07:00
xai fix xai cost calculation (#14127) 2025-09-01 11:31:56 -07:00