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)