test/llm/claude/test_claude_llm.py

# from neollm.llm.gpt.azure_llm import (
#     AzureGPT4_0613,
#     AzureGPT4T_0125,
#     AzureGPT4T_1106,
#     AzureGPT4T_20240409,
#     AzureGPT4VT_1106,
#     AzureGPT35FT,
#     AzureGPT35T16k_0613,
#     AzureGPT35T_0125,
#     AzureGPT35T_0613,
#     AzureGPT35T_1106,
#     AzureGPT432k_0613,
# )
# from neollm.types.info import APIPricing


# def test_check_price() -> None:
#     # https://azure.microsoft.com/ja-jp/pricing/details/cognitive-services/openai-service/

#     # これからのモデル --------------------------------------------------------
#     assert AzureGPT4T_20240409.dollar_per_ktoken == APIPricing(input=0.01, output=0.03)
#     # Updated --------------------------------------------------------
#     # GPT3.5T
#     assert AzureGPT35T_0125.dollar_per_ktoken == APIPricing(input=0.0005, output=0.0015)
#     # GPT4
#     assert AzureGPT4T_0125.dollar_per_ktoken == APIPricing(input=0.01, output=0.03)
#     assert AzureGPT4VT_1106.dollar_per_ktoken == APIPricing(input=0.01, output=0.03)
#     assert AzureGPT4T_1106.dollar_per_ktoken == APIPricing(input=0.01, output=0.03)
#     assert AzureGPT4_0613.dollar_per_ktoken == APIPricing(input=0.03, output=0.06)
#     assert AzureGPT432k_0613.dollar_per_ktoken == APIPricing(input=0.06, output=0.12)
#     # FT
#     assert AzureGPT35FT.dollar_per_ktoken == APIPricing(input=0.0005, output=0.0015)
#     # Legacy ---------------------------------------------------------
#     # AzureGPT35T_0301 なし
#     assert AzureGPT35T_0613.dollar_per_ktoken == APIPricing(input=0.0015, output=0.002)
#     assert AzureGPT35T16k_0613.dollar_per_ktoken == APIPricing(input=0.003, output=0.004)
#     assert AzureGPT35T_1106.dollar_per_ktoken == APIPricing(input=0.001, output=0.002)