Overview

1
from reka.client import Reka
2
3
# You can also set the API key using the REKA_API_KEY environment variable.
4
client = Reka(api_key="YOUR_API_KEY")
5
6
response = client.chat.create(
7
    messages=[
8
        {
9
            "content": "What is the fifth prime number?",
10
            "role": "user",
11
        }
12
    ],
13
    model="reka-core-20240501",
14
)
15
print(response.responses[0].message.content)

1
from reka.client import Reka
2
3
# You can also set the API key using the REKA_API_KEY environment variable.
4
client = Reka(api_key="YOUR_API_KEY")
5
6
response = client.chat.create(
7
    messages=[
8
        {
9
            "content": "Write a python one-liner to flatten a list of lists.",
10
            "role": "user",
11
        }
12
    ],
13
    model="reka-core-20240501",
14
)
15
print(response.responses[0].message.content)

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

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

1
from reka.client import Reka
2
3
client = Reka(api_key="YOUR_API_KEY")
4
5
response = client.chat.create_stream(
6
    messages=[
7
        ChatMessage(
8
            content="Write a detailed template NDA contract between two parties.",
9
            role="user",
10
        )
11
    ],
12
    max_tokens=2048,
13
    model="reka-core-20240501",
14
)
15
16
for chunk in response:
17
    print(chunk.responses[0].chunk.content)

1
import asyncio
2
from reka.client import AsyncReka
3
4
client = AsyncReka(api_key="YOUR_API_KEY")
5
max_concurrent_requests = 2
6
semaphore = asyncio.Semaphore(max_concurrent_requests)
7
8
async def respond(prompt: str) -> str:
9
    async with semaphore:
10
        response = await client.chat.create(
11
            messages=[
12
                {
13
                    "content": prompt,
14
                    "role": "user",
15
                }
16
            ],
17
            model="reka-flash",
18
        )
19
    return response.responses[0].message
20
21
async def main():
22
    prompts = [
23
        "What is your name?",
24
        "What is the fifth prime number?",
25
        "Write a python one-liner to flatten a list of lists.",
26
    ]
27
    responses = await asyncio.gather(*[respond(prompt) for prompt in prompts])
28
    return responses
29
30
asyncio.run(main())