mirror of
https://github.com/hwchase17/langchain.git
synced 2026-06-09 10:17:00 +00:00
perf(langchain): add benchmark command (#36641)
add benchmark in Makefile
This commit is contained in:
0
libs/langchain_v1/tests/benchmarks/__init__.py
Normal file
0
libs/langchain_v1/tests/benchmarks/__init__.py
Normal file
14
libs/langchain_v1/tests/benchmarks/test_create_agent.py
Normal file
14
libs/langchain_v1/tests/benchmarks/test_create_agent.py
Normal 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)
|
||||
Reference in New Issue
Block a user