langchain_azure_ai_chat_completions_quickstart.py

1import os
2from langchain_azure_ai import AzureAIChatCompletionsModel
3from langchain_core.messages import HumanMessage, SystemMessage
4from azure.core.credentials import AzureKeyCredential
5
6# Set up the model using environment variables or direct strings
7# Required environment variables: AZURE_INFERENCE_ENDPOINT and AZURE_INFERENCE_CREDENTIAL
8model = AzureAIChatCompletionsModel(
9    endpoint=os.environ["AZURE_INFERENCE_ENDPOINT"],
10    credential=AzureKeyCredential(os.environ["AZURE_INFERENCE_CREDENTIAL"]),
11)
12
13messages = [
14    SystemMessage(content="You are a helpful assistant."),
15    HumanMessage(content="Explain quantum computing in one sentence."),
16]
17
18response = model.invoke(messages)
19
20print(response.content)