Pyrit: I am not able to run Pyrit code(Automation framework by Microsoft)

I was not able to run Pyrit code that is mentioned in Microsoft documentation. Error: PYRIT\PyRIT\doc\demo\1_gandalf.py", line 51, in <module> from pyrit.chat import AzureOpenAIChat ModuleNotFoun...