vlff李飞飞 commited on
Commit
a7f9822
1 Parent(s): 5b67568
Files changed (1) hide show
  1. qwen_agent/llm/qwen_oai.py +1 -2
qwen_agent/llm/qwen_oai.py CHANGED
@@ -9,11 +9,10 @@ class QwenChatAsOAI(BaseChatModel):
9
 
10
  def __init__(self, model: str, api_key: str, model_server: str):
11
  super().__init__()
12
-
13
  if model_server.strip().lower() != 'openai':
14
  openai.api_base = os.getenv('OPENAI_API_BASE', model_server)
15
  openai.api_key = api_key.strip() or os.getenv('OPENAI_API_KEY', 'EMPTY')
16
- self.model = model
17
 
18
  def _chat_stream(
19
  self,
 
9
 
10
  def __init__(self, model: str, api_key: str, model_server: str):
11
  super().__init__()
 
12
  if model_server.strip().lower() != 'openai':
13
  openai.api_base = os.getenv('OPENAI_API_BASE', model_server)
14
  openai.api_key = api_key.strip() or os.getenv('OPENAI_API_KEY', 'EMPTY')
15
+ self.model = os.getenv('OPENAI_MODEL_NAME', model)
16
 
17
  def _chat_stream(
18
  self,