Overview

1
from openai import OpenAI
2
3
client = OpenAI(
4
    base_url="https://api.reka.ai/v1",
5
    api_key="YOUR_API_KEY",
6
)
7
8
response = client.chat.completions.create(
9
    model="reka-flash",
10
    messages=[
11
        {
12
            "content": "What is the fifth prime number?",
13
            "role": "user",
14
        }
15
    ],
16
)
17
print(response.choices[0].message.content)

1
from openai import OpenAI
2
3
client = OpenAI(
4
    base_url="https://api.reka.ai/v1",
5
    api_key="YOUR_API_KEY",
6
)
7
8
response = client.chat.completions.create(
9
    model="reka-flash",
10
    messages=[
11
        {
12
            "content": "Write a python one-liner to flatten a list of lists.",
13
            "role": "user",
14
        }
15
    ],
16
)
17
print(response.choices[0].message.content)

1
from openai import OpenAI
2
3
client = OpenAI(
4
    base_url="https://api.reka.ai/v1",
5
    api_key="YOUR_API_KEY",
6
)
7
8
response = client.chat.completions.create(
9
    model="reka-flash",
10
    messages=[
11
        {
12
            "content": "My name is Matt.",
13
            "role": "user",
14
        },
15
        {
16
            "content": "Hello Matt! How can I help you today?",
17
            "role": "assistant",
18
        },
19
        {
20
            "content": "Can you think of a couple of famous people with the same name as me?",
21
            "role": "user",
22
        },
23
    ],
24
)
25
print(response.choices[0].message.content)

1
from openai import OpenAI
2
3
client = OpenAI(
4
    base_url="https://api.reka.ai/v1",
5
    api_key="YOUR_API_KEY",
6
)
7
8
prompt = """
9
Below is a paragraph from wikipedia:
10
11
The Solar System is the gravitationally bound system of the Sun and the objects that orbit it.
12
The largest of such objects are the eight planets, in order from the Sun: four terrestrial planets named Mercury,
13
Venus, Earth and Mars, two gas giants named Jupiter and Saturn, and two ice giants named Uranus and Neptune.
14
The terrestrial planets have a definite surface and are mostly made of rock and metal. The gas giants are
15
mostly made of hydrogen and helium, while the ice giants are mostly made of 'volatile' substances such as water,
16
ammonia, and methane. In some texts, these terrestrial and giant planets are called the inner Solar System and outer
17
Solar System planets respectively.
18
19
Extract information about the planets from this paragraph as a JSON list of objects with keys 'planetName' and
20
'composition'. The 'composition' key should contain one or two words, and there should be no other keys.
21
""".strip()
22
23
json_prefix = """
24
[
25
    {
26
        "planetName":
27
""".strip()
28
29
response = client.chat.completions.create(
30
    model="reka-flash",
31
    messages=[
32
        {"role": "user", "content": prompt},
33
        {
34
            "role": "assistant",
35
            "content": (
36
                "Sure, here is a JSON object conforming to that format:\n\n"
37
                f"```json\n{json_prefix}"
38
            ),
39
        },
40
    ],
41
    max_tokens=512,
42
    temperature=0.4,
43
    stop=["```\n"],
44
)
45
print(json_prefix + response.choices[0].message.content)

1
from openai import OpenAI
2
3
client = OpenAI(
4
    base_url="https://api.reka.ai/v1",
5
    api_key="YOUR_API_KEY",
6
)
7
8
response = client.chat.completions.create(
9
    model="reka-flash",
10
    stream=True,
11
    max_tokens=2048,
12
    messages=[
13
        {
14
            "content": "Write a detailed template NDA contract between two parties.",
15
            "role": "user",
16
        }
17
    ],
18
)
19
20
for chunk in response:
21
    content = chunk.choices[0].delta.content
22
    if content:
23
        print(content, end="")