Spaces:
Running
Running
File size: 1,182 Bytes
806d7c6 d487adb 806d7c6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
import openai
class ChatGPTAPI:
def __init__(self, api_key: str = '', max_input_length: int = 1024):
if not api_key:
try:
api_key = open('data/api_key.txt', 'r').read()
except Exception as e:
raise Exception(f'ChatGPT Error: No API key provided {e}')
openai.api_key = api_key
self.max_input_length = max_input_length
def __call__(self, content: str):
assert isinstance(content, str), 'ChatGPT Error: content must be a string'
content = content.strip()
messages = [{'role': 'user', 'content': content}]
try:
resp = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=messages
)
output: str = resp['choices'][0]['message']['content']
output = output.strip()
except Exception as e:
raise Exception(f'ChatGPT Error: {e}')
return output
if __name__ == '__main__':
chatgpt = ChatGPTAPI()
r = chatgpt.truncate_string('how are you ' * 10000)
r_list = r.split(' ')
# response = chatgpt('Hello, how are you?')
# print(response)
|