From 7828db2ca8a37e5c9b73c4901295574d4036b46c Mon Sep 17 00:00:00 2001 From: YingchaoX Date: Wed, 8 Nov 2023 18:19:15 +0800 Subject: [PATCH] Update chatgpt.py MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 增加openai api地址和模型配置 --- llmriddles/llms/chatgpt.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/llmriddles/llms/chatgpt.py b/llmriddles/llms/chatgpt.py index e1adbfc..a540792 100644 --- a/llmriddles/llms/chatgpt.py +++ b/llmriddles/llms/chatgpt.py @@ -1,20 +1,23 @@ from functools import lru_cache from openai import OpenAI +import os +_LLM_URL = os.environ.get('QUESTION_LLM_URL', None) +_LLM_MODEL = os.environ.get('QUESTION_LLM_MODEL', "gpt-3.5-turbo") from .base import register_llm @lru_cache() def _get_openai_client(api_key): - return OpenAI(api_key=api_key) + return OpenAI(api_key=api_key, base_url=_LLM_URL) def ask_chatgpt(message: str, api_key: str): client = _get_openai_client(api_key) response = client.chat.completions.create( - model="gpt-3.5-turbo", + model=_LLM_MODEL, messages=[ {"role": "user", "content": message} ],