-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathcall_gpt.py
36 lines (31 loc) · 1012 Bytes
/
call_gpt.py
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
36
from openai import OpenAI
import asyncio
client = OpenAI()
model_name = 'gpt-4o-2024-05-13'
max_tokens = 4096
temperature = 0.0
messages = [
{"role": "user", "content": "大喜利しましょう。とても面白い回答をしてくださいね。"},
{"role": "assistant", "content": "おけ、任せて"},
{"role": "user", "content": "こんな台風は嫌だ、どんな台風?"}
]
response = client.chat.completions.create(
max_tokens=max_tokens,
messages=messages,
temperature=temperature,
model=model_name,
)
print(response.choices[0].message.content)
from utils import retry_with_linear_backoff
from openai import AsyncOpenAI
client = AsyncOpenAI()
@retry_with_linear_backoff(delay=60, max_retries=5)
async def main() -> None:
response = await client.chat.completions.create(
max_tokens=max_tokens,
messages=messages,
temperature=temperature,
model=model_name,
)
print(response.choices[0].message.content)
asyncio.run(main())