FictionAgent / core /openai_client.py
gdwind's picture
Upload folder using huggingface_hub
a226682 verified
from openai import OpenAI
from config import Config
class OpenAIClient:
"""OpenAI客户端单例"""
_instance = None
_client = None
def __new__(cls):
if cls._instance is None:
cls._instance = super(OpenAIClient, cls).__new__(cls)
return cls._instance
def __init__(self):
if self._client is None:
self._client = OpenAI(
api_key=Config.OPENAI_API_KEY,
base_url=Config.OPENAI_BASE_URL
)
@property
def client(self):
return self._client
@classmethod
def get_client(cls) -> OpenAI:
"""获取OpenAI客户端实例"""
instance = cls()
return instance.client