Spaces:
Sleeping
Sleeping
File size: 715 Bytes
f4b691b 6775e53 f4b691b 6775e53 f4b691b |
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 |
from openai import OpenAI
import os
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, client, 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
|