File size: 757 Bytes
991dafd bfaf40d 991dafd af5101f 991dafd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
import os
from dotenv import load_dotenv, find_dotenv
from langchain.callbacks.streaming_stdout_final_only import (
FinalStreamingStdOutCallbackHandler,
)
_ = load_dotenv(find_dotenv()) # read local .env file
openaiAPIVersion = os.getenv("OPENAI_API_VERSION") # "2023-05-15"
gpt4Model = os.getenv("AZURE_OPENAI_DEPLOYMENT_NAME_GPT4")
gpt35Model = os.getenv("AZURE_OPENAI_DEPLOYMENT_NAME_GPT35")
print(openaiAPIVersion, gpt4Model, gpt35Model)
from langchain.chat_models import AzureChatOpenAI
llm = AzureChatOpenAI(
temperature=0,
deployment_name=gpt35Model,
openai_api_version=openaiAPIVersion,
)
llm4 = AzureChatOpenAI(
temperature=0,
deployment_name=gpt4Model,
openai_api_version=openaiAPIVersion,
streaming=True,
)
|