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 |
+
)
|