fix configuration to support streaming answer with Azure

1 year ago · 5eda42ff31
parent 84168e22d0
commit 5eda42ff31
1 changed files with 4 additions and 0 deletions
--- a/application/app.py
+++ b/application/app.py
@ -159,6 +159,10 @@ def complete_stream(question, docsearch, chat_history, api_key):
    openai.api_key = api_key
    if is_azure_configured():
        logger.debug("in Azure")
+        openai.api_type = "azure"
+        openai.api_version = settings.OPENAI_API_VERSION
+        openai.api_base = settings.OPENAI_API_BASE
+        openai.engine = settings.AZURE_DEPLOYMENT_NAME
        llm = AzureChatOpenAI(
            openai_api_key=api_key,
            openai_api_base=settings.OPENAI_API_BASE,