refactor: Add MessageInput to langflow inputs

This commit is contained in:
ogabrielluiz 2024-06-17 13:01:35 -03:00
commit 71524d8219
2 changed files with 9 additions and 10 deletions

View file

@ -6,6 +6,7 @@ from .inputs import (
FloatInput, FloatInput,
HandleInput, HandleInput,
IntInput, IntInput,
MessageInput,
MultilineInput, MultilineInput,
NestedDictInput, NestedDictInput,
PromptInput, PromptInput,
@ -17,15 +18,16 @@ from .inputs import (
__all__ = [ __all__ = [
"SecretStrInput", "SecretStrInput",
"StrInput", "StrInput",
MultilineInput,
NestedDictInput,
PromptInput,
SecretStrInput,
StrInput,
TextInput,
"FileInput",
"PromptInput", "PromptInput",
"MultilineInput", "MultilineInput",
"HandleInput", "HandleInput",
"TextInput", "TextInput",
"BoolInput",
"DropdownInput",
"FloatInput",
"IntInput",
"DictInput",
"MessageInput",
"NestedDictInput",
"FileInput",
] ]

View file

@ -3,17 +3,14 @@ import json
import os import os
import re import re
from io import BytesIO from io import BytesIO
from typing import Dict
import yaml import yaml
from docstring_parser import parse
from langchain_core.language_models import BaseLanguageModel from langchain_core.language_models import BaseLanguageModel
from loguru import logger from loguru import logger
from PIL.Image import Image from PIL.Image import Image
from langflow.services.chat.config import ChatConfig from langflow.services.chat.config import ChatConfig
from langflow.services.deps import get_settings_service from langflow.services.deps import get_settings_service
from langflow.utils.util import format_dict, get_base_classes, get_default_factory
def load_file_into_dict(file_path: str) -> dict: def load_file_into_dict(file_path: str) -> dict: