Search code examples
pythonazureartificial-intelligencepy-langchain

Encountered keyerror when run LangChain program with Azure API


I encountered error when I run below codes (extracted from https://python.langchain.com/docs/tutorials/llm_chain/#using-language-models) in Jupyter Notebook:

import getpass
import os

os.environ\["AZURE_OPENAI_API_KEY"\] = "key"

from langchain_openai import AzureChatOpenAI

model = AzureChatOpenAI(
azure_endpoint=os.environ\["https://testing123.cognitiveservices.azure.com/"\],
azure_deployment=os.environ\["testing123"\],
openai_api_version=os.environ\["2021-04-30"\],
)

Error I got after run:

KeyError                                  Traceback (most recent call last)
Cell In\[16\], line 10
5 os.environ\["AZURE_OPENAI_API_KEY"\] = "key"
7 from langchain_openai import AzureChatOpenAI
9 model = AzureChatOpenAI(
\---\> 10     azure_endpoint=os.environ\["https://testing123.cognitiveservices.azure.com/"\],
11     azure_deployment=os.environ\["testing123"\],
12     openai_api_version=os.environ\["2021-04-30"\],
13 )

File \<frozen os\>:714, in __getitem__(self, key)

KeyError: 'https://testing123.cognitiveservices.azure.com/'

I have checked my Azure portal and end-point is exactly what I copied from. Don't understand why key error existed.


Solution

  • The errors says KeyError so let's fix the key. Remove those scape \ and it should work.

    Also looks like you are not accessing the environment vars below so remove the os.environ and pass the string directly.

    import getpass
    import os
    
    os.environ["AZURE_OPENAI_API_KEY"] = "d957dadfasdfafdsfsafasfasfsafasf99f4b7"
    
    from langchain_openai import AzureChatOpenAI
    
    model = AzureChatOpenAI(
      azure_endpoint="https://testing123.cognitiveservices.azure.com/",
      azure_deployment="testing123",
      openai_api_version="2021-04-30",
    )