Export VertexAIError from api instead of public types

Author: dlarocque

packages/vertexai/src/api.test.ts

@@ -14,7 +14,8 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { ModelParams, VertexAIError, VertexAIErrorCode } from './types';
+import { ModelParams, VertexAIErrorCode } from './types';
+import { VertexAIError } from './errors';
 import { getGenerativeModel } from './api';
 import { expect } from 'chai';
 import { VertexAI } from './public-types';

packages/vertexai/src/api.ts

@@ -24,15 +24,17 @@ import { VertexAI, VertexAIOptions } from './public-types';
 import {
   ModelParams,
   RequestOptions,
-  VertexAIError,
   VertexAIErrorCode
 } from './types';
+import { VertexAIError } from './errors';
 import { GenerativeModel } from './models/generative-model';
 
 export { ChatSession } from './methods/chat-session';
 
 export { GenerativeModel };
 
+export { VertexAIError } from './errors';
+
 declare module '@firebase/component' {
   interface NameServiceMapping {
     [VERTEX_TYPE]: VertexAIService;

packages/vertexai/src/methods/chat-session-helpers.ts

@@ -20,9 +20,9 @@ import {
   POSSIBLE_ROLES,
   Part,
   Role,
-  VertexAIError,
   VertexAIErrorCode
 } from '../types';
+import { VertexAIError } from '../errors';
 
 // https://ai.google.dev/api/rest/v1beta/Content#part
 

packages/vertexai/src/models/generative-model.ts

@@ -34,9 +34,9 @@ import {
   StartChatParams,
   Tool,
   ToolConfig,
-  VertexAIError,
   VertexAIErrorCode
 } from '../types';
+import { VertexAIError } from '../errors';
 import { ChatSession } from '../methods/chat-session';
 import { countTokens } from '../methods/count-tokens';
 import {

packages/vertexai/src/requests/request-helpers.ts

@@ -19,9 +19,9 @@ import {
   Content,
   GenerateContentRequest,
   Part,
-  VertexAIError,
   VertexAIErrorCode
 } from '../types';
+import { VertexAIError } from '../errors';
 
 export function formatSystemInstruction(
   input?: string | Part | Content

packages/vertexai/src/requests/request.test.ts

@@ -22,7 +22,8 @@ import chaiAsPromised from 'chai-as-promised';
 import { RequestUrl, Task, getHeaders, makeRequest } from './request';
 import { ApiSettings } from '../types/internal';
 import { DEFAULT_API_VERSION } from '../constants';
-import { VertexAIError, VertexAIErrorCode } from '../types';
+import { VertexAIErrorCode } from '../types';
+import { VertexAIError } from '../errors';
 
 use(sinonChai);
 use(chaiAsPromised);

packages/vertexai/src/requests/request.ts

@@ -15,7 +15,8 @@
  * limitations under the License.
  */
 
-import { RequestOptions, VertexAIError, VertexAIErrorCode } from '../types';
+import { RequestOptions, VertexAIErrorCode } from '../types';
+import { VertexAIError } from '../errors';
 import { ApiSettings } from '../types/internal';
 import {
   DEFAULT_API_VERSION,

packages/vertexai/src/requests/response-helpers.ts

@@ -21,9 +21,9 @@ import {
   FunctionCall,
   GenerateContentCandidate,
   GenerateContentResponse,
-  VertexAIError,
   VertexAIErrorCode
 } from '../types';
+import { VertexAIError } from '../errors';
 
 /**
  * Adds convenience helper methods to a response object, including stream

packages/vertexai/src/requests/stream-reader.ts

@@ -21,9 +21,9 @@ import {
   GenerateContentResponse,
   GenerateContentStreamResult,
   Part,
-  VertexAIError,
   VertexAIErrorCode
 } from '../types';
+import { VertexAIError } from '../errors';
 import { addHelpers } from './response-helpers';
 
 const responseLineRE = /^data\: (.*)(?:\n\n|\r\r|\r\n\r\n)/;

packages/vertexai/src/types/error.ts

@@ -90,5 +90,3 @@ export const enum VertexAIErrorCode {
   /** An error occured while parsing. */
   PARSE_FAILED = 'parse-failed'
 }
-
-export { VertexAIError } from '../errors';