litellm/tests/test_litellm/llms
Ishaan Jaff 7133488282
[Feat] VertexAI - Add BGE Embeddings support (#16033)
* Support for Custom Vertex AI Models via PSC Endpoint with api_base (#15953)

* Support for Custom Vertex AI Models via PSC Endpoint with api_base

* Add docs related psc

* remove not needed files

* remove print statemnt

* fix mypy errors

* add TextEmbeddingBGEInput

* add VertexBGEConfig

* add BGE handling

* test_vertex_ai_bge_embedding_with_custom_api_base

* fix request transform vertex BGE

* test_vertex_ai_bge_embedding_with_custom_api_base

* tes BGE

* test_is_bge_model_detection

* docs cleanup

* handling BGE URL

* fix VertexBGEConfig

* test_vertex_ai_bge_with_endpoint_id_pattern

* docs vertex BGE

* docs

* docs fix

* fix VertexAIModelRoute

* from ..common_utils import VertexAIError, get_vertex_base_model_name
add

* fix VertexAIGemmaModels

* fix get_vertex_base_model_name

* test_vertex_ai_bge_psc_endpoint_url_construction

---------

Co-authored-by: Sameer Kankute <sameer@berri.ai>
2025-11-13 12:41:00 -08:00
..
anthropic feat: Add support for Anthropic Memory Tool (#16115) 2025-11-07 19:27:28 -08:00
azure Adds support for returning Azure Content Policy error information when exceptions from Azure OpenAI occur (#16231) 2025-11-08 16:04:36 -08:00
azure_ai Removed stop param from unsupported azure models (#15229) 2025-10-06 19:56:18 -07:00
baseten/chat add testing and remove redundant function 2025-08-21 11:25:26 -07:00
bedrock Remove aws params from claude4_5 request (#16315) 2025-11-06 10:10:30 -08:00
bytez/chat Integration: Bytez as a model provider (#12121) 2025-07-12 10:50:39 -07:00
chat Litellm fix GitHub action testing (#11163) 2025-05-26 14:41:42 -07:00
cohere Guardrails - Responses API, Image Gen, Text completions, Audio transcriptions, Audio Speech, Rerank, Anthropic Messages API support via the unified apply_guardrails function (#15706) 2025-10-25 13:38:57 -07:00
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: apply provided timeout value to ClientTimeout.total (#16395) 2025-11-10 19:37:42 -08:00
dashscope feat: support dashscope tiered pricing 2025-11-03 20:27:25 -08: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 fix: return the diarized transcript when it's required in the request (#16133) 2025-11-02 09:45:18 -08:00
deepinfra fix deepinfra test 2025-09-10 19:39:23 -07:00
featherless_ai/chat test: fixes 2025-05-31 12:42:56 -07:00
fireworks_ai/chat Litellm fix GitHub action testing (#11163) 2025-05-26 14:41:42 -07:00
gemini Add Gemini image edit support (#16430) 2025-11-12 18:48:27 -08: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 fix: fix vllm test 2025-09-27 10:01:48 -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
lemonade Adding unit tests and documentation 2025-09-30 12:12:24 -06: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 Added thinking streaming support for mistral (#16434) 2025-11-10 18:41:45 -08:00
moonshot [Feat] New LLM API Integration - Add Moonshot API (Kimi) (#12551) (#12592) 2025-07-14 15:23:34 -07:00
nebius test: fixes 2025-05-31 12:42:56 -07:00
novita/chat Litellm fix GitHub action testing (#11163) 2025-05-26 14:41:42 -07:00
nscale/chat Litellm fix GitHub action testing (#11163) 2025-05-26 14:41:42 -07:00
oci/chat Add OCI Signer Authentication. Closes #16048, Closes #15654 (#16064) 2025-10-30 19:59:01 -07:00
ollama fix(ollama): Enhance chunk parsing for empty responses without 'thinking' and improve error logging (#13333) (#15717) 2025-10-21 16:59:01 -07:00
openai Guardrails - Responses API, Image Gen, Text completions, Audio transcriptions, Audio Speech, Rerank, Anthropic Messages API support via the unified apply_guardrails function (#15706) 2025-10-25 13:38:57 -07:00
openai_like/chat Sanitize null token values in provider responses (#16493) 2025-11-11 19:08:00 -08:00
openrouter/chat add tests for openrouter cost tracking 2025-10-14 13:57:39 -07:00
ovhcloud feat: Add OVHCloud AI Endpoints as a provider 2025-09-12 13:37:03 +02:00
perplexity Revert "Update perplexity cost tracking (#15743)" (#16345) 2025-11-06 19:00:45 -08: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
runwayml/videos [Feat] Add RunwayML Img Gen API support (#16557) 2025-11-12 18:20:14 -08:00
sagemaker [Oct Staging Branch] (#15460) 2025-10-17 17:52:25 -07:00
sambanova Feat/sambanova embeddings (#13308) 2025-08-12 17:15:26 -07:00
snowflake/chat feat(snowflake): add function calling support for Snowflake Cortex REST API 2025-10-05 13:08:33 +02:00
vercel_ai_gateway Fix vercel_ai_gateway entry for glm-4.6 (#16084) 2025-10-30 19:40:35 -07:00
vertex_ai [Feat] VertexAI - Add BGE Embeddings support (#16033) 2025-11-13 12:41:00 -08:00
volcengine add more test 2025-09-16 12:11:10 +08:00
wandb (feat): Add W&B Inference to LiteLLM 2025-09-11 00:07:30 +05:30
watsonx add reasoning effort and test (#16261) 2025-11-04 18:53:02 -08:00
xai Added xai responses support (#16391) 2025-11-08 12:30:09 -08:00