Spaces:
Sleeping
Sleeping
File size: 3,358 Bytes
478dec6 85d8eae 478dec6 85d8eae 478dec6 85d8eae 478dec6 85d8eae 478dec6 85d8eae 478dec6 6a0f2f3 478dec6 | 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 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 | import os
from dotenv import load_dotenv
from config.env_constant import EnvFilepath
load_dotenv(EnvFilepath.ENVPATH)
from langchain_openai import AzureChatOpenAI
# from langchain_google_genai import ChatGoogleGenerativeAI
# from langchain.callbacks import AsyncIteratorCallbackHandler
# from langfuse.callback import CallbackHandler
# langfuse_handler = CallbackHandler(
# secret_key=os.environ.get('buma--langfuse--secret-key'),
# public_key=os.environ.get('buma--langfuse--public-key'),
# host=os.environ.get('buma--langfuse__host')
# )
#######
# LLM
#######
# model_4o = AzureChatOpenAI(
# azure_endpoint=os.environ.get("azureai--endpoint--url--4o"), # or your deployment
# openai_api_version=os.environ.get("azureai--api--version--4o"), # or your api version
# deployment_name=os.environ.get("azureai--deployment--name--4o"),
# openai_api_key=os.environ.get("azureai--api-key--4o"),
# openai_api_type="azure",
# max_retries=2,
# disable_streaming=True
# )
model_4o_2 = AzureChatOpenAI(
azure_endpoint=os.environ.get("azureai__endpoint__url__4o_2"), # or your deployment
openai_api_version=os.environ.get("azureai__api__version__4o_2"), # or your api version
deployment_name=os.environ.get("azureai__deployment__name__4o_2"),
openai_api_key=os.environ.get("azureai__api_key__4o_2"),
openai_api_type="azure",
max_retries=2,
disable_streaming=True
)
model_4omini = AzureChatOpenAI(
azure_endpoint=os.environ.get("azureai__endpoint__url__4omini"), # or your deployment
openai_api_version=os.environ.get("azureai__api__version__4omini"), # or your api version
deployment_name=os.environ.get("azureai__deployment__name__4omini"),
openai_api_key=os.environ.get("azureai__api_key__4omini"),
openai_api_type="azure",
max_retries=2,
disable_streaming=True
)
model_5mini = AzureChatOpenAI(
azure_endpoint=os.environ.get("azureai__endpoint__url__5mini"), # or your deployment
openai_api_version=os.environ.get("azureai__api__version__5mini"), # or your api version
deployment_name=os.environ.get("azureai__deployment__name__5mini"),
openai_api_key=os.environ.get("azureai__api_key__5mini"),
openai_api_type="azure",
max_retries=2,
disable_streaming=True
)
model_5_1 = AzureChatOpenAI(
openai_api_key=os.environ.get("azureai__api_key__51"),
azure_endpoint=os.environ.get("azureai__endpoint__url__51"),
deployment_name=os.environ.get("azureai__deployment__name__51"),
openai_api_version=os.environ.get("azureai__api__version__51"),
openai_api_type="azure",
max_retries=2,
disable_streaming=True
)
# model_5_1codex = AzureChatOpenAI(
# azure_endpoint=os.environ.get("azureai--endpoint--url--5.1-codex"),
# openai_api_version=os.environ.get("azureai--api--version--5.1-codex"),
# deployment_name=os.environ.get("azureai--deployment--name--5.1-codex"),
# openai_api_key=os.environ.get("azureai--api-key--5.1-codex"),
# openai_api_type="azure",
# max_retries=2,
# disable_streaming=True
# )
# model_gemini = ChatGoogleGenerativeAI(
# model="gemini-2.0-flash",
# disable_streaming=False,
# # callbacks=[callback],
# verbose=True)
# response = model_gemini.invoke("Buatkan satu lagu dalam bahasa indonesia, bertema pertemanan")
# print(response.content) |