NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-13 15:14:50 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: Fixes a few samples for the breaking session service change.

Browse Source 
PiperOrigin-RevId: 759354632
This commit is contained in:
Wei Sun (Jack) 2025-05-15 16:43:47 -07:00 committed by Copybara-Service
parent 509db3f9fb
commit 04820cb0a7
2 changed files with 2 additions and 50 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
contributing/samples/hello_world/asyncio_run.py → contributing/samples/hello_world/main.py
View File

@ -14,7 +14,6 @@
import asyncio import asyncio
import time import time
import warnings
import agent import agent
from dotenv import load_dotenv from dotenv import load_dotenv
@ -27,7 +26,6 @@ from google.adk.sessions import Session
from google.genai import types from google.genai import types
load_dotenv(override=True) load_dotenv(override=True)
warnings.filterwarnings('ignore', category=UserWarning)
logs.log_to_tmp_folder() logs.log_to_tmp_folder()
@ -42,7 +40,7 @@ async def main():
artifact_service=artifact_service, artifact_service=artifact_service,
session_service=session_service, session_service=session_service,
) )
session_11 = session_service.create_session( session_11 = await session_service.create_session(
app_name=app_name, user_id=user_id_1 app_name=app_name, user_id=user_id_1
) )
@ -97,49 +95,5 @@ async def main():
print('Total time:', end_time - start_time) print('Total time:', end_time - start_time)
def main_sync():
app_name = 'my_app'
user_id_1 = 'user1'
session_service = InMemorySessionService()
artifact_service = InMemoryArtifactService()
runner = Runner(
app_name=app_name,
agent=agent.root_agent,
artifact_service=artifact_service,
session_service=session_service,
)
session_11 = session_service.create_session(
app_name=app_name, user_id=user_id_1
)
def run_prompt(session: Session, new_message: str):
content = types.Content(
role='user', parts=[types.Part.from_text(text=new_message)]
)
print('** User says:', content.model_dump(exclude_none=True))
for event in runner.run(
user_id=user_id_1,
session_id=session.id,
new_message=content,
):
if event.content.parts and event.content.parts[0].text:
print(f'** {event.author}: {event.content.parts[0].text}')
start_time = time.time()
print('Start time:', start_time)
print('------------------------------------')
run_prompt(session_11, 'Hi')
run_prompt(session_11, 'Roll a die with 100 sides.')
run_prompt(session_11, 'Roll a die again with 100 sides.')
run_prompt(session_11, 'What numbers did I got?')
end_time = time.time()
print('------------------------------------')
print('End time:', end_time)
print('Total time:', end_time - start_time)
if __name__ == '__main__': if __name__ == '__main__':
print('--------------ASYNC--------------------')
asyncio.run(main()) asyncio.run(main())
print('--------------SYNC--------------------')
main_sync()

4
contributing/samples/hello_world_litellm/asyncio_run.py → contributing/samples/hello_world_litellm/main.py
View File

@ -15,7 +15,6 @@
import asyncio import asyncio
import time import time
import warnings
import agent import agent
from dotenv import load_dotenv from dotenv import load_dotenv
@ -27,7 +26,6 @@ from google.adk.sessions import Session
from google.genai import types from google.genai import types
load_dotenv(override=True) load_dotenv(override=True)
warnings.filterwarnings('ignore', category=UserWarning)
logs.log_to_tmp_folder() logs.log_to_tmp_folder()
@ -42,7 +40,7 @@ async def main():
artifact_service=artifact_service, artifact_service=artifact_service,
session_service=session_service, session_service=session_service,
) )
session_11 = session_service.create_session( session_11 = await session_service.create_session(
app_name=app_name, user_id=user_id_1 app_name=app_name, user_id=user_id_1
) )