Skip to content
This repository has been archived by the owner on Jan 6, 2024. It is now read-only.

Commit

Permalink
docs: add detect intent with sentiment analysis snippet
Browse files Browse the repository at this point in the history
  • Loading branch information
nicain committed Jul 1, 2022
1 parent 8954254 commit 87ab6f6
Show file tree
Hide file tree
Showing 3 changed files with 134 additions and 2 deletions.
4 changes: 2 additions & 2 deletions .kokoro/test-samples-impl.sh
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ export PYTHONUNBUFFERED=1
env | grep KOKORO

# Install nox
python3.6 -m pip install --upgrade --quiet nox
python3.9 -m pip install --upgrade --quiet nox

# Use secrets acessor service account to get secrets
if [[ -f "${KOKORO_GFILE_DIR}/secrets_viewer_service_account.json" ]]; then
Expand Down Expand Up @@ -76,7 +76,7 @@ for file in samples/**/requirements.txt; do
echo "------------------------------------------------------------"

# Use nox to execute the tests for the project.
python3.6 -m nox -s "$RUN_TESTS_SESSION"
python3.9 -m nox -s "$RUN_TESTS_SESSION"
EXIT=$?

# If this is a periodic build, send the test log to the FlakyBot.
Expand Down
91 changes: 91 additions & 0 deletions samples/snippets/detect_intent_with_sentiment_analysis.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,91 @@
#!/usr/bin/env python

# Copyright 2022 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 Detect Intent Python sample, text input and sentiment analysis."""


# [START dialogflow_cx_v3_detect_intent_sentiment_analysis_async]
import uuid

from google.cloud.dialogflowcx_v3beta1.services.sessions import SessionsClient
from google.cloud.dialogflowcx_v3beta1.types import session


def run_sample():
# TODO(developer): Update these values when running the function
project_id = "YOUR-PROJECT-ID"
location = "YOUR-LOCATION-ID"
agent_id = "YOUR-AGENT-ID"
text = "Perfect!"
language_code = "en-us"

detect_intent_with_sentiment_analysis(
project_id,
location,
agent_id,
text,
language_code,
)


def detect_intent_with_sentiment_analysis(
project_id,
location,
agent_id,
text,
language_code,
):
"""Returns the result of detect intent with sentiment analysis"""

client_options = None
if location != "global":
api_endpoint = f"{location}-dialogflow.googleapis.com:443"
print(f"API Endpoint: {api_endpoint}\n")
client_options = {"api_endpoint": api_endpoint}
session_client = SessionsClient(client_options=client_options)
session_id = str(uuid.uuid4())

session_path = session_client.session_path(
project=project_id,
location=location,
agent=agent_id,
session=session_id,
)

text_input = session.TextInput(text=text)
query_input = session.QueryInput(
text=text_input,
language_code=language_code
)
query_params = session.QueryParameters(
analyze_query_text_sentiment=True,
)
request = session.DetectIntentRequest(
session=session_path,
query_input=query_input,
query_params=query_params,
)

response = session_client.detect_intent(request=request)
score = response.query_result.sentiment_analysis_result.score
print('Sentiment Score: {score}')
return score

# [END dialogflow_cx_v3_detect_intent_sentiment_analysis_async]


if __name__ == "__main__":
run_sample()
41 changes: 41 additions & 0 deletions samples/snippets/detect_intent_with_sentiment_analysis_test.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
# Copyright 2022, 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.

"""Tests for detect_intent_with_sentiment_analysis.py"""

from __future__ import absolute_import

import os

from detect_intent_with_sentiment_analysis import detect_intent_with_sentiment_analysis
import pytest


PROJECT_ID = os.getenv("GOOGLE_CLOUD_PROJECT")
AGENT_ID = os.getenv("AGENT_ID")


@pytest.mark.parametrize(
"text, expected_score_min, expected_score_max",
(['Perfect', .5, 1], ['I am not happy', -1, -.5])
)
def test_detect_intent_positive(text, expected_score_min, expected_score_max):

score = detect_intent_with_sentiment_analysis(
PROJECT_ID,
'global',
AGENT_ID,
text,
"en-us",
)
assert expected_score_min < score < expected_score_max

0 comments on commit 87ab6f6

Please sign in to comment.