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

ref: Make check_langflow_version async (#4701)

Browse source 
Make check_langflow_version async
This commit is contained in:
Christophe Bornet 2024-11-19 16:33:55 +01:00 • committed by GitHub
commit 1b39ce6f6e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/api/utils.py
View file

@ -100,7 +100,7 @@ def get_is_component_from_data(data: dict):
return data.get("is_component") return data.get("is_component")
def check_langflow_version(component: StoreComponentCreate) -> None: async def check_langflow_version(component: StoreComponentCreate) -> None:
from langflow.utils.version import get_version_info from langflow.utils.version import get_version_info
__version__ = get_version_info()["version"] __version__ = get_version_info()["version"]
@ -108,7 +108,7 @@ def check_langflow_version(component: StoreComponentCreate) -> None:
if not component.last_tested_version: if not component.last_tested_version:
component.last_tested_version = __version__ component.last_tested_version = __version__
langflow_version = get_lf_version_from_pypi() langflow_version = await get_lf_version_from_pypi()
if langflow_version is None: if langflow_version is None:
raise HTTPException(status_code=500, detail="Unable to verify the latest version of Langflow") raise HTTPException(status_code=500, detail="Unable to verify the latest version of Langflow")
if langflow_version != component.last_tested_version: if langflow_version != component.last_tested_version:

5
src/backend/base/langflow/api/v1/store.py
View file

@ -1,4 +1,3 @@
import asyncio
from typing import Annotated from typing import Annotated
from uuid import UUID from uuid import UUID
@ -68,7 +67,7 @@ async def share_component(
store_api_key: Annotated[str, Depends(get_user_store_api_key)], store_api_key: Annotated[str, Depends(get_user_store_api_key)],
) -> CreateComponentResponse: ) -> CreateComponentResponse:
try: try:
await asyncio.to_thread(check_langflow_version, component) await check_langflow_version(component)
return await get_store_service().upload(store_api_key, component) return await get_store_service().upload(store_api_key, component)
except Exception as exc: except Exception as exc:
raise HTTPException(status_code=400, detail=str(exc)) from exc raise HTTPException(status_code=400, detail=str(exc)) from exc
@ -81,7 +80,7 @@ async def update_shared_component(
store_api_key: Annotated[str, Depends(get_user_store_api_key)], store_api_key: Annotated[str, Depends(get_user_store_api_key)],
) -> CreateComponentResponse: ) -> CreateComponentResponse:
try: try:
await asyncio.to_thread(check_langflow_version, component) await check_langflow_version(component)
return await get_store_service().update(store_api_key, component_id, component) return await get_store_service().update(store_api_key, component_id, component)
except Exception as exc: except Exception as exc:
raise HTTPException(status_code=400, detail=str(exc)) from exc raise HTTPException(status_code=400, detail=str(exc)) from exc

5
src/backend/base/langflow/services/store/utils.py
View file

@ -41,9 +41,10 @@ async def update_components_with_user_data(
# Get the latest released version of langflow (https://pypi.org/project/langflow/) # Get the latest released version of langflow (https://pypi.org/project/langflow/)
def get_lf_version_from_pypi(): async def get_lf_version_from_pypi():
try: try:
response = httpx.get("https://pypi.org/pypi/langflow/json") async with httpx.AsyncClient() as client:
response = client.get("https://pypi.org/pypi/langflow/json")
if response.status_code != httpx.codes.OK: if response.status_code != httpx.codes.OK:
return None return None
return response.json()["info"]["version"] return response.json()["info"]["version"]