update

langchain-ai · Jul 26, 2024 · 729b5b0 · 729b5b0
1 parent d0f5e4b
commit 729b5b0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/libs/partners/ai21/tests/integration_tests/test_chat_models.py b/libs/partners/ai21/tests/integration_tests/test_chat_models.py
@@ -1,10 +1,11 @@
 """Test ChatAI21 chat model."""
 
 import pytest
-from langchain_ai21.chat_models import ChatAI21
 from langchain_core.messages import AIMessageChunk, HumanMessage
 from langchain_core.outputs import ChatGeneration
 from langchain_core.rate_limiters import InMemoryRateLimiter
+
+from langchain_ai21.chat_models import ChatAI21
 from tests.unit_tests.conftest import J2_CHAT_MODEL_NAME, JAMBA_CHAT_MODEL_NAME
 
 rate_limiter = InMemoryRateLimiter(requests_per_second=0.5)

diff --git a/libs/partners/ai21/tests/integration_tests/test_standard.py b/libs/partners/ai21/tests/integration_tests/test_standard.py
@@ -5,10 +5,10 @@
 
 import pytest
 from langchain_core.language_models import BaseChatModel
+from langchain_core.rate_limiters import InMemoryRateLimiter
 from langchain_standard_tests.integration_tests import ChatModelIntegrationTests
 
 from langchain_ai21 import ChatAI21
-from langchain_core.rate_limiters import InMemoryRateLimiter
 
 rate_limiter = InMemoryRateLimiter(requests_per_second=0.5)