|
22 | 22 |
|
23 | 23 | from google.adk.agents.base_agent import BaseAgent |
24 | 24 | from google.adk.agents.context_cache_config import ContextCacheConfig |
25 | | -from google.adk.agents.live_request_queue import LiveRequestQueue |
26 | 25 | from google.adk.agents.invocation_context import InvocationContext |
27 | 26 | from google.adk.agents.live_request_queue import LiveRequestQueue |
28 | 27 | from google.adk.agents.llm_agent import LlmAgent |
|
37 | 36 | from google.adk.plugins.base_plugin import BasePlugin |
38 | 37 | from google.adk.runners import Runner |
39 | 38 | from google.adk.sessions.in_memory_session_service import InMemorySessionService |
40 | | -from tests.unittests import testing_utils |
41 | 39 | from google.adk.sessions.session import Session |
42 | 40 | from google.adk.tools.function_tool import FunctionTool |
43 | 41 | from google.genai import types |
44 | 42 | import pytest |
45 | 43 |
|
| 44 | +from tests.unittests import testing_utils |
| 45 | + |
46 | 46 | TEST_APP_ID = "test_app" |
47 | 47 | TEST_USER_ID = "test_user" |
48 | 48 | TEST_SESSION_ID = "test_session" |
@@ -1579,8 +1579,9 @@ def test_new_invocation_context_for_live_with_metadata(self): |
1579 | 1579 | ) |
1580 | 1580 |
|
1581 | 1581 | test_metadata = {"user_id": "live_user", "trace_id": "live_trace"} |
| 1582 | + live_queue = LiveRequestQueue() |
1582 | 1583 | invocation_context = self.runner._new_invocation_context_for_live( |
1583 | | - mock_session, metadata=test_metadata |
| 1584 | + mock_session, live_request_queue=live_queue, metadata=test_metadata |
1584 | 1585 | ) |
1585 | 1586 |
|
1586 | 1587 | assert invocation_context.metadata == test_metadata |
|
0 commit comments