File size: 847 Bytes
df5beb9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
import requests
import json
import os
url = 'https://internlm-chat.intern-ai.org.cn/puyu/api/v1/chat/completions'
header = {
'Content-Type': 'application/json',
"Authorization": f"Bearer {os.environ['INTERNLM_API_KEY']}"
}
def complete(messages, model="internlm3-latest", temperature=None, top_p=None):
data = {
"model": model,
"messages": messages
}
if temperature is not None:
data["temperature"] = temperature
if top_p is not None:
data["top_p"] = top_p
res = requests.post(url, headers=header, data=json.dumps(data))
if res.status_code == 200:
return res.json()["choices"][0]['message']["content"]
else:
return None
if __name__ == "__main__":
response = complete(
messages=[{"role": "user", "content": "你好~"}]
)
print(response)
|