Package genai_apis

class APIFactory:
    @staticmethod
    def get_api_client(api_name, **kwargs):
        if api_name == "openai":
            try:
                from openai import AsyncOpenAI

                return OpenAIAPI(AsyncOpenAI(api_key=kwargs["api_key"]))
            except ImportError:
                raise ImportError(
                    "OpenAI library is not installed. Please install the package before proceeding."
                )
        elif api_name == "gemini":
            try:
                import google.generativeai as genai

                genai.configure(api_key=kwargs["api_key"])
                return GeminiAPI()
            except ImportError:
                raise ImportError(
                    "Google GenerativeAI library is not installed. Please install the package before proceeding."
                )
        elif api_name == "gemini-vertex":
            try:
                import vertexai

                vertexai.init(
                    project=kwargs["GCP_PROJECT_ID"],
                    location=kwargs["GCP_PROJECT_LOCATION"],
                )
                return GeminiVertexAPI()
            except ImportError:
                raise ImportError(
                    "Google google-cloud-aiplatform library is not installed. Please install the package before proceeding."
                )
        elif api_name == "anthropic":
            try:
                from anthropic import AsyncAnthropic

                return AnthropicAPI(AsyncAnthropic(api_key=kwargs["api_key"]))
            except ImportError:
                raise ImportError(
                    "Anthropic library is not installed. Please install the package before proceeding."
                )
        elif api_name == "anthropic-vertex":
            try:
                from anthropic import AsyncAnthropicVertex

                return AnthropicAPI(
                    AsyncAnthropicVertex(
                        project_id=kwargs["GCP_PROJECT_ID"],
                        region=kwargs["GCP_PROJECT_LOCATION"],
                    )
                )
            except ImportError:
                raise ImportError(
                    "Anthropic library is not installed. Please install the package before proceeding."
                )
        elif api_name == "anthropic-bedrock":
            try:
                from anthropic import AsyncAnthropicBedrock

                return AnthropicAPI(
                    AsyncAnthropicBedrock(aws_region=kwargs["AWS_REGION"])
                )
            except ImportError:
                raise ImportError(
                    "Anthropic library is not installed. Please install the package before proceeding."
                )
        else:
            raise ValueError("Unsupported API")