Tuchuanhuhuhu commited on
Commit
23d295b
1 Parent(s): 0e91c63

bugfix: Azure OpenAI

Browse files
Files changed (1) hide show
  1. modules/models/azure.py +3 -2
modules/models/azure.py CHANGED
@@ -1,4 +1,4 @@
1
- from langchain.chat_models import AzureChatOpenAI
2
  import os
3
 
4
  from .base_model import Base_Chat_Langchain_Client
@@ -14,4 +14,5 @@ class Azure_OpenAI_Client(Base_Chat_Langchain_Client):
14
  deployment_name=os.environ["AZURE_DEPLOYMENT_NAME"],
15
  openai_api_key=os.environ["AZURE_OPENAI_API_KEY"],
16
  openai_api_type="azure",
17
- )
 
 
1
+ from langchain.chat_models import AzureChatOpenAI, ChatOpenAI
2
  import os
3
 
4
  from .base_model import Base_Chat_Langchain_Client
 
14
  deployment_name=os.environ["AZURE_DEPLOYMENT_NAME"],
15
  openai_api_key=os.environ["AZURE_OPENAI_API_KEY"],
16
  openai_api_type="azure",
17
+ streaming=True
18
+ )