SDK & 示例

OpenAI SDK 兼容

使用 OpenAI 官方 SDK 接入 TENSORAXIS,並瞭解兼容邊界。

原理

TENSORAXIS 為常見 OpenAI 兼容接口提供統一入口。對於 Chat Completions、Responses、Embeddings、Images、Audio 等接口,通常只需要在初始化 SDK 時替換兩個參數:

  • api_key → 你的 TENSORAXIS API Key
  • base_urlhttps://api.tensoraxis.ai/v1

部分供應商擴展能力和異步任務會有自己的請求體約定。尤其是視頻生成,不要直接照搬上游廠商的原始 content[] 請求體,請參考 視頻生成

Python

pip install openai
from openai import OpenAI

client = OpenAI(
    api_key="your-tensoraxis-api-key",
    base_url="https://api.tensoraxis.ai/v1",
)

Node.js / TypeScript

npm install openai
import OpenAI from 'openai'

const client = new OpenAI({
  apiKey: process.env.TENSORAXIS_API_KEY,
  baseURL: 'https://api.tensoraxis.ai/v1',
})

環境變量方式(推薦)

# .env
OPENAI_API_KEY=your-tensoraxis-api-key
OPENAI_BASE_URL=https://api.tensoraxis.ai/v1

設置後,OpenAI SDK 會自動讀取環境變量:

from openai import OpenAI
client = OpenAI()

支持的接口

接口路徑說明
Chat CompletionsPOST /v1/chat/completions對話補全,支持流式輸出
ResponsesPOST /v1/responsesOpenAI Responses API
EmbeddingsPOST /v1/embeddings文本向量化
ModelsGET /v1/models可用模型列表
ImagesPOST /v1/images/generationsPOST /v1/images/edits圖片生成與編輯(取決於模型能力)
AudioPOST /v1/audio/transcriptionsPOST /v1/audio/translationsPOST /v1/audio/speech音頻轉寫、翻譯與語音生成
RerankPOST /v1/rerank重排序接口,通常使用直接 HTTP 調用
VideoPOST /v1/video/generationsGET /v1/video/generations/{task_id}異步視頻任務接口,見視頻生成
OpenAI/Sora VideoPOST /v1/videosGET /v1/videos/{task_id}OpenAI/Sora 風格視頻兼容接口

視頻接口特別說明

視頻生成是異步任務。以 Seedance / Doubao 視頻模型為例,TENSORAXIS 接收的是統一任務請求:

{
  "model": "doubao-seedance-1-0-pro-250528",
  "prompt": "一隻貓在窗邊看雨,電影感",
  "images": ["https://example.com/cat.png"],
  "metadata": {
    "resolution": "1080p",
    "ratio": "16:9",
    "duration": 5
  }
}

上游火山方舟的 content[] 會由系統適配生成,終端用戶不需要手動傳入。