Add last tested version to component creation and
handle outdated versions
This commit is contained in:
parent
978757c757
commit
e925da5671
2 changed files with 17 additions and 26 deletions
|
|
@ -1,13 +1,15 @@
|
||||||
|
import warnings
|
||||||
from typing import Annotated, Any, Dict, List, Optional, Union
|
from typing import Annotated, Any, Dict, List, Optional, Union
|
||||||
from uuid import UUID
|
from uuid import UUID
|
||||||
|
|
||||||
from fastapi import APIRouter, Depends, HTTPException, Query
|
from fastapi import APIRouter, Depends, HTTPException, Query
|
||||||
from httpx import HTTPStatusError
|
from httpx import HTTPStatusError
|
||||||
|
|
||||||
from langflow.services.auth import utils as auth_utils
|
from langflow.services.auth import utils as auth_utils
|
||||||
from langflow.services.database.models.user.user import User
|
from langflow.services.database.models.user.user import User
|
||||||
from langflow.services.deps import get_settings_service, get_store_service
|
from langflow.services.deps import get_settings_service, get_store_service
|
||||||
from langflow.services.store.schema import (
|
from langflow.services.store.schema import (
|
||||||
ComponentResponse,
|
CreateComponentResponse,
|
||||||
DownloadComponentResponse,
|
DownloadComponentResponse,
|
||||||
ListComponentResponse,
|
ListComponentResponse,
|
||||||
ListComponentResponseModel,
|
ListComponentResponseModel,
|
||||||
|
|
@ -59,8 +61,8 @@ def check_if_store_has_api_key(
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@router.post("/components/", response_model=ComponentResponse, status_code=201)
|
@router.post("/components/", response_model=CreateComponentResponse, status_code=201)
|
||||||
def create_component(
|
async def create_component(
|
||||||
component: StoreComponentCreate,
|
component: StoreComponentCreate,
|
||||||
store_service: StoreService = Depends(get_store_service),
|
store_service: StoreService = Depends(get_store_service),
|
||||||
store_api_Key: str = Depends(get_user_store_api_key),
|
store_api_Key: str = Depends(get_user_store_api_key),
|
||||||
|
|
@ -68,6 +70,11 @@ def create_component(
|
||||||
try:
|
try:
|
||||||
# Verify if this is the latest version of Langflow
|
# Verify if this is the latest version of Langflow
|
||||||
# If not, raise an error
|
# If not, raise an error
|
||||||
|
if not component.last_tested_version:
|
||||||
|
# Get the local version of Langflow
|
||||||
|
from langflow import __version__ as current_version
|
||||||
|
|
||||||
|
component.last_tested_version = current_version
|
||||||
langflow_version = get_lf_version_from_pypi()
|
langflow_version = get_lf_version_from_pypi()
|
||||||
if langflow_version is None:
|
if langflow_version is None:
|
||||||
raise HTTPException(
|
raise HTTPException(
|
||||||
|
|
@ -76,12 +83,14 @@ def create_component(
|
||||||
)
|
)
|
||||||
elif langflow_version != component.last_tested_version:
|
elif langflow_version != component.last_tested_version:
|
||||||
# If the user is using an older version of Langflow, we need to raise an error
|
# If the user is using an older version of Langflow, we need to raise an error
|
||||||
raise ValueError(
|
# raise ValueError(
|
||||||
|
warnings.warn(
|
||||||
f"Your version of Langflow ({component.last_tested_version}) is outdated."
|
f"Your version of Langflow ({component.last_tested_version}) is outdated."
|
||||||
" Please update to the latest version ({langflow_version}) and try again."
|
f" Please update to the latest version ({langflow_version}) and try again."
|
||||||
)
|
)
|
||||||
|
|
||||||
return store_service.upload(store_api_Key, component)
|
result = await store_service.upload(store_api_Key, component)
|
||||||
|
return result
|
||||||
except Exception as exc:
|
except Exception as exc:
|
||||||
raise HTTPException(status_code=400, detail=str(exc))
|
raise HTTPException(status_code=400, detail=str(exc))
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,4 +1,3 @@
|
||||||
from datetime import datetime
|
|
||||||
from typing import List, Optional
|
from typing import List, Optional
|
||||||
from uuid import UUID
|
from uuid import UUID
|
||||||
|
|
||||||
|
|
@ -15,24 +14,8 @@ class UsersLikesResponse(BaseModel):
|
||||||
liked_by_user: Optional[bool]
|
liked_by_user: Optional[bool]
|
||||||
|
|
||||||
|
|
||||||
class ComponentResponse(BaseModel):
|
class CreateComponentResponse(BaseModel):
|
||||||
id: UUID
|
id: UUID
|
||||||
status: Optional[str]
|
|
||||||
sort: Optional[int]
|
|
||||||
user_name: Optional[str]
|
|
||||||
date_created: Optional[datetime]
|
|
||||||
user_updated: Optional[UUID]
|
|
||||||
date_updated: Optional[datetime]
|
|
||||||
is_component: Optional[bool]
|
|
||||||
name: Optional[str]
|
|
||||||
description: Optional[str]
|
|
||||||
data: Optional[dict]
|
|
||||||
tags: Optional[List[int]]
|
|
||||||
liked_by_count: Optional[int]
|
|
||||||
downloads_count: Optional[int]
|
|
||||||
parent: Optional[UUID] = None
|
|
||||||
metadata: Optional[dict]
|
|
||||||
last_tested_version: Optional[str]
|
|
||||||
|
|
||||||
|
|
||||||
class TagsIdResponse(BaseModel):
|
class TagsIdResponse(BaseModel):
|
||||||
|
|
@ -88,6 +71,5 @@ class StoreComponentCreate(BaseModel):
|
||||||
tags: Optional[List[str]]
|
tags: Optional[List[str]]
|
||||||
parent: Optional[UUID] = None
|
parent: Optional[UUID] = None
|
||||||
is_component: Optional[bool]
|
is_component: Optional[bool]
|
||||||
metadata: Optional[dict]
|
last_tested_version: Optional[str] = None
|
||||||
last_tested_version: Optional[str]
|
|
||||||
public: Optional[bool] = False
|
public: Optional[bool] = False
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue