Skip to content

SDK 接入示例

Python (openai 库)

安装

bash
pip install openai

对话补全(非流式)

python
from openai import OpenAI

client = OpenAI(
    base_url="http://ok-link.com:10320/v1",
    api_key="sk-xxxxxxxxxxxx"
)

response = client.chat.completions.create(
    model="gpt-4o",
    messages=[
        {"role": "system", "content": "You are a helpful assistant."},
        {"role": "user", "content": "What is the capital of France?"}
    ],
    temperature=0.7,
    max_tokens=512
)

print(response.choices[0].message.content)
print(f"Token usage: {response.usage}")

对话补全(流式)

python
from openai import OpenAI

client = OpenAI(
    base_url="http://ok-link.com:10320/v1",
    api_key="sk-xxxxxxxxxxxx"
)

stream = client.chat.completions.create(
    model="gpt-4o",
    messages=[
        {"role": "user", "content": "Write a short poem about AI."}
    ],
    stream=True
)

for chunk in stream:
    if chunk.choices[0].delta.content:
        print(chunk.choices[0].delta.content, end="", flush=True)

向量嵌入

python
from openai import OpenAI

client = OpenAI(
    base_url="http://ok-link.com:10320/v1",
    api_key="sk-xxxxxxxxxxxx"
)

response = client.embeddings.create(
    model="text-embedding-3-small",
    input="The text to embed"
)

print(len(response.data[0].embedding))

模型列表

python
from openai import OpenAI

client = OpenAI(
    base_url="http://ok-link.com:10320/v1",
    api_key="sk-xxxxxxxxxxxx"
)

models = client.models.list()
for model in models:
    print(model.id)

Node.js (openai 库)

安装

bash
npm install openai

对话补全

javascript
import OpenAI from 'openai';

const client = new OpenAI({
  baseURL: 'http://ok-link.com:10320/v1',
  apiKey: 'sk-xxxxxxxxxxxx',
});

async function main() {
  const response = await client.chat.completions.create({
    model: 'gpt-4o',
    messages: [
      { role: 'user', content: 'Hello!' }
    ],
  });
  console.log(response.choices[0].message.content);
}

main();

cURL 命令合集

对话补全

bash
curl http://ok-link.com:10320/v1/chat/completions \
  -H "Authorization: Bearer sk-xxxxxxxxxxxx" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "gpt-4o",
    "messages": [{"role": "user", "content": "Hello!"}],
    "stream": false
  }' | jq .

流式对话

bash
curl http://ok-link.com:10320/v1/chat/completions \
  -H "Authorization: Bearer sk-xxxxxxxxxxxx" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "gpt-4o",
    "messages": [{"role": "user", "content": "Hello!"}],
    "stream": true
  }'

向量嵌入

bash
curl http://ok-link.com:10320/v1/embeddings \
  -H "Authorization: Bearer sk-xxxxxxxxxxxx" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "text-embedding-3-small",
    "input": "The text to embed"
  }' | jq .

模型列表

bash
curl http://ok-link.com:10320/v1/models \
  -H "Authorization: Bearer sk-xxxxxxxxxxxx" | jq .

科华词元平台 - AI 代理统一管理平台