ishangsf/DeepCodeGeniusWeb-tools
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

remove minimax model support

Browse Source
This commit is contained in:
bobo.yang 2024-01-31 13:01:24 +08:00
parent 9fdaf4c1db
commit 44325ab185
1 changed files with 0 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
site-packages/devchat/openai/openai_chat.py
View File

@ -7,7 +7,6 @@ from devchat.chat import Chat
from devchat.utils import get_user_info, user_id from devchat.utils import get_user_info, user_id
from .openai_message import OpenAIMessage from .openai_message import OpenAIMessage
from .openai_prompt import OpenAIPrompt from .openai_prompt import OpenAIPrompt
from .minimax_chat import chat_completion, stream_chat_completion
class OpenAIChatParameters(BaseModel, extra='ignore'): class OpenAIChatParameters(BaseModel, extra='ignore'):
@ -69,9 +68,6 @@ class OpenAIChat(Chat):
config_params['function_call'] = 'auto' config_params['function_call'] = 'auto'
config_params['stream'] = False config_params['stream'] = False
if config_params['model'].startswith('abab'):
return chat_completion(prompt.messages, config_params)
client = openai.OpenAI( client = openai.OpenAI(
api_key=os.environ.get("OPENAI_API_KEY", None), api_key=os.environ.get("OPENAI_API_KEY", None),
base_url=os.environ.get("OPENAI_API_BASE", None) base_url=os.environ.get("OPENAI_API_BASE", None)
@ -93,9 +89,6 @@ class OpenAIChat(Chat):
config_params['function_call'] = 'auto' config_params['function_call'] = 'auto'
config_params['stream'] = True config_params['stream'] = True
if config_params['model'].startswith('abab'):
return stream_chat_completion(prompt.messages, config_params)
client = openai.OpenAI( client = openai.OpenAI(
api_key=os.environ.get("OPENAI_API_KEY", None), api_key=os.environ.get("OPENAI_API_KEY", None),
base_url=os.environ.get("OPENAI_API_BASE", None) base_url=os.environ.get("OPENAI_API_BASE", None)