File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 2222
2323from google .adk .agents .base_agent import BaseAgent
2424from google .adk .agents .context_cache_config import ContextCacheConfig
25- from google .adk .agents .live_request_queue import LiveRequestQueue
2625from google .adk .agents .invocation_context import InvocationContext
2726from google .adk .agents .llm_agent import LlmAgent
2827from google .adk .agents .run_config import RunConfig
3736from google .adk .plugins .base_plugin import BasePlugin
3837from google .adk .runners import Runner
3938from google .adk .sessions .in_memory_session_service import InMemorySessionService
40- from tests .unittests import testing_utils
4139from google .adk .sessions .session import Session
4240from google .genai import types
4341import pytest
4442
43+ from tests .unittests import testing_utils
44+
4545TEST_APP_ID = "test_app"
4646TEST_USER_ID = "test_user"
4747TEST_SESSION_ID = "test_session"
@@ -1770,8 +1770,9 @@ def test_new_invocation_context_for_live_with_metadata(self):
17701770 )
17711771
17721772 test_metadata = {"user_id" : "live_user" , "trace_id" : "live_trace" }
1773+ live_queue = LiveRequestQueue ()
17731774 invocation_context = self .runner ._new_invocation_context_for_live (
1774- mock_session , metadata = test_metadata
1775+ mock_session , live_request_queue = live_queue , metadata = test_metadata
17751776 )
17761777
17771778 assert invocation_context .metadata == test_metadata
You can’t perform that action at this time.
0 commit comments