diff --git a/contact-center-insights/snippets/create_conversation_with_ttl.py b/contact-center-insights/snippets/create_conversation_with_ttl.py new file mode 100644 index 000000000000..267c71d4d65a --- /dev/null +++ b/contact-center-insights/snippets/create_conversation_with_ttl.py @@ -0,0 +1,52 @@ +# 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. +# +# Create a conversation with a TTL. +# [START contactcenterinsights_create_conversation_with_ttl] +from google.cloud import contact_center_insights_v1 +from google.protobuf import duration_pb2 + + +def create_conversation_with_ttl( + project_id: str, + transcript_uri: str = "gs://cloud-samples-data/ccai/chat_sample.json", + audio_uri: str = "gs://cloud-samples-data/ccai/voice_6912.txt", +) -> contact_center_insights_v1.Conversation: + # Construct a parent resource. + parent = contact_center_insights_v1.ContactCenterInsightsClient.common_location_path( + project_id, "us-central1" + ) + + # Construct a conversation. + conversation = contact_center_insights_v1.Conversation() + conversation.data_source.gcs_source.transcript_uri = transcript_uri + conversation.data_source.gcs_source.audio_uri = audio_uri + conversation.medium = contact_center_insights_v1.Conversation.Medium.CHAT + + # Construct a TTL. + ttl = duration_pb2.Duration() + ttl.seconds = 86400 + conversation.ttl = ttl + + # Call the Insights client to create a conversation. + insights_client = contact_center_insights_v1.ContactCenterInsightsClient() + conversation = insights_client.create_conversation( + parent=parent, conversation=conversation + ) + + print(f"Created {conversation.name}") + return conversation + + +# [END contactcenterinsights_create_conversation_with_ttl] diff --git a/contact-center-insights/snippets/create_issue_model.py b/contact-center-insights/snippets/create_issue_model.py index 4568cc43e583..b34033593e77 100644 --- a/contact-center-insights/snippets/create_issue_model.py +++ b/contact-center-insights/snippets/create_issue_model.py @@ -25,7 +25,7 @@ def create_issue_model(project_id: str) -> contact_center_insights_v1.IssueModel # 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\"" + issue_model.input_data_config.filter = 'medium="CHAT"' # Call the Insights client to create an issue model. insights_client = contact_center_insights_v1.ContactCenterInsightsClient() diff --git a/contact-center-insights/snippets/noxfile.py b/contact-center-insights/snippets/noxfile.py index b008613f03ff..0e13f5f15f27 100644 --- a/contact-center-insights/snippets/noxfile.py +++ b/contact-center-insights/snippets/noxfile.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# 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. diff --git a/contact-center-insights/snippets/set_project_ttl.py b/contact-center-insights/snippets/set_project_ttl.py index f3a98fe0ccb7..b5aa02107875 100644 --- a/contact-center-insights/snippets/set_project_ttl.py +++ b/contact-center-insights/snippets/set_project_ttl.py @@ -50,4 +50,5 @@ def set_project_ttl(project_id: str) -> None: ) ) + # [END contactcenterinsights_set_project_ttl] diff --git a/contact-center-insights/snippets/test_create_conversation_with_ttl.py b/contact-center-insights/snippets/test_create_conversation_with_ttl.py new file mode 100644 index 000000000000..ee493d4e3dc3 --- /dev/null +++ b/contact-center-insights/snippets/test_create_conversation_with_ttl.py @@ -0,0 +1,44 @@ +# 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_conversation_with_ttl + + +@pytest.fixture +def project_id(): + _, project_id = google.auth.default() + return project_id + + +@pytest.fixture +def conversation_resource(project_id): + # Create a conversation + conversation = create_conversation_with_ttl.create_conversation_with_ttl(project_id) + yield conversation + + # Delete the conversation. + insights_client = contact_center_insights_v1.ContactCenterInsightsClient() + insights_client.delete_conversation(name=conversation.name) + + +def test_create_conversation_with_ttl(capsys, conversation_resource): + conversation = conversation_resource + out, err = capsys.readouterr() + assert "Created {}".format(conversation.name) in out