From 72c7af0bc054daa0b1732605e15fd84578344f84 Mon Sep 17 00:00:00 2001 From: Bagatur <22008038+baskaryan@users.noreply.github.com> Date: Thu, 8 Feb 2024 16:39:55 -0800 Subject: [PATCH] langchain[patch]: undo redis cache import (#17275) --- libs/langchain/langchain/cache.py | 2 -- .../langchain/tests/integration_tests/cache/test_redis_cache.py | 2 +- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/libs/langchain/langchain/cache.py b/libs/langchain/langchain/cache.py index 3c249e964c5..fae1d1cf032 100644 --- a/libs/langchain/langchain/cache.py +++ b/libs/langchain/langchain/cache.py @@ -1,7 +1,6 @@ from langchain_community.cache import ( AstraDBCache, AstraDBSemanticCache, - AsyncRedisCache, CassandraCache, CassandraSemanticCache, FullLLMCache, @@ -23,7 +22,6 @@ __all__ = [ "SQLAlchemyCache", "SQLiteCache", "UpstashRedisCache", - "AsyncRedisCache", "RedisCache", "RedisSemanticCache", "GPTCache", diff --git a/libs/langchain/tests/integration_tests/cache/test_redis_cache.py b/libs/langchain/tests/integration_tests/cache/test_redis_cache.py index 846c709b971..644d0386755 100644 --- a/libs/langchain/tests/integration_tests/cache/test_redis_cache.py +++ b/libs/langchain/tests/integration_tests/cache/test_redis_cache.py @@ -4,12 +4,12 @@ from contextlib import asynccontextmanager, contextmanager from typing import AsyncGenerator, Generator, List, Optional, cast import pytest +from langchain_community.cache import AsyncRedisCache, RedisCache, RedisSemanticCache from langchain_core.embeddings import Embeddings from langchain_core.load.dump import dumps from langchain_core.messages import AIMessage, BaseMessage, HumanMessage from langchain_core.outputs import ChatGeneration, Generation, LLMResult -from langchain.cache import AsyncRedisCache, RedisCache, RedisSemanticCache from langchain.globals import get_llm_cache, set_llm_cache from tests.integration_tests.cache.fake_embeddings import ( ConsistentFakeEmbeddings,