mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-05 02:41:56 +00:00
fix(tests): mock native advisor support in interceptor assertions
Made-with: Cursor
This commit is contained in:
parent
bfd0b9fa5f
commit
9a6ba75a81
1 changed files with 6 additions and 0 deletions
|
|
@ -79,6 +79,9 @@ def test_can_handle_edge_cases():
|
|||
with patch(
|
||||
"litellm.llms.anthropic.experimental_pass_through.messages.interceptors.advisor.resolve_proxy_model_alias_to_litellm_model",
|
||||
return_value="",
|
||||
), patch(
|
||||
"litellm.llms.anthropic.experimental_pass_through.messages.interceptors.advisor.supports_native_advisor_tool",
|
||||
return_value=True,
|
||||
):
|
||||
assert h.can_handle([ADVISOR_TOOL], "openai")
|
||||
assert h.can_handle([ADVISOR_TOOL], "bedrock")
|
||||
|
|
@ -110,6 +113,9 @@ async def test_anthropic_native_interceptor_skipped():
|
|||
with patch(
|
||||
"litellm.llms.anthropic.experimental_pass_through.messages.interceptors.advisor.resolve_proxy_model_alias_to_litellm_model",
|
||||
return_value="",
|
||||
), patch(
|
||||
"litellm.llms.anthropic.experimental_pass_through.messages.interceptors.advisor.supports_native_advisor_tool",
|
||||
return_value=True,
|
||||
):
|
||||
assert not h.can_handle(
|
||||
[ADVISOR_TOOL], "anthropic"
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue