jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor APIRequest class to use 'body' instead of 'record' parameter

Browse source
This commit is contained in:
Rodrigo Nader 2024-03-06 00:50:27 -03:00
commit 9958ce76fd
1 changed files with 11 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

22
src/backend/langflow/components/helpers/APIRequest.py
View file

@ -27,9 +27,9 @@ class APIRequest(CustomComponent):
"display_name": "Headers",
"info": "The headers to send with the request.",
},
"record": {
"display_name": "Record",
"info": "The record to send with the request (for POST, PATCH, PUT).",
"body": {
"display_name": "Body",
"info": "The body to send with the request (for POST, PATCH, PUT).",
},
"timeout": {
"display_name": "Timeout",
@ -45,14 +45,14 @@ class APIRequest(CustomComponent):
method: str,
url: str,
headers: Optional[dict] = None,
record: Optional[Record] = None,
body: Optional[dict] = None,
timeout: int = 5,
) -> Record:
method = method.upper()
if method not in ["GET", "POST", "PATCH", "PUT"]:
raise ValueError(f"Unsupported method: {method}")
data = record.data if record else None
data = body if body else None
try:
response = await client.request(
method, url, headers=headers, content=data, timeout=timeout
@ -86,22 +86,22 @@ class APIRequest(CustomComponent):
method: str,
url: List[str],
headers: Optional[dict] = None,
record: Optional[Record] = None,
body: Optional[dict] = None,
timeout: int = 5,
) -> List[Record]:
if headers is None:
headers = {}
urls = url if isinstance(url, list) else [url]
records = (
record
if isinstance(record, list)
else [record] if record else [None] * len(urls)
bodies = (
body
if isinstance(body, list)
else [body] if body else [None] * len(urls)
)
async with httpx.AsyncClient() as client:
results = await asyncio.gather(
*[
self.make_request(client, method, u, headers, rec, timeout)
for u, rec in zip(urls, records)
for u, rec in zip(urls, bodies)
]
)
return results