added ora x-signed-token header
This commit is contained in:
parent
600525e550
commit
8350bc6842
4 changed files with 109 additions and 12 deletions
|
@ -3,6 +3,7 @@ from ora.typing import OraResponse
|
|||
from requests import post
|
||||
from time import time
|
||||
from random import randint
|
||||
from ora._jwt import do_jwt
|
||||
|
||||
user_id = None
|
||||
session_token = None
|
||||
|
@ -19,6 +20,15 @@ class Completion:
|
|||
cookies = {
|
||||
"cookie" : f"__Secure-next-auth.session-token={session_token}"} if session_token else {}
|
||||
|
||||
json_data = extra | {
|
||||
'chatbotId': model.id,
|
||||
'input' : prompt,
|
||||
'userId' : user_id if user_id else model.createdBy,
|
||||
'model' : model.modelName,
|
||||
'provider' : 'OPEN_AI',
|
||||
'includeHistory': includeHistory}
|
||||
|
||||
|
||||
response = post('https://ora.sh/api/conversation',
|
||||
headers = cookies | {
|
||||
"host" : "ora.sh",
|
||||
|
@ -26,14 +36,9 @@ class Completion:
|
|||
"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",
|
||||
"origin" : "https://ora.sh",
|
||||
"referer" : "https://ora.sh/chat/",
|
||||
"x-signed-token": do_jwt(json_data)
|
||||
},
|
||||
json = extra | {
|
||||
'chatbotId': model.id,
|
||||
'input' : prompt,
|
||||
'userId' : user_id if user_id else model.createdBy,
|
||||
'model' : model.modelName,
|
||||
'provider' : 'OPEN_AI',
|
||||
'includeHistory': includeHistory}).json()
|
||||
json = json_data).json()
|
||||
|
||||
if response.get('error'):
|
||||
raise Exception('''set ora.user_id and ora.session_token\napi response: %s''' % response['error'])
|
||||
|
|
75
ora/_jwt.py
Normal file
75
ora/_jwt.py
Normal file
|
@ -0,0 +1,75 @@
|
|||
import jwt
|
||||
from datetime import datetime, timedelta
|
||||
# from cryptography.hazmat.primitives import serialization
|
||||
# from cryptography.hazmat.primitives.serialization import load_pem_private_key
|
||||
# from cryptography.hazmat.backends import default_backend
|
||||
|
||||
|
||||
def do_jwt(json_data: dict):
|
||||
|
||||
private_key = b'''-----BEGIN RSA PRIVATE KEY-----
|
||||
MIIJKAIBAAKCAgEAxv9TLZP2TnsR512LqzT52N6Z9ixKmUA11jy0IXH0dEbdbfBw
|
||||
eeWrXoTuIYcY8Dkg/+q33ppfujYfb0z22bs/CZ63+jBL2UmxG/0XIzmsQlHSgJd/
|
||||
rnbERwIt7/ZjOHAcNrAzI0N11AI8AT0+M3XFOGRoIKzoc3Juxl7eyyPPEkNZMkEv
|
||||
lYfDN5AMD/+4pZ+7SCEzUCyGtBejW2P+NwTvjBxhLjIoG+m7yh81RoIBnO+Z1o5X
|
||||
ZtospuWZe1L6GNh+zezeHIyBGYgGgYbPboQ8QeHhoh+n0PuZB0GQqorqfxHjB38t
|
||||
yB4qsRGi10UNcohvFhglZk8kdMYBTd0M5ik5t4sx/ujjF57gX7dCKipHimDy7McY
|
||||
ElVLTDoSkwD/Lg3tV0utky42dL/iIMePlHfMrw/m2oAm33/dCaiAW8grNkJPjcwo
|
||||
Y8pnqpFGgAZX+6WalQCfoSStV4kYYlaq11DB6dZjDYoKLRIyH7MCAmMxms9569qe
|
||||
5gFuyQWTZgXlKoj2Zd7XIaIs5s/A6PFt7sxk8mOY/DspSbygZZCnMH3+or/8trH2
|
||||
p0fGEkqpzMKAY6TYtdYhOyTbup3VOKQwhk8b5CPuEWZutE6pT0O2O81MkuEEl/Zw
|
||||
/M1MJERTIjGAThsL0yvEn1Gi5HXl7s/5E61Yvc0ItORqio70PZcToRII27ECAwEA
|
||||
AQKCAgEAle0H3e78Q2S1uHriH7tqAdq0ZKQ6D/wwk5honkocwv4hFhNwqmY/FpdQ
|
||||
UjJWt6ZTFnzgyvXD6aedR13VHXXVqInMUtLQUoUSyuOD6yYogk7jKb76k5cnidg6
|
||||
g/A+EOdmWk2mOYs52uFUFBrwIhU44aPET9n1yAUPMKWJdcMk372eFh7GmwIOMm50
|
||||
qBkiJKaTk2RwJJdnZYfpq5FKlmlBkW5QSV3AmkcfFMkuelC4pmReoyfa8cKuoY+a
|
||||
cy+w/ccewkcTkK7LFVFGlY/b+IfoXjqwpFT1Op5UTQM420SOJ+5x/dPzyjHwODfx
|
||||
V/7OgtwH1b2bb9lwvgnwMZm5fi7RLAOC5BaSrZUb8WtVaaKURzXgdE+5LO/xXYCy
|
||||
JECbRQ5o4H4CwOc3mvJZL0O/dwPKoTccjELc8HOcogdy+hrJPXFl+oXy3yKUmf5L
|
||||
Lx13hh/kO4960TcGVQsUPV9oYB8XU5iYC1cMdlMVZAOwoLE1h/Tro0blisq6eafx
|
||||
+4ZS+COJEM+A7UgFacxdQ9C4bL5ZgjgLxMEsCIjwBN1i/bMEKpj46ulH23I57F1S
|
||||
jr6/UtMPO73c2bGcxdzRRQSI/LW5Qnb4USQsOIjYDVReLM9hDvI4OyQ2pfcgXlTL
|
||||
ODky2qivbP6WA4GKCBhaDEaeKFNDiyCqx9ObftCbRk1fWu7IP4ECggEBAOnPs88o
|
||||
DQLEaColCbh3ziogoANYMKiqaJUacnH5S5e1/aW3jgVK85NsMJT9hsODXyHup/CF
|
||||
RT3jeJA5cRj+04KI33cH2F5X9MhPB0a2Zo0Io813l95d2Wuk9rnadNCr8+h3b/nM
|
||||
HR4X+n7l0x6Y8sn60pxesYXKu8NFccUCVcGUvrrL2gsPLPB//3eqgfZuf8BCDzOB
|
||||
liO8Pzt0ELjxwxUWB9kPKLNZwVa0hq4snJThZQBrlMQcuH8BmitS5vZDVwiRLGVR
|
||||
L5z+tPJMz5wJ/dGbjyMMONCZgiXypqb1qHIGt8FEPLryQ6u+04ZszxW9QTsWqMqi
|
||||
ZvoFo0VPGkXGIWcCggEBANnh1tTCfGJSrwK1fWVhBajtn03iE5DuIkPUmL8juBq6
|
||||
LSYG8tuk+zt0RVNYLYrM2nSzU78IsuR/15XtpheDh3Fy1ZdsAe/boccdZUrLtH9h
|
||||
hRcAYUfY+E0E4U6j7FVTQCy9eNGnWJ/su2N0GDJll2BQWi8bcnL8dZqsq8pZzAjo
|
||||
7jBlOEe2xOVbCsBLfCW7tmeKCv4cc8digITGemig4NgCs6W03gJPnvnvvHMnuF3u
|
||||
8YjD9kWWEpQr37pT6QSdhwzKMAOeHbhh/CQO/sl+fBLbcYikQa0HIuvj+29w0/jv
|
||||
whVfsJxCvs6fCTMYjQE7GdTcGmCbvs+x7TrXuqeT8ycCggEAWr4Un/KAUjGd57Vm
|
||||
N2Sv6+OrloC0qdExM6UHA7roHqIwJg++G8nCDNYxaLGYiurCki3Ime1vORy+XuMc
|
||||
RMIpnoC2kcDGtZ7XTqJ1RXlnBZdz0zt2AoRT7JYid3EUYyRJTlCEceNI7bQKsRNL
|
||||
Q5XCrKce9DdAGJfdFWUvSXGljLLI70BMiHxESbazlGLle5nZFOnOcoP5nDbkJ5Pd
|
||||
JZoWx2k8dH6QokLUaW041AJWZuWvSGF4ZEBtTkV16xiKsMrjzVxiaZP/saOc4Gj1
|
||||
Li8mhiIkhEqrBjJ9s3KgQS4YSODYkjaEh12c69vsxkAWgu5nkaIysiojYyeq/Sw9
|
||||
GxVRQwKCAQAeYvTHL2iRfd6SjiUy4lkbuighgIoiCFQXCatT3PNsJtLtHsL4BwZS
|
||||
wGB6wy120iMVa30eg2QPohS7AC3N0bYuCEnpmFKc1RC26E6cI9TEfyFEl/T5RDU8
|
||||
6JVTlmD7dWTZ2ILlGmWtyCJKOIK3ZJu7/vjU4QsRJkxwiexbiDKAe5vcfAFhXwgO
|
||||
xKe3Mc/ao1dJEWN/FRDAmeg6nEOuG+G/voC3d4YO5HPTf6/Uj5GS6CQfYtUR12A3
|
||||
8fZ90f4Jer6+9ePEXWTftiqoDL9T8qPzLU+kMuRF8VzZcS472Ix3h1iWCoZjBJv/
|
||||
zQZHbgEcTtXHbfrvxkjSRopDTprljCi5AoIBAGc6M8/FH1pLgxOgS6oEGJAtErxv
|
||||
EnmELzKvfwBryphx8f0S5sHoiqli+5dqFtw5h5yy/pXrNzLi0LfpmFzxbChfO8ai
|
||||
omC/oqxU0FKqY2msFYdnfwM3PZeZ3c7LALLhWG56/fIYMtV78+cfqkRPM8nRJXaF
|
||||
Aza2YTTZGfh3x10KnSLWUmhIWUEj8VzCNW7SR0Ecqa+ordAYio4wBsq7sO3sCw8G
|
||||
Oi0/98ondhGJWL3M6FDGai8dXewt+8o0dlq95mHkNNopCWbPI71pM7u4ABPL50Yd
|
||||
spd4eADxTm2m0GR7bhVEIbYfc0aAzIoWDpVs4V3vmx+bdRbppFxV1aS/r0g=
|
||||
-----END RSA PRIVATE KEY-----'''
|
||||
|
||||
header = {
|
||||
'alg': 'RS256',
|
||||
'typ': 'JWT',
|
||||
'kid': '0d1bb0d7-45e4-445c-889e-57419470a570'
|
||||
}
|
||||
|
||||
payload = {
|
||||
**json_data,
|
||||
'iat': int(datetime.now().timestamp()),
|
||||
'exp': int((datetime.now() + timedelta(minutes=10)).timestamp()),
|
||||
'iss': 'https://ora.sh'
|
||||
}
|
||||
|
||||
return jwt.encode(payload, private_key, algorithm='RS256', headers=header)
|
12
ora/model.py
12
ora/model.py
|
@ -20,6 +20,12 @@ class CompletionModel:
|
|||
CompletionModel.description = description
|
||||
CompletionModel.slug = name
|
||||
|
||||
json_data = {
|
||||
'prompt' : system_prompt,
|
||||
'userId' : f'auto:{uuid4()}',
|
||||
'name' : name,
|
||||
'description': description}
|
||||
|
||||
headers = {
|
||||
'Origin' : 'https://ora.sh',
|
||||
'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.4 Safari/605.1.15',
|
||||
|
@ -27,11 +33,7 @@ class CompletionModel:
|
|||
'Host' : 'ora.sh',
|
||||
}
|
||||
|
||||
response = post('https://ora.sh/api/assistant', headers = headers, json = {
|
||||
'prompt' : system_prompt,
|
||||
'userId' : f'auto:{uuid4()}',
|
||||
'name' : name,
|
||||
'description': description})
|
||||
response = post('https://ora.sh/api/assistant', headers = headers, json = json_data)
|
||||
|
||||
print(response.json())
|
||||
|
||||
|
|
15
testing/ora_test.py
Normal file
15
testing/ora_test.py
Normal file
|
@ -0,0 +1,15 @@
|
|||
import ora
|
||||
|
||||
|
||||
# create model
|
||||
model = ora.CompletionModel.create(
|
||||
system_prompt = 'You are ChatGPT, a large language model trained by OpenAI. Answer as concisely as possible',
|
||||
description = 'ChatGPT Openai Language Model',
|
||||
name = 'gpt-3.5')
|
||||
|
||||
# init conversation (will give you a conversationId)
|
||||
init = ora.Completion.create(
|
||||
model = model,
|
||||
prompt = 'hello world')
|
||||
|
||||
print(init.completion.choices[0].text)
|
Loading…
Reference in a new issue