def setup_teardown():
loop = asyncio.new_event_loop()
agentName = "temp_agent_" + str(uuid.uuid4())
parent = "projects/" + PROJECT_ID + "/locations/global"
agents_client = AgentsClient()
agent = Agent(
display_name=agentName,
default_language_code="en",
time_zone="America/Los_Angeles",
)
response = agents_client.create_agent(request={
"agent": agent,
"parent": parent
})
pytest.PARENT = response.name
pytest.AGENT_ID = pytest.PARENT.split("/")[5]
print("Created Agent in setUp")
yield
delete_agent(pytest.PARENT)
loop.close()
def create_agent(project_id, display_name):
parent = "projects/" + project_id + "/locations/global"
agents_client = AgentsClient()
agent = Agent(
display_name=display_name,
default_language_code="en",
time_zone="America/Los_Angeles",
)
response = agents_client.create_agent(request={
"agent": agent,
"parent": parent
})
return response