Gemini api中转调用python代码|ieasy·未来

复制使用Python运行下面代码即可（需要替换内容里的域名url和密钥key为自己的）

import os
from openai import OpenAI

# 设置你的 openai-gemini 部署的 API 端点
# 假设你的部署地址是 https://your-deployment-url/v1
BASE_URL = "https://your-deployment-url/v1"  # 替换为你的实际部署地址

# 设置你的 Gemini API Key
# 从 Google AI Studio 获取的 Gemini API Key
GEMINI_API_KEY = "xxxxxxxxxxxxxxxxxxx"  # 替换为你的 Gemini API Key

# 初始化 OpenAI 客户端
client = OpenAI(
    api_key=GEMINI_API_KEY,
    base_url=BASE_URL
)

# 持续对话函数
def interactive_chat():
    print("欢迎使用 Gemini 聊天机器人（通过 openai-gemini 接口）！输入 'exit' 退出。（更多精彩内容可关注微信公众号：知识不打烊)")

    # 初始化消息历史，包含系统提示
    messages = [
        {"role": "system", "content": "You are a helpful assistant."}
    ]

    while True:
        # 获取用户输入
        user_input = input("\n你: ")

        # 检查是否退出
        if user_input.lower() == "exit":
            print("再见！")
            break

        # 将用户输入添加到消息历史
        messages.append({"role": "user", "content": user_input})

        try:
            # 调用 Chat Completions API
            response = client.chat.completions.create(
                model="gemini-2.0-flash",  # 指定 Gemini 模型，根据实际支持调整
                messages=messages,
                max_tokens=200,
                temperature=0.7
            )

            # 获取模型回复
            assistant_response = response.choices[0].message.content

            # 打印回复
            print(f"助手: {assistant_response}")

            # 将模型回复添加到消息历史
            messages.append({"role": "assistant", "content": assistant_response})

        except Exception as e:
            print(f"错误: {str(e)}")
            # 如果出错，继续循环，而不是退出

# 运行交互式聊天
if __name__ == "__main__":
    interactive_chat()

免责声明