From 763c7f79d4e7ba26072102943b470291111b99bf Mon Sep 17 00:00:00 2001 From: Jan-Stefan Janetzky Date: Tue, 14 Jul 2026 13:32:43 +0000 Subject: [PATCH] test(compression): isolate provider handoff setup --- .../test_pre_compress_memory_context.py | 28 +++++++++---------- 1 file changed, 14 insertions(+), 14 deletions(-) diff --git a/tests/run_agent/test_pre_compress_memory_context.py b/tests/run_agent/test_pre_compress_memory_context.py index d53b1768dcd1..6704807e8656 100644 --- a/tests/run_agent/test_pre_compress_memory_context.py +++ b/tests/run_agent/test_pre_compress_memory_context.py @@ -1,25 +1,25 @@ """Behavior contracts for the pre-compression memory-context handoff.""" -import os -from unittest.mock import MagicMock, patch +from unittest.mock import MagicMock import pytest def _make_agent(memory_manager, compressor): - with patch.dict(os.environ, {"OPENROUTER_API_KEY": "x"}): - from run_agent import AIAgent + from run_agent import AIAgent - agent = AIAgent( - api_key="", - base_url="https://openrouter.ai/api/v1", - model="test/model", - quiet_mode=True, - session_db=None, - session_id="test-session", - skip_context_files=True, - skip_memory=True, - ) + agent = AIAgent( + api_key="test-key", + provider="openrouter", + api_mode="chat_completions", + base_url="https://openrouter.ai/api/v1", + model="test/model", + quiet_mode=True, + session_db=None, + session_id="test-session", + skip_context_files=True, + skip_memory=True, + ) agent._memory_manager = memory_manager agent.context_compressor = compressor