remove extra fullModelName

google-gemini · Jun 7, 2024 · f26b607 · f26b607
1 parent 6c5b1bb
commit f26b607
Showing 1 changed file with 1 addition and 10 deletions.
diff --git a/generativeai/src/main/java/com/google/ai/client/generativeai/GenerativeModel.kt b/generativeai/src/main/java/com/google/ai/client/generativeai/GenerativeModel.kt
@@ -20,6 +20,7 @@ import android.graphics.Bitmap
 import com.google.ai.client.generativeai.common.APIController
 import com.google.ai.client.generativeai.common.CountTokensRequest
 import com.google.ai.client.generativeai.common.GenerateContentRequest
+import com.google.ai.client.generativeai.common.util.fullModelName
 import com.google.ai.client.generativeai.internal.util.toInternal
 import com.google.ai.client.generativeai.internal.util.toPublic
 import com.google.ai.client.generativeai.type.Content
@@ -252,14 +253,4 @@ internal constructor(
       .firstOrNull { it != FinishReason.STOP }
       ?.let { throw ResponseStoppedException(this) }
   }
-
-  companion object {
-    /**
-     * Ensures the model name provided has a `models/` prefix
-     *
-     * Models must be prepended with the `models/` prefix when communicating with the backend.
-     */
-    private fun fullModelName(name: String): String =
-      name.takeIf { it.contains("/") } ?: "models/$name"
-  }
 }