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

Refactor APIRequest class to use httpx.AsyncClient

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-05 21:27:05 -03:00
commit 2b8164f28d
1 changed files with 27 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,8 +1,7 @@
import asyncio
from typing import List, Optional, Union
import httpx
from typing import List, Optional
import requests
import httpx
from langflow import CustomComponent
from langflow.schema import Record
@ -42,7 +41,7 @@ class APIRequest(CustomComponent):
async def make_request(
self,
session: requests.Session,
client: httpx.AsyncClient,
method: str,
url: str,
headers: Optional[dict] = None,
@ -55,7 +54,6 @@ class APIRequest(CustomComponent):
data = record.text if record else None
try:
async with httpx.AsyncClient() as client:
response = await client.request(
method, url, headers=headers, content=data, timeout=timeout
)
@ -88,7 +86,7 @@ class APIRequest(CustomComponent):
method: str,
url: List[str],
headers: Optional[dict] = None,
record: Optional[Union[Record, List[Record]]] = None,
record: Optional[Record] = None,
timeout: int = 5,
) -> List[Record]:
if headers is None:
@ -99,11 +97,11 @@ class APIRequest(CustomComponent):
if isinstance(record, list)
else [record] if record else [None] * len(urls)
)
async with httpx.AsyncClient() as client:
results = await asyncio.gather(
*[
self.make_request(method, u, headers, doc, timeout)
for u, doc in zip(urls, records)
self.make_request(client, method, u, headers, rec, timeout)
for u, rec in zip(urls, records)
]
)
return results