Skip to content

Commit 76ced44

Browse files
authored
Merge pull request #1945 from asimurka/wire_agents_skills
LCORE-2076: Fix missing skills argument in streaming agent build
2 parents 570a66e + 281d09b commit 76ced44

2 files changed

Lines changed: 10 additions & 1 deletion

File tree

src/utils/agents/streaming.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ async def retrieve_agent_response_generator(
117117
turn_summary,
118118
)
119119

120-
agent = build_agent(context.client, responses_params)
120+
agent = build_agent(context.client, responses_params, configuration.skills)
121121

122122
return (
123123
agent_response_generator(

tests/unit/utils/agents/test_streaming.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -203,6 +203,14 @@ def patch_recording_metrics_fixture(mocker: MockerFixture) -> None:
203203
mocker.patch("utils.agents.query.recording.record_llm_call")
204204

205205

206+
@pytest.fixture(name="patch_streaming_configuration")
207+
def patch_streaming_configuration_fixture(mocker: MockerFixture) -> None:
208+
"""Patch streaming module configuration for isolated agent streaming tests."""
209+
mock_config = mocker.MagicMock()
210+
mock_config.skills = None
211+
mocker.patch("utils.agents.streaming.configuration", mock_config)
212+
213+
206214
@pytest.fixture(autouse=True, name="stream_interrupt_mocks")
207215
def stream_interrupt_mocks_fixture(mocker: MockerFixture) -> dict[str, Any]:
208216
"""Patch stream interrupt registry and deregister for wrapper tests."""
@@ -479,6 +487,7 @@ def test_part_end_native_tool_call_returns_none_when_skipped(
479487
assert not turn_state.turn_summary.tool_calls
480488

481489

490+
@pytest.mark.usefixtures("patch_streaming_configuration")
482491
class TestRetrieveAgentResponseGenerator:
483492
"""Tests for retrieve_agent_response_generator."""
484493

0 commit comments

Comments
 (0)