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

🔨 refactor(utils.py): rename get_function_custom to eval_custom_component_code for better clarity and consistency

Browse source 
✨ feat(utils.py): add type hinting to eval_custom_component_code function for better code readability and maintainability
🔨 refactor(utils.py): move import statement for PromptTemplate to the correct location for better organization
🔨 refactor(utils.py): move import statement for validate to the correct location for better organization
🔨 refactor(utils.py): remove unused imports and reorganize import statements for better organization
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-23 08:39:51 -03:00
commit 096f3ae1fc
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/langflow/interface/importing/utils.py
View file

@ -3,15 +3,16 @@
import importlib import importlib
from typing import Any, Type from typing import Any, Type
from langchain.prompts import PromptTemplate
from langchain.agents import Agent from langchain.agents import Agent
from langchain.base_language import BaseLanguageModel from langchain.base_language import BaseLanguageModel
from langchain.chains.base import Chain from langchain.chains.base import Chain
from langchain.chat_models.base import BaseChatModel from langchain.chat_models.base import BaseChatModel
from langchain.prompts import PromptTemplate
from langchain.tools import BaseTool from langchain.tools import BaseTool
from langflow.interface.custom.custom_component import CustomComponent from langflow.interface.custom.custom_component import CustomComponent
from langflow.utils import validate
from langflow.interface.wrappers.base import wrapper_creator from langflow.interface.wrappers.base import wrapper_creator
from langflow.utils import validate
def import_module(module_path: str) -> Any: def import_module(module_path: str) -> Any:
@ -180,6 +181,7 @@ def get_function(code):
return validate.create_function(code, function_name) return validate.create_function(code, function_name)
def get_function_custom(code): def eval_custom_component_code(code: str) -> CustomComponent:
"""Evaluate custom component code"""
class_name = validate.extract_class_name(code) class_name = validate.extract_class_name(code)
return validate.create_class(code, class_name) return validate.create_class(code, class_name)