import { createPiece } from "@activepieces/pieces-framework";
import { PieceCategory } from '@activepieces/pieces-framework';
import { vertexAiAuth } from "./lib/auth";
import { generateContent, generateImage, customApiCall } from "./lib/actions";

export const googleVertexai = createPiece({
  displayName: "Google Vertex AI",
  description: "Generate content and images using Gemini and Imagen models on Google Vertex AI.",
  auth: vertexAiAuth,
  minimumSupportedRelease: "0.71.4",
  logoUrl: "https://cdn.activepieces.com/pieces/google-vertexai.png",
  categories: [PieceCategory.ARTIFICIAL_INTELLIGENCE],
  authors: ["alinperghel", "onyedikachi-david","bertrandong"],
  actions: [generateContent, generateImage, customApiCall],
  triggers: [],
});

export { vertexAiAuth, GoogleVertexAIAuthValue } from "./lib/auth";
