从Google Colab调用Gemini API时遇到503错误

Question

我正在做一个项目，使用Google Colab来运行Python代码，这段代码会和Gemini API（这是谷歌云AI工具的一部分）进行互动。我的目标是自动将通话记录分类到预先定义的类别中，利用Gemini的AI来完成这项工作。

简单来说，我的工作流程是这样的：我先读取一个包含通话记录的Excel文件，然后把这些记录发送给Gemini进行分类，最后根据AI识别出的类别更新Excel文件（用0和1来标记）。

下面是我设置API和发送请求给Gemini的代码片段：

import google.generativeai as genai

GOOGLE_API_KEY = "your_api_key_here"
genai.configure(api_key=GOOGLE_API_KEY)
model = genai.GenerativeModel('gemini-pro')

def send_to_gemini(transcript):
    prompt = f"Categorize the following transcript: {transcript}"
    try:
        response = model.generate_content(prompt)
        return response.text
    except Exception as e:
        print(f"Failed to send request to Gemini: {e}")

但是，我总是遇到一个ERROR:tornado.access:503的错误，这提示可能是服务器那边出了问题：

ERROR:tornado.access:503 POST /v1beta/models/gemini-pro:generateContent (127.0.0.1) 4039.47ms

如果有人能给点建议或见解，我将非常感激。

Answer 1

错误503表示“服务不可用”。

如果你查看完整的错误信息，可能会看到“模型超载，请稍后再试。”

这肯定是谷歌那边出现了问题，而不是你的问题。你能做的事情很少。不过，你应该考虑到这个情况。通常的做法是在遇到服务器不可用的消息时，尝试在一个循环中重试，并逐渐增加重试的间隔时间。

你提到你在使用“google.generativeai”这个包，它对应的是AI Studio的服务（aistudio.google.com），而不是谷歌云。这目前只是一个免费服务，尚未正式推出。所以看起来他们还在调整，以满足预期的用户需求，并准备推出更多的付费服务。