From c472cdc692a90f26ccb84699af677b2d00a5b17e Mon Sep 17 00:00:00 2001 From: Yajun Xu Date: Sat, 1 Apr 2023 18:55:53 +0800 Subject: [PATCH] fix openai model conf issue --- model/openai/open_ai_model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/model/openai/open_ai_model.py b/model/openai/open_ai_model.py index d514f0e..fd5016b 100644 --- a/model/openai/open_ai_model.py +++ b/model/openai/open_ai_model.py @@ -18,6 +18,7 @@ class OpenAIModel(Model): openai.api_base = api_base proxy = model_conf(const.OPEN_AI).get('proxy') log.info("[OPEN_AI] api_base={}".format(openai.api_base)) + self.model = model_conf(const.OPEN_AI).get('model') def reply(self, query, context=None): @@ -49,7 +50,7 @@ class OpenAIModel(Model): def reply_text(self, query, user_id, retry_count=0): try: response = openai.Completion.create( - model="text-davinci-003", # 对话模型的名称 + model=self.model, # 对话模型的名称 prompt=query, temperature=0.9, # 值在[0,1]之间，越大表示回复越具有不确定性 max_tokens=1200, # 回复最大的字符数