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

Merge branch 'feature/store' of github.com:logspace-ai/langflow into feature/store

Browse source
This commit is contained in:
cristhianzl 2023-10-23 23:04:48 -03:00
commit 3cb0231f2e
3 changed files with 13 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/langflow/api/v1/store.py
View file

@ -41,7 +41,7 @@ def create_component(
): ):
try: try:
decrypted = auth_utils.decrypt_api_key(store_api_Key, settings_service) decrypted = auth_utils.decrypt_api_key(store_api_Key, settings_service)
return store_service.upload(decrypted, component.dict()) return store_service.upload(decrypted, component)
except Exception as exc: except Exception as exc:
raise HTTPException(status_code=400, detail=str(exc)) raise HTTPException(status_code=400, detail=str(exc))

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

@ -3,7 +3,7 @@ from langflow.services.base import Service
from typing import TYPE_CHECKING, List, Dict, Any, Optional from typing import TYPE_CHECKING, List, Dict, Any, Optional
import httpx import httpx
from httpx import HTTPError from httpx import HTTPError
from langflow.services.store.schema import ComponentResponse from langflow.services.store.schema import ComponentResponse, StoreComponentCreate
if TYPE_CHECKING: if TYPE_CHECKING:
from langflow.services.settings.service import SettingsService from langflow.services.settings.service import SettingsService
@ -106,14 +106,20 @@ class StoreService(Service):
def download(self, api_key: str, id: str) -> ComponentResponse: def download(self, api_key: str, id: str) -> ComponentResponse:
url = f"{self.components_url}/{id}" url = f"{self.components_url}/{id}"
params = {"fields": ",".join(["id", "name", "description", "data"])} params = {"fields": ",".join(["id", "name", "description", "data"])}
component = self._get(url, api_key, params)[0] component = self._get(url, api_key, params)
return ComponentResponse(**component) return ComponentResponse(**component)
def upload(self, api_key: str, component_data: Dict[str, Any]) -> ComponentResponse: def upload(
self, api_key: str, component_data: StoreComponentCreate
) -> ComponentResponse:
headers = {"Authorization": f"Bearer {api_key}"} headers = {"Authorization": f"Bearer {api_key}"}
component_dict = component_data.dict(exclude_unset=True)
# Parent is a UUID, but the store expects a string
if component_dict.get("parent"):
component_dict["parent"] = str(component_dict["parent"])
try: try:
response = httpx.post( response = httpx.post(
self.components_url, headers=headers, json=component_data self.components_url, headers=headers, json=component_dict
) )
response.raise_for_status() response.raise_for_status()
component = response.json()["data"] component = response.json()["data"]

2
src/frontend/src/controllers/API/index.ts
View file

@ -553,6 +553,7 @@ export async function saveFlowStore(newFlow: {
description?: string; description?: string;
style?: FlowStyleType; style?: FlowStyleType;
is_component?: boolean; is_component?: boolean;
parent?: string;
}): Promise<FlowType> { }): Promise<FlowType> {
try { try {
const response = await api.post(`${BASE_URL_API}store/components/`, { const response = await api.post(`${BASE_URL_API}store/components/`, {
@ -560,6 +561,7 @@ export async function saveFlowStore(newFlow: {
data: newFlow.data, data: newFlow.data,
description: newFlow.description, description: newFlow.description,
is_component: newFlow.is_component, is_component: newFlow.is_component,
parent: newFlow.parent,
}); });
if (response.status !== 201) { if (response.status !== 201) {