perf(langchain): add benchmark command (#36641)

add benchmark in Makefile
This commit is contained in:
Eugene Yurtsev
2026-04-09 16:05:17 -04:00
committed by GitHub
parent 644821e5b4
commit f0c5a28fa0
5 changed files with 23 additions and 1 deletions

View File

@@ -0,0 +1,14 @@
import pytest
from langchain_core.language_models.fake_chat_models import GenericFakeChatModel
from langchain_core.messages import AIMessage
from pytest_benchmark.fixture import BenchmarkFixture
from langchain.agents import create_agent
@pytest.mark.benchmark
def test_create_agent_instantiation(benchmark: BenchmarkFixture) -> None:
def instantiate_agent() -> None:
create_agent(model=GenericFakeChatModel(messages=iter([AIMessage(content="ok")])))
benchmark(instantiate_agent)