import%20marimo%0A%0A__generated_with%20%3D%20%220.9.17%22%0Aapp%20%3D%20marimo.App(width%3D%22medium%22)%0A%0A%0A%40app.cell%0Adef%20__()%3A%0A%20%20%20%20import%20marimo%20as%20mo%0A%20%20%20%20import%20testpackage%0A%20%20%20%20import%20dotenv%0A%0A%20%20%20%20dotenv.load_dotenv()%0A%20%20%20%20import%20os%0A%20%20%20%20import%20ell%0A%20%20%20%20ell.init(store%3D'.%2Flogdir')%0A%20%20%20%20modelname%20%3D%20os.environ%5B%22AZURE_OPENAI_CHAT_DEPLOYMENT_NAME%22%5D%0A%20%20%20%20return%20dotenv%2C%20ell%2C%20mo%2C%20modelname%2C%20os%2C%20testpackage%0A%0A%0A%40app.cell%0Adef%20__(os)%3A%0A%20%20%20%20from%20azure.identity%20import%20DefaultAzureCredential%2C%20get_bearer_token_provider%0A%0A%20%20%20%20token_provider%20%3D%20get_bearer_token_provider(%0A%20%20%20%20%20%20%20%20DefaultAzureCredential()%2C%20%22https%3A%2F%2Fcognitiveservices.azure.com%2F.default%22%0A%20%20%20%20)%0A%20%20%20%20from%20langchain_openai%20import%20AzureChatOpenAI%0A%0A%20%20%20%20model%20%3D%20AzureChatOpenAI(%0A%20%20%20%20%20%20%20%20api_version%3Dos.environ%5B%22AZURE_OPENAI_API_VERSION%22%5D%2C%0A%20%20%20%20%20%20%20%20azure_deployment%3Dos.environ%5B%22AZURE_OPENAI_CHAT_DEPLOYMENT_NAME%22%5D%2C%0A%20%20%20%20%20%20%20%20azure_ad_token_provider%3Dtoken_provider%2C%0A%20%20%20%20%20%20%20%20temperature%3D0.7%2C%0A%20%20%20%20)%0A%20%20%20%20return%20(%0A%20%20%20%20%20%20%20%20AzureChatOpenAI%2C%0A%20%20%20%20%20%20%20%20DefaultAzureCredential%2C%0A%20%20%20%20%20%20%20%20get_bearer_token_provider%2C%0A%20%20%20%20%20%20%20%20model%2C%0A%20%20%20%20%20%20%20%20token_provider%2C%0A%20%20%20%20)%0A%0A%0A%40app.cell%0Adef%20__(model)%3A%0A%20%20%20%20model.invoke(%22Tell%20a%20random%20joke%3F%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20__(ell%2C%20modelname%2C%20os%2C%20token_provider)%3A%0A%20%20%20%20import%20openai%0A%20%20%20%20azureClient%20%3D%20openai.AzureOpenAI(%0A%20%20%20%20%20%20%20%20azure_endpoint%20%3D%20os.environ%5B%22AZURE_OPENAI_ENDPOINT%22%5D%2C%0A%20%20%20%20%20%20%20%20api_version%20%3D%20%222024-05-01-preview%22%2C%0A%20%20%20%20%20%20%20%20azure_ad_token_provider%3Dtoken_provider%0A%20%20%20%20)%0A%20%20%20%20ell.config.register_model(modelname%2C%20azureClient)%0A%20%20%20%20return%20azureClient%2C%20openai%0A%0A%0A%40app.cell%0Adef%20__(ell%2C%20modelname)%3A%0A%20%20%20%20%40ell.simple(model%3Dmodelname)%0A%20%20%20%20def%20write_a_story(about%20%3A%20str)%3A%0A%20%20%20%20%20%20%20%20return%20f%22write%20me%20a%20limerick%20about%20%7Babout%7D!%22%0A%0A%20%20%20%20write_a_story(%22electric%20cars%22)%0A%20%20%20%20return%20(write_a_story%2C)%0A%0A%0A%40app.cell%0Adef%20__(mo%2C%20write_a_story)%3A%0A%20%20%20%20mo.ui.chat(write_a_story)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20__(List%2C%20Message%2C%20ell%2C%20modelname)%3A%0A%20%20%20%20%40ell.complex(model%3Dmodelname%2C%20temperature%3D0.7)%0A%20%20%20%20def%20chat_bot(message_history%3A%20List%5BMessage%5D)%20-%3E%20List%5BMessage%5D%3A%0A%20%20%20%20%20%20%20%20print(%22inside%22)%0A%20%20%20%20%20%20%20%20return%20message_history%0A%0A%20%20%20%20%40ell.simple(model%3Dmodelname)%0A%20%20%20%20def%20chat_bot(message)%3A%0A%20%20%20%20%20%20%20%20return%20message%0A%0A%20%20%20%20def%20test(messages%2C%20config)%3A%0A%20%20%20%20%20%20%20%20for%20m%20in%20messages%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20print(m)%0A%20%20%20%20%20%20%20%20print(%22sending%22)%0A%20%20%20%20%20%20%20%20cb%20%3D%20chat_bot(messages)%0A%20%20%20%20%20%20%20%20return%20%22Another%22%0A%20%20%20%20return%20chat_bot%2C%20test%0A%0A%0A%40app.cell%0Adef%20__(mo%2C%20test)%3A%0A%20%20%20%20mo.ui.chat(test)%0A%20%20%20%20return%0A%0A%0Aif%20__name__%20%3D%3D%20%22__main__%22%3A%0A%20%20%20%20app.run()%0A
0c533fb06365073011c08a8ffd38579ff8f77bbc7658594419f38279f20e7c7c