From 9f0d60ab5ae63daeaccc881d0c553b323c17c9f1 Mon Sep 17 00:00:00 2001 From: George Burton Date: Thu, 5 Sep 2024 08:20:57 +0100 Subject: [PATCH] formatting --- .../redbox_app/redbox_core/consumers.py | 20 +++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/django_app/redbox_app/redbox_core/consumers.py b/django_app/redbox_app/redbox_core/consumers.py index 212437c19..efe9f6350 100644 --- a/django_app/redbox_app/redbox_core/consumers.py +++ b/django_app/redbox_app/redbox_core/consumers.py @@ -2,7 +2,7 @@ import logging from asyncio import CancelledError from collections.abc import Mapping, Sequence -from typing import Any +from typing import Any, ClassVar from uuid import UUID from channels.db import database_sync_to_async @@ -33,10 +33,9 @@ logger.info("WEBSOCKET_SCHEME is: %s", settings.WEBSOCKET_SCHEME) - class ChatConsumer(AsyncWebsocketConsumer): - full_reply = [] - citations = [] + full_reply: ClassVar = [] + citations: ClassVar = [] route = None metadata: MetadataDetail = MetadataDetail() redbox = Redbox(env=Settings(), debug=True) @@ -97,9 +96,13 @@ async def llm_conversation(self, selected_files: Sequence[File], session: Chat, ) try: - message = await self.save_message( - session, "".join(self.full_reply), ChatRoleEnum.ai, sources=self.citations, route=self.route, metadata=self.metadata + session, + "".join(self.full_reply), + ChatRoleEnum.ai, + sources=self.citations, + route=self.route, + metadata=self.metadata, ) await self.send_to_client("end", {"message_id": message.id, "title": title, "session_id": session.id}) @@ -209,9 +212,6 @@ async def handle_documents(self, response: ClientResponse) -> Sequence[tuple[Fil files = File.objects.filter(original_file__in=s3_keys) async for file in files: - await self.send_to_client( - "source", {"url": str(file.url), "original_file_name": file.original_file_name} - ) + await self.send_to_client("source", {"url": str(file.url), "original_file_name": file.original_file_name}) for file in files: self.citations.append((file, [doc for doc in response.data if doc.s3_key == file.unique_name])) -