github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-18 00:51:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

add simple benchmark for init

Browse Source
This commit is contained in:
Chester Curme 2025-05-19 12:58:20 -04:00
parent 46c08d26c3
commit 0ba7d1f8d9
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/standard-tests/langchain_tests/unit_tests/chat_models.py
View File

@ -9,6 +9,7 @@ from typing import Any, Dict, List, Literal, Optional, Tuple, Type
from unittest import mock from unittest import mock
import pytest import pytest
from pytest_benchmark.fixture import BenchmarkFixture
from langchain_core.language_models import BaseChatModel from langchain_core.language_models import BaseChatModel
from langchain_core.load import dumpd, load from langchain_core.load import dumpd, load
from langchain_core.runnables import RunnableBinding from langchain_core.runnables import RunnableBinding
@ -850,3 +851,8 @@ class ChatModelUnitTests(ChatModelTests):
dumpd(model), valid_namespaces=model.get_lc_namespace()[:1] dumpd(model), valid_namespaces=model.get_lc_namespace()[:1]
).dict() ).dict()
) )
@pytest.mark.benchmark
def test_init_time(self, benchmark: BenchmarkFixture) -> None:
_ = benchmark(self.chat_model_class, **self.chat_model_params)