NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-12-18 11:22:22 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: Use sync request method in VertexAiSessionService. The api_client has it own event loop management.

Browse Source 
PiperOrigin-RevId: 761250268
This commit is contained in:
Shangjie Chen
2025-05-20 15:19:47 -07:00
committed by Copybara-Service
parent b299241b56
commit 53b14325ce
2 changed files with 7 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/unittests/sessions/test_vertex_ai_session_service.py
View File

@@ -125,22 +125,6 @@ class MockApiClient:
this.event_dict: dict[str, list[Any]] = {}
def request(self, http_method: str, path: str, request_dict: dict[str, Any]):
"""Mocks the API Client request method."""
if http_method == 'GET':
if re.match(SESSIONS_REGEX, path):
match = re.match(SESSIONS_REGEX, path)
return {
'sessions': [
session
for session in self.session_dict.values()
if session['userId'] == match.group(2)
],
}
raise ValueError(f'Unsupported sync path: {path}')
async def async_request(
self, http_method: str, path: str, request_dict: dict[str, Any]
):
"""Mocks the API Client request method."""
if http_method == 'GET':
if re.match(SESSION_REGEX, path):