TestOpenAIStandard Class — langchain Architecture

Architecture documentation for the TestOpenAIStandard class in test_azure_standard.py from the langchain codebase.

Class python

Entity Profile

Dependency Diagram

graph TD
  edb40573_63c9_fbcc_5b98_f024fa4e5b33["TestOpenAIStandard"]
  f4b40f56_1158_0e72_1cac_d0121e4eabd8["ChatModelUnitTests"]
  edb40573_63c9_fbcc_5b98_f024fa4e5b33 -->|extends| f4b40f56_1158_0e72_1cac_d0121e4eabd8
  4171ab6b_3b1b_6c10_eac7_da6e049d38d5["test_azure_standard.py"]
  edb40573_63c9_fbcc_5b98_f024fa4e5b33 -->|defined in| 4171ab6b_3b1b_6c10_eac7_da6e049d38d5
  2777dc48_a373_3a22_9f3e_88347f9c8fc9["chat_model_class()"]
  edb40573_63c9_fbcc_5b98_f024fa4e5b33 -->|method| 2777dc48_a373_3a22_9f3e_88347f9c8fc9
  4c7c623a_20b4_62ff_1210_961676483373["chat_model_params()"]
  edb40573_63c9_fbcc_5b98_f024fa4e5b33 -->|method| 4c7c623a_20b4_62ff_1210_961676483373
  7fd6450e_d133_a5ac_b9c0_6a794a06019e["test_bind_tool_pydantic()"]
  edb40573_63c9_fbcc_5b98_f024fa4e5b33 -->|method| 7fd6450e_d133_a5ac_b9c0_6a794a06019e
  e17203d0_454a_485c_9090_74656a104167["init_from_env_params()"]
  edb40573_63c9_fbcc_5b98_f024fa4e5b33 -->|method| e17203d0_454a_485c_9090_74656a104167

Relationship Graph

Source Code

libs/partners/openai/tests/unit_tests/chat_models/test_azure_standard.py lines 11–50

class TestOpenAIStandard(ChatModelUnitTests):
    @property
    def chat_model_class(self) -> type[BaseChatModel]:
        return AzureChatOpenAI

    @property
    def chat_model_params(self) -> dict:
        return {
            "deployment_name": "test",
            "openai_api_version": "2021-10-01",
            "azure_endpoint": "https://test.azure.com",
        }

    @pytest.mark.xfail(reason="AzureOpenAI does not support tool_choice='any'")
    def test_bind_tool_pydantic(
        self, model: BaseChatModel, my_adder_tool: BaseTool
    ) -> None:
        super().test_bind_tool_pydantic(model, my_adder_tool)

    @property
    def init_from_env_params(self) -> tuple[dict, dict, dict]:
        return (
            {
                "AZURE_OPENAI_API_KEY": "api_key",
                "AZURE_OPENAI_ENDPOINT": "https://endpoint.com",
                "AZURE_OPENAI_AD_TOKEN": "token",
                "OPENAI_ORG_ID": "org_id",
                "OPENAI_API_VERSION": "yyyy-mm-dd",
                "OPENAI_API_TYPE": "type",
            },
            {},
            {
                "openai_api_key": "api_key",
                "azure_endpoint": "https://endpoint.com",
                "azure_ad_token": "token",
                "openai_organization": "org_id",
                "openai_api_version": "yyyy-mm-dd",
                "openai_api_type": "type",
            },
        )