From 272fc9879e536f7a9ea31d8c10169cf644170769 Mon Sep 17 00:00:00 2001
From: Gal Zahavi <38544478+galz10@users.noreply.github.com>
Date: Thu, 12 Aug 2021 11:16:06 -0700
Subject: [PATCH] docs(samples): add agent creation code snippet (#146)

* add agent creation code snippet

* update test

* Updated test

* moved delete agent

* lint fix

* lint fix

* Fixed Copyright
---
 samples/snippets/create_agent.py      | 36 +++++++++++++++++++++++
 samples/snippets/create_agent_test.py | 42 +++++++++++++++++++++++++++
 2 files changed, 78 insertions(+)
 create mode 100644 samples/snippets/create_agent.py
 create mode 100644 samples/snippets/create_agent_test.py

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