Skip to content
This repository was archived by the owner on Mar 6, 2026. It is now read-only.

Commit 54921b8

Browse files
committed
chore(quickstart): added token confidence and detected languages output to quickstart sample
1 parent 678b2d1 commit 54921b8

File tree

2 files changed

+18
-0
lines changed

2 files changed

+18
-0
lines changed

samples/snippets/quickstart_sample.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,16 @@ def quickstart_sample(
105105
print(line.text)
106106
for token in page.tokens:
107107
print(token.text)
108+
# Print token confidence
109+
print(f"\tConfidence: {token.confidence:.4f}")
110+
# Print detected languages
111+
if token.detected_languages:
112+
print("\tDetected Languages:")
113+
for lang in token.detected_languages:
114+
confidence_str = f", confidence: {lang.confidence:.4f}" if hasattr(lang, "confidence") else ""
115+
print(f"\t\t- {lang.language_code}{confidence_str}")
116+
else:
117+
print("\tNo language detected")
108118

109119
# Only supported with Form Parser processor
110120
# https://cloud.google.com/document-ai/docs/form-parser

samples/snippets/test_quickstart_sample.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,8 @@ def test_quickstart_sample_gcs_bucket_prefix(capsys: pytest.CaptureFixture) -> N
3737
assert "Document structure in Cloud Storage" in out
3838
assert "Number of Pages: 1" in out
3939
assert "Number of Entities: 35" in out
40+
assert "Confidence:" in out
41+
assert "Detected Languages:" in out or "No language detected" in out
4042

4143

4244
def test_quickstart_sample_gcs_uri(capsys: pytest.CaptureFixture) -> None:
@@ -48,6 +50,8 @@ def test_quickstart_sample_gcs_uri(capsys: pytest.CaptureFixture) -> None:
4850

4951
assert "Number of Pages: 1" in out
5052
assert "Number of Entities: 35" in out
53+
assert "Confidence:" in out
54+
assert "Detected Languages:" in out or "No language detected" in out
5155

5256

5357
def test_quickstart_sample_document_path(capsys: pytest.CaptureFixture) -> None:
@@ -58,6 +62,8 @@ def test_quickstart_sample_document_path(capsys: pytest.CaptureFixture) -> None:
5862
assert "Number of Pages: 1" in out
5963
assert "Number of Entities: 0" in out
6064
assert "Form Date" in out
65+
assert "Confidence:" in out
66+
assert "Detected Languages:" in out or "No language detected" in out
6167

6268

6369
def test_quickstart_sample_documentai_document(capsys: pytest.CaptureFixture) -> None:
@@ -72,6 +78,8 @@ def test_quickstart_sample_documentai_document(capsys: pytest.CaptureFixture) ->
7278
assert "Number of Pages: 1" in out
7379
assert "Number of Entities: 0" in out
7480
assert "Form Date" in out
81+
assert "Confidence:" in out
82+
assert "Detected Languages:" in out or "No language detected" in out
7583

7684

7785
def test_quickstart_sample_batch_process_metadata(

0 commit comments

Comments
 (0)