genai: add error check to example (#227)

Fixes #225.
google · Nov 18, 2024 · f453d2d · f453d2d
1 parent 30b5de5
commit f453d2d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/genai/example_test.go b/genai/example_test.go
@@ -586,7 +586,7 @@ func ExampleGenerativeModel_CountTokens_tools() {
 	// ( total_tokens: 23 )
 
 	tools := []*genai.Tool{
-		&genai.Tool{FunctionDeclarations: []*genai.FunctionDeclaration{
+		{FunctionDeclarations: []*genai.FunctionDeclaration{
 			{Name: "add"},
 			{Name: "subtract"},
 			{Name: "multiply"},
@@ -1188,6 +1188,9 @@ func ExampleTool() {
 			"theater": "AMC16",
 		},
 	})
+	if err != nil {
+		log.Fatal(err)
+	}
 	printResponse(res)
 }