From 7b9166b5babf9ec9d3ff0db03312223afdf252cf Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 20 Feb 2024 13:19:42 -0300 Subject: [PATCH] Refactor chat service code --- src/backend/langflow/services/chat/service.py | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/src/backend/langflow/services/chat/service.py b/src/backend/langflow/services/chat/service.py index 054609f97..323b3d6f3 100644 --- a/src/backend/langflow/services/chat/service.py +++ b/src/backend/langflow/services/chat/service.py @@ -1,20 +1,13 @@ -from typing import Any, Dict +from typing import Any -from fastapi import WebSocket from langflow.services.base import Service from langflow.services.deps import get_cache_service -from .cache import cache_service - class ChatService(Service): name = "chat_service" def __init__(self): - self.active_connections: Dict[str, WebSocket] = {} - self.connection_ids: Dict[str, str] = {} - self.chat_cache = cache_service - self.chat_cache.attach(self.update) self.cache_service = get_cache_service() def set_cache(self, client_id: str, data: Any) -> bool: