mirror of
https://github.com/SevaSk/ecoute.git
synced 2024-12-19 04:37:52 +00:00
21 lines
680 B
Python
21 lines
680 B
Python
import openai
|
|
from keys import OPENAI_API_KEY
|
|
from prompts import create_prompt, INITIAL_RESPONSE
|
|
|
|
openai.api_key = OPENAI_API_KEY
|
|
|
|
class GPTResponder:
|
|
def __init__(self):
|
|
self.last_response = INITIAL_RESPONSE
|
|
|
|
def generate_response_from_transcript(self, transcript):
|
|
response = openai.ChatCompletion.create(
|
|
model="gpt-3.5-turbo-0301",
|
|
messages=[{"role": "system", "content": create_prompt(transcript)}],
|
|
temperature = 0.0
|
|
)
|
|
full_response = response.choices[0].message.content
|
|
try:
|
|
self.last_response = full_response.split('[')[1].split(']')[0]
|
|
except:
|
|
pass |