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

Refactor IMSDbLoaderComponent configuration

Browse source
This commit is contained in:
anovazzi1 2024-01-10 15:42:04 -03:00
commit 8650fde69e
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/langflow/components/documentloaders/IMSDbLoader.py
View file

@ -1,5 +1,7 @@
from langflow import CustomComponent
from langflow.field_typing import Document
from langchain_community.document_loaders.imsdb import IMSDbLoader
from typing import Dict, Optional
@ -9,8 +11,8 @@ class IMSDbLoaderComponent(CustomComponent):
def build_config(self):
return {
"metadata": {"display_name": "Metadata", "type": "dict"},
"web_path": {"display_name": "Web Page", "type": "str"},
"metadata": {"display_name": "Metadata", "field_type": "dict"},
"web_path": {"display_name": "Web Page", "field_type": "str"},
}
def build(
@ -18,6 +20,4 @@ class IMSDbLoaderComponent(CustomComponent):
metadata: Optional[Dict] = None,
web_path: str = "",
) -> Document:
# Assuming there is a class or function named `IMSDbLoader` that takes these parameters
# and returns a Document object. Replace `IMSDbLoader` with the actual class or function name.
return IMSDbLoader(metadata=metadata, web_path=web_path)