From 44325ab18520ac6bb650a3e3abdc83d7361f7e2c Mon Sep 17 00:00:00 2001 From: "bobo.yang" Date: Wed, 31 Jan 2024 13:01:24 +0800 Subject: [PATCH] remove minimax model support --- site-packages/devchat/openai/openai_chat.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/site-packages/devchat/openai/openai_chat.py b/site-packages/devchat/openai/openai_chat.py index 5aed702..81a424d 100644 --- a/site-packages/devchat/openai/openai_chat.py +++ b/site-packages/devchat/openai/openai_chat.py @@ -7,7 +7,6 @@ from devchat.chat import Chat from devchat.utils import get_user_info, user_id from .openai_message import OpenAIMessage from .openai_prompt import OpenAIPrompt -from .minimax_chat import chat_completion, stream_chat_completion class OpenAIChatParameters(BaseModel, extra='ignore'): @@ -69,9 +68,6 @@ class OpenAIChat(Chat): config_params['function_call'] = 'auto' config_params['stream'] = False - if config_params['model'].startswith('abab'): - return chat_completion(prompt.messages, config_params) - client = openai.OpenAI( api_key=os.environ.get("OPENAI_API_KEY", None), base_url=os.environ.get("OPENAI_API_BASE", None) @@ -93,9 +89,6 @@ class OpenAIChat(Chat): config_params['function_call'] = 'auto' config_params['stream'] = True - if config_params['model'].startswith('abab'): - return stream_chat_completion(prompt.messages, config_params) - client = openai.OpenAI( api_key=os.environ.get("OPENAI_API_KEY", None), base_url=os.environ.get("OPENAI_API_BASE", None)