Spaces:
Sleeping
Sleeping
File size: 1,019 Bytes
f6c123b |
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 38 39 |
from openai import OpenAI
from dotenv import load_dotenv
from prompts import UserRolePrompt, SystemRolePrompt
import os
load_dotenv()
class ChatOpenAI:
def __init__(self, model_name: str = "gpt-3.5-turbo"):
self.model_name = model_name
self.openai_api_key = os.getenv("OPENAI_API_KEY")
if self.openai_api_key is None:
raise ValueError("OPENAI_API_KEY is not set")
def run(self, messages, text_only: bool = True):
if not isinstance(messages, list):
raise ValueError("messages must be a list")
client = OpenAI()
response = client.chat.completions.create(
model=self.model_name, messages=messages
)
if text_only:
return response.choices[0].message.content
return response
if __name__ == "__main__":
chat = ChatOpenAI()
prompt = UserRolePrompt("Hello, I am a human.")
prompt = prompt.create_message()
print(prompt)
response = chat.run([prompt])
print(response)
|