You can try the gemini through genai
pip install google-generativeai genai.configure(api_key="YOUR_API_KEY") model = genai.GenerativeModel("gemini-1.5-flash")