🔧 fix(GetRequest.py): improve code readability and add support for multiple URLs and headers in GetRequest component

🔧 fix(PostRequest.py): improve code readability and add support for multiple documents, URLs, and headers in PostRequest component

src/backend/langflow/components/utilities/GetRequest.py

@@ -7,15 +7,57 @@ from typing import Optional
 
 class GetRequest(CustomComponent):
     display_name: str = "Get Request"
-    description: str = "Make a GET request to the given URL"
+    description: str = "Make a GET request to the given URL."
     output_types: list[str] = ["Document"]
     beta = True
     field_config = {
-        "url": {"display_name": "URL"},
+        "url": {
-        "headers": {"display_name": "Headers", "field_type": "code"},
+            "display_name": "URL",
+            "info": "The URL to make the request to",
+            "is_list": True,
+        },
+        "headers": {
+            "display_name": "Headers",
+            "field_type": "code",
+            "info": "The headers to send with the request.",
+        },
         "code": {"show": False},
     }
 
+    def get_document(
+        self,
+        url: str,
+        headers: Optional[dict] = None,
+    ) -> Document:
+        try:
+            if headers is None:
+                headers = {}
+            with requests.Session() as session:
+                response = session.get(url, headers=headers)
+                try:
+                    response_json = response.json()
+                    result = orjson_dumps(response_json, indent_2=False)
+                except Exception:
+                    result = response.text
+                self.repr_value = result
+                return Document(
+                    page_content=result,
+                    metadata={
+                        "source": url,
+                        "headers": headers,
+                        "status_code": response.status_code,
+                    },
+                )
+        except Exception as exc:
+            return Document(
+                page_content=str(exc),
+                metadata={
+                    "source": url,
+                    "headers": headers,
+                    "status_code": 500,
+                },
+            )
+
     def build(
         self,
         url: str,
@@ -23,7 +65,6 @@ class GetRequest(CustomComponent):
     ) -> Document:
         if headers is None:
             headers = {}
-        with requests.get(url, headers=headers) as result:
+        if not isinstance(url, list):
-            result = result.json()
+            url = [url]
-            self.repr_value = result
+        return [self.get_document(u, headers) for u in url]
-            return Document(page_content=orjson_dumps(result))

src/backend/langflow/components/utilities/PostRequest.py

@@ -7,16 +7,55 @@ from typing import Optional
 
 class PostRequest(CustomComponent):
     display_name: str = "Post Request"
-    description: str = "Make a POST request to the given URL"
+    description: str = "Make a POST request to the given URL."
     output_types: list[str] = ["Document"]
     beta = True
     field_config = {
-        "url": {"display_name": "URL"},
+        "url": {"display_name": "URL", "info": "The URL to make the request to."},
-        "headers": {"display_name": "Headers", "field_type": "code"},
+        "headers": {
+            "display_name": "Headers",
+            "field_type": "code",
+            "info": "The headers to send with the request.",
+        },
         "code": {"show": False},
         "document": {"display_name": "Document"},
     }
 
+    def post_document(
+        self,
+        document: Document,
+        url: str,
+        headers: Optional[dict] = None,
+    ) -> Document:
+        try:
+            with requests.Session() as session:
+                response = session.post(
+                    url, headers=headers, data=document.page_content
+                )
+                try:
+                    response_json = response.json()
+                    result = orjson_dumps(response_json, indent_2=False)
+                except Exception:
+                    result = response.text
+                self.repr_value = result
+                return Document(
+                    page_content=result,
+                    metadata={
+                        "source": url,
+                        "headers": headers,
+                        "status_code": response,
+                    },
+                )
+        except Exception as exc:
+            return Document(
+                page_content=str(exc),
+                metadata={
+                    "source": url,
+                    "headers": headers,
+                    "status_code": 500,
+                },
+            )
+
     def build(
         self,
         document: Document,
@@ -25,7 +64,9 @@ class PostRequest(CustomComponent):
     ) -> Document:
         if headers is None:
             headers = {}
-        with requests.post(url, headers=headers, json=document.page_content) as result:
+
-            result = result.json()
+        if not isinstance(document, list):
-            self.repr_value = result
+            document = [document]
-            return Document(page_content=orjson_dumps(result))
+        documents = [self.post_document(doc, url, headers) for doc in document]
+        self.repr_value = documents
+        return documents