16 lines
446 B
Python
16 lines
446 B
Python
|
import openai
|
||
|
|
|
||
|
|
|
||
|
|
class ChatGPT():
|
||
|
|
def __init__(self, model_path = 'gpt-3.5-turbo', api_key = None):
|
||
|
|
openai.api_key = api_key
|
||
|
|
self.model_path = model_path
|
||
|
|
|
||
|
|
def chat(self, message):
|
||
|
|
response = openai.ChatCompletion.create(
|
||
|
|
model=self.model_path,
|
||
|
|
messages=[
|
||
|
|
{"role": "user", "content": message}
|
||
|
|
]
|
||
|
|
)
|
||
|
|
return response['choices'][0]['message']['content']
|