From 0ba7d1f8d98ef5da4a48748018be4e814981fbfe Mon Sep 17 00:00:00 2001 From: Chester Curme Date: Mon, 19 May 2025 12:58:20 -0400 Subject: [PATCH] add simple benchmark for init --- .../langchain_tests/unit_tests/chat_models.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/libs/standard-tests/langchain_tests/unit_tests/chat_models.py b/libs/standard-tests/langchain_tests/unit_tests/chat_models.py index 94764939226..8e38049d4a5 100644 --- a/libs/standard-tests/langchain_tests/unit_tests/chat_models.py +++ b/libs/standard-tests/langchain_tests/unit_tests/chat_models.py @@ -9,6 +9,7 @@ from typing import Any, Dict, List, Literal, Optional, Tuple, Type from unittest import mock import pytest +from pytest_benchmark.fixture import BenchmarkFixture from langchain_core.language_models import BaseChatModel from langchain_core.load import dumpd, load from langchain_core.runnables import RunnableBinding @@ -850,3 +851,8 @@ class ChatModelUnitTests(ChatModelTests): dumpd(model), valid_namespaces=model.get_lc_namespace()[:1] ).dict() ) + + @pytest.mark.benchmark + def test_init_time(self, benchmark: BenchmarkFixture) -> None: + + _ = benchmark(self.chat_model_class, **self.chat_model_params)