scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 109

fix(lint): add missing Final import in heroku and add snowflake trailing slash tests

Browse source
This commit is contained in:
Raushankumar0720 2026-09-24 13:16:46 +05:30
parent ccb34a88d0
commit 57b00ecbf0
2 changed files with 43 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/heroku/chat/transformation.py
View file

@ -6,7 +6,7 @@ this is OpenAI compatible - no translation needed / occurs
import os
from collections.abc import Coroutine
from typing import Any, Literal, overload
from typing import Any, Final, Literal, overload
from litellm.litellm_core_utils.prompt_templates.common_utils import (
handle_messages_with_content_list_to_str_conversion,

42
tests/test_litellm/llms/test_provider_url_trailing_slash.py
View file

@ -6,6 +6,8 @@ from litellm.llms.moonshot.chat.transformation import MoonshotChatConfig
from litellm.llms.ollama.chat.transformation import OllamaChatConfig
from litellm.llms.ollama.completion.transformation import OllamaConfig
from litellm.llms.perplexity.embedding.transformation import PerplexityEmbeddingConfig
from litellm.llms.snowflake.chat.transformation import SnowflakeConfig
from litellm.llms.snowflake.embedding.transformation import SnowflakeEmbeddingConfig
from litellm.llms.topaz.image_variations.transformation import TopazImageVariationConfig
from litellm.llms.voyage.embedding.transformation import VoyageEmbeddingConfig
from litellm.llms.voyage.embedding.transformation_contextual import VoyageContextualEmbeddingConfig
@ -240,3 +242,43 @@ def test_topaz_get_complete_url_trailing_slash():
assert url_without_slash == "https://api.topazlabs.com/image/v1/enhance"
assert url_with_slash == "https://api.topazlabs.com/image/v1/enhance"
assert "//image/v1/enhance" not in url_with_slash
def test_snowflake_get_complete_url_trailing_slash():
chat_config = SnowflakeConfig()
url_without_slash = chat_config.get_complete_url(
api_base="https://myaccount.snowflakecomputing.com",
api_key=None,
model="snowflake-arctic",
optional_params={},
litellm_params={},
)
url_with_slash = chat_config.get_complete_url(
api_base="https://myaccount.snowflakecomputing.com/",
api_key=None,
model="snowflake-arctic",
optional_params={},
litellm_params={},
)
assert url_without_slash == "https://myaccount.snowflakecomputing.com/api/v2/cortex/v1/chat/completions"
assert url_with_slash == "https://myaccount.snowflakecomputing.com/api/v2/cortex/v1/chat/completions"
assert "//api/v2" not in url_with_slash
embed_config = SnowflakeEmbeddingConfig()
embed_url_without_slash = embed_config.get_complete_url(
api_base="https://myaccount.snowflakecomputing.com",
api_key=None,
model="snowflake-arctic-embed",
optional_params={},
litellm_params={},
)
embed_url_with_slash = embed_config.get_complete_url(
api_base="https://myaccount.snowflakecomputing.com/",
api_key=None,
model="snowflake-arctic-embed",
optional_params={},
litellm_params={},
)
assert embed_url_without_slash == "https://myaccount.snowflakecomputing.com/api/v2/cortex/inference:embed"
assert embed_url_with_slash == "https://myaccount.snowflakecomputing.com/api/v2/cortex/inference:embed"
assert "//api/v2" not in embed_url_with_slash