Gemini-Python示例代码

1. 资源准备

• API Key：此项配置填写在一步API官网创建API令牌，一键直达API令牌创建页面
• 创建API令牌步骤请参考API Key的获取和使用
• API Host：此项配置填写https://yibuapi.com/v1
• 查看支持的模型请参考这篇教程模型在线查询

2. 安装依赖

pip install openai

3. 实现代码

import openai

# 这里填写您在https://yibuapi.com上创建的apikey
api_key = "sk-xxxxx"
# 这里填写https://yibuapi.com/v1
base_url = "https://yibuapi.com/v1"
# 这是问题
questions = f"""
生成三个虚构的中文书名及其作者和类型的清单。 
使用以下键以 JSON 格式提供它们：book_id, title, author, genre.
"""

def get_gemini_response(question, api_key, base_url):
    try:
        client = openai.OpenAI(api_key=api_key, base_url=base_url)
        response = client.chat.completions.create(
            model="gemini-1.5-flash-8b",
            messages=[{"role": "user", "content": question}]
        )
        return response.choices[0].message.content
    except Exception as e:
        return f"请求失败: {str(e)}"

if __name__ == "__main__":
    response = get_gemini_response(questions, api_key, base_url)
    print(f"回答: {response}\n")

4. 运行效果