Source code for concepts.language.openai_utils.default_client

#! /usr/bin/env python3
# -*- coding: utf-8 -*-
# File   : default_client.py
# Author : Jiayuan Mao
# Email  : maojiayuan@gmail.com
# Date   : 08/21/2024
#
# This file is part of Project Concepts.
# Distributed under terms of the MIT license.

import openai

__all__ = ['get_default_client', 'get_default_chat_model', 'set_default_chat_model']


_DEFAULT_CLIENT = None



[docs]
def get_default_client():
    global _DEFAULT_CLIENT
    if _DEFAULT_CLIENT is None:
        _DEFAULT_CLIENT = openai.OpenAI()
    return _DEFAULT_CLIENT



_DEFAULT_CHAT_MODEL = "gpt-3.5-turbo"



[docs]
def get_default_chat_model():
    return _DEFAULT_CHAT_MODEL




[docs]
def set_default_chat_model(model):
    global _DEFAULT_CHAT_MODEL
    _DEFAULT_CHAT_MODEL = model