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(agents): use real policy data in header forwarding fixtures
Some checks are pending
LiteLLM Rust / rust-lint (push) Waiting to run
Details
LiteLLM Rust / rust-test (push) Waiting to run
Details
LiteLLM Rust / rust-wheel (push) Waiting to run
Details

Browse source
This commit is contained in:
Joshua Valluru 2026-09-30 18:57:18 -07:00
parent cbec06c183
commit 867f097a00
1 changed files with 15 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
tests/test_litellm/proxy/agent_endpoints/test_agent_headers.py
View file

@ -14,23 +14,26 @@ from unittest.mock import AsyncMock, MagicMock, patch
import pytest
from litellm.types.agents import AgentResponse
# ---------------------------------------------------------------------------
# Helper: build a minimal mock agent
# Helper: build a minimal agent
# ---------------------------------------------------------------------------
def _make_mock_agent(
static_headers=None,
extra_headers=None,
url="http://backend-agent:10001",
):
mock_agent = MagicMock()
mock_agent.agent_id = "agent-123"
mock_agent.agent_card_params = {"url": url, "name": "Test Agent"}
mock_agent.litellm_params = {}
mock_agent.static_headers = static_headers or {}
mock_agent.extra_headers = extra_headers or []
return mock_agent
static_headers: dict[str, str] | None = None,
extra_headers: list[str] | None = None,
url: str = "http://backend-agent:10001",
) -> AgentResponse:
return AgentResponse(
agent_id="agent-123",
agent_name="Test Agent",
agent_card_params={"url": url, "name": "Test Agent"},
litellm_params={},
static_headers=static_headers or {},
extra_headers=extra_headers or [],
)
def _make_mock_request(extra_headers=None, method="message/send"):