Format

Author: dlarocque

packages/vertexai/src/backend.test.ts

@@ -1,11 +1,28 @@
-import { expect } from "chai";
-import { GoogleAIBackend, VertexAIBackend } from "./backend";
-import { BackendType } from "./public-types";
-import { DEFAULT_LOCATION } from "./constants";
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { expect } from 'chai';
+import { GoogleAIBackend, VertexAIBackend } from './backend';
+import { BackendType } from './public-types';
+import { DEFAULT_LOCATION } from './constants';
 
 describe('Backend', () => {
   describe('GoogleAIBackend', () => {
-  it('sets backendType to GOOGLE_AI', () => {
+    it('sets backendType to GOOGLE_AI', () => {
       const backend = new GoogleAIBackend();
       expect(backend.backendType).to.equal(BackendType.GOOGLE_AI);
     });
@@ -32,4 +49,4 @@ describe('Backend', () => {
       expect(backend.location).to.equal(DEFAULT_LOCATION);
     });
   });
-});
+});

packages/vertexai/src/backend.ts

@@ -1,5 +1,22 @@
-import { DEFAULT_LOCATION } from "./constants";
-import { BackendType } from "./public-types";
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { DEFAULT_LOCATION } from './constants';
+import { BackendType } from './public-types';
 
 /**
  * Abstract base class representing the configuration for an AI service backend.
@@ -69,4 +86,4 @@ export class VertexAIBackend extends Backend {
       this.location = location;
     }
   }
-}
+}

packages/vertexai/src/googleai-mappers.ts

@@ -164,7 +164,8 @@ export function mapGenerateContentCandidates(
         mappedSafetyRatings = candidate.safetyRatings.map(safetyRating => {
           return {
             ...safetyRating,
-            severity: safetyRating.severity ?? HarmSeverity.HARM_SEVERITY_UNSUPPORTED,
+            severity:
+              safetyRating.severity ?? HarmSeverity.HARM_SEVERITY_UNSUPPORTED,
             probabilityScore: safetyRating.probabilityScore ?? 0,
             severityScore: safetyRating.severityScore ?? 0
           };

packages/vertexai/src/types/googleai.ts

@@ -1,4 +1,30 @@
-import { Tool, GenerationConfig, Citation, FinishReason, GroundingMetadata, PromptFeedback, SafetyRating, UsageMetadata } from '../public-types';
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import {
+  Tool,
+  GenerationConfig,
+  Citation,
+  FinishReason,
+  GroundingMetadata,
+  PromptFeedback,
+  SafetyRating,
+  UsageMetadata
+} from '../public-types';
 import { Content, Part } from './content';
 
 /**
@@ -41,4 +67,4 @@ export interface GoogleAIGenerateContentCandidate {
  */
 export interface GoogleAICitationMetadata {
   citationSources: Citation[]; // Maps to `citations`
-}
+}

packages/vertexai/src/types/responses.ts

@@ -249,7 +249,7 @@ export interface CountTokensResponse {
   /**
    * The total number of billable characters counted across all instances
    * from the request.
-   * 
+   *
    * This field is not supported in Google AI, so it will default to 0 when using Google AI.
    */
   totalBillableCharacters?: number;