scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test(ollama): cover async embedding path for explicit api_base precedence

Browse source
This commit is contained in:
Mukunda Rao Katta 2026-04-21 09:18:08 -07:00
parent 00011f4bd4
commit 404d748e07
1 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
tests/local_testing/test_completion.py
View file

@ -1354,6 +1354,22 @@ def test_ollama_embedding_explicit_api_base_takes_precedence(monkeypatch):
assert mock_embeddings.call_args.kwargs["api_base"] == "http://localhost:11434"
@pytest.mark.asyncio
async def test_ollama_aembedding_explicit_api_base_takes_precedence(monkeypatch):
monkeypatch.setattr(litellm, "api_base", "https://api.deepseek.com")
with patch("litellm.main.ollama.ollama_aembeddings") as mock_aembeddings:
mock_aembeddings.return_value = MagicMock()
await litellm.aembedding(
model="ollama/qwen3-embedding:0.6b",
input="hello",
api_base="http://localhost:11434",
)
assert mock_aembeddings.call_args.kwargs["api_base"] == "http://localhost:11434"
# ################### Hugging Face Conversational models ########################
# def hf_test_completion_conv():
# try: