From 9dbb8d3578d88363bf8efa742f65255d03684b91 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 20 Nov 2023 21:35:12 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20fix(store.py):=20remove=20unnece?= =?UTF-8?q?ssary=20comment=20in=20download=5Fcomponent=20function=20?= =?UTF-8?q?=F0=9F=94=A7=20fix(utils.py):=20replace=20requests=20library=20?= =?UTF-8?q?with=20httpx=20library=20for=20getting=20latest=20released=20ve?= =?UTF-8?q?rsion=20of=20langflow=20from=20PyPI=20to=20improve=20performanc?= =?UTF-8?q?e=20and=20reliability?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/api/v1/store.py | 3 --- src/backend/langflow/services/store/utils.py | 13 ++++++++----- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/src/backend/langflow/api/v1/store.py b/src/backend/langflow/api/v1/store.py index 090deb6c6..5e087de97 100644 --- a/src/backend/langflow/api/v1/store.py +++ b/src/backend/langflow/api/v1/store.py @@ -3,7 +3,6 @@ from typing import Annotated, List, Optional, Union from uuid import UUID from fastapi import APIRouter, Depends, HTTPException, Query - from langflow.services.auth import utils as auth_utils from langflow.services.database.models.user.user import User from langflow.services.deps import get_settings_service, get_store_service @@ -140,8 +139,6 @@ async def download_component( store_service: StoreService = Depends(get_store_service), store_api_Key: str = Depends(get_user_store_api_key), ): - # If the component is from the store, we need to get it from the store - try: component = await store_service.download(store_api_Key, component_id) except CustomException as exc: diff --git a/src/backend/langflow/services/store/utils.py b/src/backend/langflow/services/store/utils.py index ec66a1ea3..3ce4434f1 100644 --- a/src/backend/langflow/services/store/utils.py +++ b/src/backend/langflow/services/store/utils.py @@ -1,5 +1,7 @@ from typing import TYPE_CHECKING, List +import httpx + if TYPE_CHECKING: from langflow.services.store.schema import ListComponentResponse from langflow.services.store.service import StoreService @@ -40,12 +42,13 @@ async def update_components_with_user_data( # Get the latest released version of langflow (https://pypi.org/project/langflow/) def get_lf_version_from_pypi(): - import requests - - response = requests.get("https://pypi.org/pypi/langflow/json") - if response.status_code != 200: + try: + response = httpx.get("https://pypi.org/pypi/langflow/json") + if response.status_code != 200: + return None + return response.json()["info"]["version"] + except Exception: return None - return response.json()["info"]["version"] def process_component_data(nodes_list):