tools / utils /upload_test /batch_create.py
Adinosaur's picture
Upload folder using huggingface_hub
1c980b1 verified
raw
history blame contribute delete
874 Bytes
import os
from openai import OpenAI
client = OpenAI(
# 若没有配置环境变量,可用阿里云百炼API Key将下行替换为:api_key="sk-xxx"。但不建议在生产环境中直接将API Key硬编码到代码中,以减少API Key泄露风险。
api_key=os.getenv("DASHSCOPE_API_KEY"),
base_url="https://dashscope.aliyuncs.com/compatible-mode/v1", # 阿里云百炼服务的base_url
)
batch = client.batches.create(
input_file_id="oss:cn-beijing:acm-mm-reason/test/vlm_test100.jsonl", # 上传文件返回的id或OSS文件URL或OSS文件资源标识符
endpoint="/v1/chat/completions", # Embedding文本向量模型填写/v1/embeddings,测试模型batch-test-model填写/v1/chat/ds-test,其他模型填写/v1/chat/completions
completion_window="10h",
metadata={'ds_name':"vlm_test100",'ds_description':'finally test'}
)
print(batch)