scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-16 23:41:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 227

test(v2 managed agents): update import path for messages endpoints to proxy/

Browse source
This commit is contained in:
Ishaan Jaffer 2026-05-07 13:28:56 -07:00
parent f63b000b4a
commit 304a4466ee
No known key found for this signature in database
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/test_litellm/managed_agents/test_messages.py
View file

@ -27,7 +27,7 @@ from litellm.managed_agents.adapters.base import (
SandboxBadGatewayError,
SandboxUnreachableError,
)
from litellm.managed_agents.endpoints.messages import router as messages_router
from litellm.proxy.managed_agents_endpoints.messages import router as messages_router
from litellm.managed_agents.types import MessageRow
from litellm.proxy._types import UserAPIKeyAuth
from litellm.proxy.auth.user_api_key_auth import user_api_key_auth
@ -189,7 +189,7 @@ def mock_adapter(monkeypatch):
adapter.list_messages = AsyncMock(return_value=[])
adapter.abort = AsyncMock(return_value=None)
monkeypatch.setattr(
"litellm.managed_agents.endpoints.messages.get_adapter",
"litellm.proxy.managed_agents_endpoints.messages.get_adapter",
lambda sandbox_type: adapter,
)
return adapter