File size: 745 Bytes
886d623
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
import openai
import os

from dotenv import load_dotenv, find_dotenv
_ = load_dotenv(find_dotenv())
openai.api_key = os.environ['OPENAI_API_KEY']
MODEL_NAME = 'gpt-3.5-turbo'

class Assigment3:
    input_val = ''
    def __init__(self, model):
        self.model = model
        
    def get_input(self):
        self.input_val = input('Enter your message')

    def get_completion(self):
        messages = [{
            'role': 'user',
            'content': self.input_val
        }]
        response = openai.ChatCompletion.create(
            model=self.model,
            messages=messages,
            temperature=0
        )
        return response.choices[0].message.count

    
obj = Assigment3(MODEL_NAME)
print(obj.get_completion())