From c472cdc692a90f26ccb84699af677b2d00a5b17e Mon Sep 17 00:00:00 2001 From: Yajun Xu Date: Sat, 1 Apr 2023 18:55:53 +0800 Subject: [PATCH 1/2] 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, # 回复最大的字符数 From 2a6ed91a259e86e932466e18d3112a49369a2e13 Mon Sep 17 00:00:00 2001 From: Yajun Xu Date: Wed, 5 Apr 2023 09:37:34 +0800 Subject: [PATCH 2/2] set default value for open ai model --- model/openai/open_ai_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model/openai/open_ai_model.py b/model/openai/open_ai_model.py index fd5016b..1319a3c 100644 --- a/model/openai/open_ai_model.py +++ b/model/openai/open_ai_model.py @@ -18,7 +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') + self.model = model_conf(const.OPEN_AI).get('model', 'text-davinci-003') def reply(self, query, context=None):