diff --git a/samples/snippets/create_agent.py b/samples/snippets/create_agent.py new file mode 100644 index 00000000..3dfb68da --- /dev/null +++ b/samples/snippets/create_agent.py @@ -0,0 +1,36 @@ +# Copyright 2021 Google LLC +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + + +"""DialogFlow API Create Agent Sample""" + +from google.cloud.dialogflowcx_v3.services.agents.client import AgentsClient +from google.cloud.dialogflowcx_v3.types.agent import Agent + + +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 diff --git a/samples/snippets/create_agent_test.py b/samples/snippets/create_agent_test.py new file mode 100644 index 00000000..320a25cb --- /dev/null +++ b/samples/snippets/create_agent_test.py @@ -0,0 +1,42 @@ +# Copyright 2020, Google LLC +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +"""Test for create_agent""" + +from datetime import date +import os + +from google.cloud.dialogflowcx_v3.services.agents.client import AgentsClient +from google.cloud.dialogflowcx_v3.types.agent import DeleteAgentRequest + +import pytest + +from create_agent import create_agent + +PROJECT_ID = os.getenv("GOOGLE_CLOUD_PROJECT") +pytest.AGENT_PATH = "" + + +def delete_agent(name): + agents_client = AgentsClient() + request = DeleteAgentRequest(name=name) + agents_client.delete_agent(request=request) + + +def test_create_agent(): + today = date.today() + agentName = "tempAgent." + today.strftime("%d.%m.%Y") + response = create_agent(PROJECT_ID, agentName) + delete_agent(response.name) + + assert response.display_name == agentName