fix: replace any type with explicit types in post fetch processing functions

PiperOrigin-RevId: 615517662

Author: yyyu-google

src/functions/count_tokens.ts

@@ -41,7 +41,7 @@ export async function countTokens(
   apiEndpoint?: string,
   requestOptions?: RequestOptions
 ): Promise<CountTokensResponse> {
-  const response = await postRequest({
+  const response: Response | undefined = await postRequest({
     region: location,
     project: project,
     resourcePath: publisherModelEndpoint,

src/functions/post_fetch_processing.ts

@@ -267,7 +267,7 @@ function addCandidateFunctionCalls(
  * @ignore
  */
 export async function processNonStream(
-  response: any
+  response: Response | undefined
 ): Promise<GenerateContentResult> {
   if (response !== undefined) {
     // ts-ignore
@@ -287,8 +287,12 @@ export async function processNonStream(
  * @ignore
  */
 export async function processCountTokenResponse(
-  response: any
+  response: Response | undefined
 ): Promise<CountTokensResponse> {
-  // ts-ignore
-  return response.json();
+  if (response) {
+    // ts-ignore
+    return response.json();
+  }
+
+  return Promise.resolve({} as CountTokensResponse);
 }