mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-06 02:48:13 +00:00
248f0eb159
* ci: fix the litellm-tests unit job with sysmon coverage, an env allowlist and coverage upload on failure * test: replace key-dependent proxy, enterprise and mcp unit tests with synthetic values and integration and e2e coverage * test: drop key reads at the legacy proxy, enterprise and mcp paths and wire the gemini pass-through split * ci: move caching, proxy-extras, gateway and enterprise tests into tests/unit and run them from litellm-tests under their legacy flags * ci: move caching, proxy-extras, gateway and enterprise tests into tests/unit and run them from litellm-tests under their legacy flags * ci: move tests/proxy_unit_tests to tests/unit/proxy and run the proxy-db shards from litellm-tests * ci: fail the unit shard when circleci tests split errors * test: drop restating comments from the gemini pass-through split * build: point the local proxy unit targets at the nested tests/unit/proxy tree * ci: exit the unit shard cleanly when circleci tests split assigns it no files --------- Co-authored-by: yuneng <yuneng@berri.ai>
57 lines
2.6 KiB
Python
57 lines
2.6 KiB
Python
from unittest.mock import MagicMock
|
|
|
|
import litellm.skills.main as skills_main
|
|
from litellm.types.utils import LlmProviders
|
|
|
|
|
|
def test_create_skill_forwards_description_and_instructions_from_top_level_kwargs(
|
|
monkeypatch,
|
|
) -> None:
|
|
"""The REST /v1/skills form endpoint passes description/instructions as top-level
|
|
kwargs (not extra_body). Regression for a bug where the litellm_proxy dispatch
|
|
branch of create_skill() dropped both, so every LiteLLM-hosted skill was created
|
|
with description=None and instructions=None regardless of what the caller sent."""
|
|
handler = MagicMock()
|
|
monkeypatch.setattr(skills_main, "_get_litellm_skills_handler", lambda: handler)
|
|
|
|
skills_main.create_skill(
|
|
display_title="Document Translator",
|
|
description="Converts files from one language into another",
|
|
instructions="Take an uploaded document and produce it in the target language",
|
|
custom_llm_provider=LlmProviders.LITELLM_PROXY.value,
|
|
)
|
|
|
|
assert handler.create_skill_handler.call_args.kwargs["description"] == (
|
|
"Converts files from one language into another"
|
|
)
|
|
assert handler.create_skill_handler.call_args.kwargs["instructions"] == (
|
|
"Take an uploaded document and produce it in the target language"
|
|
)
|
|
|
|
|
|
def test_create_skill_forwards_description_and_instructions_from_extra_body(monkeypatch) -> None:
|
|
"""The SDK convention (see tests/unit/skills/test_skills_db.py) nests them under
|
|
extra_body instead of passing them as top-level kwargs; both paths must reach the DB."""
|
|
handler = MagicMock()
|
|
monkeypatch.setattr(skills_main, "_get_litellm_skills_handler", lambda: handler)
|
|
|
|
skills_main.create_skill(
|
|
display_title="Warehouse SQL Analyst",
|
|
extra_body={"description": "Runs SQL against the inventory database", "instructions": "Summarize results"},
|
|
custom_llm_provider=LlmProviders.LITELLM_PROXY.value,
|
|
)
|
|
|
|
assert handler.create_skill_handler.call_args.kwargs["description"] == (
|
|
"Runs SQL against the inventory database"
|
|
)
|
|
assert handler.create_skill_handler.call_args.kwargs["instructions"] == "Summarize results"
|
|
|
|
|
|
def test_create_skill_without_description_or_instructions_passes_none(monkeypatch) -> None:
|
|
handler = MagicMock()
|
|
monkeypatch.setattr(skills_main, "_get_litellm_skills_handler", lambda: handler)
|
|
|
|
skills_main.create_skill(display_title="Bare Skill", custom_llm_provider=LlmProviders.LITELLM_PROXY.value)
|
|
|
|
assert handler.create_skill_handler.call_args.kwargs["description"] is None
|
|
assert handler.create_skill_handler.call_args.kwargs["instructions"] is None
|