From 05c8f5a0f8e19e86e9dd649e81f62d8f1952c8c1 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 22 Aug 2023 16:13:07 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(GetRequest.py):=20update=20o?= =?UTF-8?q?rjson=5Fdumps=20function=20call=20to=20set=20indent=5F2=20param?= =?UTF-8?q?eter=20to=20False=20=F0=9F=90=9B=20fix(GetRequest.py):=20update?= =?UTF-8?q?=20return=20type=20annotation=20of=20get=5Fdocuments=20method?= =?UTF-8?q?=20to=20list[Document]=20=F0=9F=90=9B=20fix(PostRequest.py):=20?= =?UTF-8?q?update=20return=20type=20annotation=20of=20post=5Fdocument=20me?= =?UTF-8?q?thod=20to=20list[Document]?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/components/utilities/GetRequest.py | 6 +++--- src/backend/langflow/components/utilities/PostRequest.py | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/backend/langflow/components/utilities/GetRequest.py b/src/backend/langflow/components/utilities/GetRequest.py index 018b92751..40fe8935f 100644 --- a/src/backend/langflow/components/utilities/GetRequest.py +++ b/src/backend/langflow/components/utilities/GetRequest.py @@ -31,7 +31,7 @@ class GetRequest(CustomComponent): response = session.get(url, headers=headers, timeout=timeout) try: response_json = response.json() - result = orjson_dumps(response_json, indent=2) + result = orjson_dumps(response_json, indent_2=False) except Exception: result = response.text self.repr_value = result @@ -59,11 +59,11 @@ class GetRequest(CustomComponent): url: str, headers: Optional[dict] = None, timeout: int = 5, - ) -> Document: + ) -> list[Document]: if headers is None: headers = {} if not isinstance(url, list): - url = [url] + url: list[str] = [url] with requests.Session() as session: documents = [self.get_document(session, u, headers, timeout) for u in url] self.repr_value = documents diff --git a/src/backend/langflow/components/utilities/PostRequest.py b/src/backend/langflow/components/utilities/PostRequest.py index 60a239d34..e0687a6f7 100644 --- a/src/backend/langflow/components/utilities/PostRequest.py +++ b/src/backend/langflow/components/utilities/PostRequest.py @@ -59,12 +59,12 @@ class PostRequest(CustomComponent): document: Document, url: str, headers: Optional[dict] = None, - ) -> Document: + ) -> list[Document]: if headers is None: headers = {} if not isinstance(document, list) and isinstance(document, Document): - document = [document] + document: list[Document] = [document] else: raise ValueError("document must be a Document or a list of Documents")