From 7fe2698bf2dd307bb10a903f8b19e4470b560bdb Mon Sep 17 00:00:00 2001 From: "P. Taylor Goetz" Date: Sat, 25 May 2024 21:24:23 -0400 Subject: [PATCH] lint --- backend/app/message_types.py | 2 +- backend/app/parsing.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/backend/app/message_types.py b/backend/app/message_types.py index 9ceea94b..432e6c64 100644 --- a/backend/app/message_types.py +++ b/backend/app/message_types.py @@ -18,7 +18,7 @@ class LiberalToolMessage(ToolMessage): def _convert_pydantic_dict_to_message( - data: MessageLikeRepresentation + data: MessageLikeRepresentation, ) -> MessageLikeRepresentation: if ( isinstance(data, dict) diff --git a/backend/app/parsing.py b/backend/app/parsing.py index be18e5ee..8a84fb02 100644 --- a/backend/app/parsing.py +++ b/backend/app/parsing.py @@ -1,6 +1,6 @@ """Module contains logic for parsing binary blobs into text.""" from typing import Iterator -from langchain_core.document_loaders.blob_loaders import Blob + from langchain.text_splitter import MarkdownHeaderTextSplitter from langchain_community.document_loaders.parsers import ( BS4HTMLParser, @@ -10,6 +10,7 @@ from langchain_community.document_loaders.parsers.msword import MsWordParser from langchain_community.document_loaders.parsers.txt import TextParser from langchain_core.document_loaders import BaseBlobParser +from langchain_core.document_loaders.blob_loaders import Blob from langchain_core.documents import Document