import os | |
import streamlit as st | |
import requests | |
def get_completion_from_openai(prompt, max_tokens = None): | |
url = os.getenv('OPENAI_COMPLETION_URL') | |
headers = { | |
"Content-Type": "application/json", | |
"Authorization": "Bearer " + st.secrets["OPENAI_TOKEN"], | |
} | |
response = requests.post(url, | |
json={ | |
"model": os.getenv('OPENAI_MODEL'), | |
"max_tokens": max_tokens, | |
"messages": [ | |
{ | |
"role": "user", | |
"content": prompt | |
} | |
] | |
}, | |
headers=headers, | |
stream=False, | |
) | |
try: | |
return response.json()['choices'][0]['message']['content'] | |
except: | |
print(response.json()) | |
return "Произошла ошибка" | |