Java Shim now correctly passes prompt through (#53)

Co-authored-by: David Motsonashvili <[email protected]>
google-gemini · Feb 9, 2024 · dc21ffd · dc21ffd
1 parent 5911028
commit dc21ffd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.changes/crowd-comfort-color-burst.json b/.changes/crowd-comfort-color-burst.json
@@ -0,0 +1 @@
+{"type":"PATCH","changes":["GenerativeModelFutures now correctly passes the prompt through"]}
diff --git a/generativeai/src/main/java/com/google/ai/client/generativeai/java/GenerativeModelFutures.kt b/generativeai/src/main/java/com/google/ai/client/generativeai/java/GenerativeModelFutures.kt
@@ -74,7 +74,7 @@ abstract class GenerativeModelFutures internal constructor() {
       SuspendToFutureAdapter.launchFuture { model.generateContent(*prompt) }
 
     override fun generateContentStream(vararg prompt: Content): Publisher<GenerateContentResponse> =
-      model.generateContentStream().asPublisher()
+      model.generateContentStream(*prompt).asPublisher()
 
     override fun countTokens(vararg prompt: Content): ListenableFuture<CountTokensResponse> =
       SuspendToFutureAdapter.launchFuture { model.countTokens(*prompt) }
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		{"type":"PATCH","changes":["GenerativeModelFutures now correctly passes the prompt through"]}