docs: update classification sample to use v2 model (#378)

* docs: update classification sample to use v2 model * fix: update classification test to use v2 model Co-authored-by: Anthonios Partheniou <[email protected]>
googleapis · Sep 27, 2022 · 73670e2 · 73670e2
1 parent 3ba5c56
commit 73670e2
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 8 deletions.
diff --git a/samples/v1/language_classify_text.py b/samples/v1/language_classify_text.py
@@ -32,12 +32,12 @@ def sample_classify_text(text_content):
     Classifying Content in a String
 
     Args:
-      text_content The text content to analyze. Must include at least 20 words.
+      text_content The text content to analyze.
     """
 
     client = language_v1.LanguageServiceClient()
 
-    # text_content = 'That actor on TV makes movies in Hollywood and also stars in a variety of popular new TV shows.'
+    # text_content = "That actor on TV makes movies in Hollywood and also stars in a variety of popular new TV shows."
 
     # Available types: PLAIN_TEXT, HTML
     type_ = language_v1.Document.Type.PLAIN_TEXT
@@ -48,7 +48,16 @@ def sample_classify_text(text_content):
     language = "en"
     document = {"content": text_content, "type_": type_, "language": language}
 
-    response = client.classify_text(request = {'document': document})
+    content_categories_version = (
+        language_v1.ClassificationModelOptions.V2Model.ContentCategoriesVersion.V2)
+    response = client.classify_text(request = {
+        "document": document,
+        "classification_model_options": {
+            "v2_model": {
+                "content_categories_version": content_categories_version
+            }
+        }
+    })
     # Loop through classified categories returned from the API
     for category in response.categories:
         # Get the name of the category representing the document.

diff --git a/samples/v1/test/classifying_content.test.yaml b/samples/v1/test/classifying_content.test.yaml
@@ -16,15 +16,15 @@ test:
 
     - name: language_classify_text - Classifying Content of a text string (*custom value*)
       spec:
-      # Custom value: "Let's drink coffee and eat bagels at a coffee shop. I want muffins, croisants, coffee and baked goods."
+      # Custom value: "Dungeons and dragons and loot, oh my!"
       - call:
           sample: language_classify_text
           params:
-            text_content: {literal: "Let's drink coffee and eat bagels at a coffee shop. I want muffins, croisants, coffee and baked goods."}
+            text_content: {literal: "Dungeons and dragons and loot, oh my!"}
       - assert_contains_any:
-        - {literal: "Food"}
-        - {literal: "Drink"}
-        - {literal: "Coffee"}
+        - {literal: "Games"}
+        - {literal: "Roleplaying"}
+        - {literal: "Computer"}
 
     - name: language_classify_gcs - Classifying Content of text file in GCS (default value)
       spec: