From d13866688139601f5b9f3cf46af9c3bc9db5bfdc Mon Sep 17 00:00:00 2001 From: David Motsonashvili Date: Wed, 10 Jul 2024 13:14:48 -0700 Subject: [PATCH] add code execution samples --- .../generative/samples/code_execution.kt | 63 +++++++++ .../samples/java/code_execution.java | 123 ++++++++++++++++++ 2 files changed, 186 insertions(+) create mode 100644 samples/src/main/java/com/google/ai/client/generative/samples/code_execution.kt create mode 100644 samples/src/main/java/com/google/ai/client/generative/samples/java/code_execution.java diff --git a/samples/src/main/java/com/google/ai/client/generative/samples/code_execution.kt b/samples/src/main/java/com/google/ai/client/generative/samples/code_execution.kt new file mode 100644 index 00000000..7fcbe1e0 --- /dev/null +++ b/samples/src/main/java/com/google/ai/client/generative/samples/code_execution.kt @@ -0,0 +1,63 @@ +/* + * Copyright 2024 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. + */ + +package com.google.ai.client.generative.samples + +import com.google.ai.client.generativeai.GenerativeModel +import com.google.ai.client.generativeai.type.Tool + +// Set up your API Key +// ==================== +// +// To use the Gemini API, you'll need an API key. To learn more, see +// the "Set up your API Key section" in the [Gemini API +// quickstart](https://ai.google.dev/gemini-api/docs/quickstart?lang=android#set-up-api-key). + +suspend fun codeExecutionBasic() { + // [START code_execution_basic] + + val model = GenerativeModel( + // Specify a Gemini model appropriate for your use case + modelName = "gemini-1.5-pro", + // Access your API key as a Build Configuration variable (see "Set up your API key" above) + apiKey = BuildConfig.apiKey, + tools = listOf(Tool.CODE_EXECUTION) + ) + + val response = model.generateContent("What is the sum of the first 50 prime numbers?") + + println(response.text) + // [END code_execution_basic] +} + +suspend fun codeExecutionChat() { + // [START code_execution_chat] + + val model = GenerativeModel( + // Specify a Gemini model appropriate for your use case + modelName = "gemini-1.5-pro", + // Access your API key as a Build Configuration variable (see "Set up your API key" above) + apiKey = BuildConfig.apiKey, + tools = listOf(Tool.CODE_EXECUTION) + ) + + val chat = model.startChat() + + val response = chat.sendMessage("What is the sum of the first 50 prime numbers?") + + println(response.text) + // [END code_execution_chat] +} \ No newline at end of file diff --git a/samples/src/main/java/com/google/ai/client/generative/samples/java/code_execution.java b/samples/src/main/java/com/google/ai/client/generative/samples/java/code_execution.java new file mode 100644 index 00000000..9bc2a5cc --- /dev/null +++ b/samples/src/main/java/com/google/ai/client/generative/samples/java/code_execution.java @@ -0,0 +1,123 @@ +// Copyright 2024 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. + +package com.google.ai.client.generative.samples.java; + + +import com.google.ai.client.generativeai.GenerativeModel; +import com.google.ai.client.generativeai.java.ChatFutures; +import com.google.ai.client.generativeai.java.GenerativeModelFutures; +import com.google.ai.client.generativeai.type.Content; +import com.google.ai.client.generativeai.type.GenerateContentResponse; +import com.google.ai.client.generativeai.type.RequestOptions; +import com.google.ai.client.generativeai.type.Tool; +import com.google.common.util.concurrent.FutureCallback; +import com.google.common.util.concurrent.Futures; +import com.google.common.util.concurrent.ListenableFuture; + +import java.util.Arrays; +import java.util.Collections; +import java.util.concurrent.Executor; +import java.util.concurrent.Executors; + +// Set up your API Key +// ==================== +// +// To use the Gemini API, you'll need an API key. To learn more, see +// the "Set up your API Key section" in the [Gemini API +// quickstart](https://ai.google.dev/gemini-api/docs/quickstart?lang=android#set-up-api-key). +class CodeExecution { + + void codeExecutionBasic() { + // [START code_execution_basic] + // Specify a Gemini model appropriate for your use case + GenerativeModel gm = + new GenerativeModel( + /* modelName */ "gemini-1.5-pro", + // Access your API key as a Build Configuration variable (see "Set up your API key" + // above) + /* apiKey */ BuildConfig.apiKey, + /* generationConfig */ null, + /* safetySettings */ null, + /* requestOptions */ new RequestOptions(), + /* tools */ Collections.singletonList(Tool.Companion.getCODE_EXECUTION())); + GenerativeModelFutures model = GenerativeModelFutures.from(gm); + + Content inputContent = + new Content.Builder().addText("What is the sum of the first 50 prime numbers?").build(); + + // For illustrative purposes only. You should use an executor that fits your needs. + Executor executor = Executors.newSingleThreadExecutor(); + + ListenableFuture response = model.generateContent(inputContent); + Futures.addCallback( + response, + new FutureCallback() { + @Override + public void onSuccess(GenerateContentResponse result) { + String resultText = result.getText(); + System.out.println(resultText); + } + + @Override + public void onFailure(Throwable t) { + t.printStackTrace(); + } + }, + executor); + // [END code_execution_basic] + } + + void codeExecutionChat() { + // [START code_execution_chat] + // Specify a Gemini model appropriate for your use case + GenerativeModel gm = + new GenerativeModel( + /* modelName */ "gemini-1.5-pro", + // Access your API key as a Build Configuration variable (see "Set up your API key" + // above) + /* apiKey */ BuildConfig.apiKey, + /* generationConfig */ null, + /* safetySettings */ null, + /* requestOptions */ new RequestOptions(), + /* tools */ Collections.singletonList(Tool.Companion.getCODE_EXECUTION())); + GenerativeModelFutures model = GenerativeModelFutures.from(gm); + + Content inputContent = + new Content.Builder().addText("What is the sum of the first 50 prime numbers?").build(); + + ChatFutures chat = model.startChat(); + + // For illustrative purposes only. You should use an executor that fits your needs. + Executor executor = Executors.newSingleThreadExecutor(); + + ListenableFuture response = chat.sendMessage(inputContent); + Futures.addCallback( + response, + new FutureCallback() { + @Override + public void onSuccess(GenerateContentResponse result) { + String resultText = result.getText(); + System.out.println(resultText); + } + + @Override + public void onFailure(Throwable t) { + t.printStackTrace(); + } + }, + executor); + // [END code_execution_chat] + } +}