From 90af16018919d992dac47f9bc576f5ce95f9ea0c Mon Sep 17 00:00:00 2001 From: freddyaboulton Date: Mon, 9 Sep 2024 16:12:45 -0400 Subject: [PATCH 1/3] Fix --- gradio/components/chatbot.py | 30 +++--------------------------- 1 file changed, 3 insertions(+), 27 deletions(-) diff --git a/gradio/components/chatbot.py b/gradio/components/chatbot.py index 97e3dafbcab7..a557418743b0 100644 --- a/gradio/components/chatbot.py +++ b/gradio/components/chatbot.py @@ -434,7 +434,7 @@ def _postprocess_messages_tuples(self, value: TupleFormat) -> ChatbotDataTuples: def _postprocess_message_messages( self, message: MessageDict | ChatMessage - ) -> list[Message]: + ) -> Message: if isinstance(message, dict): message["content"] = self._postprocess_content(message["content"]) msg = Message(**message) # type: ignore @@ -450,30 +450,7 @@ def _postprocess_message_messages( f"Invalid message for Chatbot component: {message}", visible=False ) - # extract file path from message - new_messages = [] - if isinstance(msg.content, str): - for word in msg.content.split(" "): - filepath = Path(word) - try: - is_file = filepath.is_file() and filepath.exists() - except OSError: - is_file = False - if is_file: - filepath = cast( - str, move_resource_to_block_cache(filepath, block=self) - ) - mime_type = client_utils.get_mimetype(filepath) - new_messages.append( - Message( - role=msg.role, - metadata=msg.metadata, - content=FileMessage( - file=FileData(path=filepath, mime_type=mime_type) - ), - ), - ) - return [msg, *new_messages] + return msg def postprocess( self, @@ -495,9 +472,8 @@ def postprocess( return self._postprocess_messages_tuples(cast(TupleFormat, value)) self._check_format(value, "messages") processed_messages = [ - msg + self._postprocess_message_messages(cast(MessageDict, message)) for message in value - for msg in self._postprocess_message_messages(cast(MessageDict, message)) ] return ChatbotDataMessages(root=processed_messages) From 5736d5f844beeda4b7e66d02b66e90d5b342ad2e Mon Sep 17 00:00:00 2001 From: gradio-pr-bot Date: Mon, 9 Sep 2024 20:33:47 +0000 Subject: [PATCH 2/3] add changeset --- .changeset/khaki-ducks-stare.md | 5 +++++ 1 file changed, 5 insertions(+) create mode 100644 .changeset/khaki-ducks-stare.md diff --git a/.changeset/khaki-ducks-stare.md b/.changeset/khaki-ducks-stare.md new file mode 100644 index 000000000000..ae516e5c30f4 --- /dev/null +++ b/.changeset/khaki-ducks-stare.md @@ -0,0 +1,5 @@ +--- +"gradio": minor +--- + +feat:Dont move files to cache automatically in chatbot postprocess From f8806a379061d80fa77a0a4a87e2e1de10e14d59 Mon Sep 17 00:00:00 2001 From: freddyaboulton Date: Mon, 9 Sep 2024 17:07:43 -0400 Subject: [PATCH 3/3] Add code --- gradio/components/chatbot.py | 1 - 1 file changed, 1 deletion(-) diff --git a/gradio/components/chatbot.py b/gradio/components/chatbot.py index a557418743b0..05bdb45bf7eb 100644 --- a/gradio/components/chatbot.py +++ b/gradio/components/chatbot.py @@ -31,7 +31,6 @@ from gradio.data_classes import FileData, GradioModel, GradioRootModel from gradio.events import Events from gradio.exceptions import Error -from gradio.processing_utils import move_resource_to_block_cache class MetadataDict(TypedDict):