Skip to content

Commit 2bb543a

Browse files
committed
fix: resolve isort duplicate import and add missing live_request_queue argument in test
- Remove duplicate LiveRequestQueue import caused by merge conflict - Fix import ordering for tests.unittests module - Add required live_request_queue argument to test_new_invocation_context_for_live_with_metadata test
1 parent d34a225 commit 2bb543a

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tests/unittests/test_runners.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@
2222

2323
from google.adk.agents.base_agent import BaseAgent
2424
from google.adk.agents.context_cache_config import ContextCacheConfig
25-
from google.adk.agents.live_request_queue import LiveRequestQueue
2625
from google.adk.agents.invocation_context import InvocationContext
2726
from google.adk.agents.llm_agent import LlmAgent
2827
from google.adk.agents.run_config import RunConfig
@@ -37,11 +36,12 @@
3736
from google.adk.plugins.base_plugin import BasePlugin
3837
from google.adk.runners import Runner
3938
from google.adk.sessions.in_memory_session_service import InMemorySessionService
40-
from tests.unittests import testing_utils
4139
from google.adk.sessions.session import Session
4240
from google.genai import types
4341
import pytest
4442

43+
from tests.unittests import testing_utils
44+
4545
TEST_APP_ID = "test_app"
4646
TEST_USER_ID = "test_user"
4747
TEST_SESSION_ID = "test_session"
@@ -1496,8 +1496,9 @@ def test_new_invocation_context_for_live_with_metadata(self):
14961496
)
14971497

14981498
test_metadata = {"user_id": "live_user", "trace_id": "live_trace"}
1499+
live_queue = LiveRequestQueue()
14991500
invocation_context = self.runner._new_invocation_context_for_live(
1500-
mock_session, metadata=test_metadata
1501+
mock_session, live_request_queue=live_queue, metadata=test_metadata
15011502
)
15021503

15031504
assert invocation_context.metadata == test_metadata

0 commit comments

Comments
 (0)