fix: remove any type in token signature

PiperOrigin-RevId: 615514694

Author: yyyu-google

src/functions/count_tokens.ts

@@ -36,7 +36,7 @@ export async function countTokens(
   location: string,
   project: string,
   publisherModelEndpoint: string,
-  token: Promise<any>,
+  token: Promise<string | null | undefined>,
   request: CountTokensRequest,
   apiEndpoint?: string,
   requestOptions?: RequestOptions

src/functions/generate_content.ts

@@ -49,7 +49,7 @@ export async function generateContent(
   location: string,
   project: string,
   publisherModelEndpoint: string,
-  token: Promise<any>,
+  token: Promise<string | null | undefined>,
   request: GenerateContentRequest | string,
   apiEndpoint?: string,
   generationConfig?: GenerationConfig,
@@ -102,7 +102,7 @@ export async function generateContentStream(
   location: string,
   project: string,
   publisherModelEndpoint: string,
-  token: Promise<any>,
+  token: Promise<string | null | undefined>,
   request: GenerateContentRequest | string,
   apiEndpoint?: string,
   generationConfig?: GenerationConfig,

src/functions/post_request.ts

@@ -43,7 +43,7 @@ export async function postRequest({
   project: string;
   resourcePath: string;
   resourceMethod: string;
-  token: string;
+  token: string | null | undefined;
   data: GenerateContentRequest | CountTokensRequest;
   apiEndpoint?: string;
   requestOptions?: RequestOptions;

src/models/chat_session.ts

@@ -90,7 +90,7 @@ export class ChatSession {
    * fails.
    * @returns Promise of token.
    */
-  get token(): Promise<any> {
+  get token(): Promise<string | null | undefined> {
     const tokenPromise = this.googleAuth.getAccessToken().catch(e => {
       throw new GoogleAuthError(constants.CREDENTIAL_ERROR_MESSAGE, e);
     });
@@ -309,7 +309,7 @@ export class ChatSessionPreview {
    * Gets access token from GoogleAuth. Throws GoogleAuthError when fails.
    * @returns Promise of token.
    */
-  get token(): Promise<any> {
+  get token(): Promise<string | null | undefined> {
     const tokenPromise = this.googleAuth.getAccessToken().catch(e => {
       throw new GoogleAuthError(constants.CREDENTIAL_ERROR_MESSAGE, e);
     });

src/models/generative_models.ts

@@ -85,7 +85,7 @@ export class GenerativeModel {
    * fails.
    * @returns Promise of token
    */
-  get token(): Promise<any> {
+  get token(): Promise<string | null | undefined> {
     const tokenPromise = this.googleAuth.getAccessToken().catch(e => {
       throw new GoogleAuthError(constants.CREDENTIAL_ERROR_MESSAGE, e);
     });
@@ -285,7 +285,7 @@ export class GenerativeModelPreview {
    * Gets access token from GoogleAuth. Throws GoogleAuthError when fails.
    * @returns Promise of token.
    */
-  get token(): Promise<any> {
+  get token(): Promise<string | null | undefined> {
     const tokenPromise = this.googleAuth.getAccessToken().catch(e => {
       throw new GoogleAuthError(constants.CREDENTIAL_ERROR_MESSAGE, e);
     });