示例代码

ECNU 大语言模型平台提供的所有 API 都遵循 openai 规范。您可以直接使用任意兼容 openai 的第三方库来调用 ECNU 大语言模型平台的 API。

注意当 API 中包含了我们自定义的能力扩展时（例如 ecnu-max 所支持的联网检索能力），openai 库不会处理这些扩展字段，但不影响其他部分的使用。

使用 openai sdk

import os
from openai import OpenAI

client = OpenAI(
    api_key='your-api-key', 
    base_url="https://chat.ecnu.edu.cn/open/api/v1",
)
completion = client.chat.completions.create(
    model="ecnu-plus", # 模型列表：https://developer.ecnu.edu.cn/vitepress/llm/api/models.html
    messages=[
        {'role': 'system', 'content': 'You are a helpful assistant.'},
        {'role': 'user', 'content': '你是谁？'}],
    )
    
print(completion.model_dump_json())

直接调用接口

import requests

url = "https://chat.ecnu.edu.cn/open/api/v1/chat/completions"
headers = {
    "Authorization":"Bearer your-api-key",
    "Content-Type":"application/json"
}

data = {
    "model": "ecnu-max",
    "messages": [
        {"role": "system", "content": "You are a helpful assistant."},
        {"role": "user", "content": "华东师范大学是哪一年成立的？"}
    ],
    "search_mode":"enable"
}

response = requests.post(url, headers=headers, json=data)
print(response.json())