Add missing import

This commit is contained in:
Eugene Yurtsev
2024-08-07 13:50:14 -04:00
parent a1054d06ca
commit fc0fa7e8f0

View File

@@ -1,9 +1,9 @@
import time
from typing import Optional
from langchain_core.caches import InMemoryCache
from langchain_core.language_models import GenericFakeChatModel
from langchain_core.rate_limiters import InMemoryRateLimiter
from typing import Optional as Optional
def test_rate_limit_invoke() -> None:
@@ -220,7 +220,6 @@ class SerializableModel(GenericFakeChatModel):
def is_lc_serializable(cls) -> bool:
return True
SerializableModel.model_rebuild()