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)