remove with_rate_limit from public api

langchain_benchmarks/__init__.py

@@ -1,5 +1,5 @@
 from langchain_benchmarks.model_registration import model_registry
-from langchain_benchmarks.rate_limiting import RateLimiter, with_rate_limit
+from langchain_benchmarks.rate_limiting import RateLimiter
 from langchain_benchmarks.registration import registry
 from langchain_benchmarks.utils._langsmith import (
     clone_public_dataset,
@@ -13,5 +13,4 @@
     "model_registry",
     "RateLimiter",
     "registry",
-    "with_rate_limit",
 ]

langchain_benchmarks/tool_usage/agents.py

@@ -9,7 +9,7 @@
 from langchain.schema.runnable import Runnable, RunnableLambda, RunnablePassthrough
 from langchain.tools.render import format_tool_to_openai_function
 
-from langchain_benchmarks import rate_limiting, with_rate_limit
+from langchain_benchmarks import rate_limiting
 from langchain_benchmarks.schema import ToolUsageTask
 
 

tests/unit_tests/test_public_api.py

@@ -12,7 +12,6 @@ def test_public_api() -> None:
             "model_registry",
             "RateLimiter",
             "registry",
-            "with_rate_limit",
         ],
         key=lambda x: x.lower(),
     )