File size: 931 Bytes
8a469fd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from scripts.physton_prompt.get_lang import get_lang
from scripts.physton_prompt.get_translate_apis import unprotected_translate_api_config
def gen_openai(messages, api_config):
import openai
api_config = unprotected_translate_api_config('chatgpt_key', api_config)
openai.api_base = api_config.get('api_base', 'https://api.openai.com/v1')
openai.api_key = api_config.get('api_key', '')
model = api_config.get('model', 'gpt-3.5-turbo')
if not openai.api_key:
raise Exception(get_lang('is_required', {'0': 'API Key'}))
if not messages or len(messages) == 0:
raise Exception(get_lang('is_required', {'0': 'messages'}))
completion = openai.ChatCompletion.create(model=model, messages=messages, timeout=60)
if len(completion.choices) == 0:
raise Exception(get_lang('no_response_from', {'0': 'OpenAI'}))
content = completion.choices[0].message.content
return content
|