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

feat: Add timeout parameter to HTTP request in StoreService. (#3362)

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-08-15 10:50:11 -03:00 • committed by GitHub
commit cdab352bc8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 10 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/backend/base/langflow/services/store/service.py
View file

@ -97,6 +97,7 @@ class StoreService(Service):
"last_tested_version", "last_tested_version",
"private", "private",
] ]
self.timeout = 30
# Create a context manager that will use the api key to # Create a context manager that will use the api key to
# get the user data and all requests inside the context manager # get the user data and all requests inside the context manager
@ -129,7 +130,7 @@ class StoreService(Service):
headers = {} headers = {}
async with httpx.AsyncClient() as client: async with httpx.AsyncClient() as client:
try: try:
response = await client.get(url, headers=headers, params=params) response = await client.get(url, headers=headers, params=params, timeout=self.timeout)
response.raise_for_status() response.raise_for_status()
except HTTPError as exc: except HTTPError as exc:
raise exc raise exc
@ -151,7 +152,9 @@ class StoreService(Service):
try: try:
headers = {"Authorization": f"Bearer {api_key}"} headers = {"Authorization": f"Bearer {api_key}"}
async with httpx.AsyncClient() as client: async with httpx.AsyncClient() as client:
response = await client.post(webhook_url, headers=headers, json={"component_id": str(component_id)}) response = await client.post(
webhook_url, headers=headers, json={"component_id": str(component_id)}, timeout=self.timeout
)
response.raise_for_status() response.raise_for_status()
return response.json() return response.json()
except HTTPError as exc: except HTTPError as exc:
@ -359,7 +362,9 @@ class StoreService(Service):
# response = httpx.post(self.components_url, headers=headers, json=component_dict) # response = httpx.post(self.components_url, headers=headers, json=component_dict)
# response.raise_for_status() # response.raise_for_status()
async with httpx.AsyncClient() as client: async with httpx.AsyncClient() as client:
response = await client.post(self.components_url, headers=headers, json=component_dict) response = await client.post(
self.components_url, headers=headers, json=component_dict, timeout=self.timeout
)
response.raise_for_status() response.raise_for_status()
component = response.json()["data"] component = response.json()["data"]
return CreateComponentResponse(**component) return CreateComponentResponse(**component)
@ -394,7 +399,7 @@ class StoreService(Service):
# response.raise_for_status() # response.raise_for_status()
async with httpx.AsyncClient() as client: async with httpx.AsyncClient() as client:
response = await client.patch( response = await client.patch(
self.components_url + f"/{component_id}", headers=headers, json=component_dict self.components_url + f"/{component_id}", headers=headers, json=component_dict, timeout=self.timeout
) )
response.raise_for_status() response.raise_for_status()
component = response.json()["data"] component = response.json()["data"]
@ -463,6 +468,7 @@ class StoreService(Service):
self.like_webhook_url, self.like_webhook_url,
json={"component_id": str(component_id)}, json={"component_id": str(component_id)},
headers=headers, headers=headers,
timeout=self.timeout,
) )
response.raise_for_status() response.raise_for_status()
if response.status_code == 200: if response.status_code == 200: