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
def export_long_running_agent(project_id, agent_id, location):
api_endpoint = f"{location}-dialogflow.googleapis.com:443"
client_options = {"api_endpoint": api_endpoint}
agents_client = AgentsClient(client_options=client_options)
export_request = ExportAgentRequest()
export_request.name = (
f"projects/{project_id}/locations/{location}/agents/{agent_id}"
)
# export_agent returns a long running operation
operation = agents_client.export_agent(request=export_request)
# Returns the result of the operation when the operation is done
return operation.result()
def delete_agent(name):
agents_client = AgentsClient()
agent = DeleteAgentRequest(name=name)
agents_client.delete_agent(request=agent)