41 lines
1.7 KiB
Python
41 lines
1.7 KiB
Python
|
import requests
|
||
|
|
|
||
|
|
class Completion:
|
||
|
|
def create(prompt: str,
|
||
|
|
model: str = 'openai:gpt-3.5-turbo',
|
||
|
|
temperature: float = 0.7,
|
||
|
|
max_tokens: int = 200,
|
||
|
|
top_p: float = 1,
|
||
|
|
top_k: int = 1,
|
||
|
|
frequency_penalty: float = 1,
|
||
|
|
presence_penalty: float = 1,
|
||
|
|
stopSequences: list = []):
|
||
|
|
|
||
|
|
token = requests.get('https://play.vercel.ai/openai.jpeg', headers={
|
||
|
|
'authority': 'play.vercel.ai',
|
||
|
|
'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',
|
||
|
|
'referer': 'https://play.vercel.ai/',
|
||
|
|
'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'}).text.replace('=','')
|
||
|
|
|
||
|
|
print(token)
|
||
|
|
|
||
|
|
headers = {
|
||
|
|
'authority': 'play.vercel.ai',
|
||
|
|
'custom-encoding': token,
|
||
|
|
'origin': 'https://play.vercel.ai',
|
||
|
|
'referer': 'https://play.vercel.ai/',
|
||
|
|
'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'
|
||
|
|
}
|
||
|
|
|
||
|
|
for chunk in requests.post('https://play.vercel.ai/api/generate', headers=headers, stream=True, json={
|
||
|
|
'prompt': prompt,
|
||
|
|
'model': model,
|
||
|
|
'temperature': temperature,
|
||
|
|
'maxTokens': max_tokens,
|
||
|
|
'topK': top_p,
|
||
|
|
'topP': top_k,
|
||
|
|
'frequencyPenalty': frequency_penalty,
|
||
|
|
'presencePenalty': presence_penalty,
|
||
|
|
'stopSequences': stopSequences}).iter_lines():
|
||
|
|
|
||
|
|
yield (chunk)
|