Spaces:
Sleeping
Sleeping
Tuchuanhuhuhu
commited on
Commit
•
23d295b
1
Parent(s):
0e91c63
bugfix: Azure OpenAI
Browse files- 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 |
+
)
|