WRaThioN: WRTN Client For Python

Warning

This is an unofficial client. This package does not take into account WRTN's terms and conditions. Any dispute with WRTN regarding the terms and conditions is entirely your responsibility.

Installation

$ pip install WRaThioN

Usage

Requirements

• Python3 or newer
• WRTN Account

Get Cookies

To use WRaThioN, you need a refresh token and an id called '__w_id' in the cookie.

You can retrieve cookies from WRTN as follows:

1. Go to WRTN.
2. Log in with your WRTN account.
3. Open developer tools in your browser (usually by pressing 'F12').
4. You can view all cookies associated with 'https://wrtn.ai' by selecting the 'Application' tab and clicking on the 'Cookies' option.
5. Find the '__w_id' and 'refresh_token' cookies and click on them to expand their details.
6. Copy the value of each cookie.

Use Prompt

$ python3 -m WRaThioN --help
usage: WRaThioN.py [-h] [--token TOKEN] [--id ID]

WRTN AI Prompt

options:
  -h, --help     show this help message and exit
  --token TOKEN  refresh token
  --id ID        wrtn id(__w_id)

Example

from WRaThioN import WRaThioN
import argparse

def main() -> None:
    client = WRaThioN(args.token, args.id)

    client.create_chat()
    while True:
        prompt = input("You: ")

        if prompt == "!reset":
            client.create_chat()
            continue
        elif prompt == "!exit":
            client.delete_chat()
            break

        print("Bot: ", end="", flush=True)
        response = client.chat(prompt)
        print(response['message']['content'], end="", flush=True)
        print("\n")

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='WRTN AI Prompt')
    parser.add_argument('--token', type=str, help='refresh token')
    parser.add_argument('--id', type=str, help='wrtn id(__w_id))')

    args = parser.parse_args()
    
    main()