langchain Architecture Docs
Home / File/ test_azure.py — langchain Source File

test_azure.py — langchain Source File

Architecture documentation for test_azure.py, a python file in the langchain codebase. 7 imports, 0 dependents.

File python CoreAbstractions MessageSchema 7 imports 9 functions 1 classes

Entity Profile

Dependency Diagram

graph LR
  1ac9d53a_93c2_ab1e_e2d2_5444daf056b4["test_azure.py"]
  9e98f0a7_ec6e_708f_4f1b_e9428b316e1c["os"]
  1ac9d53a_93c2_ab1e_e2d2_5444daf056b4 --> 9e98f0a7_ec6e_708f_4f1b_e9428b316e1c
  6ebcaae2_3bc1_badf_b751_e164ff2776c4["unittest"]
  1ac9d53a_93c2_ab1e_e2d2_5444daf056b4 --> 6ebcaae2_3bc1_badf_b751_e164ff2776c4
  120e2591_3e15_b895_72b6_cb26195e40a6["pytest"]
  1ac9d53a_93c2_ab1e_e2d2_5444daf056b4 --> 120e2591_3e15_b895_72b6_cb26195e40a6
  d758344f_537f_649e_f467_b9d7442e86df["langchain_core.messages"]
  1ac9d53a_93c2_ab1e_e2d2_5444daf056b4 --> d758344f_537f_649e_f467_b9d7442e86df
  6e58aaea_f08e_c099_3cc7_f9567bfb1ae7["pydantic"]
  1ac9d53a_93c2_ab1e_e2d2_5444daf056b4 --> 6e58aaea_f08e_c099_3cc7_f9567bfb1ae7
  91721f45_4909_e489_8c1f_084f8bd87145["typing_extensions"]
  1ac9d53a_93c2_ab1e_e2d2_5444daf056b4 --> 91721f45_4909_e489_8c1f_084f8bd87145
  0b28cff6_d823_1571_d2bb_ec61508cc89c["langchain_openai"]
  1ac9d53a_93c2_ab1e_e2d2_5444daf056b4 --> 0b28cff6_d823_1571_d2bb_ec61508cc89c
  style 1ac9d53a_93c2_ab1e_e2d2_5444daf056b4 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

"""Test Azure OpenAI Chat API wrapper."""

import os
from unittest import mock

import pytest
from langchain_core.messages import HumanMessage
from pydantic import SecretStr
from typing_extensions import TypedDict

from langchain_openai import AzureChatOpenAI


def test_initialize_azure_openai() -> None:
    llm = AzureChatOpenAI(  # type: ignore[call-arg]
        azure_deployment="35-turbo-dev",
        openai_api_version="2023-05-15",
        azure_endpoint="my-base-url",
    )
    assert llm.deployment_name == "35-turbo-dev"
    assert llm.openai_api_version == "2023-05-15"
    assert llm.azure_endpoint == "my-base-url"


def test_initialize_more() -> None:
    llm = AzureChatOpenAI(  # type: ignore[call-arg]
        api_key="xyz",  # type: ignore[arg-type]
        azure_endpoint="my-base-url",
        azure_deployment="35-turbo-dev",
        openai_api_version="2023-05-15",
        temperature=0,
        model="gpt-35-turbo",
        model_version="0125",
    )
    assert llm.openai_api_key is not None
    assert llm.openai_api_key.get_secret_value() == "xyz"
    assert llm.azure_endpoint == "my-base-url"
    assert llm.deployment_name == "35-turbo-dev"
    assert llm.openai_api_version == "2023-05-15"
    assert llm.temperature == 0
    assert llm.stream_usage

    ls_params = llm._get_ls_params()
    assert ls_params.get("ls_provider") == "azure"
    assert ls_params.get("ls_model_name") == "gpt-35-turbo-0125"


def test_initialize_azure_openai_with_openai_api_base_set() -> None:
    with mock.patch.dict(os.environ, {"OPENAI_API_BASE": "https://api.openai.com"}):
        llm = AzureChatOpenAI(  # type: ignore[call-arg, call-arg]
            api_key="xyz",  # type: ignore[arg-type]
            azure_endpoint="my-base-url",
            azure_deployment="35-turbo-dev",
            openai_api_version="2023-05-15",
            temperature=0,
            openai_api_base=None,
        )
        assert llm.openai_api_key is not None
        assert llm.openai_api_key.get_secret_value() == "xyz"
        assert llm.azure_endpoint == "my-base-url"
// ... (117 more lines)

Domain

Subdomains

Functions

Classes

Dependencies

  • langchain_core.messages
  • langchain_openai
  • os
  • pydantic
  • pytest
  • typing_extensions
  • unittest

Source

Frequently Asked Questions

What does test_azure.py do?
test_azure.py is a source file in the langchain codebase, written in python. It belongs to the CoreAbstractions domain, MessageSchema subdomain.
What functions are defined in test_azure.py?
test_azure.py defines 9 function(s): test_chat_completions_api_uses_model_name, test_initialize_azure_openai, test_initialize_azure_openai_with_openai_api_base_set, test_initialize_more, test_max_completion_tokens_in_payload, test_max_completion_tokens_parameter, test_max_tokens_converted_to_max_completion_tokens, test_responses_api_uses_deployment_name, test_structured_output_old_model.
What does test_azure.py depend on?
test_azure.py imports 7 module(s): langchain_core.messages, langchain_openai, os, pydantic, pytest, typing_extensions, unittest.
Where is test_azure.py in the architecture?
test_azure.py is located at libs/partners/openai/tests/unit_tests/chat_models/test_azure.py (domain: CoreAbstractions, subdomain: MessageSchema, directory: libs/partners/openai/tests/unit_tests/chat_models).

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free