freks blog

export LANGCHAIN_TRACING_V2="true"
export LANGCHAIN_API_KEY = "<さっき取得したAPI key>"
export LANGCHAIN_PROJECT = "<プロジェクト名>"

import os
from langchain_openai import AzureChatOpenAI
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.output_parsers import StrOutputParser
from langchain_community.callbacks import get_openai_callback

os.environ["AZURE_OPENAI_API_KEY"] = os.getenv("AZURE_OPENAI_API_KEY", "")
os.environ["AZURE_OPENAI_ENDPOINT"] = os.getenv("AZURE_OPENAI_ENDPOINT", "")
os.environ["AZURE_OPENAI_API_VERSION"] = "2024-02-01"
os.environ["AZURE_OPENAI_CHAT_DEPLOYMENT_NAME"] = "gpt-35-turbo"
os.environ["AZURE_OPENAI_CHAT_MODEL_VERSION"] = "0613"

llm = AzureChatOpenAI(
    azure_deployment=os.environ["AZURE_OPENAI_CHAT_DEPLOYMENT_NAME"],
    api_version=os.environ["AZURE_OPENAI_API_VERSION"],
    model_version=os.environ["AZURE_OPENAI_CHAT_MODEL_VERSION"],
)

prompt = ChatPromptTemplate.from_messages(
    [
        (
            "system",
            "You are a science guy. Response in japanese.",
        ),
        ("user", "{input}"),
    ]
)
output_parser = StrOutputParser()

chain = prompt | llm | output_parser

with get_openai_callback() as cb:
    message = chain.invoke({"input": "Tell me a joke"})
    print(message)
    print(f"Total Cost (USD): ${format(cb.total_cost, '.6f')}")

export LANGCHAIN_TRACING_V2="false"