首頁 > 後端開發 > Python教學 > 將 Gemini 與 OpenAI 庫結合使用

將 Gemini 與 OpenAI 庫結合使用

Susan Sarandon
發布: 2024-11-18 04:58:01
原創
996 人瀏覽過

Using Gemini with the OpenAI Library

基於本文,我們現在可以將 Gemini 與 OpenAI 函式庫一起使用。所以,我決定在這篇文章中試試看

目前僅提供聊天完成 API 和嵌入 API。

在本文中,我嘗試使用 Python 和 JavaScript。

Python

首先,我們來搭建環境。

pip install openai python-dotenv
登入後複製

接下來,讓我們執行以下程式碼。

import os

from dotenv import load_dotenv
from openai import OpenAI


load_dotenv()
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")

client = OpenAI(
    api_key=GOOGLE_API_KEY,
    base_url="https://generativelanguage.googleapis.com/v1beta/"
)


response = client.chat.completions.create(
    model="gemini-1.5-flash",
    n=1,
    messages=[
        {"role": "system", "content": "You are a helpful assistant."},
        {
            "role": "user",
            "content": "Explain briefly(less than 30 words) to me how AI works."
        }
    ]
)

print(response.choices[0].message.content)
登入後複製

回傳了以下回應。

AI mimics human intelligence by learning patterns from data, using algorithms to solve problems and make decisions. 
登入後複製

在內容欄位中,您可以指定字串或「類型」:「文字」。

import os

from dotenv import load_dotenv
from openai import OpenAI


load_dotenv()
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")

client = OpenAI(
    api_key=GOOGLE_API_KEY,
    base_url="https://generativelanguage.googleapis.com/v1beta/"
)

response = client.chat.completions.create(
    model="gemini-1.5-flash",
    n=1,
    messages=[
        {"role": "system", "content": "You are a helpful assistant."},
        {
            "role": "user",
            "content": [
                {
                    "type": "text",
                    "text": "Explain briefly(less than 30 words) to me how AI works.",
                },
            ]
        }
    ]
)

print(response.choices[0].message.content)
登入後複製

但是,影像和音訊輸入出現錯誤。

影像輸入範例程式碼

import os

from dotenv import load_dotenv
from openai import OpenAI


load_dotenv()
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")

client = OpenAI(
    api_key=GOOGLE_API_KEY,
    base_url="https://generativelanguage.googleapis.com/v1beta/"
)

# png to base64 text
import base64
with open("test.png", "rb") as image:
    b64str = base64.b64encode(image.read()).decode("utf-8")

response = client.chat.completions.create(
    model="gemini-1.5-flash",
    # model="gpt-4o",
    n=1,
    messages=[
        {"role": "system", "content": "You are a helpful assistant."},
        {
            "role": "user",
            "content": [
                {
                    "type": "text",
                    "text": "Describe the image in the image below.",
                },
                {
                    "type": "image_url",
                    "image_url": {
                        "url": f"data:image/png;base64,{b64str}"
                    }
                }
            ]
        }
    ]
)

print(response.choices[0].message.content)
登入後複製

音訊輸入範例程式碼

import os

from dotenv import load_dotenv
from openai import OpenAI


load_dotenv()
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")

client = OpenAI(
    api_key=GOOGLE_API_KEY,
    base_url="https://generativelanguage.googleapis.com/v1beta/"
)

# png to base64 text
import base64
with open("test.wav", "rb") as audio:
    b64str = base64.b64encode(audio.read()).decode("utf-8")

response = client.chat.completions.create(
    model="gemini-1.5-flash",
    # model="gpt-4o-audio-preview", 
    n=1,
    modalities=["text"],
    messages=[
        {"role": "system", "content": "You are a helpful assistant."},
        {
            "role": "user",
            "content": [
                {
                    "type": "text",
                    "text": "What does he say?",
                },
                {
                    "type": "input_audio",
                    "input_audio": {
                        "data": b64str,
                        "format": "wav",
                    }
                }
            ]
        }
    ]
)

print(response.choices[0].message.content)
登入後複製

傳回了以下錯誤回應。

openai.BadRequestError: Error code: 400 - [{'error': {'code': 400, 'message': 'Request contains an invalid argument.', 'status': 'INVALID_ARGUMENT'}}]
登入後複製

目前僅支援文字輸入,不過後續似乎會支援圖片和音訊輸入。

JavaScript

讓我們來看看 JavaScript 範例程式碼。

首先,我們來搭建環境。

npm init -y
npm install openai
npm pkg set type=module
登入後複製

接下來,讓我們執行以下程式碼。

import OpenAI from "openai";

const GOOGLE_API_KEY = process.env.GOOGLE_API_KEY;
const openai = new OpenAI({
    apiKey: GOOGLE_API_KEY,
    baseURL: "https://generativelanguage.googleapis.com/v1beta/"
});

const response = await openai.chat.completions.create({
    model: "gemini-1.5-flash",
    messages: [
        { role: "system", content: "You are a helpful assistant." },
        {
            role: "user",
            content: "Explain briefly(less than 30 words) to me how AI works",
        },
    ],
});

console.log(response.choices[0].message.content);
登入後複製

執行程式碼時,請確保在 .env 檔案中包含 API 金鑰。 .env 檔案將在運行時載入。

node --env-file=.env run.js
登入後複製

回傳了以下回應。

AI systems learn from data, identify patterns, and make predictions or decisions based on those patterns.
登入後複製

我們可以在同一個函式庫中使用其他模型,這真是太棒了。

就我個人而言,我對此感到很高興,因為 OpenAI 讓編輯對話歷史記錄變得更加容易。

以上是將 Gemini 與 OpenAI 庫結合使用的詳細內容。更多資訊請關注PHP中文網其他相關文章!

來源:dev.to
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
作者最新文章
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板