Python 多线程批量请求 OpenAI API
为了使用 Python 多线程批量请求 OpenAI,你可以使用 Python 的内置模块 threading 和 requests。具体步骤如下:
- 导入所需的模块:
import threading
import requests
- 定义一个函数,该函数接收一个参数 url 并发送 GET 请求:
def get_url(url):
response = requests.get(url)
print(response.text)
- 定义一个函数,该函数接收一个参数 urls 列表并启动多个线程:
def get_urls(urls):
threads = []
for url in urls:
thread = threading.Thread(target=get_url, args=(url,))
threads.append(thread)
thread.start()
for thread in threads:
thread.join()
- 调用 get_urls 函数并传递一个包含多个 URL 的列表作为参数:
urls = ['https://api.openai.com/v1/engines/davinci-codex/completions', 'https://api.openai.com/v1/engines/text-davinci-002/completions']
get_urls(urls)
注意:在实际使用中,你需要根据自己的需求对 get_url 函数进行修改,例如添加请求头、请求体等。
完整代码如下:
import threading
import requests
def get_url(url):
response = requests.get(url)
print(response.text)
def get_urls(urls):
threads = []
for url in urls:
thread = threading.Thread(target=get_url, args=(url,))
threads.append(thread)
thread.start()
for thread in threads:
thread.join()
urls = ['https://api.openai.com/v1/engines/davinci-codex/completions', 'https://api.openai.com/v1/engines/text-davinci-002/completions']
get_urls(urls)
原文地址: https://www.cveoy.top/t/topic/ndY9 著作权归作者所有。请勿转载和采集!