test_groq_invoke() — langchain Function Reference

Architecture documentation for the test_groq_invoke() function in test_chat_models.py from the langchain codebase.

Function python LangChainCore LanguageModelBase

Entity Profile

LangChainCore→ LanguageModelBase→ test_groq_invoke() — langchain Function Reference

Dependency Diagram

graph TD
  7b5e8b1b_525d_41ae_dde1_6e02db71df05["test_groq_invoke()"]
  5bf2e477_37e0_3e98_4042_bc609f2f7f60["test_chat_models.py"]
  7b5e8b1b_525d_41ae_dde1_6e02db71df05 -->|defined in| 5bf2e477_37e0_3e98_4042_bc609f2f7f60
  style 7b5e8b1b_525d_41ae_dde1_6e02db71df05 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/partners/groq/tests/unit_tests/test_chat_models.py lines 169–188

def test_groq_invoke(mock_completion: dict) -> None:
    llm = ChatGroq(model="foo")
    mock_client = MagicMock()
    completed = False

    def mock_create(*args: Any, **kwargs: Any) -> Any:
        nonlocal completed
        completed = True
        return mock_completion

    mock_client.create = mock_create
    with patch.object(
        llm,
        "client",
        mock_client,
    ):
        res = llm.invoke("bar")
        assert res.content == "Bar Baz"
        assert type(res) is AIMessage
    assert completed