py http方式接入
import requests
import json
API_KEY = 'sk-abcdfasdfsdfsfsdfsdf'
API_URL = 'http://www.hichat.site/api/v1/chat/completions'
headers = {
'Authorization': f'Bearer {API_KEY}',
'Content-Type': 'application/json',
'Accept': 'text/event-stream'
}
data = {
"model": "gpt-4o",
"messages": [
{
"role": "user",
"content": "百度是一家什么样的公司"
}
],
"max_tokens": 100,
"stream": True # 启用流式输出
}
response = requests.post(API_URL, headers=headers, json=data, stream=True)
for line in response.iter_lines():
if line:
decoded_line = line.decode('utf-8').strip()
print(decoded_line)
if decoded_line.startswith('data: '):
json_data = decoded_line[len('data: '):]
if json_data == '[DONE]':
break
try:
completion = json.loads(json_data)
print(completion)
except json.JSONDecodeError as e:
print(f"JSONDecodeError: {e}")
continue
修改于 2024-07-20 02:29:39