访问Gemini API 并解决错误 503 failed to connect to all addresses

[Valdanitooooo]

首先需要科学上网

略

在Google AI Studio生成apikey

使用google登录Google AI Studio
https://aistudio.google.com/app/apikey
生成apikey

访问Gemini Pro API

import os

import google.generativeai as genai
from dotenv import load_dotenv

load_dotenv()

genai.configure(api_key=os.getenv('GOOGLE_API_KEY'), transport='rest')


def text_test():
    for m in genai.list_models():
        if 'generateContent' in m.supported_generation_methods:
            print(m.name)
    model = genai.GenerativeModel('gemini-pro')
    response = model.generate_content("你是谁?", stream=True)
    for chuck in response:
        print(chuck.text)


if __name__ == '__main__':
    text_test()

output:

models/gemini-1.0-pro
models/gemini-1.0-pro-001
models/gemini-1.0-pro-latest
models/gemini-1.0-pro-vision-latest
models/gemini-pro
models/gemini-pro-vision
我是一個由 Google 訓練的大型語言模型。

解决错误 503 failed to connect to all addresses

错误信息：

google.api_core.exceptions.RetryError: Deadline of 60.0s exceeded while calling target function, last exception: 503 failed to connect to all addresses;

解决方案

将

genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))

改为

genai.configure(api_key=os.getenv('GOOGLE_API_KEY'), transport='rest')

绕过默认的grpc请求，改用rest请求

参考链接

google-gemini/generative-ai-python#117

完整demo代码

https://github.com/Valdanitooooo/code_snippets/tree/main/llm/google/gemini