Update samples/src/main/java/com/google/ai/client/generative/samples/…

…function_calling.kt Co-authored-by: Daymon <[email protected]>
google-gemini · Jul 12, 2024 · 9156c8b · 9156c8b
1 parent ef62034
commit 9156c8b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/samples/src/main/java/com/google/ai/client/generative/samples/function_calling.kt b/samples/src/main/java/com/google/ai/client/generative/samples/function_calling.kt
@@ -46,14 +46,16 @@ suspend fun functionCalling() {
         )
     )
 
+    val usableFunctions = listOf(multiplyDefinition)
+
     val generativeModel =
         GenerativeModel(
             // Specify a Gemini model appropriate for your use case
             modelName = "gemini-1.5-flash",
             // Access your API key as a Build Configuration variable (see "Set up your API key" above)
             apiKey = BuildConfig.apiKey,
             // List the functions definitions you want to make available to the model
-            tools = listOf(Tool(listOf(multiplyDefinition)))
+            tools = listOf(Tool(usableFunctions))
         )
 
     val chat = generativeModel.startChat()