NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-16 04:02:55 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: reformat codes using autoformat.sh

Browse Source 
PiperOrigin-RevId: 764584699
This commit is contained in:
Xiang (Sean) Zhou 2025-05-28 23:25:04 -07:00 committed by Copybara-Service
parent 2b5c89b3a9
commit 0250d9e3ac
3 changed files with 69 additions and 65 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/google/adk/sessions/vertex_ai_session_service.py
View File

@ -188,10 +188,10 @@ class VertexAiSessionService(BaseSessionService):
) -> ListSessionsResponse:
reasoning_engine_id = _parse_reasoning_engine_id(app_name)
path = f"reasoningEngines/{reasoning_engine_id}/sessions"
path = f'reasoningEngines/{reasoning_engine_id}/sessions'
if user_id:
parsed_user_id = urllib.parse.quote(f'''"{user_id}"''', safe="")
path = path + f"?filter=user_id={parsed_user_id}"
parsed_user_id = urllib.parse.quote(f'''"{user_id}"''', safe='')
path = path + f'?filter=user_id={parsed_user_id}'
api_client = _get_api_client(self.project, self.location)
api_response = await api_client.async_request(

2
tests/unittests/models/test_google_llm.py
View File

@ -211,7 +211,9 @@ async def test_generate_content_async_stream_preserves_thinking_and_text_parts(
gemini_llm, llm_request
):
with mock.patch.object(gemini_llm, "api_client") as mock_client:
class MockAsyncIterator:
def __init__(self, seq):
self._iter = iter(seq)

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

@ -111,7 +111,9 @@ MOCK_SESSION = Session(
SESSION_REGEX = r'^reasoningEngines/([^/]+)/sessions/([^/]+)$'
SESSIONS_REGEX = r'^reasoningEngines/([^/]+)/sessions\?filter=user_id=%22([^%]+)%22.*$' # %22 represents double-quotes in a URL-encoded string
SESSIONS_REGEX = ( # %22 represents double-quotes in a URL-encoded string
r'^reasoningEngines/([^/]+)/sessions\?filter=user_id=%22([^%]+)%22.*$'
)
EVENTS_REGEX = r'^reasoningEngines/([^/]+)/sessions/([^/]+)/events$'
LRO_REGEX = r'^operations/([^/]+)$'
@ -156,7 +158,7 @@ class MockApiClient:
return {
'name': path,
'done': True,
'response': self.session_dict['4'] # Return the created session
'response': self.session_dict['4'], # Return the created session
}
else:
raise ValueError(f'Unsupported path: {path}')