diff --git a/contact-center-insights/snippets/create_issue_model.py b/contact-center-insights/snippets/create_issue_model.py new file mode 100644 index 000000000000..4568cc43e583 --- /dev/null +++ b/contact-center-insights/snippets/create_issue_model.py @@ -0,0 +1,41 @@ +# 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. +# +# [START contactcenterinsights_create_issue_model] +from google.cloud import contact_center_insights_v1 + + +def create_issue_model(project_id: str) -> contact_center_insights_v1.IssueModel: + # Construct a parent resource. + parent = contact_center_insights_v1.ContactCenterInsightsClient.common_location_path( + project_id, "us-central1" + ) + + # Construct an issue model. + issue_model = contact_center_insights_v1.IssueModel() + issue_model.display_name = "my-model" + issue_model.input_data_config.filter = "medium=\"CHAT\"" + + # Call the Insights client to create an issue model. + insights_client = contact_center_insights_v1.ContactCenterInsightsClient() + issue_model_operation = insights_client.create_issue_model( + parent=parent, issue_model=issue_model + ) + + issue_model = issue_model_operation.result(timeout=86400) + print(f"Created an issue model named {issue_model.name}") + return issue_model + + +# [END contactcenterinsights_create_issue_model] diff --git a/contact-center-insights/snippets/test_create_issue_model.py b/contact-center-insights/snippets/test_create_issue_model.py new file mode 100644 index 000000000000..ffa60e4f3b72 --- /dev/null +++ b/contact-center-insights/snippets/test_create_issue_model.py @@ -0,0 +1,70 @@ +# 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. +# +import google.auth + +from google.cloud import contact_center_insights_v1 + +import pytest + +import create_issue_model + +MIN_CONVERSATION_COUNT = 10000 + + +@pytest.fixture +def project_id(): + _, project_id = google.auth.default() + return project_id + + +@pytest.fixture +def insights_client(): + return contact_center_insights_v1.ContactCenterInsightsClient() + + +@pytest.fixture +def count_conversations(project_id, insights_client): + # Check if the project has the minimum number of conversations required to create an issue model. + # See https://cloud.google.com/contact-center/insights/docs/topic-model. + list_request = contact_center_insights_v1.ListConversationsRequest() + list_request.page_size = 1000 + list_request.parent = contact_center_insights_v1.ContactCenterInsightsClient.common_location_path( + project_id, "us-central1" + ) + conversations = insights_client.list_conversations(request=list_request) + conversation_count = len(list(conversations)) + + yield conversation_count + + +@pytest.fixture +def issue_model_resource(project_id, insights_client, count_conversations): + conversation_count = count_conversations + if conversation_count >= MIN_CONVERSATION_COUNT: + # Create an issue model. + issue_model = create_issue_model.create_issue_model(project_id) + yield issue_model + + # Delete the issue model. + insights_client.delete_issue_model(name=issue_model.name) + else: + yield None + + +def test_create_issue_model(capsys, issue_model_resource): + issue_model = issue_model_resource + if issue_model: + out, err = capsys.readouterr() + assert "Created {}".format(issue_model.name) in out