OpenAI SDK

npm install openai

import OpenAI from 'openai';
 
const baseURL = process.env.AI_GATEWAY_BASE_URL;
const apiKey = process.env.AI_GATEWAY_API_KEY;
const model = process.env.AI_GATEWAY_MODEL;
 
if (!baseURL || !apiKey || !model) {
  throw new Error(
    'AI_GATEWAY_BASE_URL、AI_GATEWAY_API_KEY、AI_GATEWAY_MODELを設定してください',
  );
}
 
const prompt = 'ロリポップ！AIゲートウェイについて一言で説明してください。';
 
const client = new OpenAI({
  apiKey,
  baseURL: `${baseURL}/v1`,
});
 
const completion = await client.chat.completions.create({
  model,
  messages: [
    { role: 'user', content: prompt },
  ],
});
 
console.log(completion.choices[0].message.content);

pip install openai

import os
from openai import OpenAI
 
base_url = os.environ["AI_GATEWAY_BASE_URL"]
model = os.environ["AI_GATEWAY_MODEL"]
 
client = OpenAI(
    api_key=os.environ["AI_GATEWAY_API_KEY"],
    base_url=f"{base_url}/v1",
)
 
prompt = "ロリポップ！AIゲートウェイについて一言で説明してください。"
 
completion = client.chat.completions.create(
    model=model,
    messages=[{"role": "user", "content": prompt}],
)
 
print(completion.choices[0].message.content)