summaryrefslogtreecommitdiffstats
path: root/t3nsor
diff options
context:
space:
mode:
Diffstat (limited to 't3nsor')
-rw-r--r--t3nsor/__init__.py24
-rw-r--r--t3nsor/t3nsor_test.py7
2 files changed, 27 insertions, 4 deletions
diff --git a/t3nsor/__init__.py b/t3nsor/__init__.py
index 255fea4f..f8d12055 100644
--- a/t3nsor/__init__.py
+++ b/t3nsor/__init__.py
@@ -1,6 +1,24 @@
from requests import post
from time import time
+headers = {
+ 'authority': 'www.t3nsor.tech',
+ 'accept': '*/*',
+ 'accept-language': 'en,fr-FR;q=0.9,fr;q=0.8,es-ES;q=0.7,es;q=0.6,en-US;q=0.5,am;q=0.4,de;q=0.3',
+ 'cache-control': 'no-cache',
+ 'content-type': 'application/json',
+ 'origin': 'https://www.t3nsor.tech',
+ 'pragma': 'no-cache',
+ 'referer': 'https://www.t3nsor.tech/',
+ 'sec-ch-ua': '"Chromium";v="112", "Google Chrome";v="112", "Not:A-Brand";v="99"',
+ 'sec-ch-ua-mobile': '?0',
+ 'sec-ch-ua-platform': '"macOS"',
+ 'sec-fetch-dest': 'empty',
+ 'sec-fetch-mode': 'cors',
+ 'sec-fetch-site': 'same-origin',
+ 'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36',
+}
+
class T3nsorResponse:
class Completion:
@@ -53,9 +71,7 @@ class Completion:
prompt: str = 'hello world',
messages: list = []) -> T3nsorResponse:
- raise Exception('The t3nsor.tech API is currently down.')
-
- response = post('https://www.t3nsor.tech/api/chat', json = Completion.model | {
+ response = post('https://www.t3nsor.tech/api/chat', headers = headers, json = Completion.model | {
'messages' : messages,
'key' : '',
'prompt' : prompt
@@ -91,7 +107,7 @@ class StreamCompletion:
prompt: str = 'hello world',
messages: list = []) -> T3nsorResponse:
- response = post('https://www.t3nsor.tech/api/chat', stream = True, json = Completion.model | {
+ response = post('https://www.t3nsor.tech/api/chat', headers = headers, stream = True, json = Completion.model | {
'messages' : messages,
'key' : '',
'prompt' : prompt
diff --git a/t3nsor/t3nsor_test.py b/t3nsor/t3nsor_test.py
new file mode 100644
index 00000000..eb8e2ae8
--- /dev/null
+++ b/t3nsor/t3nsor_test.py
@@ -0,0 +1,7 @@
+import t3nsor
+
+for response in t3nsor.StreamCompletion.create(
+ prompt = 'write python code to reverse a string',
+ messages = []):
+
+ print(response.completion.choices[0].text)