Make classes internal

Author: emilypgoogle

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/common/APIController.kt

@@ -73,7 +73,7 @@ internal val JSON = Json {
  * @property apiClient The value to pass in the `x-goog-api-client` header.
  * @property headerProvider A provider that generates extra headers to include in all HTTP requests.
  */
-class APIController
+internal class APIController
 internal constructor(
   private val key: String,
   model: String,
@@ -233,7 +233,7 @@ internal constructor(
   }
 }
 
-interface HeaderProvider {
+internal interface HeaderProvider {
   val timeout: Duration
 
   suspend fun generateHeaders(): Map<String, String>

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/common/Exceptions.kt

@@ -20,7 +20,7 @@ import io.ktor.serialization.JsonConvertException
 import kotlinx.coroutines.TimeoutCancellationException
 
 /** Parent class for any errors that occur. */
-sealed class GoogleGenerativeAIException(message: String, cause: Throwable? = null) :
+internal sealed class GoogleGenerativeAIException(message: String, cause: Throwable? = null) :
   RuntimeException(message, cause) {
   companion object {
 
@@ -47,15 +47,15 @@ sealed class GoogleGenerativeAIException(message: String, cause: Throwable? = nu
 }
 
 /** Something went wrong while trying to deserialize a response from the server. */
-class SerializationException(message: String, cause: Throwable? = null) :
+internal class SerializationException(message: String, cause: Throwable? = null) :
   GoogleGenerativeAIException(message, cause)
 
 /** The server responded with a non 200 response code. */
-class ServerException(message: String, cause: Throwable? = null) :
+internal class ServerException(message: String, cause: Throwable? = null) :
   GoogleGenerativeAIException(message, cause)
 
 /** The server responded that the API Key is no valid. */
-class InvalidAPIKeyException(message: String, cause: Throwable? = null) :
+internal class InvalidAPIKeyException(message: String, cause: Throwable? = null) :
   GoogleGenerativeAIException(message, cause)
 
 /**
@@ -65,7 +65,7 @@ class InvalidAPIKeyException(message: String, cause: Throwable? = null) :
  *
  * @property response the full server response for the request.
  */
-class PromptBlockedException(val response: GenerateContentResponse, cause: Throwable? = null) :
+internal class PromptBlockedException(val response: GenerateContentResponse, cause: Throwable? = null) :
   GoogleGenerativeAIException(
     "Prompt was blocked: ${response.promptFeedback?.blockReason?.name}",
     cause,
@@ -78,23 +78,23 @@ class PromptBlockedException(val response: GenerateContentResponse, cause: Throw
  * [list of regions](https://ai.google.dev/available_regions#available_regions) (countries and
  * territories) where the API is available.
  */
-class UnsupportedUserLocationException(cause: Throwable? = null) :
+internal class UnsupportedUserLocationException(cause: Throwable? = null) :
   GoogleGenerativeAIException("User location is not supported for the API use.", cause)
 
 /**
  * Some form of state occurred that shouldn't have.
  *
  * Usually indicative of consumer error.
  */
-class InvalidStateException(message: String, cause: Throwable? = null) :
+internal class InvalidStateException(message: String, cause: Throwable? = null) :
   GoogleGenerativeAIException(message, cause)
 
 /**
  * A request was stopped during generation for some reason.
  *
  * @property response the full server response for the request
  */
-class ResponseStoppedException(val response: GenerateContentResponse, cause: Throwable? = null) :
+internal class ResponseStoppedException(val response: GenerateContentResponse, cause: Throwable? = null) :
   GoogleGenerativeAIException(
     "Content generation stopped. Reason: ${response.candidates?.first()?.finishReason?.name}",
     cause,
@@ -105,17 +105,17 @@ class ResponseStoppedException(val response: GenerateContentResponse, cause: Thr
  *
  * Usually occurs due to a user specified [timeout][RequestOptions.timeout].
  */
-class RequestTimeoutException(message: String, cause: Throwable? = null) :
+internal class RequestTimeoutException(message: String, cause: Throwable? = null) :
   GoogleGenerativeAIException(message, cause)
 
 /** The quota for this API key is depleted, retry this request at a later time. */
-class QuotaExceededException(message: String, cause: Throwable? = null) :
+internal class QuotaExceededException(message: String, cause: Throwable? = null) :
   GoogleGenerativeAIException(message, cause)
 
 /** The service is not enabled for this project. Visit the Firebase Console to enable it. */
-class ServiceDisabledException(message: String, cause: Throwable? = null) :
+internal class ServiceDisabledException(message: String, cause: Throwable? = null) :
   GoogleGenerativeAIException(message, cause)
 
 /** Catch all case for exceptions not explicitly expected. */
-class UnknownException(message: String, cause: Throwable? = null) :
+internal class UnknownException(message: String, cause: Throwable? = null) :
   GoogleGenerativeAIException(message, cause)

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/common/Request.kt

@@ -25,10 +25,10 @@ import com.google.firebase.vertexai.common.util.fullModelName
 import kotlinx.serialization.SerialName
 import kotlinx.serialization.Serializable
 
-sealed interface Request
+internal sealed interface Request
 
 @Serializable
-data class GenerateContentRequest(
+internal data class GenerateContentRequest(
   val model: String? = null,
   val contents: List<Content>,
   @SerialName("safety_settings") val safetySettings: List<SafetySetting>? = null,
@@ -39,7 +39,7 @@ data class GenerateContentRequest(
 ) : Request
 
 @Serializable
-data class CountTokensRequest(
+internal data class CountTokensRequest(
   val generateContentRequest: GenerateContentRequest? = null,
   val model: String? = null,
   val contents: List<Content>? = null,

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/common/RequestOptions.kt

@@ -28,7 +28,7 @@ import kotlin.time.toDuration
  * first response.
  * @property apiVersion the api endpoint to call.
  */
-class RequestOptions(
+internal class RequestOptions(
   val timeout: Duration,
   val apiVersion: String = "v1beta",
   val endpoint: String = "https://generativelanguage.googleapis.com",

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/common/Response.kt

@@ -21,23 +21,23 @@ import com.google.firebase.vertexai.common.server.GRpcError
 import com.google.firebase.vertexai.common.server.PromptFeedback
 import kotlinx.serialization.Serializable
 
-sealed interface Response
+internal sealed interface Response
 
 @Serializable
-data class GenerateContentResponse(
+internal data class GenerateContentResponse(
   val candidates: List<Candidate>? = null,
   val promptFeedback: PromptFeedback? = null,
   val usageMetadata: UsageMetadata? = null,
 ) : Response
 
 @Serializable
-data class CountTokensResponse(val totalTokens: Int, val totalBillableCharacters: Int? = null) :
+internal data class CountTokensResponse(val totalTokens: Int, val totalBillableCharacters: Int? = null) :
   Response
 
-@Serializable data class GRpcErrorResponse(val error: GRpcError) : Response
+@Serializable internal data class GRpcErrorResponse(val error: GRpcError) : Response
 
 @Serializable
-data class UsageMetadata(
+internal data class UsageMetadata(
   val promptTokenCount: Int? = null,
   val candidatesTokenCount: Int? = null,
   val totalTokenCount: Int? = null,

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/common/client/Types.kt

@@ -21,7 +21,7 @@ import kotlinx.serialization.Serializable
 import kotlinx.serialization.json.JsonObject
 
 @Serializable
-data class GenerationConfig(
+internal data class GenerationConfig(
   val temperature: Float?,
   @SerialName("top_p") val topP: Float?,
   @SerialName("top_k") val topK: Int?,
@@ -35,19 +35,19 @@ data class GenerationConfig(
 )
 
 @Serializable
-data class Tool(
+internal data class Tool(
   val functionDeclarations: List<FunctionDeclaration>? = null,
   // This is a json object because it is not possible to make a data class with no parameters.
   val codeExecution: JsonObject? = null,
 )
 
 @Serializable
-data class ToolConfig(
+internal data class ToolConfig(
   @SerialName("function_calling_config") val functionCallingConfig: FunctionCallingConfig
 )
 
 @Serializable
-data class FunctionCallingConfig(val mode: Mode) {
+internal data class FunctionCallingConfig(val mode: Mode) {
   @Serializable
   enum class Mode {
     @SerialName("MODE_UNSPECIFIED") UNSPECIFIED,
@@ -58,10 +58,10 @@ data class FunctionCallingConfig(val mode: Mode) {
 }
 
 @Serializable
-data class FunctionDeclaration(val name: String, val description: String, val parameters: Schema)
+internal data class FunctionDeclaration(val name: String, val description: String, val parameters: Schema)
 
 @Serializable
-data class Schema(
+internal data class Schema(
   val type: String,
   val description: String? = null,
   val format: String? = null,

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/common/server/Types.kt

@@ -25,31 +25,31 @@ import kotlinx.serialization.SerialName
 import kotlinx.serialization.Serializable
 import kotlinx.serialization.json.JsonNames
 
-object BlockReasonSerializer :
+internal object BlockReasonSerializer :
   KSerializer<BlockReason> by FirstOrdinalSerializer(BlockReason::class)
 
-object HarmProbabilitySerializer :
+internal object HarmProbabilitySerializer :
   KSerializer<HarmProbability> by FirstOrdinalSerializer(HarmProbability::class)
 
-object FinishReasonSerializer :
+internal object FinishReasonSerializer :
   KSerializer<FinishReason> by FirstOrdinalSerializer(FinishReason::class)
 
 @Serializable
-data class PromptFeedback(
+internal data class PromptFeedback(
   val blockReason: BlockReason? = null,
   val safetyRatings: List<SafetyRating>? = null,
 )
 
 @Serializable(BlockReasonSerializer::class)
-enum class BlockReason {
+internal enum class BlockReason {
   UNKNOWN,
   @SerialName("BLOCKED_REASON_UNSPECIFIED") UNSPECIFIED,
   SAFETY,
   OTHER
 }
 
 @Serializable
-data class Candidate(
+internal data class Candidate(
   val content: Content? = null,
   val finishReason: FinishReason? = null,
   val safetyRatings: List<SafetyRating>? = null,
@@ -58,20 +58,20 @@ data class Candidate(
 )
 
 @Serializable
-data class CitationMetadata
+internal data class CitationMetadata
 @OptIn(ExperimentalSerializationApi::class)
-constructor(@JsonNames("citations") val citationSources: List<CitationSources>)
+internal constructor(@JsonNames("citations") val citationSources: List<CitationSources>)
 
 @Serializable
-data class CitationSources(
+internal data class CitationSources(
   val startIndex: Int = 0,
   val endIndex: Int,
   val uri: String? = null,
   val license: String? = null,
 )
 
 @Serializable
-data class SafetyRating(
+internal data class SafetyRating(
   val category: HarmCategory,
   val probability: HarmProbability,
   val blocked: Boolean? = null, // TODO(): any reason not to default to false?
@@ -81,33 +81,33 @@ data class SafetyRating(
 )
 
 @Serializable
-data class GroundingMetadata(
+internal data class GroundingMetadata(
   @SerialName("web_search_queries") val webSearchQueries: List<String>?,
   @SerialName("search_entry_point") val searchEntryPoint: SearchEntryPoint?,
   @SerialName("retrieval_queries") val retrievalQueries: List<String>?,
   @SerialName("grounding_attribution") val groundingAttribution: List<GroundingAttribution>?,
 )
 
 @Serializable
-data class SearchEntryPoint(
+internal data class SearchEntryPoint(
   @SerialName("rendered_content") val renderedContent: String?,
   @SerialName("sdk_blob") val sdkBlob: String?,
 )
 
 @Serializable
-data class GroundingAttribution(
+internal data class GroundingAttribution(
   val segment: Segment,
   @SerialName("confidence_score") val confidenceScore: Float?,
 )
 
 @Serializable
-data class Segment(
+internal data class Segment(
   @SerialName("start_index") val startIndex: Int,
   @SerialName("end_index") val endIndex: Int,
 )
 
 @Serializable(HarmProbabilitySerializer::class)
-enum class HarmProbability {
+internal enum class HarmProbability {
   UNKNOWN,
   @SerialName("HARM_PROBABILITY_UNSPECIFIED") UNSPECIFIED,
   NEGLIGIBLE,
@@ -117,7 +117,7 @@ enum class HarmProbability {
 }
 
 @Serializable
-enum class HarmSeverity {
+internal enum class HarmSeverity {
   UNKNOWN,
   @SerialName("HARM_SEVERITY_UNSPECIFIED") UNSPECIFIED,
   @SerialName("HARM_SEVERITY_NEGLIGIBLE") NEGLIGIBLE,
@@ -127,7 +127,7 @@ enum class HarmSeverity {
 }
 
 @Serializable(FinishReasonSerializer::class)
-enum class FinishReason {
+internal enum class FinishReason {
   UNKNOWN,
   @SerialName("FINISH_REASON_UNSPECIFIED") UNSPECIFIED,
   STOP,
@@ -138,6 +138,6 @@ enum class FinishReason {
 }
 
 @Serializable
-data class GRpcError(val code: Int, val message: String, val details: List<GRpcErrorDetails>)
+internal data class GRpcError(val code: Int, val message: String, val details: List<GRpcErrorDetails>)
 
-@Serializable data class GRpcErrorDetails(val reason: String? = null)
+@Serializable internal data class GRpcErrorDetails(val reason: String? = null)