编辑代码

import asyncio
import os
from openai import OpenAI

client = OpenAI(
    # 替换为您需要调用的模型服务Base Url
    base_url="https://ark.cn-beijing.volces.com/api/v3/",
    # 环境变量中配置您的API Key
    api_key=os.environ.get("46adea03-6bc0-4a71-bbe9-4902eb913bb0")
)

async def send_request():
    completion = await client.chat.completions.acreate(
        model="<Model>",
        messages=[
            {"role": "system", "content": "你是豆包，是由字节跳动开发的 AI 人工智能助手"},
            {"role": "user", "content": "常见的十字花科植物有哪些？"},
        ],
    )
    # print(completion.choices[0].message.content)  # 可以选择注释掉减少输出
    return completion


async def main():
    num_requests = 100  # 并发请求数量，可以根据需要调整
    tasks = [send_request() for _ in range(num_requests)]
    results = await asyncio.gather(*tasks)

    # 可以选择在这里处理结果，例如统计成功率、平均响应时间等
    success_count = 0
    for result in results:
      try:
        # 检查是否调用成功
        result.choices[0].message.content
        success_count +=1
      except Exception as e:
        print(f"Request failed: {e}")

    print(f"Total requests: {num_requests}, Success requests: {success_count}")



if __name__ == "__main__":
    asyncio.run(main())