diff --git a/src/backend/langflow/services/chat/service.py b/src/backend/langflow/services/chat/service.py index 82730c6ba..0d7f9ca6c 100644 --- a/src/backend/langflow/services/chat/service.py +++ b/src/backend/langflow/services/chat/service.py @@ -5,15 +5,14 @@ from typing import Any, Dict, List import orjson from fastapi import WebSocket, status -from loguru import logger -from starlette.websockets import WebSocketState - from langflow.api.v1.schemas import ChatMessage, ChatResponse, FileResponse from langflow.interface.utils import pil_to_base64 -from langflow.services import ServiceType, service_manager from langflow.services.base import Service from langflow.services.chat.cache import Subject from langflow.services.chat.utils import process_graph +from langflow.services.deps import get_cache_service +from loguru import logger +from starlette.websockets import WebSocketState from .cache import cache_service @@ -54,7 +53,7 @@ class ChatService(Service): self.chat_history = ChatHistory() self.chat_cache = cache_service self.chat_cache.attach(self.update) - self.cache_service = service_manager.get(ServiceType.CACHE_SERVICE) + self.cache_service = get_cache_service() def on_chat_history_update(self): """Send the last chat message to the client.""" @@ -240,7 +239,7 @@ class ChatService(Service): """ Get the cache for a client. """ - return self.chat_cache.get(client_id) + return self.cache_service.get(client_id) def dict_to_markdown_table(my_dict):