request-helpers

Author: russellwheatley

packages/ai/lib/requests/request-helpers.ts

@@ -15,8 +15,8 @@
  * limitations under the License.
  */
 
-import { Content, GenerateContentRequest, Part, VertexAIErrorCode } from '../types';
-import { VertexAIError } from '../errors';
+import { Content, GenerateContentRequest, Part, AIErrorCode } from '../types';
+import { AIError } from '../errors';
 
 export function formatSystemInstruction(input?: string | Part | Content): Content | undefined {
   if (input == null) {
@@ -32,7 +32,6 @@ export function formatSystemInstruction(input?: string | Part | Content): Conten
       return input as Content;
     }
   }
-
   return undefined;
 }
 
@@ -76,15 +75,15 @@ function assignRoleToPartsAndValidateSendMessageRequest(parts: Part[]): Content
   }
 
   if (hasUserContent && hasFunctionContent) {
-    throw new VertexAIError(
-      VertexAIErrorCode.INVALID_CONTENT,
+    throw new AIError(
+      AIErrorCode.INVALID_CONTENT,
       'Within a single message, FunctionResponse cannot be mixed with other type of Part in the request for sending chat message.',
     );
   }
 
   if (!hasUserContent && !hasFunctionContent) {
-    throw new VertexAIError(
-      VertexAIErrorCode.INVALID_CONTENT,
+    throw new AIError(
+      AIErrorCode.INVALID_CONTENT,
       'No Content is provided for sending chat message.',
     );
   }