AutoSAS/caller.py at master · 0ceanlight/AutoSAS · GitHub

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
37
38
39
40
from openai import OpenAI
client = OpenAI()


def caller(prompt, system = None):
    messages = [

        {
        "role": "user",
        "content": [
            {
            "type": "text",
            "text": "Hello"
            }
        ]
        },
        {
        "role": "assistant",
        "content": [
            {
            "type": "text",
            "text": "Hello! How can I assist you today?"
            }
        ]
        }
    ]

    response = client.chat.completions.create(
    model="gpt-4o-mini",
    messages= messages,
    response_format={
        "type": "text"
    },
    temperature=1,
    max_completion_tokens=2048,
    top_p=1,
    frequency_penalty=0,
    presence_penalty=0
    )