Update README.md

README.md

@@ -41,15 +41,34 @@ The current release version of Breexe-8x7B is v0.1.
 
 The API is in beta testing. If you are experiencing connectivity issues, please bear with us.
 
+Free trial API key: `'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VyX2lkIjoyLCJ1c2VybmFtZSI6ImdlbmVyYWxfcHVibGljIiwibm90ZSI6ImdlbmVyYWwgcHVibGljIn0.kCp68nRw3RSh3jbMm8FvhG0NIkStflgI1wTHLviRPQE'`
+
 Get started here:
 
 ```python
+import time
+
 from openai import OpenAI
 
-API_KEY = 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VyX2lkIjoyLCJ1c2VybmFtZSI6ImdlbmVyYWxfcHVibGljIiwibm90ZSI6ImdlbmVyYWwgcHVibGljIn0.kCp68nRw3RSh3jbMm8FvhG0NIkStflgI1wTHLviRPQE'
+API_KEY = <API_KEY_HERE>
 BASE_URL = 'https://api-mtkresearch.com/v1'
 MODEL_NAME = 'BreeXe-8x7B'
 
+client = OpenAI(base_url=BASE_URL, api_key=API_KEY)
+
+
+def ask_breexe(messages):    
+    completion = client.chat.completions.create(
+        model=MODEL_NAME,
+        messages=messages,
+        temperature=0.01,
+        top_p=0.01,
+        max_tokens=512
+    )
+    response = completion.choices[0].message.content
+    time.sleep(3)  # due to a rate limit of 200 requests per 10 minutes
+    return response
+
 messages = [
     {
         'role': 'system',
@@ -61,15 +80,7 @@ messages = [
     }
 ]
 
-client = OpenAI(base_url=BASE_URL, api_key=API_KEY)
-completion = client.chat.completions.create(
-    model=MODEL_NAME,
-    messages=messages,
-    temperature=0.01,
-    top_p=0.01,
-    max_tokens=512
-)
-response = completion.choices[0].message.content
+response = ask_breexe(messages)
 print(response)
 ```