added completions with Google search; updsated to model gemini-3-flash-preview

Author: mark-watson

gemini-llm-client/src/main/java/com/markwatson/gemini/GeminiCompletions.java

@@ -9,31 +9,47 @@
 import java.net.HttpURLConnection;
 import java.net.URI;
 import java.net.URL;
-import java.net.URLConnection;
+//import java.net.URLConnection;
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
 
 public class GeminiCompletions {
 
+    public static String model = "gemini-3-flash-preview";
+
     public static void main(String[] args) throws Exception {
         String prompt = "How much is 11 + 22?";
         String completion = getCompletion(prompt);
         System.out.println("completion: " + completion);
     }
 
     public static String getCompletion(String prompt) throws Exception {
+        String jsonBody = "{\"contents\":[{\"parts\":[{\"text\":\"" + prompt + "\"}]}]}";
+        return executeRequest(jsonBody);
+    }
+
+    public static String getCompletionWithSearch(String prompt) throws Exception {
+        String jsonBody = "{\"contents\":[{\"parts\":[{\"text\":\"" + prompt
+                + "\"}]}],\"tools\":[{\"google_search\":{}}]}";
+        return executeRequest(jsonBody);
+    }
+
+    private static String executeRequest(String jsonBody) throws Exception {
         String apiKey = System.getenv("GOOGLE_API_KEY");
         if (apiKey == null || apiKey.isEmpty()) {
             throw new IOException("GOOGLE_API_KEY environment variable not set.");
         }
-        String model = "gemini-2.5-flash";
-        URI uri = new URI("https://generativelanguage.googleapis.com/v1beta/models/" + model + ":generateContent?key=" + apiKey);
+        URI uri = new URI(
+                "https://generativelanguage.googleapis.com/v1beta/models/" + model + ":generateContent?key=" + apiKey);
         URL url = uri.toURL();
 
         HttpURLConnection connection = (HttpURLConnection) url.openConnection();
         connection.setRequestMethod("POST");
-        connection.setRequestProperty("Content-Type", "application/json");        connection.setDoOutput(true);        String jsonBody = "{\"contents\":[{\"parts\":[{\"text\":\"" + prompt + "\"}]}]}";        try (OutputStream os = connection.getOutputStream()) {            byte[] input = jsonBody.getBytes("utf-8");
+        connection.setRequestProperty("Content-Type", "application/json");
+        connection.setDoOutput(true);
+        try (OutputStream os = connection.getOutputStream()) {
+            byte[] input = jsonBody.getBytes("utf-8");
             os.write(input, 0, input.length);
         }
 
@@ -48,10 +64,10 @@ public static String getCompletion(String prompt) throws Exception {
 
         connection.disconnect();
         JSONObject jsonObject = new JSONObject(response.toString());
-        return jsonObject.getJSONArray("candidates").getJSONObject(0).getJSONObject("content").getJSONArray("parts").getJSONObject(0).getString("text");
+        return jsonObject.getJSONArray("candidates").getJSONObject(0).getJSONObject("content").getJSONArray("parts")
+                .getJSONObject(0).getString("text");
     }
 
-
     /***
      * Utilities for using the Gemini API
      */
@@ -65,6 +81,7 @@ public static String readFileToString(String filePath) throws IOException {
     public static String replaceSubstring(String originalString, String substringToReplace, String replacementString) {
         return originalString.replace(substringToReplace, replacementString);
     }
+
     public static String promptVar(String prompt0, String varName, String varValue) {
         String prompt = replaceSubstring(prompt0, varName, varValue);
         return replaceSubstring(prompt, varName, varValue);

gemini-llm-client/src/test/java/com/markwatson/gemini/GeminiCompletionsTest.java

@@ -8,35 +8,33 @@
  * Unit test for simple App.
  */
 public class GeminiCompletionsTest
-    extends TestCase
-{
+        extends TestCase {
     /**
      * Create the test case
      *
      * @param testName name of the test case
      */
-    public GeminiCompletionsTest( String testName )
-    {
-        super( testName );
+    public GeminiCompletionsTest(String testName) {
+        super(testName);
     }
 
     /**
      * @return the suite of tests being tested
      */
-    public static Test suite()
-    {
-        return new TestSuite( GeminiCompletionsTest.class );
+    public static Test suite() {
+        return new TestSuite(GeminiCompletionsTest.class);
     }
 
     /**
      * Rigourous Test :-)
-     * @throws Exception 
+     * 
+     * @throws Exception
      */
-    public void testCompletion() throws Exception
-    {
-        String r = GeminiCompletions.getCompletion("Translate the following English text to French: 'Hello, how are you?'");
+    public void testCompletion() throws Exception {
+        String r = GeminiCompletions
+                .getCompletion("Translate the following English text to French: 'Hello, how are you?'");
         System.out.println("completion: " + r);
-        assertTrue( true );
+        assertTrue(true);
     }
 
     public void testTwoShotTemplate() throws Exception {
@@ -45,10 +43,9 @@ public void testTwoShotTemplate() throws Exception {
         System.out.println("prompt0: " + prompt0);
         String prompt = GeminiCompletions.promptVar(prompt0, "{input_text}", input_text);
         System.out.println("prompt: " + prompt);
-        String r =
-        GeminiCompletions.getCompletion(prompt);
+        String r = GeminiCompletions.getCompletion(prompt);
         System.out.println("two shot extraction completion: " + r);
-        assertTrue( true );
+        assertTrue(true);
     }
 
     public void testSummarization() throws Exception {
@@ -57,9 +54,16 @@ public void testSummarization() throws Exception {
         System.out.println("prompt0: " + prompt0);
         String prompt = GeminiCompletions.promptVar(prompt0, "{input_text}", input_text);
         System.out.println("prompt: " + prompt);
-        String r =
-        GeminiCompletions.getCompletion(prompt);
+        String r = GeminiCompletions.getCompletion(prompt);
         System.out.println("summarization completion: " + r);
-        assertTrue( true );
+        assertTrue(true);
+    }
+
+    public void testCompletionWithSearch() throws Exception {
+        String prompt = "What is the current stock price of Google?";
+        String r = GeminiCompletions.getCompletionWithSearch(prompt);
+        System.out.println("Search completion: " + r);
+        assertNotNull(r);
+        assertFalse(r.isEmpty());
     }
 }